128 lines
4.6 KiB
Python
128 lines
4.6 KiB
Python
"Rules for working with transitions."
|
|
|
|
load("@bazel_skylib//lib:paths.bzl", "paths")
|
|
|
|
def _transition_platform_impl(_, attr):
|
|
return {"//command_line_option:platforms": str(attr.target_platform)}
|
|
|
|
# Transition from any input configuration to one that includes the
|
|
# --platforms command-line flag.
|
|
_transition_platform = transition(
|
|
implementation = _transition_platform_impl,
|
|
inputs = [],
|
|
outputs = ["//command_line_option:platforms"],
|
|
)
|
|
|
|
def _platform_transition_filegroup_impl(ctx):
|
|
files = []
|
|
runfiles = ctx.runfiles()
|
|
for src in ctx.attr.srcs:
|
|
files.append(src[DefaultInfo].files)
|
|
|
|
runfiles = runfiles.merge_all([src[DefaultInfo].default_runfiles for src in ctx.attr.srcs])
|
|
return [DefaultInfo(
|
|
files = depset(transitive = files),
|
|
runfiles = runfiles,
|
|
)]
|
|
|
|
platform_transition_filegroup = rule(
|
|
_platform_transition_filegroup_impl,
|
|
attrs = {
|
|
# Required to Opt-in to the transitions feature.
|
|
"_allowlist_function_transition": attr.label(
|
|
default = "@bazel_tools//tools/allowlists/function_transition_allowlist",
|
|
),
|
|
"target_platform": attr.label(
|
|
doc = "The target platform to transition the srcs.",
|
|
mandatory = True,
|
|
),
|
|
"srcs": attr.label_list(
|
|
allow_empty = False,
|
|
cfg = _transition_platform,
|
|
doc = "The input to be transitioned to the target platform.",
|
|
),
|
|
},
|
|
doc = "Transitions the srcs to use the provided platform. The filegroup will contain artifacts for the target platform.",
|
|
)
|
|
|
|
def _platform_transition_binary_impl(ctx):
|
|
# We need to forward the DefaultInfo provider from the underlying rule.
|
|
# Unfortunately, we can't do this directly, because Bazel requires that the executable to run
|
|
# is actually generated by this rule, so we need to symlink to it, and generate a synthetic
|
|
# forwarding DefaultInfo.
|
|
|
|
result = []
|
|
|
|
# ctx.attr.binary is a singleton list if this rule uses an outgoing transition.
|
|
if type(ctx.attr.binary) == type([]):
|
|
binary = ctx.attr.binary[0]
|
|
else:
|
|
binary = ctx.attr.binary
|
|
|
|
default_info = binary[DefaultInfo]
|
|
files = default_info.files
|
|
new_executable = None
|
|
original_executable = default_info.files_to_run.executable
|
|
runfiles = default_info.default_runfiles
|
|
|
|
if not original_executable:
|
|
fail("Cannot transition a 'binary' that is not executable")
|
|
|
|
new_executable_name = ctx.attr.basename if ctx.attr.basename else original_executable.basename
|
|
|
|
# In order for the symlink to have the same basename as the original
|
|
# executable (important in the case of proto plugins), put it in a
|
|
# subdirectory named after the label to prevent collisions.
|
|
new_executable = ctx.actions.declare_file(paths.join(ctx.label.name, new_executable_name))
|
|
ctx.actions.symlink(
|
|
output = new_executable,
|
|
target_file = original_executable,
|
|
is_executable = True,
|
|
)
|
|
files = depset(direct = [new_executable], transitive = [files])
|
|
runfiles = runfiles.merge(ctx.runfiles([new_executable]))
|
|
|
|
result.append(
|
|
DefaultInfo(
|
|
files = files,
|
|
runfiles = runfiles,
|
|
executable = new_executable,
|
|
),
|
|
)
|
|
|
|
return result
|
|
|
|
def _get_platform_transition_attrs(binary_cfg):
|
|
return {
|
|
"basename": attr.string(),
|
|
"binary": attr.label(allow_files = True, cfg = binary_cfg),
|
|
"target_platform": attr.label(
|
|
doc = "The target platform to transition the binary.",
|
|
mandatory = True,
|
|
),
|
|
"_allowlist_function_transition": attr.label(
|
|
default = "@bazel_tools//tools/allowlists/function_transition_allowlist",
|
|
),
|
|
}
|
|
|
|
platform_transition_binary = rule(
|
|
implementation = _platform_transition_binary_impl,
|
|
# Use an outgoing transition since the target platform of the
|
|
# platform_transition_binary doesn't matter and it results in a more
|
|
# intuitive output path (matching an untransitioned binary).
|
|
attrs = _get_platform_transition_attrs(binary_cfg = _transition_platform),
|
|
executable = True,
|
|
doc = "Transitions the binary to use the provided platform.",
|
|
)
|
|
|
|
platform_transition_test = rule(
|
|
implementation = _platform_transition_binary_impl,
|
|
attrs = _get_platform_transition_attrs(binary_cfg = "target"),
|
|
# Use an incoming transition since the target platform of the
|
|
# platform_transition_test does matter for the exec platform resolution of
|
|
# the test action.
|
|
cfg = _transition_platform,
|
|
test = True,
|
|
doc = "Transitions the test to use the provided platform.",
|
|
)
|