KT-53551 KT-52213 KT-58476 Fix handling of suspend functional types with context receivers

compiler/testData/codegen/box/extensionFunctions/contextReceivers/kt52213.kt

@@ -0,0 +1,20 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// WITH_STDLIB
+// WITH_COROUTINES
+
+interface Context
+
+class Receiver
+
+interface Action {
+    context (Context)
+    fun run()
+}
+
+fun execute(block: suspend context(Context) Receiver.(Action) -> Unit) = Unit
+
+fun box(): String {
+    execute { it.run() }
+    return "OK"
+}

compiler/testData/codegen/box/extensionFunctions/contextReceivers/kt53551.kt

@@ -0,0 +1,35 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// WITH_STDLIB
+// WITH_COROUTINES
+
+import helpers.*
+import kotlin.coroutines.*
+import kotlin.coroutines.intrinsics.*
+
+fun box(): String {
+    withFooBar { fooBarBaz() }
+    return "OK"
+}
+
+fun <T> runBlocking(c: suspend () -> T): T {
+    var res: T? = null
+    c.startCoroutine(Continuation(EmptyCoroutineContext) {
+        res = it.getOrThrow()
+    })
+    return res!!
+}
+
+class Foo
+class Bar
+class Baz
+
+context(Foo, Bar)
+fun Baz.fooBarBaz() { }
+
+fun withFooBar(block: suspend context(Foo, Bar) Baz.() -> Unit) {
+    val foo = Foo()
+    val bar = Bar()
+    val baz = Baz()
+    runBlocking { block(foo, bar, baz) }
+}

compiler/testData/codegen/box/extensionFunctions/contextReceivers/kt58476.kt

@@ -0,0 +1,20 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// WITH_STDLIB
+// WITH_COROUTINES
+
+class InContext
+class MyReciever {
+    public suspend fun MyOutput.innerFun(): Int = 123
+}
+class MyOutput
+
+// 2 - Declare the caller that calls the suspended function in a context
+public fun caller(block: suspend context(InContext) MyReciever.() -> Int): MyOutput = MyOutput()
+
+fun box(): String {
+    val out1 = caller {  MyOutput().innerFun() }
+    val out2 = with (InContext()) { caller {  MyOutput().innerFun() } }
+    val out3 = caller { with (InContext()) { MyOutput().innerFun() } }
+    return "OK"
+}