Add diagnostic for missing sequence expr in for-loop

Author: kimdv

Sources/SwiftParser/Statements.swift

@@ -616,7 +616,14 @@ extension Parser {
 
     let (unexpectedBeforeInKeyword, inKeyword) = self.expect(.keyword(.in))
 
-    let expr = self.parseExpression(.basic)
+    // if there is no expression, like `switch { default: return false }` then left brace will be parsed as
+    // an `RawClosureExprSyntax`.
+    let expr: RawExprSyntax
+    if self.at(.leftBrace) {
+      expr = RawExprSyntax(RawMissingExprSyntax(arena: self.arena))
+    } else {
+      expr = self.parseExpression(.basic)
+    }
 
     // Parse the 'where' expression if present.
     let whereClause: RawWhereClauseSyntax?

Sources/SwiftParserDiagnostics/ParseDiagnosticsGenerator.swift

@@ -583,7 +583,12 @@ public class ParseDiagnosticsGenerator: SyntaxAnyVisitor {
         ] as [Syntax?]).compactMap({ $0 }),
         handledNodes: [node.inKeyword.id, node.sequenceExpr.id, unexpectedCondition.id]
       )
+    } else {  // If it's not a C-style for loop
+      if node.sequenceExpr.is(MissingExprSyntax.self) {
+        addDiagnostic(node.sequenceExpr, .expectedSequenceExpressionInForEachLoop, handledNodes: [node.sequenceExpr.id])
+      }
     }
+
     return .visitChildren
   }
 

Sources/SwiftParserDiagnostics/ParserDiagnosticMessages.swift

@@ -131,6 +131,9 @@ extension DiagnosticMessage where Self == StaticParserError {
   public static var expectedExpressionAfterTry: Self {
     .init("expected expression after 'try'")
   }
+  public static var expectedSequenceExpressionInForEachLoop: Self {
+    .init("expected Sequence expression for for-each loop")
+  }
   public static var initializerInPattern: Self {
     .init("unexpected initializer in pattern; did you mean to use '='?")
   }

Tests/SwiftParserTest/translated/RecoveryTests.swift

@@ -459,13 +459,10 @@ final class RecoveryTests: XCTestCase {
     AssertParse(
       """
       for 1️⃣{
-      }2️⃣
+      }
       """,
       diagnostics: [
-        // TODO: Old parser expected error on line 1: expected pattern
-        // TODO: Old parser expected error on line 1: expected Sequence expression for for-each loop
-        DiagnosticSpec(locationMarker: "1️⃣", message: "expected pattern and 'in' in 'for' statement"),
-        DiagnosticSpec(locationMarker: "2️⃣", message: "expected code block in 'for' statement"),
+        DiagnosticSpec(locationMarker: "1️⃣", message: "expected pattern, 'in', and expression in 'for' statement")
       ]
     )
   }
@@ -475,12 +472,10 @@ final class RecoveryTests: XCTestCase {
       """
       for1️⃣
       {
-      }2️⃣
+      }
       """,
       diagnostics: [
-        // TODO: Old parser expected error on line 1: expected Sequence expression for for-each loop
-        DiagnosticSpec(locationMarker: "1️⃣", message: "expected pattern and 'in' in 'for' statement"),
-        DiagnosticSpec(locationMarker: "2️⃣", message: "expected code block in 'for' statement"),
+        DiagnosticSpec(locationMarker: "1️⃣", message: "expected pattern, 'in', and expression in 'for' statement")
       ]
     )
   }
@@ -489,12 +484,10 @@ final class RecoveryTests: XCTestCase {
     AssertParse(
       """
       for i 1️⃣{
-      }2️⃣
+      }
       """,
       diagnostics: [
-        // TODO: Old parser expected error on line 1: expected Sequence expression for for-each loop
-        DiagnosticSpec(locationMarker: "1️⃣", message: "expected 'in' in 'for' statement"),
-        DiagnosticSpec(locationMarker: "2️⃣", message: "expected code block in 'for' statement"),
+        DiagnosticSpec(locationMarker: "1️⃣", message: "expected 'in' and expression in 'for' statement")
       ]
     )
   }
@@ -503,12 +496,10 @@ final class RecoveryTests: XCTestCase {
     AssertParse(
       """
       for var i 1️⃣{
-      }2️⃣
+      }
       """,
       diagnostics: [
-        // TODO: Old parser expected error on line 1: expected Sequence expression for for-each loop
-        DiagnosticSpec(locationMarker: "1️⃣", message: "expected 'in' in 'for' statement"),
-        DiagnosticSpec(locationMarker: "2️⃣", message: "expected code block in 'for' statement"),
+        DiagnosticSpec(locationMarker: "1️⃣", message: "expected 'in' and expression in 'for' statement")
       ]
     )
   }
@@ -517,14 +508,11 @@ final class RecoveryTests: XCTestCase {
     AssertParse(
       """
       for 1️⃣in 2️⃣{
-      }3️⃣
+      }
       """,
       diagnostics: [
-        // TODO: Old parser expected error on line 1: expected pattern
-        // TODO: Old parser expected error on line 1: expected Sequence expression for for-each loop
         DiagnosticSpec(locationMarker: "1️⃣", message: "keyword 'in' cannot be used as an identifier here"),
-        DiagnosticSpec(locationMarker: "2️⃣", message: "expected 'in' in 'for' statement"),
-        DiagnosticSpec(locationMarker: "3️⃣", message: "expected code block in 'for' statement"),
+        DiagnosticSpec(locationMarker: "2️⃣", message: "expected 'in' and expression in 'for' statement"),
       ]
     )
   }
@@ -544,25 +532,24 @@ final class RecoveryTests: XCTestCase {
   func testRecovery41() {
     AssertParse(
       """
-      for 1️⃣for in {
-      }2️⃣
+      for 1️⃣for in 2️⃣{
+      }
       """,
       diagnostics: [
         DiagnosticSpec(locationMarker: "1️⃣", message: "keyword 'for' cannot be used as an identifier here"),
-        DiagnosticSpec(locationMarker: "2️⃣", message: "expected code block in 'for' statement"),
+        DiagnosticSpec(locationMarker: "2️⃣", message: "expected Sequence expression for for-each loop"),
       ]
     )
   }
 
   func testRecovery42() {
     AssertParse(
       """
-      for i in {
-      }1️⃣
+      for i in 1️⃣{
+      }
       """,
       diagnostics: [
-        // TODO: Old parser expected error on line 1: expected Sequence expression for for-each loop
-        DiagnosticSpec(message: "expected code block in 'for' statement")
+        DiagnosticSpec(message: "expected Sequence expression for for-each loop")
       ]
     )
   }