FIR: fix label on non-lambda anonymous function

compiler/fir/raw-fir/light-tree2fir/src/org/jetbrains/kotlin/fir/lightTree/converter/DeclarationsConverter.kt

@@ -1507,14 +1507,15 @@ class DeclarationsConverter(
         val functionSymbol: FirFunctionSymbol<*>
         val isAnonymousFunction = identifier == null && isLocal
         val functionBuilder = if (isAnonymousFunction) {
-            val labelName = functionDeclaration.getLabelName() ?: context.calleeNamesForLambda.lastOrNull()?.identifier
-            target = FirFunctionTarget(labelName = labelName, isLambda = false)
             functionSymbol = FirAnonymousFunctionSymbol()
             FirAnonymousFunctionBuilder().apply {
                 source = functionSource
                 receiverTypeRef = receiverType
                 symbol = functionSymbol
                 isLambda = false
+                label = context.getLastLabel(functionDeclaration)
+                val labelName = label?.name ?: context.calleeNamesForLambda.lastOrNull()?.identifier
+                target = FirFunctionTarget(labelName = labelName, isLambda = false)
             }
         } else {
             val functionName = identifier.nameAsSafeName()

compiler/fir/raw-fir/psi2fir/src/org/jetbrains/kotlin/fir/builder/RawFirBuilder.kt

@@ -1174,7 +1174,8 @@ open class RawFirBuilder(
                     receiverTypeRef = receiverType
                     symbol = FirAnonymousFunctionSymbol()
                     isLambda = false
-                    labelName = function.getLabelName() ?: context.calleeNamesForLambda.lastOrNull()?.identifier
+                    label = context.getLastLabel(function)
+                    labelName = label?.name ?: context.calleeNamesForLambda.lastOrNull()?.identifier
                 }
             } else {
                 FirSimpleFunctionBuilder().apply {

compiler/testData/diagnostics/tests/labels/labeledFunctionLiteral.fir.kt

@@ -1,17 +1,17 @@
 // !DIAGNOSTICS: -UNUSED_VARIABLE
 
 val funLit = lambda@ fun String.() {
-    val d1 = this<!UNRESOLVED_LABEL!>@lambda<!>
+    val d1 = this@lambda
 }
 
 fun test() {
-    val funLit = lambda@ fun String.(): <!UNRESOLVED_LABEL!>String<!> {
-        return this<!UNRESOLVED_LABEL!>@lambda<!>
+    val funLit = lambda@ fun String.(): String {
+        return this@lambda
     }
 }
 
 fun lambda() {
-    val funLit = lambda@ fun String.(): <!UNRESOLVED_LABEL!>String<!> {
-        return this<!UNRESOLVED_LABEL!>@lambda<!>
+    val funLit = lambda@ fun String.(): String {
+        return this@lambda
     }
 }