ASTGen: Translate operator declarations

Author: AnthonyLatsis

include/swift/AST/CASTBridging.h

@@ -412,6 +412,21 @@ BridgedDeclContextAndDecl ExtensionDecl_create(
     BridgedArrayRef cInheritedTypes, void *_Nullable opaqueGenericWhereClause,
     BridgedSourceRange cBraceRange);
 
+typedef enum ENUM_EXTENSIBILITY_ATTR(closed) {
+  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,
                               BridgedArrayRef cParameters,

lib/AST/CASTBridging.cpp

@@ -747,6 +747,46 @@ 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

@@ -3,6 +3,7 @@ import CASTBridging
 // Needed to use SyntaxTransformVisitor's visit method.
 @_spi(SyntaxTransformVisitor)
 import SwiftSyntax
+import SwiftDiagnostics
 
 // MARK: - TypeDecl
 
@@ -262,6 +263,48 @@ extension ASTGenVisitor {
   }
 }
 
+// MARK: - OperatorDecl
+
+extension BridgedOperatorFixity {
+  fileprivate init?(from tokenKind: TokenKind) {
+    switch tokenKind {
+    case .keyword(.infix): self = .infix
+    case .keyword(.prefix): self = .prefix
+    case .keyword(.postfix): self = .postfix
+    default: return nil
+    }
+  }
+}
+
+extension ASTGenVisitor {
+  func visit(_ node: OperatorDeclSyntax) -> ASTNode {
+    let (name, nameLoc) = node.name.bridgedIdentifierAndSourceLoc(in: self)
+    let (precedenceGroupName, precedenceGroupLoc) = (node.operatorPrecedenceAndTypes?.precedenceGroup).bridgedIdentifierAndSourceLoc(in: self)
+
+    let fixity: BridgedOperatorFixity
+    if let value = BridgedOperatorFixity(from: node.fixitySpecifier.tokenKind) {
+      fixity = value
+    } else {
+      fixity = .infix
+      self.diagnose(Diagnostic(node: node.fixitySpecifier, message: UnexpectedTokenKindError(token: node.fixitySpecifier)))
+    }
+
+    return .decl(
+      OperatorDecl_create(
+        self.ctx,
+        self.declContext,
+        fixity,
+        self.bridgedSourceLoc(for: node.operatorKeyword),
+        name,
+        nameLoc,
+        self.bridgedSourceLoc(for: node.operatorPrecedenceAndTypes?.colon),
+        precedenceGroupName,
+        precedenceGroupLoc
+      )
+    )
+  }
+}
+
 extension ASTGenVisitor {
   @inline(__always)
   func visit(_ node: MemberBlockItemListSyntax) -> BridgedArrayRef {

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 ⎩^-^⎩