Add semantic highlighting for function parameter labels

Author: ahoppen

Sources/SourceKitLSP/Swift/SemanticTokens.swift

@@ -155,6 +155,8 @@ extension SyntaxClassification {
       return (.comment, [])
     case .docLineComment, .docBlockComment:
       return (.comment, .documentation)
+    case .parameterLabel:
+      return (.function, [])
     }
   }
 }

Tests/SourceKitLSPTests/SemanticTokensTests.swift

@@ -430,7 +430,7 @@ final class SemanticTokensTests: XCTestCase {
       [
         Token(line: 0, utf16index: 0, length: 4, kind: .keyword),
         Token(line: 0, utf16index: 5, length: 1, kind: .identifier),
-        Token(line: 0, utf16index: 7, length: 1, kind: .identifier),
+        Token(line: 0, utf16index: 7, length: 1, kind: .function),
         Token(line: 0, utf16index: 10, length: 3, kind: .struct, modifiers: .defaultLibrary),
         Token(line: 0, utf16index: 17, length: 1, kind: .identifier),
         Token(line: 0, utf16index: 20, length: 6, kind: .struct, modifiers: .defaultLibrary),
@@ -833,13 +833,34 @@ final class SemanticTokensTests: XCTestCase {
         Token(line: 2, utf16index: 7, length: 8, kind: .identifier),
         Token(line: 4, utf16index: 0, length: 4, kind: .keyword),
         Token(line: 4, utf16index: 5, length: 1, kind: .identifier),
-        Token(line: 5, utf16index: 4, length: 1, kind: .identifier),
+        Token(line: 5, utf16index: 4, length: 1, kind: .function),
         Token(line: 5, utf16index: 7, length: 7, kind: .actor),
-        Token(line: 6, utf16index: 4, length: 1, kind: .identifier),
+        Token(line: 6, utf16index: 4, length: 1, kind: .function),
         Token(line: 6, utf16index: 7, length: 8, kind: .struct),
       ]
     )
   }
+
+  func testArgumentLabels() async throws {
+    let text = """
+      func foo(arg: Int) {}
+      foo(arg: 1)
+      """
+
+    let tokens = try await openAndPerformSemanticTokensRequest(text: text)
+    XCTAssertEqual(
+      tokens,
+      [
+        Token(line: 0, utf16index: 0, length: 4, kind: .keyword),
+        Token(line: 0, utf16index: 5, length: 3, kind: .identifier),
+        Token(line: 0, utf16index: 9, length: 3, kind: .function),
+        Token(line: 0, utf16index: 14, length: 3, kind: .struct, modifiers: .defaultLibrary),
+        Token(line: 1, utf16index: 0, length: 3, kind: .function),
+        Token(line: 1, utf16index: 4, length: 3, kind: .function),
+        Token(line: 1, utf16index: 9, length: 1, kind: .number),
+      ]
+    )
+  }
 }
 
 extension Token {