load("//java/kotlin-extractor:versions.bzl", "VERSIONS")
load("//misc/bazel:lfs.bzl", "lfs_smudge")

_kotlin_dep_build = """
load("@rules_kotlin//kotlin:jvm.bzl", "kt_jvm_import")

kt_jvm_import(
    name = "{name}",
    jar = "{name}.jar",
    visibility = ["//visibility:public"],
)
"""

_empty_zip = "PK\005\006\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000\000"

def _get_dep(repository_ctx, name):
    return repository_ctx.path(Label("//java/kotlin-extractor/deps:%s" % name))

def _kotlin_dep_impl(repository_ctx):
    _, _, name = repository_ctx.name.rpartition("+")
    lfs_smudge(repository_ctx, [_get_dep(repository_ctx, name + ".jar")])

    # for some reason rules_kotlin warns about these jars missing, this is to silence those warnings
    repository_ctx.file("empty.zip", _empty_zip)
    for jar in (
        "annotations-13.0.jar",
        "kotlin-stdlib.jar",
        "kotlin-reflect.jar",
        "kotlin-script-runtime.jar",
        "trove4j.jar",
    ):
        repository_ctx.symlink("empty.zip", jar)
    repository_ctx.file("BUILD.bazel", _kotlin_dep_build.format(name = name))

_kotlin_dep = repository_rule(
    implementation = _kotlin_dep_impl,
)

def _walk(dir):
    res = []
    next_dirs = [dir]

    # loops must be bounded in starlark
    for i in range(100):
        current_dirs = next_dirs
        next_dirs = []
        for d in current_dirs:
            children = d.readdir()
            next_dirs.extend([c for c in children if c.is_dir])
            res.extend([c for c in children if not c.is_dir])
        if not next_dirs:
            return res
    fail("%s directory too deep" % dir)

def _embeddable_source_impl(repository_ctx):
    src_dir = repository_ctx.path(Label("//java/kotlin-extractor:src"))
    repository_ctx.watch_tree(src_dir)
    for src in _walk(src_dir):
        contents = repository_ctx.read(src)
        contents = contents.replace(
            "import com.intellij",
            "import org.jetbrains.kotlin.com.intellij",
        )
        repository_ctx.file(str(src).replace(str(src_dir), "src"), contents)
    repository_ctx.symlink(
        Label("//java/kotlin-extractor:BUILD.bazel"),
        "BUILD.bazel",
    )

_embeddable_source = repository_rule(implementation = _embeddable_source_impl)

def _get_version(repository_ctx, available = []):
    default_version = repository_ctx.getenv("CODEQL_KOTLIN_SINGLE_VERSION")
    if default_version:
        return default_version
    repository_ctx.watch(Label("//java/kotlin-extractor:dev/.kotlinc_version"))
    version_picker = repository_ctx.path(Label("//java/kotlin-extractor:pick-kotlin-version.py"))
    python = repository_ctx.which("python3") or repository_ctx.which("python")

    # use the kotlinc wrapper as fallback
    path = repository_ctx.getenv("PATH")
    path_to_add = repository_ctx.path(Label("//java/kotlin-extractor:dev"))
    if not path:
        path = str(path_to_add)
    elif repository_ctx.os.name == "windows":
        path = "%s;%s" % (path, path_to_add)
    else:
        path = "%s:%s" % (path, path_to_add)
    res = repository_ctx.execute([python, version_picker] + available, environment = {"PATH": path})
    if res.return_code != 0:
        fail(res.stderr)
    return res.stdout.strip()

def _defaults_impl(repository_ctx):
    default_version = _get_version(repository_ctx)
    default_variant = "standalone"
    if repository_ctx.getenv("CODEQL_KOTLIN_SINGLE_VERSION_EMBEDDABLE") in ("true", "1"):
        default_variant = "embeddable"
    available_version = _get_version(repository_ctx, VERSIONS)
    info = struct(
        version = default_version,
        variant = default_variant,
        extractor_version = available_version,
    )
    repository_ctx.file(
        "defaults.bzl",
        "kotlin_extractor_defaults = %s\n" % repr(info),
    )
    repository_ctx.file("BUILD.bazel")

_defaults = repository_rule(implementation = _defaults_impl)

def _kotlin_deps_impl(module_ctx):
    for v in VERSIONS:
        for lib in ("compiler", "compiler-embeddable", "stdlib"):
            _kotlin_dep(name = "kotlin-%s-%s" % (lib, v))
    _embeddable_source(name = "codeql_kotlin_embeddable")
    _defaults(name = "codeql_kotlin_defaults")
    return module_ctx.extension_metadata(
        root_module_direct_deps = "all",
        root_module_direct_dev_deps = [],
    )

kotlin_extractor_deps = module_extension(implementation = _kotlin_deps_impl)