Merge pull request #15795 from aschwaighofer/tbd_resilient_enum

Author: swift-ci

lib/TBDGen/TBDGen.cpp

@@ -299,6 +299,17 @@ void TBDGenVisitor::visitProtocolDecl(ProtocolDecl *PD) {
 #endif
 }
 
+void TBDGenVisitor::visitEnumDecl(EnumDecl *ED) {
+  if (!ED->isResilient(SwiftModule, ResilienceExpansion::Minimal))
+    return;
+
+  // Emit resilient tags.
+  for (auto *elt : ED->getAllElements()) {
+    auto entity = LinkEntity::forEnumCase(elt);
+    addSymbol(entity);
+  }
+}
+
 static void enumeratePublicSymbolsAndWrite(ModuleDecl *M, FileUnit *singleFile,
                                            StringSet &symbols,
                                            bool hasMultipleIGMs,

lib/TBDGen/TBDGenVisitor.h

@@ -100,6 +100,8 @@ class TBDGenVisitor : public ASTVisitor<TBDGenVisitor> {
 
   void visitVarDecl(VarDecl *VD);
 
+  void visitEnumDecl(EnumDecl *ED);
+
   void visitDecl(Decl *D) {}
 };
 } // end namespace tbdgen

test/TBD/enum.swift

@@ -0,0 +1,17 @@
+// RUN: %target-swift-frontend -emit-ir -o- -parse-as-library -module-name test -validate-tbd-against-ir=missing %s
+// RUN: %target-swift-frontend -enable-resilience -emit-ir -o- -parse-as-library -module-name test -validate-tbd-against-ir=missing %s
+
+class C {
+}
+
+enum SinglePayload {
+  case A
+  case B(C)
+  case D
+}
+
+enum MultiPayload {
+  case A
+  case B(C)
+  case D(C)
+}