feat(uv): Create a UV driver

Author: arrdem

.github/workflows/ci.yaml

@@ -33,6 +33,7 @@ jobs:
           "e2e/repository-rule-deps",
           "e2e/smoke",
           "e2e/system-interpreter",
+          "e2e/uv-deps",
           "examples/uv_pip_compile"
         ]
       # TODO: Build Windows tools and add to toolchain

Cargo.lock

@@ -15,8 +15,9 @@ bazel_dep(name = "aspect_bazel_lib", version = "2.16.0")  # TODO(alexeagle): rem
 bazel_dep(name = "aspect_tools_telemetry", version = "0.2.8")
 bazel_dep(name = "bazel_lib", version = "3.0.0-rc.0")
 bazel_dep(name = "bazel_skylib", version = "1.4.2")
-bazel_dep(name = "rules_python", version = "0.29.0")
-bazel_dep(name = "platforms", version = "0.0.7")
+bazel_dep(name = "rules_python", version = "1.6.3")
+bazel_dep(name = "rules_python_gazelle_plugin", version = "1.6.3")
+bazel_dep(name = "platforms", version = "1.0.0")
 
 bazel_lib = use_extension("@aspect_bazel_lib//lib:extensions.bzl", "toolchains")
 bazel_lib.expand_template()
@@ -28,7 +29,7 @@ use_repo(tel, "aspect_tools_telemetry_report")
 python = use_extension("@rules_python//python/extensions:python.bzl", "python", dev_dependency = True)
 python.toolchain(
     is_default = False,
-    python_version = "3.9",
+    python_version = "3.13",
 )
 
 tools = use_extension("//py:extensions.bzl", "py_tools")
@@ -115,3 +116,29 @@ oci.pull(
     tag = "latest",
 )
 use_repo(oci, "ubuntu", "ubuntu_linux_amd64", "ubuntu_linux_arm64_v8")
+
+# This is what `rules_uv` really boils down to, so vendoring that
+bazel_dep(name = "rules_multitool", version = "0.11.0")
+
+multitool = use_extension("@rules_multitool//multitool:extension.bzl", "multitool")
+multitool.hub(lockfile = "//uv/private/uv:uv.lock.json")
+use_repo(multitool, "multitool")
+
+# Used as part of the sdist flow
+bazel_dep(name = "tar.bzl", version = "0.6.0")
+
+# The toml tool that pip2 uses
+tomltool = use_extension("//uv/private/tomltool:extension.bzl", "tomltool")
+use_repo(
+    tomltool,
+    "aspect_rules_py_tomltool_aarch64_linux_gnu",
+    "aspect_rules_py_tomltool_aarch64_linux_musl",
+    "aspect_rules_py_tomltool_aarch64_osx_libsystem",
+    "aspect_rules_py_tomltool_x86_64_linux_gnu",
+    "aspect_rules_py_tomltool_x86_64_linux_musl",
+    "aspect_rules_py_tomltool_x86_64_osx_libsystem",
+)
+
+# Mandatory pip2 host configuration
+host = use_extension("//uv/private/host:extension.bzl", "host_platform")
+use_repo(host, "aspect_rules_py_pip_host")

MODULE.bazel

@@ -37,8 +37,13 @@ stardoc_with_diff_test(
 )
 
 stardoc_with_diff_test(
-    name = "venv",
-    bzl_library_target = "//py/private/py_venv:py_venv",
+    name = "unstable",
+    bzl_library_target = "//py/unstable:defs",
+)
+
+stardoc_with_diff_test(
+    name = "uv",
+    bzl_library_target = "//uv/private:extension",
 )
 
 update_docs(name = "update")

docs/BUILD.bazel

@@ -0,0 +1,54 @@
+"Bazel dependencies"
+
+bazel_dep(name = "aspect_rules_py", version = "0.0.0", dev_dependency = True)
+local_path_override(
+    module_name = "aspect_rules_py",
+    path = "../..",
+)
+
+# Because we use a prerelease of rules_py, we must compile the rust tools from source.
+bazel_dep(name = "rules_rust", version = "0.53.0")
+
+rust = use_extension("@rules_rust//rust:extensions.bzl", "rust")
+rust.toolchain(
+    edition = "2021",
+    versions = ["1.81.0"],
+)
+use_repo(rust, "rust_toolchains")
+
+register_toolchains("@rust_toolchains//:all")
+
+bazel_dep(name = "rules_python", version = "0.29.0", dev_dependency = True)
+
+python = use_extension("@rules_python//python/extensions:python.bzl", "python")
+python.toolchain(
+    configure_coverage_tool = True,
+    python_version = "3.13",
+)
+python.toolchain(
+    configure_coverage_tool = True,
+    python_version = "3.11",
+)
+
+# Testing
+uv = use_extension("@aspect_rules_py//uv:extension.bzl", "uv")
+uv.declare_hub(hub_name = "pip")
+uv.declare_venv(
+    hub_name = "pip",
+    venv_name = "default",
+)
+uv.lockfile(
+    hub_name = "pip",
+    lockfile = "//:uv-default.lock",
+    venv_name = "default",
+)
+uv.declare_venv(
+    hub_name = "pip",
+    venv_name = "airflow",
+)
+uv.lockfile(
+    hub_name = "pip",
+    lockfile = "//:uv-airflow.lock",
+    venv_name = "airflow",
+)
+use_repo(uv, "pip")