Merge pull request #33544 from CodaFi/coordination-in-name-only

Partially Revert #33371

Author: CodaFi

lib/Sema/DerivedConformanceCaseIterable.cpp

@@ -69,6 +69,15 @@ static ArraySliceType *computeAllCasesType(NominalTypeDecl *enumDecl) {
   return ArraySliceType::get(enumType);
 }
 
+static Type deriveCaseIterable_AllCases(DerivedConformance &derived) {
+  // enum SomeEnum : CaseIterable {
+  //   @derived
+  //   typealias AllCases = [SomeEnum]
+  // }
+  auto *rawInterfaceType = computeAllCasesType(cast<EnumDecl>(derived.Nominal));
+  return derived.getConformanceContext()->mapTypeIntoContext(rawInterfaceType);
+}
+
 ValueDecl *DerivedConformance::deriveCaseIterable(ValueDecl *requirement) {
   // Conformance can't be synthesized in an extension.
   if (checkAndDiagnoseDisallowedContext(requirement))
@@ -102,3 +111,18 @@ ValueDecl *DerivedConformance::deriveCaseIterable(ValueDecl *requirement) {
 
   return propDecl;
 }
+
+Type DerivedConformance::deriveCaseIterable(AssociatedTypeDecl *assocType) {
+  // Check that we can actually derive CaseIterable for this type.
+  if (!canDeriveConformance(Nominal))
+    return nullptr;
+
+  if (assocType->getName() == Context.Id_AllCases) {
+    return deriveCaseIterable_AllCases(*this);
+  }
+
+  Context.Diags.diagnose(assocType->getLoc(),
+                         diag::broken_case_iterable_requirement);
+  return nullptr;
+}
+

lib/Sema/DerivedConformances.h

@@ -163,6 +163,12 @@ class DerivedConformance {
   /// \returns the derived member, which will also be added to the type.
   ValueDecl *deriveCaseIterable(ValueDecl *requirement);
 
+  /// Derive a CaseIterable type witness for an enum if it has no associated
+  /// values for any of its cases.
+  ///
+  /// \returns the derived member, which will also be added to the type.
+  Type deriveCaseIterable(AssociatedTypeDecl *assocType);
+
   /// Determine if a RawRepresentable requirement can be derived for a type.
   ///
   /// This is implemented for non-empty enums without associated values,

lib/Sema/TypeCheckProtocol.cpp

@@ -5771,6 +5771,8 @@ TypeChecker::deriveTypeWitness(DeclContext *DC,
   switch (*knownKind) {
   case KnownProtocolKind::RawRepresentable:
     return std::make_pair(derived.deriveRawRepresentable(AssocType), nullptr);
+  case KnownProtocolKind::CaseIterable:
+    return std::make_pair(derived.deriveCaseIterable(AssocType), nullptr);
   case KnownProtocolKind::Differentiable:
     return derived.deriveDifferentiable(AssocType);
   default: