wip

Author: aignas

python/uv/private/BUILD.bazel

@@ -14,6 +14,7 @@
 
 load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
 load("//python:py_binary.bzl", "py_binary")
+load("//python/private:bzlmod_enabled.bzl", "BZLMOD_ENABLED")  # buildifier: disable=bzl-visibility
 
 exports_files(
     srcs = [
@@ -110,3 +111,12 @@ py_binary(
     visibility = ["//visibility:public"],
     deps = ["//python/runfiles"],
 )
+
+filegroup(
+    name = "pip_compile_template",
+    srcs = select({
+        "@platforms//os:windows": ["//python:none"],
+        "//conditions:default": ["pip_compile.sh"],
+    }),
+    target_compatible_with = [] if BZLMOD_ENABLED else ["@platforms//:incompatible"],
+)

python/uv/private/lock.bzl

@@ -18,12 +18,9 @@
 load("@bazel_skylib//rules:diff_test.bzl", "diff_test")
 load("@bazel_skylib//rules:expand_template.bzl", "expand_template")
 load("//python:py_binary.bzl", "py_binary")
-load("//python/private:bzlmod_enabled.bzl", "BZLMOD_ENABLED")  # buildifier: disable=bzl-visibility
 
 visibility(["//..."])
 
-_REQUIREMENTS_TARGET_COMPATIBLE_WITH = [] if BZLMOD_ENABLED else ["@platforms//:incompatible"]
-
 def _impl(ctx):
     args = ctx.actions.args()
 
@@ -67,6 +64,85 @@ _lock = rule(
     },
 )
 
+_uv_toolchain = Label("//python/uv:uv_toolchain_type")
+_py_toolchain = Label("//python:toolchain_type")
+
+_LockInfo = provider(
+    doc = "",
+    fields = {
+        "args": "",
+        "py": "",
+        "srcs": "",
+        "template": "",
+        "uv": "",
+    }
+)
+
+def _impl2(ctx):
+    args = ctx.attr.args
+    srcs = ctx.attr.srcs
+
+    toolchain_info = ctx.toolchains[_uv_toolchain]
+    uv = toolchain_info.uv_toolchain_info.uv[DefaultInfo].files_to_run.executable
+
+    py_runtime = ctx.toolchains[_py_toolchain]
+    fail(py_runtime)
+    py = ""
+
+    return [
+        DefaultInfo(files = depset([ctx.outputs.out])),
+        _LockInfo(
+            args = args,
+            srcs = srcs,
+            uv = uv,
+            py = py,
+            template = ctx.attr._template,
+        ),
+    ]
+
+_lock2 = rule(
+    implementation = _impl2,
+    doc = """\
+""",
+    attrs = {
+        "args": attr.string_list(),
+        "env": attr.string_dict(),
+        "existing_output": attr.label(
+            mandatory = False, allow_single_file = True,
+            doc = "An already existing output file that is used as a basis for futher modifications and the locking is not done from scratch",
+        ),
+        "output": attr.output(mandatory = False),
+        "python_version": attr.string(doc = "TODO: how do I create a transition thing?"),
+        "srcs": attr.label_list(mandatory = True, allow_files = True),
+        "_template": attr.label(default = "//python/uv/private:pip_compile_template"),
+    },
+    toolchains = [
+        _uv_toolchain,
+        _py_toolchain,
+    ],
+)
+
+def _impl3(ctx):
+    fail()
+
+_lock_run = rule(
+    implementation = _impl3,
+    doc = """\
+""",
+    attrs = {
+        "lock": attr.label(
+            doc = "",
+            providers = [_LockInfo],
+        ),
+        "out": attr.string(),
+    },
+    toolchains = [
+        _uv_toolchain,
+        _py_toolchain,
+    ],
+)
+
+
 def _maybe_file(path):
     """A small function to return a list of existing outputs.
 
@@ -161,22 +237,20 @@ def lock(*, name, srcs, out, args = [], **kwargs):
         deps = ["//python/runfiles"],
     )
 
-    _lock(
+    _lock2(
         name = name,
         srcs = srcs,
         # Check if the output file already exists, if yes, first copy it to the
         # output file location in order to make `uv` not change the requirements if
         # we are just running the command.
-        maybe_out = maybe_out,
-        out = out + ".new",
+        existing_output = maybe_out,
+        output = out + ".new",
         tags = [
             "local",
             "manual",
             "no-cache",
         ],
         args = args,
-        target_compatible_with = _REQUIREMENTS_TARGET_COMPATIBLE_WITH,
-        cmd = locker_target,
     )
 
     if maybe_out:

python/uv/private/pip_compile.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+set -euxo pipefail
+
+exec uv --output-file "$@"