[AST] RuntimeMetadata: Allow using enums as custom runtime metadata types

Author: xedin

test/IDE/complete_decl_attribute.swift

@@ -138,9 +138,9 @@ actor MyGlobalActor {
 // KEYWORD4-NEXT:             Keyword/None:                       globalActor[#Enum Attribute#]; name=globalActor
 // KEYWORD4-NEXT:             Keyword/None:                       preconcurrency[#Enum Attribute#]; name=preconcurrency
 // KEYWORD4-NEXT:             Keyword/None:                       typeWrapper[#Enum Attribute#]; name=typeWrapper
+// KEYWORD4-NEXT:             Keyword/None:                       runtimeMetadata[#Enum Attribute#]; name=runtimeMetadata
 // KEYWORD4-NEXT:             End completions
 
-
 @#^KEYWORD5^# struct S{}
 // KEYWORD5:                  Begin completions
 // KEYWORD5-NEXT:             Keyword/None:                       available[#Struct Attribute#]; name=available{{$}}

test/IRGen/Inputs/runtime_attrs.swift

@@ -11,3 +11,24 @@ public struct TestAmbiguity {
 @Ignore
 public protocol Ignored {
 }
+
+@runtimeMetadata
+public enum EnumFlag<B, V> {
+case type(B.Type)
+case method((B) -> V)
+case property(KeyPath<B, V>)
+case function(() -> V)
+}
+
+extension EnumFlag {
+  public init(attachedTo: KeyPath<B, V>) { self = .property(attachedTo) }
+  public init(attachedTo: @escaping (B) -> V) { self = .method(attachedTo) }
+}
+
+extension EnumFlag where V == Void {
+  public init(attachedTo: B.Type) { self = .type(attachedTo) }
+}
+
+extension EnumFlag where B == Void {
+  public init(attachedTo: @escaping () -> V) { self = .function(attachedTo) }
+}

test/IRGen/runtime_attributes.swift

@@ -134,6 +134,17 @@
 // CHECK-SAME: @"$s18runtime_attributes9OuterTypeV15outerMutatingFnSiycvpfaAA19FlagForInnerMethodsHF"
 // CHECK-SAME: section "__TEXT, __swift5_rattrs, regular"
 
+// CHECK: @"$s3RAD8EnumFlagOHa" = internal constant
+// CHECK-SAME: i32 0
+// CHECK-SAME: %swift.type_descriptor** @"got.$s3RAD8EnumFlagOMn"
+// CHECK-SAME: i32 5
+// CHECK-SAME: @"$s18runtime_attributes14globalEnumTestSi_SaySSGtSgycvpfa3RAD0D4FlagHF"
+// CHECK-SAME: @"$s18runtime_attributes12EnumTypeTestV1xSivpfa3RAD0C4FlagHF"
+// CHECK-SAME: @"$s18runtime_attributes12EnumTypeTestV8testInstyycvpfa3RAD0C4FlagHF"
+// CHECK-SAME: @"$s18runtime_attributes12EnumTypeTestV10testStaticSiycvpZfa3RAD0C4FlagHF"
+// CHECK-SAME: @"$s18runtime_attributes12EnumTypeTestAaBVmvpfa3RAD0C4FlagHF"
+// CHECK-SAME: section "__TEXT, __swift5_rattrs, regular"
+
 import RAD
 
 @runtimeMetadata
@@ -315,3 +326,18 @@ extension OuterType {
   // CHECK-LABEL: define hidden swiftcc void @"$s18runtime_attributes9OuterTypeV15outerMutatingFnSiycvpfaAA19FlagForInnerMethods"(%T18runtime_attributes19FlagForInnerMethodsVySiGSg* noalias nocapture sret(%T18runtime_attributes19FlagForInnerMethodsVySiGSg) %0)
   @FlagForInnerMethods mutating func outerMutatingFn() -> Int { 42 }
 }
+
+// CHECK-LABEL: define hidden swiftcc void @"$s18runtime_attributes14globalEnumTestSi_SaySSGtSgycvpfa3RAD0D4Flag"(%T3RAD8EnumFlagOyytSi_SaySSGtSgGSg* noalias nocapture sret(%T3RAD8EnumFlagOyytSi_SaySSGtSgGSg) %0)
+@EnumFlag func globalEnumTest() -> (Int, [String])? {
+  nil
+}
+
+@EnumFlag struct EnumTypeTest {
+  // CHECK-LABEL: define hidden swiftcc void @"$s18runtime_attributes12EnumTypeTestV1xSivpfa3RAD0C4Flag"(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVSiGSg* noalias nocapture sret(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVSiGSg) %0)
+  @EnumFlag var x: Int = 42
+  // CHECK-LABEL: define hidden swiftcc void @"$s18runtime_attributes12EnumTypeTestV8testInstyycvpfa3RAD0C4Flag"(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVytGSg* noalias nocapture sret(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVytGSg) %0)
+  @EnumFlag func testInst() {}
+  // CHECK-LABEL: define hidden swiftcc void @"$s18runtime_attributes12EnumTypeTestV10testStaticSiycvpZfa3RAD0C4Flag"(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVmSiGSg* noalias nocapture sret(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVmSiGSg) %0)
+  @EnumFlag static func testStatic() -> Int { 42 }
+}
+// CHECK-LABEL: define hidden swiftcc void @"$s18runtime_attributes12EnumTypeTestAaBVmvpfa3RAD0C4Flag"(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVytGSg* noalias nocapture sret(%T3RAD8EnumFlagOy18runtime_attributes0B8TypeTestVytGSg) %0)

test/type/runtime_discoverable_attrs.swift

@@ -232,3 +232,35 @@ struct TestSelfUse {
   @Flag(Self.description) var x: Int = 42
   @Flag(Self.description) func test() {}
 }
+
+
+@runtimeMetadata
+enum EnumFlag<B, V> {
+  case type(B.Type)
+  case method((B) -> V)
+  case property(KeyPath<B, V>)
+  case function(() -> V)
+}
+
+extension EnumFlag {
+  init(attachedTo: KeyPath<B, V>) { self = .property(attachedTo) }
+  init(attachedTo: @escaping (B) -> V) { self = .method(attachedTo) }
+}
+
+extension EnumFlag where V == Void {
+  init(attachedTo: B.Type) { self = .type(attachedTo) }
+}
+
+extension EnumFlag where B == Void {
+  init(attachedTo: @escaping () -> V) { self = .function(attachedTo) }
+}
+
+@EnumFlag func globalEnumTest() -> (Int, [String])? {
+  nil
+}
+
+@EnumFlag struct EnumTypeTest {
+  @EnumFlag var x: Int = 42
+  @EnumFlag func testInst() {}
+  @EnumFlag static func testStatic() -> Int { 42 }
+}

utils/gyb_syntax_support/AttributeKinds.py

@@ -718,7 +718,7 @@ def __init__(self, name, swift_name=None):
                         code=138),
 
     SimpleDeclAttribute('runtimeMetadata', 'RuntimeMetadata',
-                        OnStruct, OnClass,
+                        OnStruct, OnClass, OnEnum,
                         ABIBreakingToAdd, ABIBreakingToRemove, APIBreakingToAdd, APIBreakingToRemove,  # noqa: E501
                         code=139)
 ]