ASTGen: Translate import declarations

Author: AnthonyLatsis

include/swift/AST/CASTBridging.h

@@ -256,12 +256,6 @@ BridgedIdentifier ASTContext_getIdentifier(BridgedASTContext cContext,
 _Bool ASTContext_langOptsHasFeature(BridgedASTContext cContext,
                                     BridgedFeature feature);
 
-void *ImportDecl_create(BridgedASTContext cContext,
-                        BridgedDeclContext cDeclContext,
-                        BridgedSourceLoc cImportLoc, char kind,
-                        BridgedSourceLoc cKindLoc, BridgedArrayRef path,
-                        BridgedArrayRef cPathLocs);
-
 void *TopLevelCodeDecl_createStmt(BridgedASTContext cContext,
                                   BridgedDeclContext cDeclContext,
                                   BridgedSourceLoc cStartLoc, void *element,
@@ -440,6 +434,24 @@ void *OperatorDecl_create(BridgedASTContext cContext,
                           BridgedIdentifier cPrecedenceGroupName,
                           BridgedSourceLoc cPrecedenceGroupLoc);
 
+typedef enum ENUM_EXTENSIBILITY_ATTR(open) {
+  BridgedImportKindModule,
+  BridgedImportKindType,
+  BridgedImportKindStruct,
+  BridgedImportKindClass,
+  BridgedImportKindEnum,
+  BridgedImportKindProtocol,
+  BridgedImportKindVar,
+  BridgedImportKindFunc,
+} BridgedImportKind;
+
+void *ImportDecl_create(BridgedASTContext cContext,
+                        BridgedDeclContext cDeclContext,
+                        BridgedSourceLoc cImportKeywordLoc,
+                        BridgedImportKind cImportKind,
+                        BridgedSourceLoc cImportKindLoc,
+                        BridgedArrayRef cImportPathElements);
+
 void *GenericParamList_create(BridgedASTContext cContext,
                               BridgedSourceLoc cLeftAngleLoc,
                               BridgedArrayRef cParameters,

lib/AST/CASTBridging.cpp

@@ -186,6 +186,7 @@ BridgedIdentifier ASTContext_getIdentifier(BridgedASTContext cContext,
   return {convertASTContext(cContext).getIdentifier(str).getAsOpaquePointer()};
 }
 
+<<<<<<< HEAD
 bool ASTContext_langOptsHasFeature(BridgedASTContext cContext,
                                    BridgedFeature feature) {
   return convertASTContext(cContext).LangOpts.hasFeature((Feature)feature);
@@ -215,6 +216,9 @@ void *ImportDecl_create(BridgedASTContext cContext,
 }
 
 BridgedSourceLoc SourceLoc_advanced(BridgedSourceLoc cLoc, SwiftInt len) {
+=======
+BridgedSourceLoc SourceLoc_advanced(BridgedSourceLoc cLoc, long len) {
+>>>>>>> 2841aca3260 (ASTGen: Translate import declarations)
   SourceLoc loc = convertSourceLoc(cLoc).getAdvancedLoc(len);
   return {loc.getOpaquePointerValue()};
 }
@@ -822,6 +826,28 @@ void *OperatorDecl_create(BridgedASTContext cContext,
   return static_cast<Decl *>(decl);
 }
 
+void *ImportDecl_create(BridgedASTContext cContext,
+                        BridgedDeclContext cDeclContext,
+                        BridgedSourceLoc cImportKeywordLoc,
+                        BridgedImportKind cImportKind,
+                        BridgedSourceLoc cImportKindLoc,
+                        BridgedArrayRef cImportPathElements) {
+  ImportPath::Builder builder;
+  for (auto &element :
+       convertArrayRef<BridgedIdentifierAndSourceLoc>(cImportPathElements)) {
+    builder.push_back(convertIdentifier(element.name),
+                      convertSourceLoc(element.nameLoc));
+  }
+
+  ASTContext &context = convertASTContext(cContext);
+  auto *decl = ImportDecl::create(
+      context, convertDeclContext(cDeclContext),
+      convertSourceLoc(cImportKeywordLoc), static_cast<ImportKind>(cImportKind),
+      convertSourceLoc(cImportKindLoc), std::move(builder).get());
+
+  return static_cast<Decl *>(decl);
+}
+
 void *OptionalTypeRepr_create(BridgedASTContext cContext, void *base,
                               BridgedSourceLoc cQuestionLoc) {
   ASTContext &context = convertASTContext(cContext);

lib/ASTGen/Sources/ASTGen/Decls.swift

@@ -351,6 +351,53 @@ extension ASTGenVisitor {
   }
 }
 
+// MARK: - ImportDecl
+
+extension BridgedImportKind {
+  fileprivate init?(from tokenKind: TokenKind) {
+    switch tokenKind {
+    case .keyword(.typealias): self = .type
+    case .keyword(.struct): self = .struct
+    case .keyword(.class): self = .class
+    case .keyword(.enum): self = .enum
+    case .keyword(.protocol): self = .protocol
+    case .keyword(.var), .keyword(.let): self = .var
+    case .keyword(.func): self = .func
+    default: return nil
+    }
+  }
+}
+
+extension ASTGenVisitor {
+  func visit(_ node: ImportDeclSyntax) -> ASTNode {
+    let importKind: BridgedImportKind
+    switch node.importKindSpecifier {
+    case let specifier?:
+      guard let value = BridgedImportKind(from: specifier.tokenKind) else {
+        self.diagnose(Diagnostic(node: specifier, message: UnexpectedTokenKindError(token: specifier)))
+        fallthrough
+      }
+
+      importKind = value
+    case nil:
+      importKind = .module
+    }
+
+    return .decl(
+      ImportDecl_create(
+        self.ctx,
+        self.declContext,
+        self.bridgedSourceLoc(for: node.importKeyword),
+        importKind,
+        self.bridgedSourceLoc(for: node.importKindSpecifier),
+        node.path.map(in: self) {
+          $0.name.bridgedIdentifierAndSourceLoc(in: self) as BridgedIdentifierAndSourceLoc
+        }
+      )
+    )
+  }
+}
+
 extension ASTGenVisitor {
   func visit(_ node: MemberBlockItemListSyntax) -> BridgedArrayRef {
     node.map(in: self) { self.visit($0).rawValue }

test/ASTGen/verify-parse.swift

@@ -14,6 +14,16 @@
 
 // NB: Ridiculous formatting to test that we do not include leading trivia in locations.
 
+import
+Swift
+import typealias Swift.Codable
+import enum Swift.Optional
+import struct Swift.Array
+import class Swift.KeyPath
+import protocol Swift.Sequence
+import func Swift.max
+import var Swift._playgroundPrintHook
+
 func
 test1
 (