...

_archive/openrpc/generator/mdbook/generate_mdbook.py

@@ -0,0 +1,90 @@
+import argparse
+import json
+import os
+
+from jinja2 import Environment, FileSystemLoader
+
+from ....openrpc.tools import get_pydantic_type, get_return_type, topological_sort
+
+script_dir = os.path.dirname(os.path.abspath(__file__))
+
+
+def generate_models(openrpc_spec: dict) -> str:
+    schema_dict = openrpc_spec["components"]["schemas"]
+    sorted_classes = topological_sort(schema_dict)
+
+    env = Environment(loader=FileSystemLoader(script_dir), trim_blocks=True, lstrip_blocks=True)
+    template = env.get_template("templates/mdbook/structs.jinja")
+    model_code = template.render(
+        sorted_classes=sorted_classes,
+        schema_dict=schema_dict,
+        get_pydantic_type=get_pydantic_type,
+    )
+
+    return model_code
+
+
+def generate_model(model_name: str, schema: dict) -> str:
+    env = Environment(loader=FileSystemLoader(script_dir))
+    template = env.get_template("templates/vlang/struct.jinja")
+    model_code = template.render(model_name=model_name, schema=schema, get_pydantic_type=get_pydantic_type)
+
+    return model_code
+
+
+def generate_api_methods(openrpc_spec: dict) -> str:
+    env = Environment(loader=FileSystemLoader(script_dir), trim_blocks=True, lstrip_blocks=True)
+    template = env.get_template("templates/mdbook/methods.jinja")
+
+    code = template.render(
+        spec=openrpc_spec,
+        methods=openrpc_spec.get("methods", []),
+        get_return_type=get_return_type,
+        get_pydantic_type=get_pydantic_type,
+    )
+
+    return code
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Generate API code from OpenRPC specification")
+    parser.add_argument(
+        "-s",
+        "--spec",
+        help="Path to the specs (expressed in our own V format)",
+        default="~/code/git.threefold.info/projectmycelium/hero_server/generatorexamples/example1/specs",
+    )
+    parser.add_argument(
+        "-o",
+        "--output",
+        default="/tmp/generator/mdbook",
+        help="Output file path (default: /tmp/generator/mdbook)",
+    )
+    args = parser.parse_args()
+
+    spec_file = os.path.expanduser(args.spec)
+    output_dir = os.path.expanduser(args.output)
+
+    if not os.path.isfile(spec_file):
+        print(f"Error: OpenRPC specification file '{spec_file}' does not exist.")
+        return
+
+    with open(spec_file) as file:
+        openrpc_spec = json.load(file)
+
+    code_models = generate_models(openrpc_spec)
+    code_methods = generate_api_methods(openrpc_spec)
+
+    os.makedirs(os.path.dirname(output_dir), exist_ok=True)
+
+    # Write the generated code to a file
+    with open(f"{output_dir}/models.md", "w") as file:
+        file.write(code_models)
+    with open(f"{output_dir}/methods.md", "w") as file:
+        file.write(code_methods)
+
+    print(f"Generated API code has been written to {output_dir}")
+
+
+if __name__ == "__main__":
+    main()