KTIJ-31409 [kotlin] Handle UNRESOLVED_REFERENCE_WRONG_RECEIVER diagnostic in K2 Mode auto-import implementation

`ImportQuickFixFactories` is responsible for the "on the fly" auto-import,
and `KotlinDiagnosticHighlightVisitor` is responsible for providing the
auto-import options in the regular Alt+Enter menu

^KTIJ-31409 Fixed

GitOrigin-RevId: c1257fba60379026336e696b20d62d38e61738e8

plugins/kotlin/code-insight/fixes-k2/src/org/jetbrains/kotlin/idea/k2/codeinsight/fixes/KotlinK2QuickFixRegistrar.kt

@@ -582,6 +582,7 @@ class KotlinK2QuickFixRegistrar : KotlinQuickFixRegistrar() {
 
     override val importOnTheFlyList: KotlinQuickFixesList = KtQuickFixesListBuilder.registerPsiQuickFix {
         registerFactory(ImportQuickFixFactories.unresolvedReferenceFactory)
+        registerFactory(ImportQuickFixFactories.unresolvedReferenceWrongReceiverFactory)
         registerFactory(ImportQuickFixFactories.invisibleReferenceFactory)
     }
 }

plugins/kotlin/code-insight/fixes-k2/src/org/jetbrains/kotlin/idea/k2/codeinsight/fixes/imprt/ImportQuickFixFactories.kt

@@ -16,4 +16,11 @@ object ImportQuickFixFactories {
     val unresolvedReferenceFactory = KotlinQuickFixFactory.IntentionBased { diagnostic: KaFirDiagnostic.UnresolvedReference ->
         ImportQuickFixProvider.getFixes(diagnostic.psi)
     }
+
+    /**
+     * See KDoc for [unresolvedReferenceFactory].
+     */
+    val unresolvedReferenceWrongReceiverFactory = KotlinQuickFixFactory.IntentionBased { diagnostic: KaFirDiagnostic.UnresolvedReferenceWrongReceiver ->
+        ImportQuickFixProvider.getFixes(diagnostic.psi)
+    }
 }

plugins/kotlin/code-insight/fixes-k2/tests/test/org/jetbrains/kotlin/idea/k2/codeinsight/fixes/HighLevelQuickFixMultiFileTestGenerated.java

@@ -999,6 +999,11 @@ public abstract class HighLevelQuickFixMultiFileTestGenerated extends AbstractHi
             public void testWithSmartCastQualifier() throws Exception {
                 runTest("../../../idea/tests/testData/quickfix/autoImports/withSmartCastQualifier.before.Main.kt");
             }
+
+            @TestMetadata("wrongReceiverType.test")
+            public void testWrongReceiverType() throws Exception {
+                runTest("../../../idea/tests/testData/quickfix/autoImports/wrongReceiverType.test");
+            }
         }
     }
 

plugins/kotlin/fir/tests/test/org/jetbrains/kotlin/idea/fir/imports/K2AutoImportTestGenerated.java

@@ -184,4 +184,9 @@ public class K2AutoImportTestGenerated extends AbstractK2AutoImportTest {
     public void testWithoutAutoImport() throws Exception {
         runTest("../../idea/tests/testData/editor/autoImport/withoutAutoImport/");
     }
+
+    @TestMetadata("wrongReceiverType")
+    public void testWrongReceiverType() throws Exception {
+        runTest("../../idea/tests/testData/editor/autoImport/wrongReceiverType/");
+    }
 }

plugins/kotlin/highlighting/highlighting-k2/src/org/jetbrains/kotlin/idea/highlighting/KotlinDiagnosticHighlightVisitor.kt

