remove SpecItemCompletionProvider, move code to ReferenceProvider

pontem-network · Nov 8, 2024 · e9dc260 · e9dc260
1 parent 136d6f1
commit e9dc260
Show file tree

Hide file tree

Showing 7 changed files with 47 additions and 66 deletions.
diff --git a/src/main/kotlin/org/move/lang/core/completion/CommonCompletionContributor.kt b/src/main/kotlin/org/move/lang/core/completion/CommonCompletionContributor.kt
@@ -14,7 +14,7 @@ import org.move.lang.core.psi.ext.prevNonWsSibling
 class CommonCompletionContributor: CompletionContributor() {
     init {
         extend(CompletionType.BASIC, PrimitiveTypesCompletionProvider)
-        extend(CompletionType.BASIC, SpecItemCompletionProvider)
+//        extend(CompletionType.BASIC, SpecItemCompletionProvider)
 
         // addresses
         extend(CompletionType.BASIC, NamedAddressInUseStmtCompletionProvider)

diff --git a/src/main/kotlin/org/move/lang/core/completion/LookupElements.kt b/src/main/kotlin/org/move/lang/core/completion/LookupElements.kt
@@ -10,9 +10,7 @@ import com.intellij.psi.PsiElement
 import com.intellij.psi.util.PsiTreeUtil
 import org.move.lang.core.psi.*
 import org.move.lang.core.psi.ext.*
-import org.move.lang.core.resolve.ScopeEntry
 import org.move.lang.core.resolve2.ref.ResolutionContext
-import org.move.lang.core.types.infer.*
 import org.move.lang.core.types.ty.Ty
 
 const val KEYWORD_PRIORITY = 80.0
@@ -114,6 +112,16 @@ open class DefaultInsertHandler(val completionCtx: MvCompletionContext? = null):
         item: LookupElement
     ) {
         val document = context.document
+
+        val itemSpecRef = context.getElementOfType<MvItemSpecRef>()
+        if (itemSpecRef != null) {
+            // inserting item in `spec /*caret*/ {}`, no need for the signature, just insert with leading space
+            if (!context.alreadyHasSpace) {
+                context.addSuffix(" ")
+            }
+            return
+        }
+
         when (element) {
             is MvFunctionLike -> {
                 // no suffix for imports

diff --git a/src/main/kotlin/org/move/lang/core/completion/providers/ReferenceCompletionProvider.kt b/src/main/kotlin/org/move/lang/core/completion/providers/ReferenceCompletionProvider.kt
@@ -7,6 +7,7 @@ import com.intellij.patterns.PlatformPatterns
 import com.intellij.psi.PsiElement
 import com.intellij.util.ProcessingContext
 import org.move.lang.core.completion.MvCompletionContext
+import org.move.lang.core.psi.MvItemSpecRef
 import org.move.lang.core.psi.MvLabel
 import org.move.lang.core.psi.MvPatBinding
 import org.move.lang.core.psi.MvPatField
@@ -17,6 +18,7 @@ import org.move.lang.core.psiElement
 import org.move.lang.core.resolve.RsResolveProcessor
 import org.move.lang.core.resolve.collectCompletionVariants
 import org.move.lang.core.resolve.ref.MvReferenceElement
+import org.move.lang.core.resolve.ref.processItemSpecRefResolveVariants
 import org.move.lang.core.resolve.wrapWithFilter
 import org.move.lang.core.resolve2.processLabelResolveVariants
 import org.move.lang.core.resolve2.processPatBindingResolveVariants
@@ -46,9 +48,9 @@ object ReferenceCompletionProvider: MvCompletionProvider() {
     fun addCompletionVariants(
         element: MvReferenceElement,
         result: CompletionResultSet,
-        context: MvCompletionContext,
+        completionCtx: MvCompletionContext,
     ) {
-        collectCompletionVariants(result, context) {
+        collectCompletionVariants(result, completionCtx) {
             val processor0 = filterCompletionVariantsByVisibility(element, it)
             // todo: filter test functions
             when (element) {
@@ -59,7 +61,10 @@ object ReferenceCompletionProvider: MvCompletionProvider() {
                     val processor = skipAlreadyProvidedFields(element, processor0)
                     processPatBindingResolveVariants(element, true, processor)
                 }
+                // loop labels
                 is MvLabel -> processLabelResolveVariants(element, it)
+                // `spec ITEM {}` module items, where ITEM is a reference to the function/struct/enum
+                is MvItemSpecRef -> processItemSpecRefResolveVariants(element, it)
             }
         }
     }

diff --git a/src/main/kotlin/org/move/lang/core/completion/providers/SpecItemCompletionProvider.kt b/src/main/kotlin/org/move/lang/core/completion/providers/SpecItemCompletionProvider.kt
diff --git a/src/main/kotlin/org/move/lang/core/psi/ext/MvItemSpec.kt b/src/main/kotlin/org/move/lang/core/psi/ext/MvItemSpec.kt
@@ -11,10 +11,6 @@ val MvItemSpec.funcItem get() = this.item as? MvFunction
 
 val MvModuleItemSpec.itemSpecBlock: MvSpecCodeBlock? get() = this.childOfType()
 
-val MvModule.mslSpecifiableItems: List<MvNameIdentifierOwner>
-    get() =
-        this.allFunctions().chain(this.structs()).toList()
-
 val MvItemSpec.module: MvModule?
     get() {
         val parent = this.parent

diff --git a/src/main/kotlin/org/move/lang/core/psi/ext/MvModule.kt b/src/main/kotlin/org/move/lang/core/psi/ext/MvModule.kt
@@ -1,8 +1,6 @@
 package org.move.lang.core.psi.ext
 
-import com.intellij.ide.projectView.PresentationData
 import com.intellij.lang.ASTNode
-import com.intellij.navigation.ItemPresentation
 import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.Key
 import com.intellij.psi.stubs.IStubElementType
@@ -51,16 +49,8 @@ fun MvModule.allFunctions(): List<MvFunction> {
     return stub?.childrenStubsOfType<MvFunctionStub>()?.map { it.psi } ?: functionList
 }
 
-fun MvModule.allNonTestFunctions(): List<MvFunction> =
-//    allFunctions().filter { f -> !f.isTest }
-    this.allFunctions().filter { f -> !f.hasTestAttr }
-//    getProjectPsiDependentCache(this) {
-//    }
-
-fun MvModule.testFunctions(): List<MvFunction> =
-    getProjectPsiDependentCache(this) {
-        it.allFunctions().filter { f -> f.hasTestAttr }
-    }
+fun MvModule.allNonTestFunctions(): List<MvFunction> = this.allFunctions().filter { f -> !f.hasTestAttr }
+fun MvModule.testFunctions(): List<MvFunction> = this.allFunctions().filter { f -> f.hasTestAttr }
 
 val MvModule.isBuiltins: Boolean get() = this.name == "builtins" && (this.address(null)?.is0x0 ?: false)
 val MvModule.isSpecBuiltins: Boolean

diff --git a/src/main/kotlin/org/move/lang/core/resolve/ref/MvItemSpecRefReferenceImpl.kt b/src/main/kotlin/org/move/lang/core/resolve/ref/MvItemSpecRefReferenceImpl.kt
@@ -2,17 +2,37 @@ package org.move.lang.core.resolve.ref
 
 import org.move.lang.core.psi.MvItemSpecRef
 import org.move.lang.core.psi.MvNamedElement
+import org.move.lang.core.psi.ext.allNonTestFunctions
 import org.move.lang.core.psi.ext.itemSpec
 import org.move.lang.core.psi.ext.module
-import org.move.lang.core.psi.ext.mslSpecifiableItems
+import org.move.lang.core.psi.ext.structs
+import org.move.lang.core.resolve.RsResolveProcessor
+import org.move.lang.core.resolve.SimpleScopeEntry
+import org.move.lang.core.resolve.collectResolveVariants
+import org.move.lang.core.resolve.processAll
 
 class MvItemSpecRefReferenceImpl(element: MvItemSpecRef): MvPolyVariantReferenceCached<MvItemSpecRef>(element) {
 
-    override fun multiResolveInner(): List<MvNamedElement> {
-        val module = element.itemSpec.module ?: return emptyList()
-        val referenceName = element.referenceName
-        return module.mslSpecifiableItems
-            .filter { it.name == referenceName }
-    }
+    override fun multiResolveInner(): List<MvNamedElement> =
+        collectResolveVariants(element.referenceName) {
+            processItemSpecRefResolveVariants(element, it)
+        }
+}
 
+fun processItemSpecRefResolveVariants(
+    itemSpecRef: MvItemSpecRef,
+    processor: RsResolveProcessor
+): Boolean {
+    val module = itemSpecRef.itemSpec.module ?: return false
+    val mslEnabledItems =
+        listOf(
+            module.allNonTestFunctions(),
+            module.structs(),
+            module.enumList,
+        ).flatten()
+    return processor.processAll(
+        mslEnabledItems.mapNotNull {
+            it.name?.let { name -> SimpleScopeEntry(name, it, ALL_NAMESPACES) }
+        }
+    )
 }