1 files changed, 63 insertions, 0 deletions
diff --git a/doc/arm/_ext/mergegrammar.py b/doc/arm/_ext/mergegrammar.py
new file mode 100644
index 0000000..b0500eb
--- /dev/null
+++ b/doc/arm/_ext/mergegrammar.py
@@ -0,0 +1,63 @@
+############################################################################
+# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
+#
+# SPDX-License-Identifier: MPL-2.0
+#
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, you can obtain one at https://mozilla.org/MPL/2.0/.
+#
+# See the COPYRIGHT file distributed with this work for additional
+# information regarding copyright ownership.
+############################################################################
+
+# Depends on CWD - Sphinx plugin
+
+import json
+from pathlib import Path
+
+import parsegrammar
+
+
+def read_zone():
+    zone_grammars = {}
+    for file in Path("../misc/").glob("*.zoneopt"):
+        # in-view is not really a zone type
+        if file.stem == "in-view":
+            zone_type = "in-view"
+        else:
+            zone_type = f"type {file.stem}"
+
+        with file.open(encoding="ascii") as fp:
+            zonegrammar = parsegrammar.parse_mapbody(fp)
+            assert len(zonegrammar) == 1
+            assert "zone" in zonegrammar
+            zone_grammars[zone_type] = zonegrammar["zone"]
+            zone_grammars[zone_type]["_pprint_name"] = "zone"
+
+    return {"zone": {"_mapbody": zone_grammars, "_ignore_this_level": True}}
+
+
+def read_main():
+    with Path("../misc/options").open(encoding="ascii") as fp:
+        optgrammar = parsegrammar.parse_mapbody(fp)
+    return optgrammar
+
+
+def combine():
+    zones = read_zone()
+    assert zones
+    rest = read_main()
+    assert rest
+    rest.update(zones)
+
+    # this is a terrible hack
+    # but cfg_test cannot print zone grammars inside view
+    rest["view"]["_mapbody"].update(zones)
+
+    return rest
+
+
+if __name__ == "__main__":
+    full_grammar = combine()
+    print(json.dumps(full_grammar))