@@ -159,6 +159,7 @@ class KotlinDiagnosticHighlightVisitor : HighlightVisitor {
 
         if (
             diagnostic is KaFirDiagnostic.UnresolvedReference ||
+            diagnostic is KaFirDiagnostic.UnresolvedReferenceWrongReceiver ||
             diagnostic is KaFirDiagnostic.UnresolvedImport
         ) {
             psiElement.registerKotlinUnresolvedReferenceKind(KotlinUnresolvedReferenceKind.Regular)
@@ -167,6 +168,7 @@ class KotlinDiagnosticHighlightVisitor : HighlightVisitor {
         if (
             diagnostic is KaFirDiagnostic.UnresolvedImport ||
             diagnostic is KaFirDiagnostic.UnresolvedReference ||
+            diagnostic is KaFirDiagnostic.UnresolvedReferenceWrongReceiver ||
             diagnostic is KaFirDiagnostic.DelegateSpecialFunctionMissing ||
             diagnostic is KaFirDiagnostic.DelegateSpecialFunctionNoneApplicable
         ) {

plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/imports/K1AutoImportTestGenerated.java

@@ -184,4 +184,9 @@ public class K1AutoImportTestGenerated extends AbstractK1AutoImportTest {
     public void testWithoutAutoImport() throws Exception {
         runTest("testData/editor/autoImport/withoutAutoImport/");
     }
+
+    @TestMetadata("wrongReceiverType")
+    public void testWrongReceiverType() throws Exception {
+        runTest("testData/editor/autoImport/wrongReceiverType/");
+    }
 }

plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/quickfix/QuickFixMultiFileTestGenerated.java

@@ -1111,6 +1111,11 @@ public abstract class QuickFixMultiFileTestGenerated extends AbstractQuickFixMul
             public void testWithSmartCastQualifier() throws Exception {
                 runTest("testData/quickfix/autoImports/withSmartCastQualifier.before.Main.kt");
             }
+
+            @TestMetadata("wrongReceiverType.test")
+            public void testWrongReceiverType() throws Exception {
+                runTest("testData/quickfix/autoImports/wrongReceiverType.test");
+            }
         }
     }
 

plugins/kotlin/idea/tests/testData/editor/autoImport/wrongReceiverType/dependency/dependency.kt

@@ -0,0 +1,5 @@
+package dependency
+
+class FromDependency
+
+fun FromDependency.extension() {}

plugins/kotlin/idea/tests/testData/editor/autoImport/wrongReceiverType/wrongReceiverType.kt

@@ -0,0 +1,11 @@
+package main
+
+import dependency.FromDependency
+
+class UnrelatedClass
+
+fun UnrelatedClass.extension()
+
+fun test(f: FromDependency) {
+    f.extension() // UNRESOLVED_REFERENCE_WRONG_RECEIVER
+}

plugins/kotlin/idea/tests/testData/editor/autoImport/wrongReceiverType/wrongReceiverType.kt.after

@@ -0,0 +1,12 @@
+package main
+
+import dependency.FromDependency
+import dependency.extension
+
+class UnrelatedClass
+
+fun UnrelatedClass.extension()
+
+fun test(f: FromDependency) {
+    f.extension() // UNRESOLVED_REFERENCE_WRONG_RECEIVER
+}

plugins/kotlin/idea/tests/testData/quickfix/autoImports/wrongReceiverType.test

@@ -0,0 +1,30 @@
+// FILE: first.before.kt
+// "Import extension function 'String.extension'" "true"
+// ERROR: Unresolved reference. None of the following candidates is applicable because of receiver type mismatch: <br>public fun Int.extension(): Unit defined in test in file first.before.kt
+
+package test
+
+fun Int.extension() {}
+
+fun test(str: String) {
+    str.<caret>extension()
+}
+
+// FILE: second.kt
+package dependency
+
+fun String.extension() {}
+
+// FILE: first.after.kt
+// "Import extension function 'String.extension'" "true"
+// ERROR: Unresolved reference. None of the following candidates is applicable because of receiver type mismatch: <br>public fun Int.extension(): Unit defined in test in file first.before.kt
+
+package test
+
+import dependency.extension
+
+fun Int.extension() {}
+
+fun test(str: String) {
+    str.<caret>extension()
+}