Changes needed for rules_js bzlmod support

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

@@ -7,14 +7,24 @@ load(
  "register_jq_toolchains",
  "register_yq_toolchains",
 )
+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)
 
+ 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
  )