Added completion action to exclude classes/packages.

#KT-2413 fixed
2015-07-07 13:53:01 +03:00
parent 0451debda4
commit b98aa76325
2 changed files with 69 additions and 0 deletions
@@ -0,0 +1,68 @@
+/*
+ * Copyright 2010-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.jetbrains.kotlin.idea.completion
+
+import com.intellij.codeInsight.completion.ExcludeFromCompletionLookupActionProvider
+import com.intellij.codeInsight.daemon.impl.actions.AddImportAction
+import com.intellij.codeInsight.lookup.Lookup
+import com.intellij.codeInsight.lookup.LookupActionProvider
+import com.intellij.codeInsight.lookup.LookupElement
+import com.intellij.codeInsight.lookup.LookupElementAction
+import com.intellij.openapi.project.Project
+import com.intellij.psi.*
+import com.intellij.psi.util.PsiUtil
+import com.intellij.util.Consumer
+import org.jetbrains.kotlin.idea.core.completion.DeclarationLookupObject
+import org.jetbrains.kotlin.idea.imports.importableFqName
+import org.jetbrains.kotlin.name.FqName
+
+public class KotlinExcludeFromCompletionLookupActionProvider : LookupActionProvider {
+    override fun fillActions(element: LookupElement, lookup: Lookup, consumer: Consumer<LookupElementAction>) {
+        val lookupObject = element.getObject() as? DeclarationLookupObject ?: return
+
+        lookupObject.descriptor?.let {
+            it.importableFqName?.let {
+                addExcludes(consumer, lookup.getPsiFile().getProject(), it)
+                return
+            }
+        }
+
+        lookupObject.psiElement?.let {
+            val fakeLookupElement = object : LookupElement() {
+                override fun getLookupString() = ""
+                override fun getObject() = it
+            }
+
+            ExcludeFromCompletionLookupActionProvider().fillActions(fakeLookupElement, lookup, consumer)
+        }
+    }
+
+    private fun addExcludes(consumer: Consumer<LookupElementAction>, project: Project, fqName: FqName) {
+        for (s in AddImportAction.getAllExcludableStrings(fqName.asString())) {
+            consumer.consume(ExcludeFromCompletionAction(project, s))
+        }
+    }
+
+    private class ExcludeFromCompletionAction(
+            private val project: Project,
+            private val exclude: String
+    ) : LookupElementAction(null, "Exclude '$exclude' from completion") {
+        override fun performLookupAction(): LookupElementAction.Result {
+            AddImportAction.excludeFromImport(project, exclude)
+            return LookupElementAction.Result.HIDE_LOOKUP
+        }
+    }
+}
@@ -322,6 +322,7 @@
    <completion.confidence language="jet" implementationClass="com.intellij.codeInsight.completion.AlwaysFocusLookup" order="last"/>

    <lookup.charFilter implementation="org.jetbrains.kotlin.idea.completion.KotlinCompletionCharFilter"/>
+    <lookup.actionProvider implementation="org.jetbrains.kotlin.idea.completion.KotlinExcludeFromCompletionLookupActionProvider"/>

    <psi.referenceContributor language="jet" implementation="org.jetbrains.kotlin.idea.references.JetReferenceContributor"/>