[Performance] Minimize RawTokenKindSubset.allCases usage

'allCases' is not trivial because it creates an Array. We should avoid
creating temporary array if possible.

Author: rintaro

Sources/SwiftParser/Attributes.swift

@@ -311,6 +311,15 @@ extension Parser {
       case integerLiteral
       case selfKeyword
 
+      init?(lexeme: Lexer.Lexeme) {
+        switch lexeme.tokenKind {
+        case .identifier: self = .identifier
+        case .integerLiteral: self = .integerLiteral
+        case .selfKeyword: self = .selfKeyword
+        default: return nil
+        }
+      }
+
       var rawTokenKind: RawTokenKind {
         switch self {
         case .identifier: return .identifier

Sources/SwiftParser/Declarations.swift

@@ -393,10 +393,14 @@ extension Parser {
           case postfixOperator
           case prefixOperator
 
-          func accepts(lexeme: Lexer.Lexeme) -> Bool {
-            switch self {
-            case .colon: return true
-            default: return lexeme.tokenText == "=="
+          init?(lexeme: Lexer.Lexeme) {
+            switch (lexeme.tokenKind, lexeme.tokenText) {
+            case (.colon, _): self = .colon
+            case (.spacedBinaryOperator, "=="): self = .spacedBinaryOperator
+            case (.unspacedBinaryOperator, "=="): self = .unspacedBinaryOperator
+            case (.postfixOperator, "=="): self = .postfixOperator
+            case (.prefixOperator, "=="): self = .prefixOperator
+            default: return nil
             }
           }
 
@@ -2045,6 +2049,14 @@ extension Parser {
       case poundErrorKeyword
       case poundWarningKeyword
 
+      init?(lexeme: Lexer.Lexeme) {
+        switch lexeme.tokenKind {
+        case .poundErrorKeyword: self = .poundErrorKeyword
+        case .poundWarningKeyword: self = .poundWarningKeyword
+        default: return nil
+        }
+      }
+
       var rawTokenKind: RawTokenKind {
         switch self {
         case .poundErrorKeyword: return .poundErrorKeyword

Sources/SwiftParser/Expressions.swift

@@ -173,6 +173,21 @@ extension Parser {
       case arrow
       case throwsKeyword
 
+      init?(lexeme: Lexer.Lexeme) {
+        switch lexeme.tokenKind {
+        case .spacedBinaryOperator: self = .spacedBinaryOperator
+        case .unspacedBinaryOperator: self = .unspacedBinaryOperator
+        case .infixQuestionMark: self = .infixQuestionMark
+        case .equal: self = .equal
+        case .isKeyword: self = .isKeyword
+        case .asKeyword: self = .asKeyword
+        case .identifier where lexeme.tokenText == "async": self = .async
+        case .arrow: self = .arrow
+        case .throwsKeyword: self = .throwsKeyword
+        default: return nil
+        }
+      }
+
       var rawTokenKind: RawTokenKind {
         switch self {
         case .spacedBinaryOperator: return .spacedBinaryOperator

Sources/SwiftParser/Lookahead.swift

@@ -283,20 +283,6 @@ extension Parser.Lookahead {
     "_opaqueReturnTypeOf",
   ]
 
-  func isParenthesizedUnowned() -> Bool {
-    assert(self.atContextualKeyword("unowned") && self.peek().tokenKind == .leftParen,
-           "Invariant violated")
-
-    // Look ahead to parse the parenthesized expression.
-    var lookahead = self.lookahead()
-    lookahead.expectIdentifierWithoutRecovery()
-    guard lookahead.consume(if: .leftParen) != nil else {
-      return false
-    }
-    return lookahead.at(.identifier)
-        && lookahead.peek().tokenKind == .rightParen
-        && (lookahead.atContextualKeyword("safe") || lookahead.atContextualKeyword("unsafe"))
-  }
 }
 
 extension Parser.Lookahead {
@@ -365,6 +351,18 @@ extension Parser.Lookahead {
       case poundElseKeyword
       case poundElseifKeyword
 
+      init?(lexeme: Lexer.Lexeme) {
+        switch lexeme.tokenKind {
+        case .leftParen: self = .leftParen
+        case .leftBrace: self = .leftBrace
+        case .leftSquareBracket: self = .leftSquareBracket
+        case .poundIfKeyword: self = .poundIfKeyword
+        case .poundElseKeyword: self = .poundElseKeyword
+        case .poundElseifKeyword: self = .poundElseifKeyword
+        default: return nil
+        }
+      }
+
       var rawTokenKind: RawTokenKind {
         switch self {
         case .leftParen: return .leftParen

Sources/SwiftParser/Names.swift

@@ -227,15 +227,15 @@ extension Lexer.Lexeme {
   }
 
   func isContextualPunctuator(_ name: SyntaxText) -> Bool {
-    return Operator(self) != nil && self.tokenText == name
+    return Operator(lexeme: self) != nil && self.tokenText == name
   }
 
   var isKeyword: Bool {
     self.tokenKind.isKeyword
   }
 
   func starts(with symbol: SyntaxText) -> Bool {
-    guard Operator(self) != nil || self.tokenKind.isPunctuation else {
+    guard Operator(lexeme: self) != nil || self.tokenKind.isPunctuation else {
       return false
     }
 

Sources/SwiftParser/Patterns.swift

@@ -52,6 +52,17 @@ extension Parser {
       case letKeyword
       case varKeyword
 
+      init?(lexeme: Lexer.Lexeme) {
+        switch lexeme.tokenKind {
+        case .leftParen: self = .leftParen
+        case .wildcardKeyword: self = .wildcardKeyword
+        case .identifier: self = .identifier
+        case .letKeyword: self = .letKeyword
+        case .varKeyword: self = .varKeyword
+        default: return nil
+        }
+      }
+
       var rawTokenKind: RawTokenKind {
         switch self {
         case .leftParen: return .leftParen
@@ -213,6 +224,17 @@ extension Parser.Lookahead {
       case varKeyword
       case leftParen
 
+      init?(lexeme: Lexer.Lexeme) {
+        switch lexeme.tokenKind {
+        case .identifier: self = .identifier
+        case .wildcardKeyword: self = .wildcardKeyword
+        case .letKeyword: self = .letKeyword
+        case .varKeyword: self = .varKeyword
+        case .leftParen: self = .leftParen
+        default: return nil
+        }
+      }
+
       var rawTokenKind: RawTokenKind {
         switch self {
         case .identifier: return .identifier