From 39c8059abd90c1d0f4884f2e1ea7ad0197ad17c0 Mon Sep 17 00:00:00 2001
From: "Lukacs T. Berki" <lberki@google.com>
Date: Wed, 16 Oct 2024 10:38:16 +0000
Subject: [PATCH] Add the necessary methods for FakeStarlarkAttrModuleApi.

---
 .../FakeStarlarkAttrModuleApi.java            | 26 +++++++++++++++++++
 .../rendering/proto/stardoc_output.proto      |  1 +
 2 files changed, 27 insertions(+)

diff --git a/src/main/java/com/google/devtools/build/skydoc/fakebuildapi/FakeStarlarkAttrModuleApi.java b/src/main/java/com/google/devtools/build/skydoc/fakebuildapi/FakeStarlarkAttrModuleApi.java
index a56bdf7f2fb00d..a716693b3f5f1f 100644
--- a/src/main/java/com/google/devtools/build/skydoc/fakebuildapi/FakeStarlarkAttrModuleApi.java
+++ b/src/main/java/com/google/devtools/build/skydoc/fakebuildapi/FakeStarlarkAttrModuleApi.java
@@ -131,6 +131,32 @@ public Descriptor labelListAttribute(
         AttributeType.LABEL_LIST, toTrimmedString(doc), mandatory, allNameGroups, defaultList);
   }
 
+  @Override
+  public Descriptor stringKeyedLabelDictAttribute(
+      Boolean allowEmpty,
+      Object defaultList,
+      Object doc,
+      Object allowFiles,
+      Object allowRules,
+      Sequence<?> providers,
+      Sequence<?> flags,
+      Boolean mandatory,
+      Object cfg,
+      Sequence<?> aspects,
+      StarlarkThread thread)
+      throws EvalException {
+    List<List<String>> allNameGroups = new ArrayList<>();
+    if (providers != null) {
+      allNameGroups = allProviderNameGroups(providers, thread);
+    }
+    return new FakeDescriptor(
+        AttributeType.LABEL_DICT_UNARY,
+        toTrimmedString(doc),
+        mandatory,
+        allNameGroups,
+        defaultList);
+  }
+
   @Override
   public Descriptor labelKeyedStringDictAttribute(
       Boolean allowEmpty,
diff --git a/src/main/java/com/google/devtools/build/skydoc/rendering/proto/stardoc_output.proto b/src/main/java/com/google/devtools/build/skydoc/rendering/proto/stardoc_output.proto
index a165a33581414a..c4b4f8fffd261d 100644
--- a/src/main/java/com/google/devtools/build/skydoc/rendering/proto/stardoc_output.proto
+++ b/src/main/java/com/google/devtools/build/skydoc/rendering/proto/stardoc_output.proto
@@ -68,6 +68,7 @@ enum AttributeType {
   STRING_LIST_DICT = 11;
   OUTPUT = 12;
   OUTPUT_LIST = 13;
+  LABEL_DICT_UNARY = 14;
 }
 
 // Representation of a Starlark rule definition.