Sema: EmittedMembersRequest forces SynthesizeMainFunctionRequest

Author: slavapestov

lib/Sema/TypeCheckDecl.cpp

@@ -2523,6 +2523,12 @@ EmittedMembersRequest::evaluate(Evaluator &evaluator,
   forceConformance(Context.getProtocol(KnownProtocolKind::Hashable));
   forceConformance(Context.getProtocol(KnownProtocolKind::Differentiable));
 
+  // If the class has a @main attribute, we need to force synthesis of the
+  // $main function.
+  (void) evaluateOrDefault(Context.evaluator,
+                           SynthesizeMainFunctionRequest{CD},
+                           nullptr);
+
   for (auto *member : CD->getMembers()) {
     if (auto *var = dyn_cast<VarDecl>(member)) {
       // The projected storage wrapper ($foo) might have dynamically-dispatched

test/SILGen/attr_main_class.swift

@@ -0,0 +1,11 @@
+// RUN: %target-swift-emit-silgen -parse-as-library %s | %FileCheck %s
+
+@main class Horse {
+  static func main() {}
+}
+
+// CHECK-LABEL: sil hidden [ossa] @$s15attr_main_class5HorseC5$mainyyFZ : $@convention(method) (@thick Horse.Type) -> () {
+
+// CHECK-LABEL: sil [ossa] @main : $@convention(c) (Int32, UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>) -> Int32 {
+// CHECK: function_ref @$s15attr_main_class5HorseC5$mainyyFZ : $@convention(method) (@thick Horse.Type) -> ()
+// CHECK: }

test/SILGen/attr_main_class_2.swift

@@ -0,0 +1,11 @@
+// RUN: %target-swift-emit-silgen -parse-as-library %s | %FileCheck %s
+
+@main class Horse {
+  class func main() {}
+}
+
+// CHECK-LABEL: sil hidden [ossa] @$s17attr_main_class_25HorseC5$mainyyFZ : $@convention(method) (@thick Horse.Type) -> () {
+
+// CHECK-LABEL: sil [ossa] @main : $@convention(c) (Int32, UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>) -> Int32 {
+// CHECK: function_ref @$s17attr_main_class_25HorseC5$mainyyFZ : $@convention(method) (@thick Horse.Type) -> ()
+// CHECK: }

test/SILGen/attr_main_enum.swift

@@ -0,0 +1,11 @@
+// RUN: %target-swift-emit-silgen -parse-as-library %s | %FileCheck %s
+
+@main enum Horse {
+  static func main() {}
+}
+
+// CHECK-LABEL: sil hidden [ossa] @$s14attr_main_enum5HorseO5$mainyyFZ : $@convention(method) (@thin Horse.Type) -> () {
+
+// CHECK-LABEL: sil [ossa] @main : $@convention(c) (Int32, UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>) -> Int32 {
+// CHECK: function_ref @$s14attr_main_enum5HorseO5$mainyyFZ : $@convention(method) (@thin Horse.Type) -> ()
+// CHECK: }

test/SILGen/attr_main_struct.swift

@@ -0,0 +1,11 @@
+// RUN: %target-swift-emit-silgen -parse-as-library %s | %FileCheck %s
+
+@main struct Horse {
+  static func main() {}
+}
+
+// CHECK-LABEL: sil hidden [ossa] @$s16attr_main_struct5HorseV5$mainyyFZ : $@convention(method) (@thin Horse.Type) -> () {
+
+// CHECK-LABEL: sil [ossa] @main : $@convention(c) (Int32, UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>) -> Int32 {
+// CHECK: function_ref @$s16attr_main_struct5HorseV5$mainyyFZ : $@convention(method) (@thin Horse.Type) -> ()
+// CHECK: }