From 048f048c75b89d441649a3694a419c90c66782fb Mon Sep 17 00:00:00 2001
From: Ivo List <ilist@google.com>
Date: Thu, 8 Feb 2024 13:33:31 +0100
Subject: [PATCH] [7.1.0] Fixes for experimental extend rule and subrule
 functionality (#21237)

Contains cherry-picks:

d083f3d4b2 Reify outputs of initializer within the package
723a3ac6f2 Allow reading the name of the target in the initializer
48b975ac72 Fix dict types in initializers
c51991600c Make Google proto_library and extension of Bazel
proto_library


Fixes: https://github.com/bazelbuild/bazel/issues/20782
---
 .../starlark/StarlarkRuleClassFunctions.java  |  19 ++-
 .../starlark/StarlarkRuleContext.java         |   4 +-
 .../devtools/build/lib/packages/Type.java     |  17 +-
 .../proto/ProtoCompileActionBuilder.java      |  25 ---
 .../build/lib/rules/proto/ProtoConstants.java |   2 +-
 .../StarlarkRuleFunctionsApi.java             |   4 +-
 .../starlark/builtins_bzl/bazel/exports.bzl   |   2 +
 .../starlark/builtins_bzl/common/exports.bzl  |   2 -
 .../common/proto/proto_library.bzl            |  10 +-
 .../common/proto/proto_semantics.bzl          |  10 --
 .../rules/proto/BazelProtoLibraryTest.java    |   4 +-
 .../StarlarkRuleClassFunctionsTest.java       | 147 +++++++++++++++---
 12 files changed, 173 insertions(+), 73 deletions(-)
 delete mode 100644 src/main/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilder.java

diff --git a/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleClassFunctions.java b/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleClassFunctions.java
index 31488ce1f47a1c..38e73767fd39f9 100644
--- a/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleClassFunctions.java
+++ b/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleClassFunctions.java
@@ -516,7 +516,9 @@ public static StarlarkRuleFunction createRule(
     }
 
     // Verify the child against parent's allowlist
-    if (parent != null && parent.getExtendableAllowlist() != null) {
+    if (parent != null
+        && parent.getExtendableAllowlist() != null
+        && !bzlFile.getRepository().getNameWithAt().equals("@_builtins")) {
       builder.addAllowlistChecker(EXTEND_RULE_ALLOWLIST_CHECKER);
       Attribute.Builder<Label> allowlistAttr =
           attr("$allowlist_extend_rule", LABEL)
@@ -1071,7 +1073,7 @@ public Object call(StarlarkThread thread, Tuple args, Dict<String, Object> kwarg
           // The less magic the better. Do not give in those temptations!
           Dict.Builder<String, Object> initializerKwargs = Dict.builder();
           for (var attr : currentRuleClass.getAttributes()) {
-            if (attr.isPublic() && attr.starlarkDefined()) {
+            if ((attr.isPublic() && attr.starlarkDefined()) || attr.getName().equals("name")) {
               if (kwargs.containsKey(attr.getName())) {
                 Object value = kwargs.get(attr.getName());
                 if (value == Starlark.NONE) {
@@ -1101,6 +1103,12 @@ public Object call(StarlarkThread thread, Tuple args, Dict<String, Object> kwarg
                   : Dict.cast(ret, String.class, Object.class, "rule's initializer return value");
 
           for (var arg : newKwargs.keySet()) {
+            if (arg.equals("name")) {
+              if (!kwargs.get("name").equals(newKwargs.get("name"))) {
+                throw Starlark.errorf("Initializer can't change the name of the target");
+              }
+              continue;
+            }
             checkAttributeName(arg);
             if (arg.startsWith("_")) {
               // allow setting private attributes from initializers in builtins
@@ -1126,8 +1134,11 @@ public Object call(StarlarkThread thread, Tuple args, Dict<String, Object> kwarg
                         currentRuleClass.getName(),
                         attr,
                         value,
-                        // Reify to the location of the initializer definition
-                        currentRuleClass.getLabelConverterForInitializer());
+                        // Reify to the location of the initializer definition (except for outputs)
+                        attr.getType() == BuildType.OUTPUT
+                                || attr.getType() == BuildType.OUTPUT_LIST
+                            ? pkgContext.getBuilder().getLabelConverter()
+                            : currentRuleClass.getLabelConverterForInitializer());
             kwargs.putEntry(nativeName, reifiedValue);
           }
         }
diff --git a/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleContext.java b/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleContext.java
index 3168c814de7431..24ef380e53f78d 100644
--- a/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleContext.java
+++ b/src/main/java/com/google/devtools/build/lib/analysis/starlark/StarlarkRuleContext.java
@@ -465,7 +465,7 @@ public boolean isDefaultExecutableCreated() {
    */
   public void close() {
     // Check super was called
-    if (ruleClassUnderEvaluation.getStarlarkParent() != null && !superCalled) {
+    if (ruleClassUnderEvaluation.getStarlarkParent() != null && !superCalled && !isForAspect()) {
       ruleContext.ruleError("'super' was not called.");
     }
 
@@ -595,7 +595,7 @@ public Object callParent(StarlarkThread thread) throws EvalException, Interrupte
       BuiltinRestriction.failIfCalledOutsideAllowlist(thread, ALLOWLIST_RULE_EXTENSION_API);
     }
     checkMutable("super()");
-    if (aspectDescriptor != null) {
+    if (isForAspect()) {
       throw Starlark.errorf("Can't use 'super' call in an aspect.");
     }
     if (ruleClassUnderEvaluation.getStarlarkParent() == null) {
diff --git a/src/main/java/com/google/devtools/build/lib/packages/Type.java b/src/main/java/com/google/devtools/build/lib/packages/Type.java
index 54577d36082d7e..697376891c9964 100644
--- a/src/main/java/com/google/devtools/build/lib/packages/Type.java
+++ b/src/main/java/com/google/devtools/build/lib/packages/Type.java
@@ -606,9 +606,24 @@ public Map<KeyT, ValueT> convert(Object x, Object what, LabelConverter labelConv
     @Override
     public Object copyAndLiftStarlarkValue(
         Object x, Object what, @Nullable LabelConverter labelConverter) throws ConversionException {
-      return Dict.immutableCopyOf(convert(x, what, labelConverter));
+      if (!(x instanceof Map)) {
+        throw new ConversionException(this, x, what);
+      }
+      Map<?, ?> o = (Map<?, ?>) x;
+      // It's possible that #convert() calls transform non-equal keys into equal ones so we can't
+      // just use ImmutableMap.Builder() here (that throws on collisions).
+      LinkedHashMap<Object, Object> result = new LinkedHashMap<>();
+      for (Map.Entry<?, ?> elem : o.entrySet()) {
+        result.put(
+            keyType.copyAndLiftStarlarkValue(elem.getKey(), "dict key element", labelConverter),
+            valueType.copyAndLiftStarlarkValue(
+                elem.getValue(), "dict value element", labelConverter));
+      }
+      return Dict.immutableCopyOf(result);
     }
 
+
+
     @Override
     public Map<KeyT, ValueT> concat(Iterable<Map<KeyT, ValueT>> iterable) {
       Dict.Builder<KeyT, ValueT> output = new Dict.Builder<>();
diff --git a/src/main/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilder.java b/src/main/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilder.java
deleted file mode 100644
index 0b029d35e9c259..00000000000000
--- a/src/main/java/com/google/devtools/build/lib/rules/proto/ProtoCompileActionBuilder.java
+++ /dev/null
@@ -1,25 +0,0 @@
-// Copyright 2016 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.
-
-package com.google.devtools.build.lib.rules.proto;
-
-import com.google.common.annotations.VisibleForTesting;
-
-/** Constructs actions to run the protocol compiler to generate sources from .proto files. */
-public class ProtoCompileActionBuilder {
-
-  @VisibleForTesting
-  public static final String STRICT_DEPS_FLAG_TEMPLATE =
-      "--direct_dependencies_violation_msg=" + ProtoConstants.STRICT_PROTO_DEPS_VIOLATION_MESSAGE;
-}
diff --git a/src/main/java/com/google/devtools/build/lib/rules/proto/ProtoConstants.java b/src/main/java/com/google/devtools/build/lib/rules/proto/ProtoConstants.java
index 089c9fe3bc9031..65329225c3e7a8 100644
--- a/src/main/java/com/google/devtools/build/lib/rules/proto/ProtoConstants.java
+++ b/src/main/java/com/google/devtools/build/lib/rules/proto/ProtoConstants.java
@@ -43,7 +43,7 @@ public final class ProtoConstants {
    * it with the .proto file that violates strict proto deps.
    */
   static final String STRICT_PROTO_DEPS_VIOLATION_MESSAGE =
-      "%%s is imported, but %1$s doesn't directly depend on a proto_library that 'srcs' it.";
+      "--direct_dependencies_violation_msg=%%s is imported, but %1$s doesn't directly depend on a proto_library that 'srcs' it.";
 
   private ProtoConstants() {}
 }
diff --git a/src/main/java/com/google/devtools/build/lib/starlarkbuildapi/StarlarkRuleFunctionsApi.java b/src/main/java/com/google/devtools/build/lib/starlarkbuildapi/StarlarkRuleFunctionsApi.java
index 1e402377349296..89c18897a2fbbe 100644
--- a/src/main/java/com/google/devtools/build/lib/starlarkbuildapi/StarlarkRuleFunctionsApi.java
+++ b/src/main/java/com/google/devtools/build/lib/starlarkbuildapi/StarlarkRuleFunctionsApi.java
@@ -466,8 +466,8 @@ Object provider(Object doc, Object fields, Object init, StarlarkThread thread)
             doc =
                 "Experimental: the Stalark function initializing the attributes of the rule. "
                     + "<p>The function is called at load time for each instance of the rule. It's "
-                    + "called with values of public attributes defined by the rule (not with "
-                    + "generic attributes, for example <code>name</code> or <code>tags</code>). "
+                    + "called with <code>name</code> and the values of public attributes defined by"
+                    + "the rule (not with generic attributes, for example <code>tags</code>). "
                     + "<p>It has to return a dictionary from the attribute names to the desired "
                     + "values. The attributes that are not returned are unaffected. Returning "
                     + "<code>None</code> as value results in using the default value specified in "
diff --git a/src/main/starlark/builtins_bzl/bazel/exports.bzl b/src/main/starlark/builtins_bzl/bazel/exports.bzl
index 2c11416a181ea0..63d6b5b6e9db38 100644
--- a/src/main/starlark/builtins_bzl/bazel/exports.bzl
+++ b/src/main/starlark/builtins_bzl/bazel/exports.bzl
@@ -19,6 +19,7 @@ load("@_builtins//:common/java/java_import.bzl", "java_import")
 load("@_builtins//:common/java/java_library.bzl", "JAVA_LIBRARY_ATTRS", "bazel_java_library_rule", "java_library", "make_sharded_java_library")
 load("@_builtins//:common/java/java_plugin.bzl", "java_plugin")
 load("@_builtins//:common/java/proto/java_proto_library.bzl", "java_proto_library")
+load("@_builtins//:common/proto/proto_library.bzl", "proto_library")
 load("@_builtins//:common/python/py_binary_macro.bzl", "py_binary")
 load("@_builtins//:common/python/py_internal.bzl", "py_internal")
 load("@_builtins//:common/python/py_library_macro.bzl", "py_library")
@@ -38,6 +39,7 @@ exported_toplevels = {
     "py_internal": py_internal,
 }
 exported_rules = {
+    "proto_library": proto_library,
     "java_library": java_library,
     "java_plugin": java_plugin,
     "java_import": java_import,
diff --git a/src/main/starlark/builtins_bzl/common/exports.bzl b/src/main/starlark/builtins_bzl/common/exports.bzl
index 4b6fa0349005ac..0bc0ce41b4d828 100755
--- a/src/main/starlark/builtins_bzl/common/exports.bzl
+++ b/src/main/starlark/builtins_bzl/common/exports.bzl
@@ -35,7 +35,6 @@ load("@_builtins//:common/objc/objc_library.bzl", "objc_library")
 load("@_builtins//:common/proto/proto_common.bzl", "proto_common_do_not_use")
 load("@_builtins//:common/proto/proto_info.bzl", "ProtoInfo")
 load("@_builtins//:common/proto/proto_lang_toolchain.bzl", "proto_lang_toolchain")
-load("@_builtins//:common/proto/proto_library.bzl", "proto_library")
 load("@_builtins//:common/python/providers.bzl", "PyCcLinkParamsProvider", "PyInfo", "PyRuntimeInfo")
 load("@_builtins//:common/python/py_runtime_macro.bzl", "py_runtime")
 load(":common/java/java_common.bzl", "java_common")
@@ -75,7 +74,6 @@ exported_rules = {
     "objc_import": objc_import,
     "objc_library": objc_library,
     "j2objc_library": j2objc_library,
-    "proto_library": proto_library,
     "cc_shared_library": cc_shared_library,
     "cc_binary": cc_binary,
     "cc_test": cc_test,
diff --git a/src/main/starlark/builtins_bzl/common/proto/proto_library.bzl b/src/main/starlark/builtins_bzl/common/proto/proto_library.bzl
index bb6ac60287947e..022ecdfa372b05 100644
--- a/src/main/starlark/builtins_bzl/common/proto/proto_library.bzl
+++ b/src/main/starlark/builtins_bzl/common/proto/proto_library.bzl
@@ -36,7 +36,7 @@ def _check_srcs_package(target_package, srcs):
 def _get_import_prefix(ctx):
     """Gets and verifies import_prefix attribute if it is declared."""
 
-    import_prefix = ctx.attr.import_prefix if hasattr(ctx.attr, "import_prefix") else ""
+    import_prefix = ctx.attr.import_prefix
 
     if not paths.is_normalized(import_prefix):
         fail("should be normalized (without uplevel references or '.' path segments)", attr = "import_prefix")
@@ -61,8 +61,6 @@ def _get_strip_import_prefix(ctx):
     return strip_import_prefix.removesuffix("/")
 
 def _proto_library_impl(ctx):
-    semantics.preprocess(ctx)
-
     # Verifies attributes.
     _check_srcs_package(ctx.label.package, ctx.attr.srcs)
     srcs = ctx.files.srcs
@@ -247,6 +245,7 @@ proto_library = rule(
             providers = [ProtoInfo],
         ),
         "strip_import_prefix": attr.string(default = "/"),
+        "import_prefix": attr.string(),
         "allow_exports": attr.label(
             cfg = "exec",
             providers = [PackageSpecificationInfo],
@@ -263,9 +262,8 @@ proto_library = rule(
             allow_files = True,
             default = configuration_field("proto", "proto_compiler"),
         ),
-    }) | semantics.EXTRA_ATTRIBUTES,
-    fragments = ["proto"] + semantics.EXTRA_FRAGMENTS,
+    }),
+    fragments = ["proto"],
     provides = [ProtoInfo],
-    exec_groups = semantics.EXEC_GROUPS,
     toolchains = toolchains.use_toolchain(semantics.PROTO_TOOLCHAIN),
 )
diff --git a/src/main/starlark/builtins_bzl/common/proto/proto_semantics.bzl b/src/main/starlark/builtins_bzl/common/proto/proto_semantics.bzl
index d34f5412f99354..80590a3583c7b1 100644
--- a/src/main/starlark/builtins_bzl/common/proto/proto_semantics.bzl
+++ b/src/main/starlark/builtins_bzl/common/proto/proto_semantics.bzl
@@ -16,17 +16,8 @@
 Proto Semantics
 """
 
-def _preprocess(ctx):
-    pass
-
 semantics = struct(
     PROTO_TOOLCHAIN = "@rules_proto//proto:toolchain_type",
-    PROTO_COMPILER_LABEL = "@bazel_tools//tools/proto:protoc",
-    EXTRA_ATTRIBUTES = {
-        "import_prefix": attr.string(),
-    },
-    EXTRA_FRAGMENTS = [],
-    preprocess = _preprocess,
     # This constant is used in ProtoCompileActionBuilder to generate an error message that's
     # displayed when a strict proto deps violation occurs.
     #
@@ -37,6 +28,5 @@ semantics = struct(
         "--direct_dependencies_violation_msg=" +
         "%%s is imported, but %s doesn't directly depend on a proto_library that 'srcs' it."
     ),
-    EXEC_GROUPS = {},
     allowlist_different_package = None,
 )
diff --git a/src/test/java/com/google/devtools/build/lib/rules/proto/BazelProtoLibraryTest.java b/src/test/java/com/google/devtools/build/lib/rules/proto/BazelProtoLibraryTest.java
index d9ca872aa9f74d..8f355152cc415f 100644
--- a/src/test/java/com/google/devtools/build/lib/rules/proto/BazelProtoLibraryTest.java
+++ b/src/test/java/com/google/devtools/build/lib/rules/proto/BazelProtoLibraryTest.java
@@ -225,7 +225,7 @@ public void testDescriptorSetOutput_strictDeps() throws Exception {
         .containsAtLeast("--direct_dependencies", "x/nodeps.proto")
         .inOrder();
     assertThat(getGeneratingSpawnAction(getDescriptorOutput("//x:nodeps")).getRemainingArguments())
-        .contains(String.format(ProtoCompileActionBuilder.STRICT_DEPS_FLAG_TEMPLATE, "//x:nodeps"));
+        .contains(String.format(ProtoConstants.STRICT_PROTO_DEPS_VIOLATION_MESSAGE, "//x:nodeps"));
 
     assertThat(
             getGeneratingSpawnAction(getDescriptorOutput("//x:withdeps")).getRemainingArguments())
@@ -275,7 +275,7 @@ public void testDescriptorSetOutput_strictDeps_disabled() throws Exception {
       assertThat(arg).doesNotContain("--direct_dependencies=");
       assertThat(arg)
           .doesNotContain(
-              String.format(ProtoCompileActionBuilder.STRICT_DEPS_FLAG_TEMPLATE, "//x:foo_proto"));
+              String.format(ProtoConstants.STRICT_PROTO_DEPS_VIOLATION_MESSAGE, "//x:foo_proto"));
     }
   }
 
diff --git a/src/test/java/com/google/devtools/build/lib/starlark/StarlarkRuleClassFunctionsTest.java b/src/test/java/com/google/devtools/build/lib/starlark/StarlarkRuleClassFunctionsTest.java
index d12add6d7f643b..1ced7947625c11 100644
--- a/src/test/java/com/google/devtools/build/lib/starlark/StarlarkRuleClassFunctionsTest.java
+++ b/src/test/java/com/google/devtools/build/lib/starlark/StarlarkRuleClassFunctionsTest.java
@@ -82,6 +82,7 @@
 import java.io.IOException;
 import java.util.Arrays;
 import java.util.List;
+import java.util.Map;
 import java.util.Optional;
 import java.util.stream.Stream;
 import javax.annotation.Nullable;
@@ -2793,7 +2794,7 @@ public void initializer_basic() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
         "MyInfo = provider()",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  return {'deps': deps + ['//:added']}",
         "def impl(ctx): ",
         "  return [MyInfo(",
@@ -2821,12 +2822,122 @@ public void initializer_basic() throws Exception {
     assertThat((List<String>) info.getValue("deps")).containsExactly("@@//:initial", "@@//:added");
   }
 
+  @Test
+  public void initializer_nameUnchanged() throws Exception {
+    scratch.file(
+        "initializer_testing/b.bzl",
+        "def initializer(name, **kwargs):",
+        "  if name != 'my_target':",
+        "     fail()",
+        "  return {'name': name} | kwargs",
+        "MyInfo = provider()",
+        "def impl(ctx): ",
+        "  pass",
+        "my_rule = rule(impl, initializer = initializer)");
+    scratch.file(
+        "initializer_testing/BUILD", //
+        "load(':b.bzl','my_rule')",
+        "my_rule(name = 'my_target')");
+
+    getConfiguredTarget("//initializer_testing:my_target");
+
+    assertNoEvents();
+  }
+
+  @Test
+  public void initializer_nameChanged() throws Exception {
+    scratch.file(
+        "initializer_testing/b.bzl",
+        "def initializer(name, **kwargs):",
+        "  return {'name': 'my_new_name'}",
+        "def impl(ctx): ",
+        "  pass",
+        "my_rule = rule(impl, initializer = initializer)");
+    scratch.file(
+        "initializer_testing/BUILD", //
+        "load(':b.bzl','my_rule')",
+        "my_rule(name = 'my_target')");
+
+    reporter.removeHandler(failFastHandler);
+    reporter.addHandler(ev.getEventCollector());
+    getConfiguredTarget("//initializer_testing:my_target");
+
+    ev.assertContainsError("Error in my_rule: Initializer can't change the name of the target");
+  }
+
+  @Test
+  @SuppressWarnings("unchecked")
+  public void initializer_stringListDict() throws Exception {
+    scratch.file(
+        "initializer_testing/b.bzl",
+        "def initializer(**kwargs):",
+        "  return {}",
+        "MyInfo = provider()",
+        "def impl(ctx): ",
+        "  return [MyInfo(dict = ctx.attr.dict)]",
+        "my_rule = rule(impl,",
+        "  initializer = initializer,",
+        "  attrs = {",
+        "    'dict': attr.string_list_dict(),",
+        "  })");
+    scratch.file(
+        "initializer_testing/BUILD", //
+        "load(':b.bzl','my_rule')",
+        "my_rule(name = 'my_target', dict = {'k': ['val']})");
+
+    ConfiguredTarget myTarget = getConfiguredTarget("//initializer_testing:my_target");
+    StructImpl info =
+        (StructImpl)
+            myTarget.get(
+                new StarlarkProvider.Key(
+                    Label.parseCanonical("//initializer_testing:b.bzl"), "MyInfo"));
+
+    assertThat(((Map<String, List<String>>) info.getValue("dict")).keySet()).containsExactly("k");
+    assertThat(((Map<String, List<String>>) info.getValue("dict")).get("k")).containsExactly("val");
+  }
+
+  @Test
+  @SuppressWarnings("unchecked")
+  public void initializer_labelKeyedStringDict() throws Exception {
+    scratch.file(
+        "BUILD", //
+        "filegroup(name = 'key')");
+    scratch.file(
+        "initializer_testing/b.bzl",
+        "def initializer(**kwargs):",
+        "  return {}",
+        "MyInfo = provider()",
+        "def impl(ctx): ",
+        "  return [MyInfo(dict = ctx.attr.dict)]",
+        "my_rule = rule(impl,",
+        "  initializer = initializer,",
+        "  attrs = {",
+        "    'dict': attr.label_keyed_string_dict(),",
+        "  })");
+    scratch.file(
+        "initializer_testing/BUILD", //
+        "load(':b.bzl','my_rule')",
+        "my_rule(name = 'my_target', dict = {'//:key': 'val'})");
+
+    ConfiguredTarget myTarget = getConfiguredTarget("//initializer_testing:my_target");
+    ConfiguredTarget key = getConfiguredTarget("//:key");
+    StructImpl info =
+        (StructImpl)
+            myTarget.get(
+                new StarlarkProvider.Key(
+                    Label.parseCanonical("//initializer_testing:b.bzl"), "MyInfo"));
+
+    assertThat(((Map<ConfiguredTarget, String>) info.getValue("dict")).keySet())
+        .containsExactly(key);
+    assertThat(((Map<ConfiguredTarget, String>) info.getValue("dict")).get(key)).isEqualTo("val");
+  }
+
   @Test
   public void initializer_legacyAnyType() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
         "MyInfo = provider()",
-        "def initializer(tristate = -1):",
+        "def initializer(name, tristate = -1):",
         "  return {'tristate': int(tristate)}",
         "def impl(ctx): ",
         "  return [MyInfo(tristate = ctx.attr.tristate)]",
@@ -2886,7 +2997,7 @@ public void initializer_withSelect() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
         "MyInfo = provider()",
-        "def initializer(srcs = []):",
+        "def initializer(name, srcs = []):",
         "  return {'srcs': srcs + ['b.ml']}",
         "def impl(ctx): ",
         "  return [MyInfo(",
@@ -2946,7 +3057,7 @@ public void initializer_overridesAttributeDefault() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
         "MyInfo = provider()",
-        "def initializer(deps = ['//:initializer_default']):",
+        "def initializer(name, deps = ['//:initializer_default']):",
         "  return {'deps': deps}",
         "def impl(ctx): ",
         "  return [MyInfo(",
@@ -2981,7 +3092,7 @@ public void initializer_returningNoneSetsDefault() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
         "MyInfo = provider()",
-        "def initializer(deps = ['//:initializer_default']):",
+        "def initializer(name, deps = ['//:initializer_default']):",
         "  return {'deps': None}",
         "def impl(ctx): ",
         "  return [MyInfo(",
@@ -3011,7 +3122,7 @@ public void initializer_omittedValueIsNotPassed() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
         "MyInfo = provider()",
-        "def initializer(srcs):",
+        "def initializer(name, srcs):",
         "  return {'srcs': srcs}",
         "def impl(ctx): ",
         "  pass",
@@ -3038,7 +3149,7 @@ public void initializer_noneValueIsNotPassed() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
         "MyInfo = provider()",
-        "def initializer(srcs):",
+        "def initializer(name, srcs):",
         "  return {'srcs': srcs}",
         "def impl(ctx): ",
         "  pass",
@@ -3063,7 +3174,7 @@ public void initializer_noneValueIsNotPassed() throws Exception {
   public void initializer_incorrectReturnType() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = []):",
+        "def initializer(name, srcs = []):",
         "  return [srcs]",
         "def impl(ctx): ",
         "  pass",
@@ -3088,7 +3199,7 @@ public void initializer_incorrectReturnType() throws Exception {
   public void initializer_incorrectReturnDicts() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = []):",
+        "def initializer(name, srcs = []):",
         "  return {True: srcs}",
         "def impl(ctx): ",
         "  pass",
@@ -3114,7 +3225,7 @@ public void initializer_failsSettingBaseAttribute() throws Exception {
     // 'args' is an attribute defined for all executable rules
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  return {'srcs': srcs, 'deps': deps, 'args': ['a']}",
         "def impl(ctx): ",
         "  pass",
@@ -3141,7 +3252,7 @@ public void initializer_failsSettingBaseAttribute() throws Exception {
   public void initializer_failsSettingPrivateAttribute_outsideBuiltins() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  return {'srcs': srcs, '_tool': ':my_tool'}",
         "def impl(ctx): ",
         "  pass",
@@ -3172,7 +3283,7 @@ public void initializer_settingPrivateAttribute_insideBuiltins() throws Exceptio
     scratch.file("initializer_testing/builtins/BUILD", "filegroup(name='my_tool')");
     scratch.file(
         "initializer_testing/builtins/b.bzl",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  return {'srcs': srcs, '_tool': ':my_tool'}",
         "MyInfo = provider()",
         "def impl(ctx): ",
@@ -3203,7 +3314,7 @@ public void initializer_settingPrivateAttribute_insideBuiltins() throws Exceptio
   public void initializer_failsSettingUnknownAttr() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  return {'srcs': srcs, 'my_deps': deps}",
         "def impl(ctx): ",
         "  pass",
@@ -3229,7 +3340,7 @@ public void initializer_failsSettingUnknownAttr() throws Exception {
   public void initializer_failsCreatingAnotherRule() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  native.java_library(name = 'jl', srcs = ['a.java'])",
         "  return {'srcs': srcs, 'deps': deps}",
         "def impl(ctx): ",
@@ -3257,7 +3368,7 @@ public void initializer_failsCreatingAnotherRule() throws Exception {
   public void initializer_failsWithExistingRules() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  native.existing_rules()",
         "  return {'srcs': srcs, 'deps': deps}",
         "def impl(ctx): ",
@@ -3284,7 +3395,7 @@ public void initializer_failsWithExistingRules() throws Exception {
   public void initializer_withFails() throws Exception {
     scratch.file(
         "initializer_testing/b.bzl",
-        "def initializer(srcs = [], deps = []):",
+        "def initializer(name, srcs = [], deps = []):",
         "  fail('Fail called in initializer')",
         "  return {'srcs': srcs, 'deps': deps}",
         "def impl(ctx): ",
@@ -3413,7 +3524,7 @@ public void extendRule_withInitializers() throws Exception {
     scratch.file(
         "extend_rule_testing/parent/parent.bzl",
         "ParentInfo = provider()",
-        "def _parent_initializer(srcs, deps):", // only parents attributes
+        "def _parent_initializer(name, srcs, deps):", // only parents attributes
         "  return {'deps': deps + ['//extend_rule_testing:parent_dep']}",
         "def _impl(ctx):",
         "  return [ParentInfo()]",
@@ -3430,7 +3541,7 @@ public void extendRule_withInitializers() throws Exception {
         "extend_rule_testing/child.bzl",
         "load('//extend_rule_testing/parent:parent.bzl', 'parent_library')",
         "ChildInfo = provider()",
-        "def _child_initializer(srcs, deps, runtime_deps = []):",
+        "def _child_initializer(name, srcs, deps, runtime_deps = []):",
         "  return {'deps': deps + [':child_dep'], 'runtime_deps': runtime_deps + [':runtime_dep']}",
         "def _impl(ctx):",
         "  return ctx.super() + [ChildInfo(",