K2 Scripting: add support for script implicit receivers

compiler/fir/fir2ir/src/org/jetbrains/kotlin/fir/backend/Fir2IrVisitor.kt

@@ -217,18 +217,28 @@ class Fir2IrVisitor(
                 declarationStorage.createIrVariable(parameter, irScript, givenOrigin = IrDeclarationOrigin.SCRIPT_CALL_PARAMETER)
             }
 
-            irScript.thisReceiver = script.contextReceivers.find { it.customLabelName?.asString() == SCRIPT_SPECIAL_NAME_STRING }?.let { receiver ->
-                receiver.convertWithOffsets { startOffset, endOffset ->
-                    irFactory.createValueParameter(
-                        startOffset, endOffset, IrDeclarationOrigin.INSTANCE_RECEIVER, SpecialNames.THIS, receiver.typeRef.toIrType(),
-                        isAssignable = false, IrValueParameterSymbolImpl(), UNDEFINED_PARAMETER_INDEX,
-                        varargElementType = null,
-                        isCrossinline = false, isNoinline = false,
-                        isHidden = false,
-                    ).also {
-                        it.parent = irScript
+            // NOTE: index should correspond to one generated in the collectTowerDataElementsForScript
+            irScript.implicitReceiversParameters = script.contextReceivers.mapIndexedNotNull { index, receiver ->
+                val isSelf = receiver.customLabelName?.asString() == SCRIPT_SPECIAL_NAME_STRING
+                val name =
+                    if (isSelf) SpecialNames.THIS
+                    else Name.identifier("${receiver.labelName?.asStringStripSpecialMarkers() ?: SCRIPT_RECEIVER_NAME_PREFIX}_$index")
+                val origin = if (isSelf) IrDeclarationOrigin.INSTANCE_RECEIVER else IrDeclarationOrigin.SCRIPT_IMPLICIT_RECEIVER
+                val irReceiver =
+                    receiver.convertWithOffsets { startOffset, endOffset ->
+                        irFactory.createValueParameter(
+                            startOffset, endOffset, origin, name, receiver.typeRef.toIrType(), isAssignable = false,
+                            IrValueParameterSymbolImpl(),
+                            if (isSelf) UNDEFINED_PARAMETER_INDEX else index,
+                            varargElementType = null, isCrossinline = false, isNoinline = false, isHidden = false
+                        ).also {
+                            it.parent = irScript
+                        }
                     }
-                }
+                if (isSelf) {
+                    irScript.thisReceiver = irReceiver
+                    null
+                } else irReceiver
             }
 
             conversionScope.withParent(irScript) {

libraries/scripting/jvm-host-test/test/kotlin/script/experimental/jvmhost/test/CachingTest.kt

@@ -110,7 +110,7 @@ class CachingTest : TestCase() {
     }
 
     @Test
-    fun testImplicitReceiversWithJarCache() = expectTestToFailOnK2 {
+    fun testImplicitReceiversWithJarCache() {
         withTempDir("scriptingTestJarCache") { cacheDir ->
             val cache = TestCompiledScriptJarsCache(cacheDir)
             Assert.assertTrue(cache.baseDir.listFiles()!!.isEmpty())

libraries/scripting/jvm-host-test/test/kotlin/script/experimental/jvmhost/test/ConstructorArgumentOrderTest.kt

@@ -24,7 +24,7 @@ class ConstructorArgumentsOrderTest {
     }
 
     @Test
-    fun testScriptWithImplicitReceiver() = expectTestToFailOnK2 {
+    fun testScriptWithImplicitReceiver() {
         val res = evalString<ScriptWithImplicitReceiver>("""println(receiverString)""") {
             implicitReceivers(ImplicitReceiverClass("Hello Receiver!"))
         }

libraries/scripting/jvm-host-test/test/kotlin/script/experimental/jvmhost/test/ScriptingHostTest.kt

@@ -223,6 +223,27 @@ class ScriptingHostTest : TestCase() {
         Assert.assertEquals(greeting, output)
     }
 
+    @Test
+    fun testSimpleScriptWithImplicitReceiver() {
+        val greeting = listOf("3")
+        val script = "println(length)"
+        val definition = createJvmScriptDefinitionFromTemplate<SimpleScriptTemplate>(
+            compilation = {
+                implicitReceivers(String::class)
+            },
+            evaluation = {
+                implicitReceivers("abc")
+            }
+        )
+        val output = captureOut {
+            val retVal = BasicJvmScriptingHost().eval(
+                script.toScriptSource(), definition.compilationConfiguration, definition.evaluationConfiguration
+            ).valueOrThrow().returnValue
+            if (retVal is ResultValue.Error) throw retVal.error
+        }.lines()
+        Assert.assertEquals(greeting, output)
+    }
+
     @Test
     fun testProvidedPropertiesNullability() = expectTestToFailOnK2 {
         val stringType = KotlinType(String::class)