Add 'operator' Intention: Support header/impl functions

#KT-18851 Fixed

idea/src/org/jetbrains/kotlin/idea/intentions/AddOperatorModifierIntention.kt

@@ -20,6 +20,7 @@ import com.intellij.openapi.editor.Editor
 import com.intellij.openapi.util.TextRange
 import org.jetbrains.kotlin.descriptors.FunctionDescriptor
 import org.jetbrains.kotlin.idea.caches.resolve.resolveToDescriptorIfAny
+import org.jetbrains.kotlin.idea.refactoring.withHeaderImplementations
 import org.jetbrains.kotlin.lexer.KtTokens
 import org.jetbrains.kotlin.psi.KtNamedFunction
 import org.jetbrains.kotlin.util.OperatorChecks
@@ -33,7 +34,7 @@ class AddOperatorModifierIntention : SelfTargetingRangeIntention<KtNamedFunction
     }
 
     override fun applyTo(element: KtNamedFunction, editor: Editor?) {
-        element.addModifier(KtTokens.OPERATOR_KEYWORD)
+        element.withHeaderImplementations().forEach { it.addModifier(KtTokens.OPERATOR_KEYWORD) }
     }
 }
 

idea/testData/multiModuleQuickFix/addOperatorByHeader/header/header.kt

@@ -0,0 +1,5 @@
+// "Add 'operator' modifier" "true"
+
+header class Foo {
+    fun <caret>unaryMinus()
+}

idea/testData/multiModuleQuickFix/addOperatorByHeader/header/header.kt.after

@@ -0,0 +1,5 @@
+// "Add 'operator' modifier" "true"
+
+header class Foo {
+    operator fun unaryMinus()
+}

idea/testData/multiModuleQuickFix/addOperatorByHeader/js/js.kt

@@ -0,0 +1,5 @@
+impl class Foo {
+    impl fun unaryMinus() {
+
+    }
+}

idea/testData/multiModuleQuickFix/addOperatorByHeader/js/js.kt.after

@@ -0,0 +1,5 @@
+impl class Foo {
+    impl operator fun unaryMinus() {
+
+    }
+}

idea/testData/multiModuleQuickFix/addOperatorByHeader/jvm/jvm.kt

@@ -0,0 +1,5 @@
+impl class Foo {
+    impl fun unaryMinus() {
+
+    }
+}

idea/testData/multiModuleQuickFix/addOperatorByHeader/jvm/jvm.kt.after

@@ -0,0 +1,5 @@
+impl class Foo {
+    impl operator fun unaryMinus() {
+
+    }
+}

idea/testData/multiModuleQuickFix/addOperatorByImpl/header/header.kt

@@ -0,0 +1,3 @@
+header class Foo {
+    fun unaryMinus()
+}

idea/testData/multiModuleQuickFix/addOperatorByImpl/header/header.kt.after

@@ -0,0 +1,3 @@
+header class Foo {
+    operator fun unaryMinus()
+}

idea/testData/multiModuleQuickFix/addOperatorByImpl/js/js.kt

@@ -0,0 +1,7 @@
+// "Add 'operator' modifier" "true"
+
+impl class Foo {
+    impl fun <caret>unaryMinus() {
+
+    }
+}

idea/testData/multiModuleQuickFix/addOperatorByImpl/js/js.kt.after

@@ -0,0 +1,7 @@
+// "Add 'operator' modifier" "true"
+
+impl class Foo {
+    impl operator fun unaryMinus() {
+
+    }
+}

idea/testData/multiModuleQuickFix/addOperatorByImpl/jvm/jvm.kt

@@ -0,0 +1,5 @@
+impl class Foo {
+    impl fun unaryMinus() {
+
+    }
+}

idea/testData/multiModuleQuickFix/addOperatorByImpl/jvm/jvm.kt.after

@@ -0,0 +1,5 @@
+impl class Foo {
+    impl operator fun unaryMinus() {
+
+    }
+}

idea/tests/org/jetbrains/kotlin/idea/quickfix/QuickFixMultiModuleTest.kt

@@ -123,4 +123,10 @@ class QuickFixMultiModuleTest : AbstractQuickFixMultiModuleTest() {
 
     @Test
     fun testMemberValToExtensionByImpl() = doTestHeaderWithJvmAndJs()
+
+    @Test
+    fun testAddOperatorByHeader() = doTestHeaderWithJvmAndJs()
+
+    @Test
+    fun testAddOperatorByImpl() = doTestHeaderWithJvmAndJs()
 }