[kotlin] Properly retarget unary operators

#KTIJ-30032 Fixed

GitOrigin-RevId: 154fa9b03ed09f9e6fdbada465217a9764934be3

plugins/kotlin/idea/tests/test/org/jetbrains/kotlin/idea/refactoring/move/MoveTestGenerated.java

@@ -692,6 +692,11 @@ public abstract class MoveTestGenerated extends AbstractMoveTest {
         public void testKotlin_unaffectedQualifiedReferences_UnaffectedQualifiedReferences() throws Exception {
             runTest("testData/refactoring/moveTopLevel/kotlin/unaffectedQualifiedReferences/unaffectedQualifiedReferences.test");
         }
+
+        @TestMetadata("kotlin/unaryOperatorReference/unaryOperatorReference.test")
+        public void testKotlin_unaryOperatorReference_UnaryOperatorReference() throws Exception {
+            runTest("testData/refactoring/moveTopLevel/kotlin/unaryOperatorReference/unaryOperatorReference.test");
+        }
     }
 
     @RunWith(JUnit3RunnerWithInners.class)

plugins/kotlin/idea/tests/testData/refactoring/bindToElement/operationReference/UnaryOperator.kt

@@ -0,0 +1,19 @@
+// FILE: test/UnaryOperator.kt
+// BIND_TO test.bar.not
+package test
+
+import test.foo.not
+
+fun foo(x: String): Boolean {
+    return <caret>!x
+}
+
+// FILE: test/foo/Test.kt
+package test.foo
+
+operator fun String.not() = false
+
+// FILE: test/bar/Test.kt
+package test.bar
+
+operator fun String.not() = false

plugins/kotlin/idea/tests/testData/refactoring/bindToElement/operationReference/UnaryOperator.kt.after

@@ -0,0 +1,11 @@
+// FILE: test/UnaryOperator.kt
+// BIND_TO test.bar.not
+package test
+
+import test.bar.not
+import test.foo.not
+
+fun foo(x: String): Boolean {
+    return <caret>!x
+}
+

plugins/kotlin/idea/tests/testData/refactoring/moveTopLevel/kotlin/unaryOperatorReference/after/foo/DeclSite.kt

@@ -0,0 +1,3 @@
+package foo
+
+fun doFoo(): Boolean { return false }

plugins/kotlin/idea/tests/testData/refactoring/moveTopLevel/kotlin/unaryOperatorReference/after/otherBar/FooBar.kt

@@ -0,0 +1,7 @@
+package otherBar
+
+import foo.doFoo
+
+fun fooBar() {
+    if (!doFoo()) return
+}

plugins/kotlin/idea/tests/testData/refactoring/moveTopLevel/kotlin/unaryOperatorReference/before/bar/UseSite.kt

@@ -0,0 +1,7 @@
+package bar
+
+import foo.doFoo
+
+fun foo<caret>Bar() {
+    if (!doFoo()) return
+}

plugins/kotlin/idea/tests/testData/refactoring/moveTopLevel/kotlin/unaryOperatorReference/before/foo/DeclSite.kt

@@ -0,0 +1,3 @@
+package foo
+
+fun doFoo(): Boolean { return false }

plugins/kotlin/idea/tests/testData/refactoring/moveTopLevel/kotlin/unaryOperatorReference/unaryOperatorReference.test

@@ -0,0 +1,7 @@
+{
+  "mainFile": "bar/UseSite.kt",
+  "type": "MOVE_KOTLIN_TOP_LEVEL_DECLARATIONS",
+  "targetPackage": "otherBar",
+  "enabledInK1": "true",
+  "enabledInK2": "true"
+}

plugins/kotlin/refactorings/kotlin.refactorings.k2/src/org/jetbrains/kotlin/idea/k2/refactoring/K2ReferenceMutateService.kt

@@ -228,7 +228,15 @@ internal class K2ReferenceMutateService : KtReferenceMutateServiceBase() {
         val isInfix = analyze(targetElement) { (targetElement.getFunctionLikeSymbol() as? KaNamedFunctionSymbol)?.isInfix == true }
         val isOperator = analyze(targetElement) { (targetElement.getFunctionLikeSymbol() as? KaNamedFunctionSymbol)?.isOperator == true }
         val replacedExpr = if (isOperator) {
-            replaced(psiFactory.createOperationName(OperatorNameConventions.TOKENS_BY_OPERATOR_NAME[Name.identifier(shortName)] ?: shortName))
+            val identifier = Name.identifier(shortName)
+            val isUnary = OperatorNameConventions.UNARY_OPERATION_NAMES.contains(identifier)
+            val operator = OperatorNameConventions.TOKENS_BY_OPERATOR_NAME[identifier] ?: shortName
+            val newOperator = if (isUnary) {
+                (psiFactory.createExpression("${operator}0") as KtUnaryExpression).operationReference
+            } else {
+                psiFactory.createOperationName(operator)
+            }
+            replaced(newOperator)
         } else if (isInfix) {
             replaced(psiFactory.createOperationName(shortName))
         } else {

plugins/kotlin/refactorings/kotlin.refactorings.move.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/move/K2MoveTopLevelTestGenerated.java

@@ -449,4 +449,9 @@ public class K2MoveTopLevelTestGenerated extends AbstractK2MoveTopLevelTest {
     public void testKotlin_unaffectedQualifiedReferences_UnaffectedQualifiedReferences() throws Exception {
         runTest("../../idea/tests/testData/refactoring/moveTopLevel/kotlin/unaffectedQualifiedReferences/unaffectedQualifiedReferences.test");
     }
+
+    @TestMetadata("kotlin/unaryOperatorReference/unaryOperatorReference.test")
+    public void testKotlin_unaryOperatorReference_UnaryOperatorReference() throws Exception {
+        runTest("../../idea/tests/testData/refactoring/moveTopLevel/kotlin/unaryOperatorReference/unaryOperatorReference.test");
+    }
 }

plugins/kotlin/refactorings/kotlin.refactorings.tests.k2/test/org/jetbrains/kotlin/idea/k2/refactoring/bindToElement/K2BindToElementTestGenerated.java

@@ -816,6 +816,11 @@ public abstract class K2BindToElementTestGenerated extends AbstractK2BindToEleme
                 runTest("../../idea/tests/testData/refactoring/bindToElement/operationReference/UnQualified.kt");
             }
 
+            @TestMetadata("UnaryOperator.kt")
+            public void testUnaryOperator() throws Exception {
+                runTest("../../idea/tests/testData/refactoring/bindToElement/operationReference/UnaryOperator.kt");
+            }
+
             @TestMetadata("WithRegularCall.kt")
             public void testWithRegularCall() throws Exception {
                 runTest("../../idea/tests/testData/refactoring/bindToElement/operationReference/WithRegularCall.kt");