mirror of
https://github.com/NoRedInk/noredink-ui.git
synced 2024-12-02 11:02:56 +03:00
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
||
|
|
#
|
||
|
|
# This source code is licensed under both the MIT license found in the
|
||
|
|
# LICENSE-MIT file in the root directory of this source tree and the Apache
|
||
|
|
# License, Version 2.0 found in the LICENSE-APACHE file in the root directory
|
||
|
|
# of this source tree.
|
||
|
|
|
||
|
|
load("@prelude//utils:utils.bzl", "expect")
|
||
|
|
|
||
|
|
WorkerToolInfo = provider(
|
||
|
|
fields = [
|
||
|
|
"command", # "cmd_args"
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
def worker_tool(ctx: "context") -> ["provider"]:
|
||
|
|
"""
|
||
|
|
worker_tool() rule implementation
|
||
|
|
|
||
|
|
Args:
|
||
|
|
ctx: rule analysis context
|
||
|
|
Returns:
|
||
|
|
list of created providers (DefaultInfo with an empty output and TemplatePlaceholderInfo with $(worker) macro key)
|
||
|
|
"""
|
||
|
|
|
||
|
|
executable = ctx.attrs.exe
|
||
|
|
worker_tool_run_info = executable[RunInfo]
|
||
|
|
expect(worker_tool_run_info != None, "Worker tool executable must have a RunInfo!")
|
||
|
|
|
||
|
|
worker_tool_runner = ctx.attrs._worker_tool_runner[RunInfo]
|
||
|
|
worker_tool_cmd = cmd_args(worker_tool_runner)
|
||
|
|
worker_tool_cmd.add("--worker-tool")
|
||
|
|
worker_tool_cmd.add(worker_tool_run_info)
|
||
|
|
|
||
|
|
worker_args = ctx.attrs.args
|
||
|
|
if worker_args:
|
||
|
|
worker_args_file, _ = ctx.actions.write(
|
||
|
|
"worker_tool_args",
|
||
|
|
worker_args,
|
||
|
|
allow_args = True,
|
||
|
|
)
|
||
|
|
|
||
|
|
worker_tool_cmd.add("--worker-args-file")
|
||
|
|
worker_tool_cmd.add(worker_args_file)
|
||
|
|
|
||
|
|
worker_env = ctx.attrs.env
|
||
|
|
if worker_env:
|
||
|
|
env_args = []
|
||
|
|
for key, value in worker_env.items():
|
||
|
|
env_args.append(key)
|
||
|
|
env_args.append(value)
|
||
|
|
|
||
|
|
env_args_file, _ = ctx.actions.write(
|
||
|
|
"worker_tool_envs",
|
||
|
|
env_args,
|
||
|
|
allow_args = True,
|
||
|
|
)
|
||
|
|
|
||
|
|
worker_tool_cmd.add("--worker-env-file")
|
||
|
|
worker_tool_cmd.add(env_args_file)
|
||
|
|
|
||
|
|
return [
|
||
|
|
DefaultInfo(),
|
||
|
|
TemplatePlaceholderInfo(keyed_variables = {
|
||
|
|
"worker": worker_tool_cmd,
|
||
|
|
}),
|
||
|
|
WorkerToolInfo(
|
||
|
|
command = worker_tool_cmd,
|
||
|
|
),
|
||
|
|
]
|