Handle missing comma i func params

Author: kimdv

Sources/SwiftParser/Parameters.swift

@@ -156,7 +156,16 @@ extension Parser {
       defaultValue = nil
     }
 
-    let trailingComma = self.consume(if: .comma)
+    let canParseIdentifier: Bool = withLookahead {
+      $0.canParseTypeIdentifier(allowKeyword: false)
+    }
+
+    var trailingComma: Token?
+    if canParseIdentifier && !self.at(.rightParen) && !self.at(.comma) {
+      trailingComma = Token(missing: .comma, arena: self.arena)
+    } else {
+      trailingComma = self.consume(if: .comma)
+    }
 
     return RawFunctionParameterSyntax(
       attributes: attrs,

Tests/SwiftParserTest/DeclarationTests.swift

@@ -1516,7 +1516,8 @@ final class DeclarationTests: ParserTestCase {
           leftSquare: .leftSquareToken(),
           element: IdentifierTypeSyntax(name: .identifier("third")),
           rightSquare: .rightSquareToken(presence: .missing)
-        )
+        ),
+        trailingComma: .commaToken(presence: .missing)
       ),
       diagnostics: [
         DiagnosticSpec(
@@ -1532,11 +1533,12 @@ final class DeclarationTests: ParserTestCase {
         ),
         DiagnosticSpec(
           locationMarker: "4️⃣",
-          message: "unexpected code 'fourth: Int' in parameter clause"
+          message: "expected ',' in parameter",
+          fixIts: ["insert ','"]
         ),
       ],
       fixedSource: """
-        func foo(first second: [third]fourth: Int) {}
+        func foo(first second: [third], fourth: Int) {}
         """
     )
   }
@@ -3402,4 +3404,36 @@ final class DeclarationTests: ParserTestCase {
       ]
     )
   }
+
+  func testMissingCommaInParameters() {
+    assertParse(
+      "func a(foo: Bar1️⃣ foo2: Bar2) {}",
+      diagnostics: [
+        DiagnosticSpec(
+          message: "expected ',' in parameter",
+          fixIts: ["insert ','"]
+        ),
+      ],
+      fixedSource: "func a(foo: Bar, foo2: Bar2) {}"
+    )
+  }
+
+  func testMissingMultipleCommasInParameters() {
+    assertParse(
+      "func a(foo: Bar1️⃣ foo2: Bar2, foo3: Bar32️⃣ foo4: Bar4) {}",
+      diagnostics: [
+        DiagnosticSpec(
+          locationMarker: "1️⃣",
+          message: "expected ',' in parameter",
+          fixIts: ["insert ','"]
+        ),
+        DiagnosticSpec(
+          locationMarker: "2️⃣",
+          message: "expected ',' in parameter",
+          fixIts: ["insert ','"]
+        ),
+      ],
+      fixedSource: "func a(foo: Bar, foo2: Bar2, foo3: Bar3, foo4: Bar4) {}"
+    )
+  }
 }