ASTGen: Translate operator declarations

Author: AnthonyLatsis

include/swift/AST/CASTBridging.h

@@ -459,6 +459,23 @@ BridgedDeclContextAndDecl ExtensionDecl_create(
     BridgedArrayRef cInheritedTypes,
     void *_Nullable opaqueGenericWhereClause);
 
+typedef enum ENUM_EXTENSIBILITY_ATTR(open) {
+  BridgedOperatorFixityInfix,
+  BridgedOperatorFixityPrefix,
+  BridgedOperatorFixityPostfix,
+} BridgedOperatorFixity;
+
+void *OperatorDecl_create(
+    BridgedASTContext cContext,
+    BridgedDeclContext cDeclContext,
+    BridgedOperatorFixity cFixity,
+    BridgedSourceLoc cOperatorKeywordLoc,
+    BridgedIdentifier cName,
+    BridgedSourceLoc cNameLoc,
+    BridgedSourceLoc cColonLoc,
+    BridgedIdentifier cPrecedenceGroupName,
+    BridgedSourceLoc cPrecedenceGroupLoc);
+
 void *GenericParamList_create(
     BridgedASTContext cContext,
     BridgedSourceLoc cLeftAngleLoc,

lib/AST/CASTBridging.cpp

@@ -800,6 +800,52 @@ BridgedDeclContextAndDecl ExtensionDecl_create(
   return {bridgeDeclContext(decl), static_cast<Decl *>(decl)};
 }
 
+void *OperatorDecl_create(
+    BridgedASTContext cContext,
+    BridgedDeclContext cDeclContext,
+    BridgedOperatorFixity cFixity,
+    BridgedSourceLoc cOperatorKeywordLoc,
+    BridgedIdentifier cName,
+    BridgedSourceLoc cNameLoc,
+    BridgedSourceLoc cColonLoc,
+    BridgedIdentifier cPrecedenceGroupName,
+    BridgedSourceLoc cPrecedenceGroupLoc) {
+  assert(bool(cColonLoc.raw) == (bool)cPrecedenceGroupName.raw);
+  assert(bool(cColonLoc.raw) == (bool)cPrecedenceGroupLoc.raw);
+
+  ASTContext &context = convertASTContext(cContext);
+  auto operatorKeywordLoc = convertSourceLoc(cOperatorKeywordLoc);
+  auto name = convertIdentifier(cName);
+  auto nameLoc = convertSourceLoc(cNameLoc);
+  auto *declContext = convertDeclContext(cDeclContext);
+
+  OperatorDecl *decl = nullptr;
+  switch (cFixity) {
+  case BridgedOperatorFixityInfix:
+    decl = new (context) InfixOperatorDecl(
+        declContext,
+        operatorKeywordLoc,
+        name,
+        nameLoc,
+        convertSourceLoc(cColonLoc),
+        convertIdentifier(cPrecedenceGroupName),
+        convertSourceLoc(cPrecedenceGroupLoc));
+    break;
+  case BridgedOperatorFixityPrefix:
+    assert(!cColonLoc.raw);
+    decl = new (context)
+        PrefixOperatorDecl(declContext, operatorKeywordLoc, name, nameLoc);
+    break;
+  case BridgedOperatorFixityPostfix:
+    assert(!cColonLoc.raw);
+    decl = new (context)
+        PostfixOperatorDecl(declContext, operatorKeywordLoc, name, nameLoc);
+    break;
+  }
+
+  return static_cast<Decl *>(decl);
+}
+
 void *OptionalTypeRepr_create(BridgedASTContext cContext, void *base,
                               BridgedSourceLoc cQuestionLoc) {
   ASTContext &context = convertASTContext(cContext);

lib/ASTGen/Sources/ASTGen/Decls.swift

@@ -301,3 +301,39 @@ extension ASTGenVisitor {
     return .decl(out.decl)
   }
 }
+
+// MARK: - OperatorDecl
+
+extension BridgedOperatorFixity {
+  fileprivate init(tokenKind: TokenKind) {
+    switch tokenKind {
+    case .keyword(.infix): self = .infix
+    case .keyword(.prefix): self = .prefix
+    case .keyword(.postfix): self = .postfix
+    default: fatalError("Unknown operator fixity")
+    }
+  }
+}
+
+extension ASTGenVisitor {
+  func visit(_ node: OperatorDeclSyntax) -> ASTNode {
+    let (name, nameLoc) = self.bridgedIdentifierAndSourceLoc(for: node.identifier)
+    let (precedenceGroupName, precedenceGroupLoc) = self.bridgedIdentifierAndSourceLoc(
+      for: node.operatorPrecedenceAndTypes?.precedenceGroup
+    )
+
+    return .decl(
+      OperatorDecl_create(
+        self.ctx,
+        self.declContext,
+        BridgedOperatorFixity(tokenKind: node.fixity.tokenKind),
+        self.bridgedSourceLoc(for: node.operatorKeyword),
+        name,
+        nameLoc,
+        self.bridgedSourceLoc(for: node.operatorPrecedenceAndTypes?.colon),
+        precedenceGroupName,
+        precedenceGroupLoc
+      )
+    )
+  }
+}

test/ASTGen/verify-parse.swift

@@ -159,3 +159,15 @@ extension
 Class: Proto2 where T: Proto1 {
   func method2(_ b: Bool) {}
 }
+
+prefix
+operator ⎭^-^⎭
+
+infix
+operator
+  ~^-^~
+  :
+  AdditionPrecedence
+
+postfix
+operator ⎩^-^⎩