91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
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()
|