[IR] Generate field names from context parameters correctly

#KT-77713

Author: zhelenskiy

analysis/low-level-api-fir/tests-gen/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLBlackBoxTestGenerated.java

@@ -135,6 +135,9 @@ open class LocalDeclarationsLowering(
     val oldParameterToNew: MutableMap<IrValueParameter, IrValueParameter> = mutableMapOf(),
 ) : BodyLoweringPass {
 
+    open val invalidChars: Set<Char>
+        get() = emptySet()
+
     override fun lower(irFile: IrFile) {
         runOnFilePostfix(irFile)
     }
@@ -1151,7 +1154,10 @@ open class LocalDeclarationsLowering(
                 }
             }
 
-            val base = if (declaration.name.isSpecial) {
+            val baseAsContextParameter = (declaration as? IrValueParameter)?.let {
+                (declaration.parent as IrFunction).anonymousContextParameterName(declaration, invalidChars)
+            }
+            val base = baseAsContextParameter ?: if (declaration.name.isSpecial) {
                 declaration.name.asStringStripSpecialMarkers()
             } else {
                 declaration.name.asString()

analysis/low-level-api-fir/tests-gen/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLReversedBlackBoxTestGenerated.java

@@ -9,6 +9,7 @@ import org.jetbrains.kotlin.backend.common.lower.*
 import org.jetbrains.kotlin.backend.common.phaser.PhaseDescription
 import org.jetbrains.kotlin.backend.jvm.JvmBackendContext
 import org.jetbrains.kotlin.backend.jvm.JvmLoweredDeclarationOrigin
+import org.jetbrains.kotlin.backend.jvm.ir.anonymousContextParameterName
 import org.jetbrains.kotlin.descriptors.DescriptorVisibilities
 import org.jetbrains.kotlin.descriptors.DescriptorVisibility
 import org.jetbrains.kotlin.ir.builders.declarations.addValueParameter
@@ -24,6 +25,7 @@ import org.jetbrains.kotlin.ir.util.parentAsClass
 import org.jetbrains.kotlin.ir.util.parentDeclarationsWithSelf
 import org.jetbrains.kotlin.load.java.JavaDescriptorVisibilities
 import org.jetbrains.kotlin.name.NameUtils
+import org.jetbrains.kotlin.resolve.jvm.JvmConstants
 import org.jetbrains.kotlin.utils.filterIsInstanceAnd
 
 /**
@@ -47,6 +49,9 @@ internal class JvmLocalDeclarationsLowering(override val context: JvmBackendCont
     newParameterToOld = context.evaluatorData?.localDeclarationsData?.newParameterToOld ?: mutableMapOf(),
     oldParameterToNew = context.evaluatorData?.localDeclarationsData?.oldParameterToNew ?: mutableMapOf(),
 ) {
+    override val invalidChars: Set<Char>
+        get() = JvmConstants.INVALID_CHARS
+
     override fun getReplacementSymbolForCaptured(container: IrDeclaration, symbol: IrValueSymbol): IrValueSymbol {
         if (context.evaluatorData?.evaluatorGeneratedFunction == container && !symbol.owner.parentDeclarationsWithSelf.contains(container)) {
             val newParameter = (container as IrFunction).addValueParameter {

compiler/fir/fir2ir/tests-gen/org/jetbrains/kotlin/test/runners/codegen/FirLightTreeBlackBoxCodegenTestGenerated.java

@@ -503,19 +503,7 @@ fun IrFunction.extensionReceiverName(config: JvmBackendConfig): String {
         AsmUtil.LABELED_THIS_PARAMETER + mangleNameIfNeeded(callableName.asString())
 }
 
-private fun String.replaceInvalidChars() =
-    JvmConstants.INVALID_CHARS.fold(this) { acc, ch -> if (ch in acc) acc.replace(ch, '_') else acc }
-
-fun IrFunction.anonymousContextParameterName(parameter: IrValueParameter): String? {
-    if (parameter.kind != IrParameterKind.Context || parameter.origin != UNDERSCORE_PARAMETER) return null
-    val contextParameterNames = parameters
-        .filter { it.kind == IrParameterKind.Context && it.origin == UNDERSCORE_PARAMETER }
-        .associateWith { it.type.erasedUpperBound.name.asString().replaceInvalidChars() }
-    val nameGroups = contextParameterNames.entries.groupBy({ it.value }, { it.key })
-    val baseName = contextParameterNames[parameter]
-    val currentNameGroup = nameGroups[baseName]!!
-    return if (currentNameGroup.size == 1) "\$context-$baseName" else "\$context-$baseName#${currentNameGroup.indexOf(parameter) + 1}"
-}
+fun IrFunction.anonymousContextParameterName(parameter: IrValueParameter): String? = anonymousContextParameterName(parameter, JvmConstants.INVALID_CHARS)
 
 fun IrFunction.isBridge(): Boolean =
     origin == IrDeclarationOrigin.BRIDGE || origin == IrDeclarationOrigin.BRIDGE_SPECIAL

compiler/fir/fir2ir/tests-gen/org/jetbrains/kotlin/test/runners/codegen/FirPsiBlackBoxCodegenTestGenerated.java

@@ -19,6 +19,7 @@ import org.jetbrains.kotlin.ir.builders.IrBuilderWithScope
 import org.jetbrains.kotlin.ir.builders.declarations.*
 import org.jetbrains.kotlin.ir.builders.irImplicitCast
 import org.jetbrains.kotlin.ir.declarations.*
+import org.jetbrains.kotlin.ir.declarations.IrDeclarationOrigin.Companion.UNDERSCORE_PARAMETER
 import org.jetbrains.kotlin.ir.expressions.*
 import org.jetbrains.kotlin.ir.expressions.impl.*
 import org.jetbrains.kotlin.ir.overrides.FakeOverrideBuilderStrategy
@@ -35,6 +36,7 @@ import org.jetbrains.kotlin.util.OperatorNameConventions
 import org.jetbrains.kotlin.utils.*
 import org.jetbrains.kotlin.utils.addToStdlib.assignFrom
 import java.io.StringWriter
+import kotlin.collections.get
 
 /**
  * Binds all arguments represented in the IR to the parameters of the accessed function.
@@ -1592,3 +1594,18 @@ val IrSimpleFunction.isTrivialGetter: Boolean
 
         return (receiver as? IrGetValue)?.symbol?.owner === this.dispatchReceiverParameter
     }
+
+
+private fun String.replaceInvalidChars(invalidChars: Set<Char>) =
+    invalidChars.fold(this) { acc, ch -> if (ch in acc) acc.replace(ch, '_') else acc }
+
+fun IrFunction.anonymousContextParameterName(parameter: IrValueParameter, invalidChars: Set<Char>): String? {
+    if (parameter.kind != IrParameterKind.Context || parameter.origin != UNDERSCORE_PARAMETER) return null
+    val contextParameterNames = parameters
+        .filter { it.kind == IrParameterKind.Context && it.origin == UNDERSCORE_PARAMETER }
+        .associateWith { it.type.erasedUpperBound.name.asString().replaceInvalidChars(invalidChars) }
+    val nameGroups = contextParameterNames.entries.groupBy({ it.value }, { it.key })
+    val baseName = contextParameterNames[parameter]
+    val currentNameGroup = nameGroups[baseName]!!
+    return if (currentNameGroup.size == 1) "\$context-$baseName" else "\$context-$baseName#${currentNameGroup.indexOf(parameter) + 1}"
+}

compiler/ir/backend.common/src/org/jetbrains/kotlin/backend/common/lower/LocalDeclarationsLowering.kt

@@ -0,0 +1,20 @@
+// ISSUE: KT-77713
+// WITH_STDLIB
+// IGNORE_BACKEND_K1: ANY
+// LANGUAGE: +ContextParameters
+// CHECK_BYTECODE_TEXT
+
+context(s: String)
+fun foo() = s
+
+fun bar(f: () -> String) = f()
+
+inline fun baz(crossinline f: () -> String) = bar { f() }
+
+context(_: String)
+fun qux() = baz { foo() }
+
+fun box() = context("OK") { qux() }
+
+
+// 0 unused var\$inlined