From dd0bb9abf15f2d569fb95726349932a6ead010fd Mon Sep 17 00:00:00 2001
From: Vladimir Dolzhenko <Vladimir.Dolzhenko@jetbrains.com>
Date: Mon, 6 Apr 2020 09:53:24 +0000
Subject: [PATCH] Do not resolve references pasted into the same origin

#KT-37993 Fixed
---
 .../BasicKotlinReferenceTransferableData.kt   |  3 +-
 .../KotlinCopyPasteReferenceProcessor.kt      | 34 ++++++++++++++++++-
 2 files changed, 35 insertions(+), 2 deletions(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/codeInsight/BasicKotlinReferenceTransferableData.kt b/idea/src/org/jetbrains/kotlin/idea/codeInsight/BasicKotlinReferenceTransferableData.kt
index d093da7914d..9dc9a0fe8f9 100644
--- a/idea/src/org/jetbrains/kotlin/idea/codeInsight/BasicKotlinReferenceTransferableData.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/codeInsight/BasicKotlinReferenceTransferableData.kt
@@ -16,7 +16,8 @@ class BasicKotlinReferenceTransferableData(
     val imports: List<String>,
     val sourceTextOffset: Int,
     val sourceText: String,
-    val textRanges: List<TextRange>
+    val textRanges: List<TextRange>,
+    val locationFqName: String?
 ) : TextBlockTransferableData, Cloneable, Serializable {
     override fun getFlavor() = dataFlavor
     override fun getOffsetCount() = 0
diff --git a/idea/src/org/jetbrains/kotlin/idea/codeInsight/KotlinCopyPasteReferenceProcessor.kt b/idea/src/org/jetbrains/kotlin/idea/codeInsight/KotlinCopyPasteReferenceProcessor.kt
index d47b39611ca..8e41701ef09 100644
--- a/idea/src/org/jetbrains/kotlin/idea/codeInsight/KotlinCopyPasteReferenceProcessor.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/codeInsight/KotlinCopyPasteReferenceProcessor.kt
@@ -117,6 +117,11 @@ class KotlinCopyPasteReferenceProcessor : CopyPastePostProcessor<BasicKotlinRefe
         val text = file.text.substring(offsetDelta)
         val ranges = startOffsets.indices.map { TextRange(startOffsets[it], endOffsets[it]) }
 
+        val locationFqName = if (startOffsets.size == 1) {
+            val fqName = file.namedDeclarationFqName(startOffsets[0] - 1)
+            fqName?.takeIf { it == file.namedDeclarationFqName(endOffsets[0] + 1) }
+        } else null
+
         return listOf(
             BasicKotlinReferenceTransferableData(
                 sourceFileUrl = file.virtualFile.url,
@@ -124,11 +129,17 @@ class KotlinCopyPasteReferenceProcessor : CopyPastePostProcessor<BasicKotlinRefe
                 imports = imports,
                 sourceTextOffset = offsetDelta,
                 sourceText = text,
-                textRanges = ranges
+                textRanges = ranges,
+                locationFqName = locationFqName
             )
         )
     }
 
+    private fun KtFile.namedDeclarationFqName(offset: Int): String? =
+        if (offset in 0 until textLength)
+            PsiTreeUtil.getNonStrictParentOfType(this.findElementAt(offset), KtNamedDeclaration::class.java)?.fqName?.asString()
+        else null
+
     private fun collectReferenceData(
         textRanges: List<TextRange>,
         file: KtFile,
@@ -267,9 +278,30 @@ class KotlinCopyPasteReferenceProcessor : CopyPastePostProcessor<BasicKotlinRefe
         val file = PsiDocumentManager.getInstance(project).getPsiFile(document)
         if (file !is KtFile) return
 
+        if (isPastedToTheSameOrigin(file, caretOffset, values)) return
+
         processReferenceData(project, editor, file, bounds.startOffset, values.single())
     }
 
+    private fun isPastedToTheSameOrigin(
+        file: KtFile,
+        caretOffset: Int,
+        values: List<BasicKotlinReferenceTransferableData>
+    ): Boolean {
+        if (values.size == 1 && values[0].sourceFileUrl == file.virtualFile.url &&
+            values[0].locationFqName != null &&
+            // check locationFqName on position before pasted snippet
+            values[0].locationFqName == file.namedDeclarationFqName(caretOffset - 1)
+        ) {
+            val currentImports = file.importDirectives.map { it.text }.toSet()
+            val originalImports = values.flatMap { it.imports }.toSet()
+            if (currentImports == originalImports) {
+                return true
+            }
+        }
+        return false
+    }
+
     private fun processReferenceData(
         project: Project,
         editor: Editor,