Add diagnostic for dictionary type

Author: kimdv

Sources/SwiftParser/Types.swift

@@ -1071,6 +1071,22 @@ extension Parser {
         )
       }
 
+        if let colon = self.consume(if: .colon) {
+            let secondType = self.parseSimpleType()
+            let rightSquareBracket = self.consume(if: .rightSquareBracket) ?? self.missingToken(.rightSquareBracket)
+
+            result = RawTypeSyntax(
+                RawDictionaryTypeSyntax(
+                    leftSquareBracket: self.missingToken(.leftSquareBracket),
+                    keyType: result,
+                    colon: colon,
+                    valueType: secondType,
+                    rightSquareBracket: rightSquareBracket,
+                    arena: self.arena
+                )
+            )
+        }
+
       if !self.currentToken.isAtStartOfLine {
         if self.at(.postfixQuestionMark) {
           result = RawTypeSyntax(self.parseOptionalType(result))

Tests/SwiftParserTest/translated/RecoveryTests.swift

@@ -1122,23 +1122,29 @@ final class RecoveryTests: XCTestCase {
     assertParse(
       """
       struct ErrorTypeInVarDeclDictionaryType {
-        let a1: String1️⃣:
-        let a2: String2️⃣: Int]
-        let a3: String3️⃣: [Int]
-        let a4: String4️⃣: Int
+        let a1: 1️⃣String:2️⃣
+        let a2: 3️⃣String: Int]
+        let a3: 4️⃣String: [Int]5️⃣
+        let a4: 6️⃣String: Int7️⃣
       }
       """,
       diagnostics: [
-        // TODO: Old parser expected error on line 2: unexpected ':' in type; did you mean to write a dictionary type?, Fix-It replacements: 11 - 11 = '['
-        DiagnosticSpec(locationMarker: "1️⃣", message: "consecutive declarations on a line must be separated by ';'"),
-        DiagnosticSpec(locationMarker: "1️⃣", message: "unexpected code ':' before variable"),
-        // TODO: Old parser expected error on line 3: unexpected ':' in type; did you mean to write a dictionary type?, Fix-It replacements: 11 - 11 = '['
-        DiagnosticSpec(locationMarker: "2️⃣", message: "unexpected code ': Int]' before variable"),
-        // TODO: Old parser expected error on line 4: unexpected ':' in type; did you mean to write a dictionary type?, Fix-It replacements: 11 - 11 = '[', 24 - 24 = ']'
-        DiagnosticSpec(locationMarker: "3️⃣", message: "unexpected code ': [Int]' before variable"),
-        // TODO: Old parser expected error on line 5: unexpected ':' in type; did you mean to write a dictionary type?, Fix-It replacements: 11 - 11 = '[', 22 - 22 = ']'
-        DiagnosticSpec(locationMarker: "4️⃣", message: "unexpected code ': Int' in struct"),
-      ]
+        DiagnosticSpec(locationMarker: "1️⃣", message: "expected '[' to start dictionary type"),
+        DiagnosticSpec(locationMarker: "2️⃣", message: "expected value type and ']' to end dictionary type"),
+        DiagnosticSpec(locationMarker: "3️⃣", message: "expected '[' to start dictionary type"),
+        DiagnosticSpec(locationMarker: "4️⃣", message: "expected '[' to start dictionary type"),
+        DiagnosticSpec(locationMarker: "5️⃣", message: "expected ']' to end dictionary type"),
+        DiagnosticSpec(locationMarker: "6️⃣", message: "expected '[' to start dictionary type"),
+        DiagnosticSpec(locationMarker: "7️⃣", message: "expected ']' to end dictionary type"),
+      ],
+      fixedSource: """
+      struct ErrorTypeInVarDeclDictionaryType {
+        let a1: [String: <#type#> ]
+        let a2: [String: Int]
+        let a3: [String: [Int]]
+        let a4: [String: Int]
+      }
+      """
     )
   }