mirror of
https://github.com/carbon-language/carbon-lang.git
synced 2026-09-24 21:40:12 +01:00
Jon Ross-Perkins
5e293ad97f
SCRIPT_LOCATION contains `bazel-out`, so this drops the portions shared between the script and tool locations before removing the suffix.
68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
# Part of the Carbon Language project, under the Apache License v2.0 with LLVM
|
|
# Exceptions. See /LICENSE for license information.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
"""Supports running a tool from the install filegroup."""
|
|
|
|
_RUN_TOOL_TMPL = """#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
|
|
# These will be relative locations in bazel-out.
|
|
_SCRIPT_LOCATION = "{0}"
|
|
_TOOL_LOCATION = "{1}"
|
|
|
|
# Drop shared parent portions so that we're working with a minimum suffix.
|
|
# This is so that both `bazel-out` and manual `bazel-bin` invocations work.
|
|
script_parts = _SCRIPT_LOCATION.split('/')
|
|
tool_parts = _TOOL_LOCATION.split('/')
|
|
while script_parts and tool_parts and script_parts[0] == tool_parts[0]:
|
|
del script_parts[0]
|
|
del tool_parts[0]
|
|
script_suffix = '/' + '/'.join(script_parts)
|
|
tool_suffix = '/' + '/'.join(tool_parts)
|
|
|
|
# Make sure we have the expected structure.
|
|
if not __file__.endswith(script_suffix):
|
|
exit(
|
|
"Unable to figure out path:\\n"
|
|
f" __file__: {{__file__}}\\n"
|
|
f" script: {{script_suffix}}\\n"
|
|
)
|
|
|
|
# Run the tool using the absolute path, forwarding arguments.
|
|
tool_path = __file__.removesuffix(script_suffix) + tool_suffix
|
|
os.execv(tool_path, [tool_path] + sys.argv[1:])
|
|
"""
|
|
|
|
def _run_tool_impl(ctx):
|
|
content = _RUN_TOOL_TMPL.format(ctx.outputs.executable.path, ctx.file.tool.path)
|
|
ctx.actions.write(
|
|
output = ctx.outputs.executable,
|
|
is_executable = True,
|
|
content = content,
|
|
)
|
|
return [
|
|
DefaultInfo(
|
|
runfiles = ctx.runfiles(files = ctx.files.data),
|
|
),
|
|
RunEnvironmentInfo(environment = ctx.attr.env),
|
|
]
|
|
|
|
run_tool = rule(
|
|
doc = "Helper for running a tool in a filegroup.",
|
|
implementation = _run_tool_impl,
|
|
attrs = {
|
|
"data": attr.label_list(allow_files = True),
|
|
"env": attr.string_dict(),
|
|
"tool": attr.label(
|
|
allow_single_file = True,
|
|
executable = True,
|
|
cfg = "target",
|
|
mandatory = True,
|
|
),
|
|
},
|
|
executable = True,
|
|
)
|