import subprocess
import sys
import urllib.request
from os import makedirs
from os.path import join, exists

from importlib_resources import files, as_file

from ..vendor.appdirs import user_cache_dir

_openapi_jar_basename = "openapi-generator-cli-6.6.0.jar"
_openapi_jar_url = f"https://repo1.maven.org/maven2/org/openapitools/openapi-generator-cli/6.6.0/{_openapi_jar_basename}"


def is_java_installed():
    try:
        command = "java -version"
        result = subprocess.check_output(command, stderr=subprocess.STDOUT, shell=True, text=True)
        return "version" in result.lower()
    except subprocess.CalledProcessError:
        return False


def main():
    if not is_java_installed():
        print("java must be installed to generate openapi clients automatically", file=sys.stderr)
        raise FileNotFoundError("java")

    cache_dir = user_cache_dir(appname="comfyui")
    jar = join(cache_dir, _openapi_jar_basename)

    if not exists(jar):
        makedirs(cache_dir, exist_ok=True)
        print(f"downloading {_openapi_jar_basename} to {jar}", file=sys.stderr)
        urllib.request.urlretrieve(_openapi_jar_url, jar)

    with as_file(files('comfy.api').joinpath('openapi.yaml')) as openapi_schema:
        with as_file(files('comfy.api').joinpath('openapi_python_config.yaml')) as python_config:
            cmds = [
                "java",
                "--add-opens", "java.base/java.io=ALL-UNNAMED",
                "--add-opens", "java.base/java.util=ALL-UNNAMED",
                "--add-opens", "java.base/java.lang=ALL-UNNAMED",
                "-jar", jar,
                "generate",
                "--input-spec", openapi_schema,
                "--global-property", "models",
                "--config", python_config
            ]
            subprocess.check_output(cmds)


if __name__ == "__main__":
    main()