ASTGen: Translate initializer declarations

Author: AnthonyLatsis

include/swift/AST/CASTBridging.h

@@ -359,6 +359,18 @@ BridgedDeclContextAndDecl FuncDecl_create(
     void *_Nullable opaqueReturnType,
     void *_Nullable opaqueGenericWhereClause);
 
+BridgedDeclContextAndDecl ConstructorDecl_create(
+    BridgedASTContext cContext,
+    BridgedDeclContext cDeclContext,
+    BridgedSourceLoc cInitKeywordLoc,
+    BridgedSourceLoc cFailabilityMarkLoc,
+    _Bool isIUO,
+    void *_Nullable opaqueGenericParams,
+    void *opaqueParameterList,
+    BridgedSourceLoc cAsyncLoc,
+    BridgedSourceLoc cThrowsLoc,
+    void *_Nullable opaqueGenericWhereClause);
+
 BridgedDeclContextAndDecl DestructorDecl_create(
     BridgedASTContext cContext,
     BridgedDeclContext cDeclContext,

lib/AST/CASTBridging.cpp

@@ -481,6 +481,47 @@ BridgedDeclContextAndDecl FuncDecl_create(
   return {bridgeDeclContext(decl), static_cast<Decl *>(decl)};
 }
 
+BridgedDeclContextAndDecl ConstructorDecl_create(
+    BridgedASTContext cContext,
+    BridgedDeclContext cDeclContext,
+    BridgedSourceLoc cInitKeywordLoc,
+    BridgedSourceLoc cFailabilityMarkLoc,
+    bool isIUO,
+    void *_Nullable opaqueGenericParams,
+    void *opaqueParameterList,
+    BridgedSourceLoc cAsyncLoc,
+    BridgedSourceLoc cThrowsLoc,
+    void *_Nullable opaqueGenericWhereClause) {
+  assert((bool)cFailabilityMarkLoc.raw || !isIUO);
+
+  ASTContext &context = convertASTContext(cContext);
+
+  auto *parameterList = static_cast<ParameterList *>(opaqueParameterList);
+  auto declName =
+      DeclName(context, DeclBaseName::createConstructor(), parameterList);
+  auto asyncLoc = convertSourceLoc(cAsyncLoc);
+  auto throwsLoc = convertSourceLoc(cThrowsLoc);
+  auto failabilityMarkLoc = convertSourceLoc(cFailabilityMarkLoc);
+
+  auto *decl = new (context) ConstructorDecl(
+      declName,
+      convertSourceLoc(cInitKeywordLoc),
+      failabilityMarkLoc.isValid(),
+      failabilityMarkLoc,
+      asyncLoc.isValid(),
+      asyncLoc,
+      throwsLoc.isValid(),
+      throwsLoc,
+      parameterList,
+      static_cast<GenericParamList *>(opaqueGenericParams),
+      convertDeclContext(cDeclContext));
+  decl->setTrailingWhereClause(
+      static_cast<TrailingWhereClause *>(opaqueGenericWhereClause));
+  decl->setImplicitlyUnwrappedOptional(isIUO);
+
+  return {bridgeDeclContext(decl), static_cast<Decl *>(decl)};
+}
+
 BridgedDeclContextAndDecl DestructorDecl_create(
     BridgedASTContext cContext,
     BridgedDeclContext cDeclContext,

lib/ASTGen/Sources/ASTGen/Decls.swift

@@ -296,6 +296,29 @@ extension ASTGenVisitor {
     return .decl(declAndDeclContext.decl)
   }
 
+  func visit(_ node: InitializerDeclSyntax) -> ASTNode {
+    let isIUO = node.optionalMark?.tokenKind == .exclamationMark
+
+    let declAndDeclContext = ConstructorDecl_create(
+      self.ctx,
+      self.declContext,
+      self.bridgedSourceLoc(for: node.initKeyword),
+      self.bridgedSourceLoc(for: node.optionalMark),
+      isIUO,
+      node.genericParameterClause.map(self.visit)?.rawValue,
+      self.visit(node.signature.input).rawValue,
+      self.bridgedSourceLoc(for: node.signature.effectSpecifiers?.asyncSpecifier),
+      self.bridgedSourceLoc(for: node.signature.effectSpecifiers?.throwsSpecifier),
+      node.genericWhereClause.map(self.visit)?.rawValue
+    )
+
+    if let body = node.body {
+      self.visitFunctionCodeBlock(body, declContextAndDecl: declAndDeclContext)
+    }
+
+    return .decl(declAndDeclContext.decl)
+  }
+
   func visit(_ node: DeinitializerDeclSyntax) -> ASTNode {
     let declAndDeclContext = DestructorDecl_create(
       self.ctx,

test/ASTGen/verify-parse.swift

@@ -152,6 +152,12 @@ Class<T>: Proto1 where T: Proto3 {
   deinit {
     if true {}
   }
+
+  init?<U>(_ u: U) where U: Proto1 {
+    if true {}
+  }
+
+  init!(i: Int) {}
 }
 
 actor