ASTGen: Translate enumeration declarations

Author: AnthonyLatsis

include/swift/AST/CASTBridging.h

@@ -358,6 +358,13 @@ void *TypeAliasDecl_create(BridgedASTContext cContext,
 void IterableDeclContext_setParsedMembers(BridgedArrayRef members,
                                           void *opaqueDecl);
 
+BridgedDeclContextAndDecl EnumDecl_create(
+    BridgedASTContext cContext, BridgedDeclContext cDeclContext,
+    BridgedSourceLoc cEnumKeywordLoc, BridgedIdentifier cName,
+    BridgedSourceLoc cNameLoc, void *_Nullable opaqueGenericParamList,
+    BridgedArrayRef cInheritedTypes, void *_Nullable opaqueGenericWhereClause,
+    BridgedSourceRange cBraceRange);
+
 BridgedDeclContextAndDecl StructDecl_create(
     BridgedASTContext cContext, BridgedDeclContext cDeclContext,
     BridgedSourceLoc cStructKeywordLoc, BridgedIdentifier cName,

lib/AST/CASTBridging.cpp

@@ -573,6 +573,27 @@ convertToInheritedEntries(BridgedArrayRef cInheritedTypes) {
   return inheritedEntries;
 }
 
+BridgedDeclContextAndDecl EnumDecl_create(
+    BridgedASTContext cContext, BridgedDeclContext cDeclContext,
+    BridgedSourceLoc cEnumKeywordLoc, BridgedIdentifier cName,
+    BridgedSourceLoc cNameLoc, void *_Nullable opaqueGenericParamList,
+    BridgedArrayRef cInheritedTypes, void *_Nullable opaqueGenericWhereClause,
+    BridgedSourceRange cBraceRange) {
+  ASTContext &context = convertASTContext(cContext);
+
+  auto *decl = new (context)
+      EnumDecl(convertSourceLoc(cEnumKeywordLoc), convertIdentifier(cName),
+               convertSourceLoc(cNameLoc),
+               context.AllocateCopy(convertToInheritedEntries(cInheritedTypes)),
+               static_cast<GenericParamList *>(opaqueGenericParamList),
+               convertDeclContext(cDeclContext));
+  decl->setTrailingWhereClause(
+      static_cast<TrailingWhereClause *>(opaqueGenericWhereClause));
+  decl->setBraces(convertSourceRange(cBraceRange));
+
+  return {bridgeDeclContext(decl), static_cast<Decl *>(decl)};
+}
+
 BridgedDeclContextAndDecl StructDecl_create(
     BridgedASTContext cContext, BridgedDeclContext cDeclContext,
     BridgedSourceLoc cStructKeywordLoc, BridgedIdentifier cName,

lib/ASTGen/Sources/ASTGen/Decls.swift

@@ -25,6 +25,28 @@ extension ASTGenVisitor {
     )
   }
 
+  public func visit(_ node: EnumDeclSyntax) -> ASTNode {
+    let (name, nameLoc) = node.name.bridgedIdentifierAndSourceLoc(in: self)
+
+    let decl = EnumDecl_create(
+      self.ctx,
+      self.declContext,
+      self.bridgedSourceLoc(for: node.enumKeyword),
+      name,
+      nameLoc,
+      self.visit(node.genericParameterClause)?.rawValue,
+      self.visit(node.inheritanceClause?.inheritedTypes),
+      self.visit(node.genericWhereClause)?.rawValue,
+      BridgedSourceRange(startToken: node.memberBlock.leftBrace, endToken: node.memberBlock.rightBrace, in: self)
+    )
+
+    self.withDeclContext(decl.asDeclContext) {
+      IterableDeclContext_setParsedMembers(self.visit(node.memberBlock.members), decl.asDecl)
+    }
+
+    return .decl(decl.asDecl)
+  }
+
   public func visit(_ node: StructDeclSyntax) -> ASTNode {
     let (name, nameLoc) = node.name.bridgedIdentifierAndSourceLoc(in: self)
 

test/ASTGen/verify-parse.swift

@@ -104,6 +104,11 @@ Alias<T>
 =
 String where T: Proto1
 
+enum
+Enum<T>: Proto1 where T: Proto1 {
+  func method(_ b: Bool) {}
+}
+
 struct
 Struct
 <