119 lines
4.2 KiB
Python
119 lines
4.2 KiB
Python
load("//bazel:utils.bzl", "generate_noop_toolchain", "get_toolchain_subs", "retry_download_and_extract")
|
|
load("//bazel/toolchains/cc/mongo_linux:mongo_toolchain_version.bzl", "TOOLCHAIN_MAP")
|
|
load("//bazel/toolchains/cc/mongo_linux:mongo_mold.bzl", "MOLD_MAP")
|
|
|
|
def _toolchain_download(ctx):
|
|
distro, arch, substitutions = get_toolchain_subs(ctx)
|
|
|
|
toolchain_key = "{distro}_{arch}".format(distro = distro, arch = arch)
|
|
|
|
if toolchain_key in TOOLCHAIN_MAP[ctx.attr.version]:
|
|
toolchain_info = TOOLCHAIN_MAP[ctx.attr.version][toolchain_key]
|
|
urls = toolchain_info["url"]
|
|
sha = toolchain_info["sha"]
|
|
|
|
ctx.report_progress("downloading {} mongo toolchain {}".format(toolchain_key, urls))
|
|
print("downloading {} mongo toolchain {}".format(toolchain_key, urls))
|
|
retry_download_and_extract(
|
|
ctx = ctx,
|
|
tries = 5,
|
|
url = urls,
|
|
sha256 = sha,
|
|
)
|
|
|
|
if arch in MOLD_MAP:
|
|
print("Downloading mold from {}.".format(MOLD_MAP[arch]["url"]))
|
|
retry_download_and_extract(
|
|
ctx = ctx,
|
|
tries = 5,
|
|
url = MOLD_MAP[arch]["url"],
|
|
sha256 = MOLD_MAP[arch]["sha"],
|
|
)
|
|
|
|
ctx.report_progress("generating toolchain " + ctx.attr.version + " build file")
|
|
ctx.template(
|
|
"BUILD.bazel",
|
|
ctx.attr.build_tpl,
|
|
substitutions = substitutions,
|
|
)
|
|
ctx.template(
|
|
"mongo_toolchain_flags.bzl",
|
|
ctx.attr.flags_tpl,
|
|
substitutions = substitutions,
|
|
)
|
|
|
|
else:
|
|
generate_noop_toolchain(ctx, substitutions)
|
|
ctx.report_progress("Mongo toolchain " + ctx.attr.version + " not supported on this platform. Platform key not found: " + toolchain_key)
|
|
|
|
return None
|
|
|
|
toolchain_download = repository_rule(
|
|
implementation = _toolchain_download,
|
|
attrs = {
|
|
"os": attr.string(
|
|
values = ["macos", "linux", "windows"],
|
|
doc = "Host operating system.",
|
|
),
|
|
"arch": attr.string(
|
|
values = ["amd64", "aarch64", "amd64", "x86_64", "ppc64le", "s390x"],
|
|
doc = "Host architecture.",
|
|
),
|
|
"version": attr.string(
|
|
values = ["v5"],
|
|
doc = "Mongodbtoolchain version.",
|
|
mandatory = True,
|
|
),
|
|
"flags_tpl": attr.label(
|
|
doc = "Label denoting the toolchain flags template.",
|
|
mandatory = True,
|
|
),
|
|
"build_tpl": attr.label(
|
|
default = "//bazel/toolchains/cc/mongo_linux:mongo_toolchain.BUILD.tmpl",
|
|
doc = "Label denoting the BUILD file template that gets installed in the repo.",
|
|
),
|
|
},
|
|
)
|
|
|
|
def setup_mongo_toolchains():
|
|
toolchain_download(
|
|
name = "mongo_toolchain_v5",
|
|
version = "v5",
|
|
flags_tpl = "//bazel/toolchains/cc/mongo_linux:mongo_toolchain_flags_v5.bzl",
|
|
)
|
|
|
|
native.register_toolchains(
|
|
"@mongo_toolchain_v5//:all",
|
|
)
|
|
|
|
# Defines aliases for key targets inside the toolchain the user has chosen via
|
|
# //bazel/config:mongo_toolchain_version. For example, clang_tidy is a target inside the
|
|
# toolchains, so this function defines an alias called clang_tidy that points to
|
|
# @mongo_toolchain_vN//:clang_tidy, where N depends on the config value. This lets us use
|
|
# the name //:clang_tidy to refer to the clang_tidy of the configured toolchain.
|
|
def setup_mongo_toolchain_aliases():
|
|
# Map from target's name inside the toolchain to the name we want to alias it to.
|
|
toolchain_targets = {
|
|
"llvm_symbolizer": "llvm_symbolizer",
|
|
"clang_format": "clang_format",
|
|
"clang_tidy": "clang_tidy",
|
|
"mongo_toolchain": "mongo_toolchain",
|
|
"all_files": "toolchain_files",
|
|
}
|
|
selects = {}
|
|
for target in toolchain_targets:
|
|
selects[target] = {}
|
|
|
|
for target in toolchain_targets:
|
|
selects[target]["//bazel/config:mongo_toolchain_v5"] = "@mongo_toolchain_v5//:{}".format(target)
|
|
|
|
for target, local_alias in toolchain_targets.items():
|
|
native.alias(
|
|
name = local_alias,
|
|
actual = select(selects[target]),
|
|
)
|
|
|
|
setup_mongo_toolchains_extension = module_extension(
|
|
implementation = lambda ctx: setup_mongo_toolchains(),
|
|
)
|