Fix wrong fix it for replacing to module for OriginallyDefinedInArgumentsSyntax

Author: kimdv

Sources/SwiftParserDiagnostics/ParseDiagnosticsGenerator.swift

@@ -993,6 +993,35 @@ public class ParseDiagnosticsGenerator: SyntaxAnyVisitor {
     return handleMissingSyntax(node, additionalHandledNodes: [node.placeholder.id])
   }
 
+  override open func visit(_ node: OriginallyDefinedInArgumentsSyntax) -> SyntaxVisitorContinueKind {
+    if shouldSkip(node) {
+      return .skipChildren
+    }
+    if let token = node.unexpectedBetweenModuleLabelAndColon?.onlyToken(where: { $0.tokenKind.isIdentifier }),
+      node.moduleLabel.presence == .missing
+    {
+      addDiagnostic(
+        node,
+        MissingNodesError(missingNodes: [Syntax(node.moduleLabel)]),
+        fixIts: [
+          FixIt(
+            message: ReplaceTokensFixIt(
+              replaceTokens: [token],
+              replacements: [node.moduleLabel]
+            ),
+            changes: [
+              FixIt.MultiNodeChange.makeMissing(token),
+              FixIt.MultiNodeChange.makePresentBeforeTrivia(node.moduleLabel),
+            ]
+          )
+        ],
+        handledNodes: [node.moduleLabel.id, token.id]
+      )
+    }
+
+    return .visitChildren
+  }
+
   public override func visit(_ node: OperatorDeclSyntax) -> SyntaxVisitorContinueKind {
     if shouldSkip(node) {
       return .skipChildren

Tests/SwiftParserTest/AttributeTests.swift

@@ -425,16 +425,20 @@ final class AttributeTests: XCTestCase {
       @_expose(Cxx, 1️⃣baz2️⃣) func foo() {}
       """,
       diagnostics: [
-        DiagnosticSpec(locationMarker: "1️⃣",
-                       message: #"expected '"' in string literal"#,
-                       fixIts: [#"insert '"'"#]),
-        DiagnosticSpec(locationMarker: "2️⃣",
-                       message: #"expected '"' to end string literal"#,
-                      fixIts: [#"insert '"'"#]),
+        DiagnosticSpec(
+          locationMarker: "1️⃣",
+          message: #"expected '"' in string literal"#,
+          fixIts: [#"insert '"'"#]
+        ),
+        DiagnosticSpec(
+          locationMarker: "2️⃣",
+          message: #"expected '"' to end string literal"#,
+          fixIts: [#"insert '"'"#]
+        ),
       ],
-      fixedSource:"""
-      @_expose(Cxx, "baz") func foo() {}
-      """
+      fixedSource: """
+        @_expose(Cxx, "baz") func foo() {}
+        """
     )
   }
 
@@ -500,13 +504,25 @@ final class AttributeTests: XCTestCase {
 
     assertParse(
       """
-      @_unavailableFromAsync(message: 1️⃣abc)
+      @_unavailableFromAsync(message: 1️⃣abc2️⃣)
       func foo() {}
       """,
       diagnostics: [
-        DiagnosticSpec(message: "expected string literal to end @_unavailableFromAsync argument", fixIts: ["insert string literal"]),
-        DiagnosticSpec(message: "unexpected code 'abc' in attribute"),
-      ]
+        DiagnosticSpec(
+          locationMarker: "1️⃣",
+          message: #"expected '"' in string literal"#,
+          fixIts: [#"insert '"'"#]
+        ),
+        DiagnosticSpec(
+          locationMarker: "2️⃣",
+          message: #"expected '"' to end string literal"#,
+          fixIts: [#"insert '"'"#]
+        ),
+      ],
+      fixedSource: """
+        @_unavailableFromAsync(message: "abc")
+        func foo() {}
+        """
     )
   }
 

Tests/SwiftParserTest/translated/OriginalDefinedInAttrTests.swift

@@ -31,9 +31,15 @@ final class OriginalDefinedInAttrTests: XCTestCase {
       public func foo1() {}
       """#,
       diagnostics: [
-        DiagnosticSpec(message: "expected 'module' in @_originallyDefinedIn arguments", fixIts: ["insert 'module'"]),
-        DiagnosticSpec(message: "unexpected code 'modulename' before @_originallyDefinedIn arguments"),
-      ]
+        DiagnosticSpec(
+          message: "expected 'module' in @_originallyDefinedIn arguments",
+          fixIts: ["replace 'modulename' with 'module'"]
+        )
+      ],
+      fixedSource: #"""
+        @_originallyDefinedIn(module: "foo", OSX 13.13)
+        public func foo1() {}
+        """#
     )
   }
 
@@ -53,20 +59,31 @@ final class OriginalDefinedInAttrTests: XCTestCase {
       public class ToplevelClass1 {}
       """#,
       diagnostics: [
-        DiagnosticSpec(locationMarker: "1️⃣", message: "expected ',' and version list in @_originallyDefinedIn arguments", fixIts: ["insert ',' and version list"])
+        DiagnosticSpec(
+          message: "expected ',' and version list in @_originallyDefinedIn arguments",
+          fixIts: ["insert ',' and version list"]
+        )
       ]
     )
   }
 
   func testOriginalDefinedInAttr5() {
     assertParse(
       """
-      @_originallyDefinedIn(1️⃣OSX 13.13.3)
+      @_originallyDefinedIn(1️⃣OSX 2️⃣13.13.3)
       public class ToplevelClass2 {}
       """,
       diagnostics: [
-        DiagnosticSpec(message: "expected 'module:', string literal, and ',' in @_originallyDefinedIn arguments", fixIts: ["insert 'module:', string literal, and ','"])
-      ]
+        DiagnosticSpec(locationMarker: "1️⃣", message: "expected 'module:' in @_originallyDefinedIn arguments", fixIts: ["insert 'module:'"]),
+        DiagnosticSpec(locationMarker: "1️⃣", message: #"expected '"' in string literal"#, fixIts: [#"insert '"'"#]),
+        DiagnosticSpec(locationMarker: "2️⃣", message: #"expected '"' to end string literal"#, fixIts: [#"insert '"'"#]),
+        DiagnosticSpec(locationMarker: "2️⃣", message: "expected ',' in @_originallyDefinedIn arguments", fixIts: ["insert ','"]),
+        DiagnosticSpec(locationMarker: "2️⃣", message: "expected platform in version restriction", fixIts: ["insert platform"]),
+      ],
+      fixedSource: """
+        @_originallyDefinedIn(module: "OSX" , <#identifier#>13.13.3)
+        public class ToplevelClass2 {}
+        """
     )
   }
 
@@ -172,5 +189,4 @@ final class OriginalDefinedInAttrTests: XCTestCase {
       """
     )
   }
-
 }