[Compile Time Constant Extraction] Extract result builder-type information (#63091)

Author: jPaolantonio

Diff for: lib/ConstExtract/ConstExtract.cpp

@@ -646,14 +646,46 @@ void writeEnumCases(
   });
 }
 
+void writeResultBuilderInformation(llvm::json::OStream &JSON,
+                                   const swift::NominalTypeDecl *TypeDecl,
+                                   const swift::VarDecl *VarDecl) {
+  if (auto *attr = VarDecl->getAttachedResultBuilder()) {
+    JSON.attributeObject("resultBuilder", [&] {
+      JSON.attribute("type", toFullyQualifiedTypeNameString(attr->getType()));
+    });
+
+    return;
+  }
+
+  for (ProtocolDecl *Decl :
+       TypeDecl->getLocalProtocols(ConformanceLookupKind::All)) {
+    for (auto Member : Decl->getMembers()) {
+      if (auto *VD = dyn_cast<swift::VarDecl>(Member)) {
+        if (VD->getName() != VarDecl->getName())
+          continue;
+
+        if (auto *attr = VD->getAttachedResultBuilder()) {
+          JSON.attributeObject("resultBuilder", [&] {
+            JSON.attribute("type",
+                           toFullyQualifiedTypeNameString(attr->getType()));
+          });
+        }
+
+        return;
+      }
+    }
+  }
+}
+
 bool writeAsJSONToFile(const std::vector<ConstValueTypeInfo> &ConstValueInfos,
                        llvm::raw_fd_ostream &OS) {
   llvm::json::OStream JSON(OS, 2);
   JSON.array([&] {
     for (const auto &TypeInfo : ConstValueInfos) {
       JSON.object([&] {
         const auto *TypeDecl = TypeInfo.TypeDecl;
-        JSON.attribute("typeName", toFullyQualifiedTypeNameString(TypeDecl->getDeclaredInterfaceType()));
+        JSON.attribute("typeName", toFullyQualifiedTypeNameString(
+                                       TypeDecl->getDeclaredInterfaceType()));
         JSON.attribute(
             "kind",
             TypeDecl->getDescriptiveKindName(TypeDecl->getDescriptiveKind())
@@ -672,6 +704,7 @@ bool writeAsJSONToFile(const std::vector<ConstValueTypeInfo> &ConstValueInfos,
               writeFileInformation(JSON, decl);
               writeValue(JSON, PropertyInfo.Value);
               writeAttributes(JSON, PropertyInfo.PropertyWrappers);
+              writeResultBuilderInformation(JSON, TypeDecl, decl);
             });
           }
         });

Diff for: test/ConstExtraction/ExtractResultBuilders.swift

@@ -0,0 +1,102 @@
+// RUN: %empty-directory(%t)
+// RUN: %empty-directory(%t/inputs)
+// RUN: echo "[FooProvider]" > %t/inputs/protocols.json
+
+// RUN: %target-swift-frontend -typecheck -emit-const-values-path %t/ExtractResultBuilders.swiftconstvalues -const-gather-protocols-file %t/inputs/protocols.json -primary-file %s
+// RUN: cat %t/ExtractResultBuilders.swiftconstvalues 2>&1 | %FileCheck %s
+
+public struct Foo {
+    let name: String
+}
+
+@resultBuilder
+public enum FooBuilder {
+    public static func buildExpression(_ component: Foo) -> Foo {
+        component
+    }
+
+    public static func buildBlock(_ components: Foo...) -> [Foo] {
+        components
+    }
+}
+
+public protocol FooProvider {
+    @FooBuilder
+    static var foos: [Foo] { get }
+}
+
+public struct MyFooProvider: FooProvider {
+    @FooBuilder
+    public static var foos: [Foo] {
+        Foo(name: "AAA")
+        Foo(name: "BBB")
+    }
+
+    @FooBuilder
+    public static var fooTwo: [Foo] {
+        Foo(name: "111")
+        Foo(name: "222")
+    }
+}
+
+public struct MyFooProviderInferred: FooProvider {
+    public static var foos: [Foo] {
+        Foo(name: "CCC")
+        Foo(name: "DDD")
+    }
+}
+
+// CHECK: [
+// CHECK-NEXT:    {
+// CHECK-NEXT:      "typeName": "ExtractResultBuilders.MyFooProvider",
+// CHECK-NEXT:      "kind": "struct",
+// CHECK-NEXT:      "file": "{{.*}}test{{/|\\\\}}ConstExtraction{{/|\\\\}}ExtractResultBuilders.swift",
+// CHECK-NEXT:      "line": 28,
+// CHECK-NEXT:      "properties": [
+// CHECK-NEXT:        {
+// CHECK-NEXT:          "label": "foos",
+// CHECK-NEXT:          "type": "[ExtractResultBuilders.Foo]",
+// CHECK-NEXT:          "isStatic": "true",
+// CHECK-NEXT:          "isComputed": "true",
+// CHECK-NEXT:          "file": "{{.*}}test{{/|\\\\}}ConstExtraction{{/|\\\\}}ExtractResultBuilders.swift",
+// CHECK-NEXT:          "line": 30,
+// CHECK-NEXT:          "valueKind": "Runtime",
+// CHECK-NEXT:          "resultBuilder": {
+// CHECK-NEXT:            "type": "ExtractResultBuilders.FooBuilder"
+// CHECK-NEXT:          }
+// CHECK-NEXT:        },
+// CHECK-NEXT:        {
+// CHECK-NEXT:          "label": "fooTwo",
+// CHECK-NEXT:          "type": "[ExtractResultBuilders.Foo]",
+// CHECK-NEXT:          "isStatic": "true",
+// CHECK-NEXT:          "isComputed": "true",
+// CHECK-NEXT:          "file": "{{.*}}test{{/|\\\\}}ConstExtraction{{/|\\\\}}ExtractResultBuilders.swift",
+// CHECK-NEXT:          "line": 36,
+// CHECK-NEXT:          "valueKind": "Runtime",
+// CHECK-NEXT:          "resultBuilder": {
+// CHECK-NEXT:            "type": "ExtractResultBuilders.FooBuilder"
+// CHECK-NEXT:          }
+// CHECK-NEXT:        }
+// CHECK-NEXT:      ]
+// CHECK-NEXT:    },
+// CHECK-NEXT:    {
+// CHECK-NEXT:      "typeName": "ExtractResultBuilders.MyFooProviderInferred",
+// CHECK-NEXT:      "kind": "struct",
+// CHECK-NEXT:      "file": "{{.*}}test{{/|\\\\}}ConstExtraction{{/|\\\\}}ExtractResultBuilders.swift",
+// CHECK-NEXT:      "line": 42,
+// CHECK-NEXT:      "properties": [
+// CHECK-NEXT:        {
+// CHECK-NEXT:          "label": "foos",
+// CHECK-NEXT:          "type": "[ExtractResultBuilders.Foo]",
+// CHECK-NEXT:          "isStatic": "true",
+// CHECK-NEXT:          "isComputed": "true",
+// CHECK-NEXT:          "file": "{{.*}}test{{/|\\\\}}ConstExtraction{{/|\\\\}}ExtractResultBuilders.swift",
+// CHECK-NEXT:          "line": 43,
+// CHECK-NEXT:          "valueKind": "Runtime",
+// CHECK-NEXT:          "resultBuilder": {
+// CHECK-NEXT:            "type": "ExtractResultBuilders.FooBuilder"
+// CHECK-NEXT:          }
+// CHECK-NEXT:        }
+// CHECK-NEXT:      ]
+// CHECK-NEXT:    }
+// CHECK-NEXT:  ]