blitz_api/gen_client_libs.py

import importlib
import json
import os
import shutil
from subprocess import PIPE, Popen

from fastapi.openapi.utils import get_openapi

# npm install @openapitools/openapi-generator-cli -g
# sudo apt install default-jre

out_path = os.path.abspath(os.path.join("../../", "blitz_api_client_libraries"))

mod = importlib.import_module("app.main")

app = getattr(mod, "app")
version = "v1"

if version:
    for route in app.router.routes:
        if version in route.path:
            app = route.app
            break


def _dart_dio_post_action():
    # dart-dio has to generate some code after the initial generation.
    # flutter pub get && flutter pub run build_runner build --delete-conflicting-outputs
    print("Post action for dart-dio")
    p = os.path.abspath(os.path.join(out_path, "clients/dart-dio"))

    print("Running flutter pub get")
    process = Popen(["flutter", "pub", "get"], stdout=PIPE, stderr=PIPE, cwd=p)
    _, stderr = process.communicate()
    if stderr:
        print(stderr)

    print("Generating code")
    process = Popen(
        [
            "flutter",
            "pub",
            "run",
            "build_runner",
            "build",
            "--delete-conflicting-outputs",
        ],
        stdout=PIPE,
        stderr=PIPE,
        cwd=p,
    )
    _, stderr = process.communicate()
    if stderr:
        print(stderr)


def _generate(generator: str, langOptions: list = []):
    p = os.path.abspath(os.path.join(out_path, f"clients/{generator}"))

    print(f"Removing {p}")
    if os.path.exists(p):
        shutil.rmtree(p)

    print(f"Recreating directory {p}")
    os.makedirs(p, exist_ok=True)

    print(f"Generating client files {generator}")

    process = Popen(
        [
            "openapi-generator-cli",
            "generate",
            "-i",
            "openapi.json",
            "-g",
            generator,
            "-o",
            p,
            *langOptions,
        ],
        stdout=PIPE,
        stderr=PIPE,
    )
    _, stderr = process.communicate()
    if stderr:
        print(stderr)


def main():
    # Gets the OpenAPI specs.
    specs = get_openapi(
        title=app.title if app.title else None,
        version=app.version if app.version else None,
        openapi_version=app.openapi_version if app.openapi_version else None,
        description=app.description if app.description else None,
        routes=app.routes if app.routes else None,
    )

    with open("openapi.json", "w") as f:
        json.dump(specs, f, indent=2)

    dartOpts = [
        "--additional-properties=pubName=blitz_api_client",
        "--additional-properties=pubVersion=0.5.1",
    ]

    _generate("dart-dio", dartOpts)
    _generate("dart", dartOpts)
    _generate("go")
    _generate("javascript")
    _generate("kotlin")
    _generate("python")
    _generate("typescript-axios")
    _generate("typescript-fetch")

    _dart_dio_post_action()


if __name__ == "__main__":
    main()