feat(intellij): support for more semantic highlight tokens

Author: d-biehl

intellij-client/build.gradle.kts

@@ -4,6 +4,7 @@ import org.jetbrains.intellij.platform.gradle.Constants.Constraints
 import org.jetbrains.intellij.platform.gradle.IntelliJPlatformType
 import org.jetbrains.intellij.platform.gradle.TestFrameworkType
 import org.jetbrains.intellij.platform.gradle.tasks.PrepareSandboxTask
+import org.jetbrains.intellij.platform.gradle.tasks.aware.IntelliJPlatformVersionAware
 import org.jetbrains.kotlin.gradle.dsl.JvmTarget
 import org.jetbrains.kotlin.gradle.dsl.KotlinVersion
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
@@ -204,6 +205,13 @@ tasks.withType<KotlinCompile> {
     prepareSandboxTask(prepareSandboxConfig)
 }
 
+@Suppress("unused") val runIdePyCharmCommunityEAP by intellijPlatformTesting.runIde.registering {
+    type = IntelliJPlatformType.PyCharmCommunity
+    version = "LATEST-EAP-SNAPSHOT"
+    
+    prepareSandboxTask(prepareSandboxConfig)
+}
+
 @Suppress("unused") val runIdeIntellijIdeaC by intellijPlatformTesting.runIde.registering {
     type = IntelliJPlatformType.IntellijIdeaCommunity
 

intellij-client/src/main/kotlin/dev/robotcode/robotcode4ij/configuration/RobotCodeColorSettingsPage.kt

@@ -17,6 +17,8 @@ class RobotCodeColorSettingsPage : ColorSettingsPage {
         AttributesDescriptor("Test case name", Colors.TESTCASE_NAME),
         AttributesDescriptor("Keyword name", Colors.KEYWORD_NAME),
         AttributesDescriptor("Keyword call", Colors.KEYWORD_CALL),
+        AttributesDescriptor("Keyword call inner", Colors.KEYWORD_CALL_INNER),
+        AttributesDescriptor("Name call", Colors.NAME_CALL),
         AttributesDescriptor("Setting", Colors.SETTING),
         AttributesDescriptor("Setting import", Colors.SETTING_IMPORT),
         AttributesDescriptor("Control flow", Colors.CONTROL_FLOW),

intellij-client/src/main/kotlin/dev/robotcode/robotcode4ij/highlighting/Colors.kt

@@ -16,6 +16,10 @@ object Colors {
         createTextAttributesKey("ROBOTFRAMEWORK_KEYWORD_NAME", DefaultLanguageHighlighterColors.FUNCTION_DECLARATION)
     val KEYWORD_CALL: TextAttributesKey =
         createTextAttributesKey("ROBOTFRAMEWORK_KEYWORD_CALL", DefaultLanguageHighlighterColors.FUNCTION_CALL)
+    val KEYWORD_CALL_INNER: TextAttributesKey =
+        createTextAttributesKey("ROBOTFRAMEWORK_KEYWORD_CALL_INNER", DefaultLanguageHighlighterColors.FUNCTION_CALL)
+    val NAME_CALL: TextAttributesKey =
+        createTextAttributesKey("ROBOTFRAMEWORK_NAME_CALL", DefaultLanguageHighlighterColors.FUNCTION_CALL)
 
     val SETTING: TextAttributesKey =
         createTextAttributesKey("ROBOTFRAMEWORK_SETTING", DefaultLanguageHighlighterColors.KEYWORD)

intellij-client/src/main/kotlin/dev/robotcode/robotcode4ij/lsp/RobotCodeSemanticTokensColorsProvider.kt

@@ -1,7 +1,9 @@
 package dev.robotcode.robotcode4ij.lsp
 
+import com.intellij.openapi.diagnostic.thisLogger
 import com.intellij.openapi.editor.colors.TextAttributesKey
 import com.intellij.psi.PsiFile
+import com.redhat.devtools.lsp4ij.features.semanticTokens.DefaultSemanticTokensColorsProvider
 import com.redhat.devtools.lsp4ij.features.semanticTokens.SemanticTokensColorsProvider
 import dev.robotcode.robotcode4ij.highlighting.Colors
 
@@ -22,7 +24,8 @@ private val mapping by lazy {
         "testcaseName" to Colors.TESTCASE_NAME,
         "keywordName" to Colors.KEYWORD_NAME,
         "keywordCall" to Colors.KEYWORD_CALL,
-        
+        "keywordCallInner" to Colors.KEYWORD_CALL_INNER,
+        "nameCall" to Colors.NAME_CALL,
         "argument" to Colors.ARGUMENT,
         "embeddedArgument" to Colors.EMBEDDED_ARGUMENT,
         "namedArgument" to Colors.NAMED_ARGUMENT,
@@ -37,11 +40,16 @@ private val mapping by lazy {
     )
 }
 
-class RobotCodeSemanticTokensColorsProvider : SemanticTokensColorsProvider {
+class RobotCodeSemanticTokensColorsProvider : DefaultSemanticTokensColorsProvider() {
     override fun getTextAttributesKey(
         tokenType: String, tokenModifiers: MutableList<String>, file: PsiFile
     ): TextAttributesKey? {
-        return mapping[tokenType]
+        val result = mapping[tokenType] ?: super.getTextAttributesKey(tokenType, tokenModifiers, file)
+        
+        return result ?: run {
+            thisLogger().warn("Unknown token type: $tokenType")
+            null
+        }
     }
 }