mirror of
https://github.com/github/codeql.git
synced 2026-05-05 21:55:19 +02:00
b07fa70133
By adding `java/kotlinc-extractor/deps/dev` to `PATH`, one gets a `kotlinc` wrapper that takes care of downloading and extracting the desired version of `kotlinc` on demand. The desired version can be selected with `kotlinc --select x.y.z`, or left to the current default of `1.9.0`. Moreover, this default version is integrated with the Bazel build, so that when using this wrapper, changes in the selected version will be picked up to define the default single version kotlin extractor build, without needing to do anything else (like `bazel fetch --force` or similar). Selected and installed version data is stored in `.gitignore`d files in the same directory, and can be cleared with `kotlinc --clear`.
133 lines
4.7 KiB
Python
133 lines
4.7 KiB
Python
load("//java/kotlin-extractor:versions.bzl", "VERSIONS", "version_less")
|
|
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:
|
|
break
|
|
return res
|
|
|
|
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_default_version(repository_ctx):
|
|
default_version = repository_ctx.getenv("CODEQL_KOTLIN_SINGLE_VERSION")
|
|
if default_version:
|
|
return default_version
|
|
kotlin_plugin_versions = repository_ctx.path(Label("//java/kotlin-extractor:current_kotlin_version.py"))
|
|
python = repository_ctx.which("python3") or repository_ctx.which("python")
|
|
env = {}
|
|
repository_ctx.watch(Label("//java/kotlin-extractor/deps:dev/.kotlinc_selected_version"))
|
|
if not repository_ctx.which("kotlinc"):
|
|
# take default from the kotlinc wrapper
|
|
path = repository_ctx.getenv("PATH")
|
|
path_to_add = repository_ctx.path(Label("//java/kotlin-extractor/deps: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)
|
|
env["PATH"] = path
|
|
res = repository_ctx.execute([python, kotlin_plugin_versions], environment = env)
|
|
if res.return_code != 0:
|
|
fail(res.stderr)
|
|
return res.stdout.strip()
|
|
|
|
def _get_available_version(version):
|
|
for available_version in reversed(VERSIONS):
|
|
if not version_less(version, available_version):
|
|
return available_version
|
|
fail("no available version found for version %s among:\n %s" % (version, " ".join(VERSIONS)))
|
|
|
|
def _defaults_impl(repository_ctx):
|
|
default_version = _get_default_version(repository_ctx)
|
|
default_variant = "standalone"
|
|
if repository_ctx.getenv("CODEQL_KOTLIN_SINGLE_VERSION_EMBEDDABLE") in ("true", "1"):
|
|
default_variant = "embeddable"
|
|
available_version = _get_available_version(default_version)
|
|
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)
|