Parse SwitchExprSyntax

Update the parser to support parsing SwitchExprSyntax.
For now, this only updates the existing statement
parsing, and does not attempt to parse in
expression position.

Author: hamishknight

Sources/SwiftParser/Expressions.swift

@@ -2571,18 +2571,20 @@ extension Parser {
   }
 }
 
-// MARK: Switch Statements
+// MARK: Switch Statements/Expressions
 
 extension Parser {
-  /// Parse a switch statement.
+  /// Parse a switch statement/expression.
   ///
   /// Grammar
   /// =======
   ///
-  ///     switch-statement → 'switch' expression '{' switch-cases? '}'
+  ///     switch-expression → 'switch' expression '{' switch-cases? '}'
   ///     switch-cases → switch-case switch-cases?
   @_spi(RawSyntax)
-  public mutating func parseSwitchStatement(switchHandle: RecoveryConsumptionHandle) -> RawSwitchStmtSyntax {
+  public mutating func parseSwitchExpression(
+    switchHandle: RecoveryConsumptionHandle
+  ) -> RawSwitchExprSyntax {
     let (unexpectedBeforeSwitchKeyword, switchKeyword) = self.eat(switchHandle)
 
     let subject = self.parseExpression(.basic)
@@ -2591,7 +2593,7 @@ extension Parser {
     let cases = self.parseSwitchCases(allowStandaloneStmtRecovery: !lbrace.isMissing)
 
     let (unexpectedBeforeRBrace, rbrace) = self.expectRightBrace(leftBrace: lbrace, introducer: switchKeyword)
-    return RawSwitchStmtSyntax(
+    return RawSwitchExprSyntax(
       unexpectedBeforeSwitchKeyword,
       switchKeyword: switchKeyword,
       expression: subject,

Sources/SwiftParser/Statements.swift

@@ -106,7 +106,12 @@ extension Parser {
     case (.guardKeyword, let handle)?:
       return label(self.parseGuardStatement(guardHandle: handle), with: optLabel)
     case (.switchKeyword, let handle)?:
-      return label(self.parseSwitchStatement(switchHandle: handle), with: optLabel)
+      let switchExpr = self.parseSwitchExpression(switchHandle: handle)
+      let switchStmt = RawExpressionStmtSyntax(
+        expression: RawExprSyntax(switchExpr),
+        arena: self.arena
+      )
+      return label(switchStmt, with: optLabel)
     case (.breakKeyword, let handle)?:
       return label(self.parseBreakStatement(breakHandle: handle), with: optLabel)
     case (.continueKeyword, let handle)?:

Tests/SwiftSyntaxBuilderTest/SwitchTests.swift

@@ -16,7 +16,7 @@ import SwiftSyntaxBuilder
 
 final class SwitchTests: XCTestCase {
   func testSwitch() {
-    let syntax = SwitchStmt(expression: Expr("count")) {
+    let syntax = SwitchExpr(expression: Expr("count")) {
       for num in 1..<3 {
         SwitchCase("case \(literal: num):") {
           Expr("print(count)")