Support automatic labeling for infix calls

#KT-8877 Fixed

compiler/frontend/src/org/jetbrains/kotlin/types/expressions/LabelResolver.kt

@@ -59,7 +59,11 @@ object LabelResolver {
 
     private fun getLabelForFunctionalExpression(element: KtExpression): Name? {
         val parent = element.parent
-        return if (parent is KtLabeledExpression) getLabelNameIfAny(parent) else getCallerName(element)
+        return when (parent) {
+            is KtLabeledExpression -> getLabelNameIfAny(parent)
+            is KtBinaryExpression -> parent.operationReference.getReferencedNameAsName()
+            else -> getCallerName(element)
+        }
     }
 
     private fun getExpressionUnderLabel(labeledExpression: KtExpression): KtExpression {

compiler/testData/codegen/box/labels/infixCallLabelling.kt

@@ -0,0 +1,24 @@
+fun test(x: Int): Int {
+    x myMap {
+        return@myMap
+    }
+
+    return 0
+}
+
+fun myMap(x: Int): Int {
+    x myMap {
+        return@myMap
+    }
+
+    return 0
+}
+
+infix fun Int.myMap(x: () -> Unit) {}
+
+fun box(): String {
+    test(0)
+    myMap(0)
+
+    return "OK"
+}

compiler/testData/diagnostics/tests/labels/automaticLabelFromInfixOperator.kt

@@ -0,0 +1,17 @@
+fun test(x: List<Int>): Int {
+    x myMap {
+        return@myMap
+    }
+
+    return 0
+}
+
+fun myMap(x: List<Int>): Int {
+    x myMap {
+        return<!LABEL_NAME_CLASH!>@myMap<!>
+    }
+
+    return 0
+}
+
+infix fun List<Int>.myMap(<!UNUSED_PARAMETER!>x<!>: () -> Unit) {}

compiler/testData/diagnostics/tests/labels/automaticLabelFromInfixOperator.txt

@@ -0,0 +1,5 @@
+package
+
+public fun myMap(/*0*/ x: kotlin.collections.List<kotlin.Int>): kotlin.Int
+public fun test(/*0*/ x: kotlin.collections.List<kotlin.Int>): kotlin.Int
+public infix fun kotlin.collections.List<kotlin.Int>.myMap(/*0*/ x: () -> kotlin.Unit): kotlin.Unit

compiler/tests-ir-jvm/tests/org/jetbrains/kotlin/codegen/ir/IrBlackBoxCodegenTestGenerated.java

@@ -10070,6 +10070,12 @@ public class IrBlackBoxCodegenTestGenerated extends AbstractIrBlackBoxCodegenTes
             doTest(fileName);
         }
 
+        @TestMetadata("infixCallLabelling.kt")
+        public void testInfixCallLabelling() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/infixCallLabelling.kt");
+            doTest(fileName);
+        }
+
         @TestMetadata("labeledDeclarations.kt")
         public void testLabeledDeclarations() throws Exception {
             String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/labeledDeclarations.kt");

compiler/tests/org/jetbrains/kotlin/checkers/DiagnosticsTestGenerated.java

@@ -12610,6 +12610,12 @@ public class DiagnosticsTestGenerated extends AbstractDiagnosticsTest {
                 KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("compiler/testData/diagnostics/tests/labels"), Pattern.compile("^(.+)\\.kt$"), TargetBackend.ANY, true);
             }
 
+            @TestMetadata("automaticLabelFromInfixOperator.kt")
+            public void testAutomaticLabelFromInfixOperator() throws Exception {
+                String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/diagnostics/tests/labels/automaticLabelFromInfixOperator.kt");
+                doTest(fileName);
+            }
+
             @TestMetadata("kt1703.kt")
             public void testKt1703() throws Exception {
                 String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/diagnostics/tests/labels/kt1703.kt");

compiler/tests/org/jetbrains/kotlin/codegen/BlackBoxCodegenTestGenerated.java

@@ -10070,6 +10070,12 @@ public class BlackBoxCodegenTestGenerated extends AbstractBlackBoxCodegenTest {
             doTest(fileName);
         }
 
+        @TestMetadata("infixCallLabelling.kt")
+        public void testInfixCallLabelling() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/infixCallLabelling.kt");
+            doTest(fileName);
+        }
+
         @TestMetadata("labeledDeclarations.kt")
         public void testLabeledDeclarations() throws Exception {
             String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/labeledDeclarations.kt");

compiler/tests/org/jetbrains/kotlin/codegen/LightAnalysisModeTestGenerated.java

@@ -10070,6 +10070,12 @@ public class LightAnalysisModeTestGenerated extends AbstractLightAnalysisModeTes
             doTest(fileName);
         }
 
+        @TestMetadata("infixCallLabelling.kt")
+        public void testInfixCallLabelling() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/infixCallLabelling.kt");
+            doTest(fileName);
+        }
+
         @TestMetadata("labeledDeclarations.kt")
         public void testLabeledDeclarations() throws Exception {
             String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/labeledDeclarations.kt");

js/js.tests/test/org/jetbrains/kotlin/js/test/semantics/JsCodegenBoxTestGenerated.java

@@ -11601,6 +11601,12 @@ public class JsCodegenBoxTestGenerated extends AbstractJsCodegenBoxTest {
             doTest(fileName);
         }
 
+        @TestMetadata("infixCallLabelling.kt")
+        public void testInfixCallLabelling() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/infixCallLabelling.kt");
+            doTest(fileName);
+        }
+
         @TestMetadata("labeledDeclarations.kt")
         public void testLabeledDeclarations() throws Exception {
             String fileName = KotlinTestUtils.navigationMetadata("compiler/testData/codegen/box/labels/labeledDeclarations.kt");