Expose bindgen toolchain through bzlmod

Author: jwnrt

MODULE.bazel

@@ -174,8 +174,11 @@ register_toolchains(
     "//proto/prost:default_prost_toolchain",
 )
 
+bindgen = use_extension("//bindgen:extension.bzl", "bindgen")
+use_repo(bindgen, "bindgen_toolchains")
+
 register_toolchains(
-    "//bindgen:default_bindgen_toolchain",
+    "@bindgen_toolchains//:default_bindgen_toolchain",
 )
 
 rust_host_tools = use_extension("//rust:extensions.bzl", "rust_host_tools")

bindgen/BUILD.bazel

@@ -1,5 +1,4 @@
 load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
-load("//bindgen:defs.bzl", "rust_bindgen_toolchain")
 
 package(default_visibility = ["//visibility:public"])
 
@@ -11,21 +10,18 @@ bzl_library(
     name = "bzl_lib",
     srcs = glob(["**/*.bzl"]),
     deps = [
-        "//bindgen/3rdparty:bzl_lib",
-        "//bindgen/private:bzl_lib",
-        "//rust:bzl_lib",
+        "@rules_rust//bindgen/3rdparty:bzl_lib",
+        "@rules_rust//bindgen/private:bzl_lib",
+        "@rules_rust//rust:bzl_lib",
     ],
 )
 
-rust_bindgen_toolchain(
+alias(
     name = "default_bindgen_toolchain_impl",
-    bindgen = "//bindgen/3rdparty:bindgen",
-    clang = "@llvm-project//clang:clang",
-    libclang = "@llvm-project//clang:libclang",
+    actual = "//bindgen/toolchain:default_bindgen_toolchain_impl",
 )
 
-toolchain(
+alias(
     name = "default_bindgen_toolchain",
-    toolchain = "default_bindgen_toolchain_impl",
-    toolchain_type = "//bindgen:toolchain_type",
+    actual = "//bindgen/toolchain:default_bindgen_toolchain",
 )

bindgen/extension.bzl

@@ -0,0 +1,24 @@
+"""Module extension for accessing a bindgen toolchain"""
+
+load(":repositories.bzl", "rust_bindgen_dependencies")
+load(":transitive_repositories.bzl", "rust_bindgen_transitive_dependencies")
+
+def _bindgen_impl(_):
+    rust_bindgen_transitive_dependencies()
+    rust_bindgen_dependencies()
+    bindgen_toolchains(
+        name = "bindgen_toolchains",
+        build = "//bindgen/toolchain:BUILD.bazel",
+    )
+
+bindgen = module_extension(implementation = _bindgen_impl)
+
+def _bindgen_toolchains_impl(ctx):
+    ctx.file("BUILD.bazel", ctx.read(ctx.attr.build))
+
+bindgen_toolchains = repository_rule(
+    implementation = _bindgen_toolchains_impl,
+    attrs = {
+        "build": attr.label(mandatory = True),
+    },
+)

bindgen/toolchain/BUILD.bazel

@@ -0,0 +1,16 @@
+load("@rules_rust//bindgen:defs.bzl", "rust_bindgen_toolchain")
+
+package(default_visibility = ["//visibility:public"])
+
+rust_bindgen_toolchain(
+    name = "default_bindgen_toolchain_impl",
+    bindgen = "@rules_rust//bindgen/3rdparty:bindgen",
+    clang = "@llvm-project//clang:clang",
+    libclang = "@llvm-project//clang:libclang",
+)
+
+toolchain(
+    name = "default_bindgen_toolchain",
+    toolchain = "default_bindgen_toolchain_impl",
+    toolchain_type = "@rules_rust//bindgen:toolchain_type",
+)

bindgen/transitive_repositories.bzl

@@ -8,7 +8,6 @@ def rust_bindgen_transitive_dependencies():
 
     llvm_configure(
         name = "llvm-project",
-        repo_mapping = {"@llvm_zlib": "@zlib"},
         targets = [
             "AArch64",
             "X86",