refactor: pip-compile requirements for tools/publish

Author: aignas

MODULE.bazel

@@ -59,8 +59,8 @@ pip.parse(
     hub_name = "rules_python_publish_deps",
     python_version = "3.11",
     requirements_by_platform = {
-        "//tools/publish:requirements.txt": "linux_*",
         "//tools/publish:requirements_darwin.txt": "osx_*",
+        "//tools/publish:requirements_linux.txt": "linux_*",
         "//tools/publish:requirements_windows.txt": "windows_*",
     },
 )

examples/bzlmod/MODULE.bazel.lock

@@ -0,0 +1,15 @@
+load("@rules_multirun//:defs.bzl", "multirun")
+
+multirun(
+    name = "requirements_publish.update",
+    commands = [
+        "//tools/publish:{}.update".format(r)
+        for r in [
+            "requirements_universal",
+            "requirements",
+            "requirements_darwin",
+            "requirements_windows",
+            "requirements_linux",
+        ]
+    ],
+)

tools/private/BUILD.bazel

@@ -0,0 +1,29 @@
+# Copyright 2024 The Bazel Authors. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""A simple macro to lock the requirements for twine
+"""
+
+load("//python/uv/private:lock.bzl", "lock")  # buildifier: disable=bzl-visibility
+
+def publish_deps(*, name, outs, **kwargs):
+    """Generate all of the requirements files for all platforms."""
+    for out, platform in outs.items():
+        lock(
+            name = out.replace(".txt", ""),
+            out = out,
+            universal = platform == "",
+            args = [] if not platform else ["--python-platform=" + platform],
+            **kwargs
+        )

tools/private/publish_deps.bzl

@@ -1,13 +1,6 @@
-load("//python:pip.bzl", "compile_pip_requirements")
 load("//python/config_settings:transition.bzl", "py_binary")
 load("//python/entry_points:py_console_script_binary.bzl", "py_console_script_binary")
-
-compile_pip_requirements(
-    name = "requirements",
-    src = "requirements.in",
-    requirements_darwin = "requirements_darwin.txt",
-    requirements_windows = "requirements_windows.txt",
-)
+load("//tools/private:publish_deps.bzl", "publish_deps")
 
 py_console_script_binary(
     name = "twine",
@@ -28,7 +21,24 @@ filegroup(
         "BUILD.bazel",
         "requirements.txt",
         "requirements_darwin.txt",
+        "requirements_linux.txt",
+        "requirements_universal.txt",
         "requirements_windows.txt",
     ],
-    visibility = ["//tools:__pkg__"],
+    visibility = ["//tools:__subpackages__"],
+)
+
+# Run bazel run //tools/private:requirements_publish.update to update the outs
+publish_deps(
+    name = "requirements",
+    srcs = ["requirements.in"],
+    outs = {
+        "requirements.txt": "linux",  # TODO: maybe deprecate
+        "requirements_darwin.txt": "macos",
+        "requirements_linux.txt": "linux",
+        "requirements_universal.txt": "",  # universal
+        "requirements_windows.txt": "windows",
+    },
+    upgrade = False,
+    visibility = ["//tools:__subpackages__"],
 )