[Compile Time Constant Extraction] Add extraction computed variables in extensions (#62465)

Author: jPaolantonio

lib/ConstExtract/ConstExtract.cpp

@@ -205,6 +205,7 @@ ConstantValueInfoRequest::evaluate(Evaluator &Evaluator,
     Properties.push_back(
         {Property, extractPropertyInitializationValue(Property)});
   }
+
   for (auto Member : Decl->getMembers()) {
     auto *VD = dyn_cast<VarDecl>(Member);
     // Ignore plain stored properties collected above,
@@ -214,6 +215,14 @@ ConstantValueInfoRequest::evaluate(Evaluator &Evaluator,
     Properties.push_back({VD, extractPropertyInitializationValue(VD)});
   }
 
+  for (auto Extension: Decl->getExtensions()) {
+    for (auto Member : Extension->getMembers()) {
+      if (auto *VD = dyn_cast<VarDecl>(Member)) {
+        Properties.push_back({VD, extractPropertyInitializationValue(VD)});
+      }
+    }
+  }
+
   return ConstValueTypeInfo{Decl, Properties};
 }
 

test/ConstExtraction/fields.swift

@@ -154,6 +154,13 @@
 // CHECK-NEXT:        "isComputed": "true",
 // CHECK-NEXT:        "valueKind": "RawLiteral",
 // CHECK-NEXT:        "value": "3"
+// CHECK-NEXT:      },
+// CHECK-NEXT:      {
+// CHECK-NEXT:        "label": "e1",
+// CHECK-NEXT:        "type": "fields.Foo.Boo",
+// CHECK-NEXT:        "isStatic": "false",
+// CHECK-NEXT:        "isComputed": "true",
+// CHECK-NEXT:        "valueKind": "Runtime"
 // CHECK-NEXT:      }
 // CHECK-NEXT:    ]
 // CHECK-NEXT:  }
@@ -178,6 +185,12 @@ public struct Foo {
     let p13: Int = adder(2, 3)
 }
 
+extension Foo {
+    struct Boo {}
+
+    var e1: Boo { Boo() }
+}
+
 func adder(_ x: Int, _ y: Int) -> Int {
     x + y
 }