Merge pull request #1261 from kimdv/kimdv/add-node-with-body-to-switch

Add init with `PartialSyntaxNodeString` on `SwitchStmtSyntax`

Author: kimdv

Sources/SwiftSyntaxBuilder/SyntaxNodeWithBody.swift

@@ -93,7 +93,7 @@ extension ExtensionDeclSyntax: HasTrailingMemberDeclBlock {}
 extension ProtocolDeclSyntax: HasTrailingMemberDeclBlock {}
 extension StructDeclSyntax: HasTrailingMemberDeclBlock {}
 
-// MARK: - IfStmt
+// MARK: - IfStmtSyntax
 // IfStmtSyntax is a special scenario as we also have the `else` body or an if-else
 // So we cannot conform to `HasTrailingCodeBlock`
 
@@ -105,10 +105,21 @@ public extension IfStmtSyntax {
     self.elseKeyword = elseBody != nil ? .keyword(.else) : nil
   }
 
-  init(_ signature: String, @CodeBlockItemListBuilder bodyBuilder: () -> CodeBlockItemListSyntax, elseIf: IfStmtSyntax) {
+  init(_ signature: PartialSyntaxNodeString, @CodeBlockItemListBuilder bodyBuilder: () -> CodeBlockItemListSyntax, elseIf: IfStmtSyntax) {
     self = "\(signature) {}"
     self.body = CodeBlockSyntax(statements: bodyBuilder())
     self.elseBody = .ifStmt(elseIf)
     self.elseKeyword = elseBody != nil ? .keyword(.else) : nil
   }
 }
+
+// MARK: - SwitchStmtSyntax
+// SwitchStmtSyntax is a special scenario as it don't have body or members
+// So we cannot conform to `HasTrailingCodeBlock` or `HasTrailingMemberDeclBlock`
+
+public extension SwitchStmtSyntax {
+  init(_ signature: PartialSyntaxNodeString, @SwitchCaseListBuilder casesBuilder: () -> SwitchCaseListSyntax = { SwitchCaseListSyntax([]) }) {
+    self = "\(signature) {}"
+    self.cases = casesBuilder()
+  }
+}

Tests/SwiftSyntaxBuilderTest/SwitchTests.swift

@@ -41,4 +41,31 @@ final class SwitchTests: XCTestCase {
       """
     )
   }
+
+  func testSwitchStmtSyntaxWithStringParsing() {
+    let syntax = SwitchStmtSyntax("switch count") {
+      for num in 1..<3 {
+        SwitchCaseSyntax("case \(literal: num):") {
+          ExprSyntax("print(count)")
+        }
+      }
+      SwitchCaseSyntax("default:") {
+        BreakStmtSyntax("break")
+      }
+    }
+
+    AssertBuildResult(
+      syntax,
+      """
+      switch count {
+      case 1:
+          print(count)
+      case 2:
+          print(count)
+      default:
+          break
+      }
+      """
+    )
+  }
 }