feat(completion): improve file reference completion by using editor h…

…istory and project directory. #101
unit-mesh · Mar 14, 2024 · 78d95fb · 78d95fb
1 parent 1d2e422
commit 78d95fb
Showing 1 changed file with 25 additions and 10 deletions.
diff --git a/...ng/src/main/kotlin/cc/unitmesh/devti/language/completion/FileReferenceLanguageProvider.kt b/...ng/src/main/kotlin/cc/unitmesh/devti/language/completion/FileReferenceLanguageProvider.kt
@@ -1,10 +1,13 @@
 package cc.unitmesh.devti.language.completion
 
-import com.intellij.codeInsight.completion.CompletionParameters
-import com.intellij.codeInsight.completion.CompletionProvider
-import com.intellij.codeInsight.completion.CompletionResultSet
+import com.intellij.codeInsight.completion.*
 import com.intellij.codeInsight.lookup.LookupElementBuilder
+import com.intellij.ide.presentation.VirtualFilePresentation
+import com.intellij.openapi.fileEditor.impl.EditorHistoryManager
+import com.intellij.openapi.project.guessProjectDir
+import com.intellij.openapi.vfs.VirtualFile
 import com.intellij.util.ProcessingContext
+import java.io.File
 
 class FileReferenceLanguageProvider : CompletionProvider<CompletionParameters>() {
     companion object {
@@ -16,13 +19,25 @@ class FileReferenceLanguageProvider : CompletionProvider<CompletionParameters>()
         context: ProcessingContext,
         result: CompletionResultSet,
     ) {
-        // sample file: "file1", "file2"
-        listOf("file1", "file2").forEach {
-            result.addElement(
-                LookupElementBuilder.create(it)
-                    .withTypeText("file", true)
-            )
+        val project = parameters.position.project
+        val basePath = project.guessProjectDir()?.path ?: return
+
+        val editorHistoryManager = EditorHistoryManager.getInstance(project)
+        val fileList: List<VirtualFile> = editorHistoryManager.fileList
+
+        fileList.forEach {
+            val removePrefix = it.path.removePrefix(basePath)
+            val relativePath: String = removePrefix.removePrefix(File.separator)
+
+            val element = LookupElementBuilder.create(relativePath)
+                .withIcon(VirtualFilePresentation.getIcon(it))
+                .withInsertHandler { context, _ ->
+                    context.editor.caretModel.moveCaretRelatively(
+                        1, 0, false, false, false
+                    )
+                }
+
+            result.addElement(element)
         }
     }
-
 }