fix(devin-lang): improve code highlighting and completion #101

The commit fixes an issue with the DevInSyntaxHighlighter where the LANGUAGE_ID token was incorrectly highlighted as an instance field. It has been corrected to be highlighted as a constant. Additionally, the DevInCompletionContributor now uses appropriate patterns and imports to enhance the completion feature. Furthermore, the DevInTypedHandler has been updated to handle the `'` character more effectively during code completion.

Author: phodal

exts/devin-lang/src/main/kotlin/cc/unitmesh/devti/language/DevInTypedHandler.kt

@@ -2,14 +2,12 @@
 package cc.unitmesh.devti.language
 
 import cc.unitmesh.devti.language.psi.DevInFile
-import cc.unitmesh.devti.language.psi.DevInTypes
 import com.intellij.codeInsight.AutoPopupController
 import com.intellij.codeInsight.editorActions.TypedHandlerDelegate
 import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiDocumentManager
 import com.intellij.psi.PsiFile
-import com.intellij.psi.util.elementType
 
 class DevInTypedHandler : TypedHandlerDelegate() {
     override fun checkAutoPopup(charTyped: Char, project: Project, editor: Editor, file: PsiFile): Result {
@@ -19,15 +17,15 @@ class DevInTypedHandler : TypedHandlerDelegate() {
 
         return when (charTyped) {
             '`' -> {
-                val offset = editor.caretModel.primaryCaret.offset
-                if (offset == 0) {
-                    return Result.CONTINUE
-                }
-
-                val element = file.findElementAt(offset - 1)
-                if (element?.elementType == DevInTypes.CODE_CONTENT || element?.elementType == DevInTypes.CODE_BLOCK_END) {
-                    return Result.CONTINUE
-                }
+//                val offset = editor.caretModel.primaryCaret.offset
+//                if (offset == 0) {
+//                    return Result.CONTINUE
+//                }
+//
+//                val element = file.findElementAt(offset - 1)
+//                if (element?.elementType == DevInTypes.CODE_CONTENT || element?.elementType == DevInTypes.CODE_BLOCK_END) {
+//                    return Result.CONTINUE
+//                }
 
                 PsiDocumentManager.getInstance(project).commitDocument(editor.document)
                 AutoPopupController.getInstance(project).autoPopupMemberLookup(editor, null)

exts/devin-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/DevInCompletionContributor.kt

@@ -5,13 +5,18 @@ import cc.unitmesh.devti.language.psi.DevInTypes
 import com.intellij.codeInsight.completion.CompletionContributor
 import com.intellij.codeInsight.completion.CompletionInitializationContext
 import com.intellij.codeInsight.completion.CompletionType
+import com.intellij.patterns.*
 import com.intellij.patterns.PlatformPatterns.psiElement
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiWhiteSpace
+import com.intellij.psi.tree.TokenSet
+import com.intellij.util.ProcessingContext
 
 class DevInCompletionContributor : CompletionContributor() {
     private val INPUT_DUMMY_IDENTIFIER = "DevInDummy"
 
     init {
-        extend(CompletionType.BASIC, psiElement(DevInTypes.LANGUAGE_ID), CodeLanguageProvider())
+        extend(CompletionType.BASIC, declarationPattern(), CodeLanguageProvider())
         extend(CompletionType.BASIC, psiElement(DevInTypes.VARIABLE_ID), CustomVariableProvider())
     }
 
@@ -20,4 +25,8 @@ class DevInCompletionContributor : CompletionContributor() {
             context.dummyIdentifier = INPUT_DUMMY_IDENTIFIER
         }
     }
+
+    fun declarationPattern(): PsiElementPattern.Capture<PsiElement> =
+        psiElement()
+            .and(psiElement(DevInTypes.LANGUAGE_ID))
 }

exts/devin-lang/src/main/kotlin/cc/unitmesh/devti/language/highlight/DevInSyntaxHighlighter.kt

@@ -25,7 +25,7 @@ class DevInSyntaxHighlighter : SyntaxHighlighter {
 
             ATTRIBUTES[DevInTypes.CODE_BLOCK_START] = DefaultLanguageHighlighterColors.KEYWORD
             ATTRIBUTES[DevInTypes.CODE_BLOCK_END] = DefaultLanguageHighlighterColors.KEYWORD
-            ATTRIBUTES[DevInTypes.LANGUAGE_ID] = DefaultLanguageHighlighterColors.INSTANCE_FIELD
+            ATTRIBUTES[DevInTypes.LANGUAGE_ID] = DefaultLanguageHighlighterColors.CONSTANT
         }
     }