[FE] Support functional literals with context receivers

compiler/frontend/src/org/jetbrains/kotlin/resolve/FunctionDescriptorResolver.kt

@@ -198,9 +198,11 @@ class FunctionDescriptorResolver(
             }
 
         val contextReceivers = function.contextReceivers
-        val contextReceiverTypes = contextReceivers.mapNotNull { it.typeReference() }.map {
+        val contextReceiverTypes =
-            typeResolver.resolveType(headerScope, it, trace, true)
+            if (function is KtFunctionLiteral) expectedFunctionType.getContextReceiversTypes()
-        }
+            else contextReceivers
+                .mapNotNull { it.typeReference() }
+                .map { typeResolver.resolveType(headerScope, it, trace, true) }
 
         val valueParameterDescriptors =
             createValueParameterDescriptors(function, functionDescriptor, headerScope, trace, expectedFunctionType, inferenceSession)
@@ -351,6 +353,9 @@ class FunctionDescriptorResolver(
     private fun KotlinType.getReceiverType(): KotlinType? =
         if (functionTypeExpected()) this.getReceiverTypeFromFunctionType() else null
 
+    private fun KotlinType.getContextReceiversTypes(): List<KotlinType> =
+        if (functionTypeExpected()) this.getContextReceiverTypesFromFunctionType() else emptyList()
+
     private fun KotlinType.getValueParameters(owner: FunctionDescriptor): List<ValueParameterDescriptor>? =
         if (functionTypeExpected()) {
             createValueParametersForInvokeInFunctionType(owner, this.getValueParameterTypesFromFunctionType())

compiler/testData/diagnostics/tests/extensions/contextReceivers/functionalType.fir.kt

@@ -1,6 +1,10 @@
 class Param
-class C
+class C {
-class R
+    val c = 42
+}
+class R {
+    val r = 42
+}
 
 context(C)
 fun R.f1(g: context(C) R.(Param) -> Unit) {
@@ -23,10 +27,20 @@ fun f4(g: context(C) () -> Unit) {
 }
 
 fun test() {
-    val lf1: context(C) R.(Param) -> Unit = { _ -> }
+    val lf1: context(C) R.(Param) -> Unit = { _ ->
-    val lf2: context(C) (Param) -> Unit = { _ -> }
+        r
-    val lf3: context(C) R.() -> Unit = { }
+        c
-    val lf4: context(C) () -> Unit = { }
+    }
+    val lf2: context(C) (Param) -> Unit = { _ ->
+        c
+    }
+    val lf3: context(C) R.() -> Unit = {
+        r
+        c
+    }
+    val lf4: context(C) () -> Unit = {
+        c
+    }
 
     with(C()) {
         with(R()) {

compiler/testData/diagnostics/tests/extensions/contextReceivers/functionalType.kt

@@ -1,6 +1,10 @@
 class Param
-class C
+class C {
-class R
+    val c = 42
+}
+class R {
+    val r = 42
+}
 
 context(C)
 fun R.f1(g: context(C) R.(Param) -> Unit) {
@@ -23,10 +27,20 @@ fun f4(g: context(C) () -> Unit) {
 }
 
 fun test() {
-    val lf1: context(C) R.(Param) -> Unit = { _ -> }
+    val lf1: context(C) R.(Param) -> Unit = { _ ->
-    val lf2: context(C) (Param) -> Unit = { _ -> }
+        r
-    val lf3: context(C) R.() -> Unit = { }
+        c
-    val lf4: context(C) () -> Unit = { }
+    }
+    val lf2: context(C) (Param) -> Unit = { _ ->
+        c
+    }
+    val lf3: context(C) R.() -> Unit = {
+        r
+        c
+    }
+    val lf4: context(C) () -> Unit = {
+        c
+    }
 
     with(C()) {
         with(R()) {

compiler/testData/diagnostics/tests/extensions/contextReceivers/functionalType.txt

@@ -8,6 +8,7 @@ public fun R.f3(/*0*/ g: @kotlin.ContextFunctionTypeParams(count = 1) (R.() -> k
 
 public final class C {
     public constructor C()
+    public final val c: kotlin.Int = 42
     public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
     public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
     public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
@@ -22,6 +23,7 @@ public final class Param {
 
 public final class R {
     public constructor R()
+    public final val r: kotlin.Int = 42
     public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
     public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
     public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String