feat: allow additional common args to be passed to update_docs

e2e/bzlmod/BUILD.bazel

@@ -12,6 +12,7 @@ load("@aspect_bazel_lib//lib:diff_test.bzl", "diff_test")
 load("@aspect_bazel_lib//lib:jq.bzl", "jq")
 load("@aspect_bazel_lib//lib:yq.bzl", "yq")
 load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
+load("@aspect_bazel_lib_host//:defs.bzl", "host")
 
 bzl_library(
     name = "defs",
@@ -86,3 +87,9 @@ diff_test(
     # Source directories are not support on remote execution.
     tags = ["no-remote-exec"],
 )
+
+genrule(
+    name = "host_bazel_version",
+    outs = ["host_bazel_version.txt"],
+    cmd = "echo '%s' > $@" % host.bazel_version,
+)

e2e/bzlmod/MODULE.bazel

@@ -12,3 +12,9 @@ local_path_override(
     module_name = "aspect_bazel_lib",
     path = "../..",
 )
+
+ext = use_extension("@aspect_bazel_lib//lib:extensions.bzl", "ext")
+
+ext.host()
+
+use_repo(ext, "aspect_bazel_lib_host")

lib/extensions.bzl

@@ -9,14 +9,22 @@ load(
 )
 load("//lib/private:host_repo.bzl", "host_repo")
 
-def _toolchain_extension(_):
+def _toolchain_extension(mctx):
     register_copy_directory_toolchains(register = False)
     register_copy_to_directory_toolchains(register = False)
     register_jq_toolchains(register = False)
     register_yq_toolchains(register = False)
-    host_repo(name = "aspect_bazel_lib_host")
+
+    create_host_repo = False
+    for module in mctx.modules:
+        if len(module.tags.host) > 0:
+            create_host_repo = True
+
+    if create_host_repo:
+        host_repo(name = "aspect_bazel_lib_host")
 
 # TODO: some way for users to control repo name/version of the tools installed
 ext = module_extension(
     implementation = _toolchain_extension,
+    tag_classes = {"host": tag_class(attrs = {})},
 )

lib/private/docs.bzl

@@ -27,7 +27,7 @@ def stardoc_with_diff_test(
         **kwargs
     )
 
-def update_docs(name = "update"):
+def update_docs(name = "update", **kwargs):
     """Stamps an executable run for writing all stardocs declared with stardoc_with_diff_test to the source tree.
 
     This is to be used in tandem with `stardoc_with_diff_test()` to produce a convenient workflow
@@ -45,6 +45,7 @@ def update_docs(name = "update"):
 
     Args:
         name: the name of executable target
+        **kwargs: Other common named parameters such as `tags` or `visibility`
     """
 
     update_files = {}
@@ -60,4 +61,5 @@ def update_docs(name = "update"):
     write_source_files(
         name = name,
         files = update_files,
+        **kwargs
     )