propagate env

Author: aignas

python/uv/private/lock.bzl

@@ -55,6 +55,7 @@ _RunLockInfo = provider(
     doc = "",
     fields = {
         "args": "",
+        "env": "",
         "py": "",
         "srcs": "",
         "template": "",
@@ -77,7 +78,7 @@ def _lock_impl(ctx):
     ctx.actions.expand_template(
         template = ctx.files._template[0],
         substitutions = {
-            "\"{{args}}\"": " ",
+            '"{{args}}"': "",
             "{{bazel_out}}": output.path,
             "{{src_out}}": existing_output[0].path,
             "{{uv}}": uv.path,
@@ -132,6 +133,7 @@ def _lock_impl(ctx):
         DefaultInfo(files = depset([output])),
         _RunLockInfo(
             args = run_args + args,
+            env = ctx.attr.env,
             srcs = srcs,
             uv = uv,
             py = py_runtime,
@@ -145,10 +147,12 @@ _lock = rule(
 """,
     attrs = {
         "args": attr.string_list(
-            doc = "",
+            doc = "Args to pass to `uv`.",
         ),
         "env": attr.string_dict(
-            doc = "",
+            doc = """\
+The env variables that are set when executing `uv`.
+""",
         ),
         "existing_output": attr.label(
             mandatory = False,
@@ -221,7 +225,7 @@ def _lock_run_impl(ctx):
     ctx.actions.expand_template(
         template = info.template,
         substitutions = {
-            "\"{{args}}\"": " ".join([shell.quote(a) for a in args]),
+            '"{{args}}"': " ".join([shell.quote(a) for a in args]),
             "{{src_out}}": "{}/{}".format(ctx.label.package, ctx.attr.output),
             "{{uv}}": uv.short_path,
         },
@@ -237,11 +241,15 @@ def _lock_run_impl(ctx):
             ],
         ),
     )
+    run_environment_info = RunEnvironmentInfo(
+        environment = info.env,
+    )
     return [
         DefaultInfo(
             executable = executable,
             runfiles = runfiles,
         ),
+        run_environment_info,
     ]
 
 _lock_run = rule(
@@ -342,8 +350,6 @@ def lock(*, name, srcs, out, args = [], env = None, **kwargs):
         name = locker_target,
         lock = name,
         output = out,
-        # TODO @aignas 2025-03-13: allow customizing the env
-        # env = env,
         tags = ["manual"],
     )