Imports on paste: added checking of symbols to import into test + fixed bug on copy/paste import/package directive
This commit is contained in:
Valentin Kipyatkov
+14
-5
@@ -68,8 +68,8 @@ import java.util.ArrayList
|
|||||||
public class KotlinCopyPasteReferenceProcessor() : CopyPastePostProcessor<KotlinReferenceTransferableData>() {
|
public class KotlinCopyPasteReferenceProcessor() : CopyPastePostProcessor<KotlinReferenceTransferableData>() {
|
||||||
private val LOG = Logger.getInstance(javaClass<KotlinCopyPasteReferenceProcessor>())
|
private val LOG = Logger.getInstance(javaClass<KotlinCopyPasteReferenceProcessor>())
|
||||||
|
|
||||||
private val IGNORE_REFERENCES_INSIDE: Array<Class<out JetElement>?> = array(
|
private val IGNORE_REFERENCES_INSIDE: Array<Class<out JetElement>> = array(
|
||||||
javaClass<JetImportDirective>(),
|
javaClass<JetImportList>(),
|
||||||
javaClass<JetPackageDirective>()
|
javaClass<JetPackageDirective>()
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -138,10 +138,12 @@ public class KotlinCopyPasteReferenceProcessor() : CopyPastePostProcessor<Kotlin
|
|||||||
startOffsets: IntArray,
|
startOffsets: IntArray,
|
||||||
endOffsets: IntArray
|
endOffsets: IntArray
|
||||||
) {
|
) {
|
||||||
if (PsiTreeUtil.getParentOfType(element, *IGNORE_REFERENCES_INSIDE) != null) return
|
if (PsiTreeUtil.getNonStrictParentOfType(element, *IGNORE_REFERENCES_INSIDE) != null) return
|
||||||
|
|
||||||
element.accept(object : PsiElementVisitor() {
|
element.accept(object : PsiElementVisitor() {
|
||||||
override fun visitElement(element: PsiElement) {
|
override fun visitElement(element: PsiElement) {
|
||||||
|
if (element.javaClass in IGNORE_REFERENCES_INSIDE) return
|
||||||
|
|
||||||
element.acceptChildren(this)
|
element.acceptChildren(this)
|
||||||
|
|
||||||
val reference = element.getReference() as? JetReference ?: return
|
val reference = element.getReference() as? JetReference ?: return
|
||||||
@@ -322,13 +324,15 @@ public class KotlinCopyPasteReferenceProcessor() : CopyPastePostProcessor<Kotlin
|
|||||||
}
|
}
|
||||||
|
|
||||||
private fun showRestoreReferencesDialog(project: Project, referencesToRestore: List<ReferenceToRestoreData>): Collection<ReferenceToRestoreData> {
|
private fun showRestoreReferencesDialog(project: Project, referencesToRestore: List<ReferenceToRestoreData>): Collection<ReferenceToRestoreData> {
|
||||||
|
val fqNames = referencesToRestore.map { it.refData.fqName }.toSortedSet()
|
||||||
|
|
||||||
|
declarationsToImportSuggested = fqNames
|
||||||
|
|
||||||
val shouldShowDialog = CodeInsightSettings.getInstance().ADD_IMPORTS_ON_PASTE == CodeInsightSettings.ASK
|
val shouldShowDialog = CodeInsightSettings.getInstance().ADD_IMPORTS_ON_PASTE == CodeInsightSettings.ASK
|
||||||
if (!shouldShowDialog || referencesToRestore.isEmpty()) {
|
if (!shouldShowDialog || referencesToRestore.isEmpty()) {
|
||||||
return referencesToRestore
|
return referencesToRestore
|
||||||
}
|
}
|
||||||
|
|
||||||
val fqNames = referencesToRestore.map { it.refData.fqName }.toSortedSet()
|
|
||||||
|
|
||||||
val dialog = RestoreReferencesDialog(project, fqNames.copyToArray())
|
val dialog = RestoreReferencesDialog(project, fqNames.copyToArray())
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
@@ -345,4 +349,9 @@ public class KotlinCopyPasteReferenceProcessor() : CopyPastePostProcessor<Kotlin
|
|||||||
if (getContainingFile() != fileCopiedFrom) return false
|
if (getContainingFile() != fileCopiedFrom) return false
|
||||||
return toTextRanges(startOffsets, endOffsets).any { this.range in it }
|
return toTextRanges(startOffsets, endOffsets).any { this.range in it }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
companion object {
|
||||||
|
// for tests
|
||||||
|
public var declarationsToImportSuggested: Collection<String> = emptyList()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,7 @@
|
|||||||
|
a.E.ENTRY
|
||||||
|
a.Outer.Inner
|
||||||
|
a.Outer.Nested
|
||||||
|
a.Outer.NestedAnnotation
|
||||||
|
a.Outer.NestedEnum
|
||||||
|
a.Outer.NestedObj
|
||||||
|
a.Outer.NestedTrait
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
a.A
|
||||||
|
a.A.Companion
|
||||||
|
a.B
|
||||||
|
a.B.Companion
|
||||||
|
a.T
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
a.Outer.Companion.Nested
|
||||||
|
a.Outer.Companion.NestedAnnotation
|
||||||
|
a.Outer.Companion.NestedEnum
|
||||||
|
a.Outer.Companion.NestedObj
|
||||||
|
a.Outer.Companion.NestedTrait
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.a
|
||||||
|
a.a.Companion
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.A
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.A
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.A
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.A
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
a.Outer
|
||||||
|
a.Outer.Inner.II
|
||||||
|
a.Outer.Nested.NI
|
||||||
|
a.Outer.Nested.NN
|
||||||
|
a.Outer.Nested.NN2
|
||||||
|
a.with
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
A
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
a.A
|
||||||
|
a.get
|
||||||
|
a.set
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
java.JavaClass
|
||||||
|
java.JavaClass.NestedClass
|
||||||
|
java.JavaClass.staticMethod
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
d.A
|
||||||
|
d.ClassObject.Companion
|
||||||
|
d.E.ENTRY
|
||||||
|
d.O1
|
||||||
|
d.O2
|
||||||
|
d.Outer.Inner
|
||||||
|
d.Outer.Nested
|
||||||
|
d.Outer.NestedAnnotation
|
||||||
|
d.Outer.NestedEnum
|
||||||
|
d.Outer.NestedObj
|
||||||
|
d.Outer.NestedTrait
|
||||||
|
d.T
|
||||||
|
d.c
|
||||||
|
d.ext
|
||||||
|
d.g
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
kotlin.Int
|
||||||
|
kotlin.concurrent.FunctionalList
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
a.A.A
|
||||||
|
a.A.B
|
||||||
|
a.A.C
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
a.f
|
||||||
|
a.g
|
||||||
|
a.p
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
a.A
|
||||||
|
a.infix
|
||||||
|
a.plus
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.ext
|
||||||
|
a.p
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
a.ext
|
||||||
|
a.infix
|
||||||
|
a.minus
|
||||||
|
a.p
|
||||||
|
a.plus
|
||||||
@@ -0,0 +1,4 @@
|
|||||||
|
a.A
|
||||||
|
a.hasNext
|
||||||
|
a.iterator
|
||||||
|
a.next
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.g
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.A
|
||||||
|
a.get
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
d.A
|
||||||
|
d.ClassObject.Companion
|
||||||
|
d.E.ENTRY
|
||||||
|
d.O1
|
||||||
|
d.O2
|
||||||
|
d.Outer.Inner
|
||||||
|
d.Outer.Nested
|
||||||
|
d.Outer.NestedAnnotation
|
||||||
|
d.Outer.NestedEnum
|
||||||
|
d.Outer.NestedObj
|
||||||
|
d.Outer.NestedTrait
|
||||||
|
d.T
|
||||||
|
d.c
|
||||||
|
d.ext
|
||||||
|
d.g
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
a.A.Nested
|
||||||
|
a.ext
|
||||||
|
a.f
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
a.Outer
|
||||||
|
a.Outer.Inner
|
||||||
|
a.Outer.Nested
|
||||||
|
a.Outer.NestedAnnotation
|
||||||
|
a.Outer.NestedEnum
|
||||||
|
a.Outer.NestedObj
|
||||||
|
a.Outer.NestedTrait
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.A
|
||||||
|
a.invoke
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.JavaF
|
||||||
|
a.JavaM
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
a.A
|
||||||
|
a.component1
|
||||||
|
a.component2
|
||||||
@@ -0,0 +1,4 @@
|
|||||||
|
a.A
|
||||||
|
a.B
|
||||||
|
a.hasNext
|
||||||
|
a.next
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.b
|
||||||
|
a.d
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
a.E.ENTRY
|
||||||
|
a.Outer.Inner
|
||||||
|
a.Outer.Nested
|
||||||
|
a.Outer.NestedAnnotation
|
||||||
|
a.Outer.NestedEnum
|
||||||
|
a.Outer.NestedObj
|
||||||
|
a.Outer.NestedTrait
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
a.A
|
||||||
|
a.B
|
||||||
|
a.T
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.b
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.Foo
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.A
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.b
|
||||||
|
a.c
|
||||||
@@ -0,0 +1 @@
|
|||||||
|
a.A
|
||||||
@@ -0,0 +1,2 @@
|
|||||||
|
a.A
|
||||||
|
a.T
|
||||||
@@ -17,22 +17,20 @@
|
|||||||
package org.jetbrains.kotlin.idea.codeInsight
|
package org.jetbrains.kotlin.idea.codeInsight
|
||||||
|
|
||||||
import com.intellij.openapi.actionSystem.IdeActions
|
import com.intellij.openapi.actionSystem.IdeActions
|
||||||
import com.intellij.openapi.util.TextRange
|
|
||||||
import com.intellij.openapi.util.io.FileUtil
|
import com.intellij.openapi.util.io.FileUtil
|
||||||
import org.jetbrains.kotlin.checkers.DebugInfoUtil
|
import org.jetbrains.kotlin.checkers.DebugInfoUtil
|
||||||
import org.jetbrains.kotlin.diagnostics.Diagnostic
|
|
||||||
import org.jetbrains.kotlin.diagnostics.DiagnosticUtils
|
import org.jetbrains.kotlin.diagnostics.DiagnosticUtils
|
||||||
import org.jetbrains.kotlin.diagnostics.Errors
|
import org.jetbrains.kotlin.diagnostics.Errors
|
||||||
import org.jetbrains.kotlin.diagnostics.rendering.DefaultErrorMessages
|
import org.jetbrains.kotlin.diagnostics.rendering.DefaultErrorMessages
|
||||||
import org.jetbrains.kotlin.idea.AbstractCopyPasteTest
|
import org.jetbrains.kotlin.idea.AbstractCopyPasteTest
|
||||||
import org.jetbrains.kotlin.idea.PluginTestCaseBase
|
import org.jetbrains.kotlin.idea.PluginTestCaseBase
|
||||||
import org.jetbrains.kotlin.idea.caches.resolve.*
|
import org.jetbrains.kotlin.idea.caches.resolve.analyze
|
||||||
|
import org.jetbrains.kotlin.idea.caches.resolve.analyzeFully
|
||||||
import org.jetbrains.kotlin.psi.JetFile
|
import org.jetbrains.kotlin.psi.JetFile
|
||||||
import org.jetbrains.kotlin.psi.JetReferenceExpression
|
import org.jetbrains.kotlin.psi.JetReferenceExpression
|
||||||
import org.jetbrains.kotlin.resolve.BindingContext
|
|
||||||
import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
|
import org.jetbrains.kotlin.resolve.lazy.BodyResolveMode
|
||||||
import org.jetbrains.kotlin.test.InTextDirectivesUtils
|
import org.jetbrains.kotlin.test.InTextDirectivesUtils
|
||||||
|
import org.jetbrains.kotlin.test.JetTestUtils
|
||||||
import java.io.File
|
import java.io.File
|
||||||
import kotlin.test.fail
|
import kotlin.test.fail
|
||||||
|
|
||||||
@@ -61,10 +59,15 @@ public abstract class AbstractInsertImportOnPasteTest : AbstractCopyPasteTest()
|
|||||||
myFixture.configureByFile(testFileName)
|
myFixture.configureByFile(testFileName)
|
||||||
myFixture.performEditorAction(cutOrCopy)
|
myFixture.performEditorAction(cutOrCopy)
|
||||||
|
|
||||||
|
KotlinCopyPasteReferenceProcessor.declarationsToImportSuggested = emptyList()
|
||||||
|
|
||||||
val toFileName = testFileName.replace(".kt", ".to.kt")
|
val toFileName = testFileName.replace(".kt", ".to.kt")
|
||||||
val toFile = configureToFile(toFileName)
|
val toFile = configureToFile(toFileName)
|
||||||
performNotWriteEditorAction(IdeActions.ACTION_PASTE)
|
performNotWriteEditorAction(IdeActions.ACTION_PASTE)
|
||||||
|
|
||||||
|
val namesToImportDump = KotlinCopyPasteReferenceProcessor.declarationsToImportSuggested.joinToString("\n")
|
||||||
|
JetTestUtils.assertEqualsToFile(File(path.replace(".kt", ".expected.names")), namesToImportDump)
|
||||||
|
|
||||||
myFixture.checkResultByFile(testFileName.replace(".kt", ".expected.kt"))
|
myFixture.checkResultByFile(testFileName.replace(".kt", ".expected.kt"))
|
||||||
|
|
||||||
if (!InTextDirectivesUtils.isDirectiveDefined(FileUtil.loadFile(testFile, true), ALLOW_UNRESOLVED_DIRECTIVE)) {
|
if (!InTextDirectivesUtils.isDirectiveDefined(FileUtil.loadFile(testFile, true), ALLOW_UNRESOLVED_DIRECTIVE)) {
|
||||||
|
|||||||
Reference in New Issue
Block a user