[PSI2IR] Check how contextual functional type works with IR

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

@@ -16199,6 +16199,12 @@ public class FirBlackBoxCodegenTestGenerated extends AbstractFirBlackBoxCodegenT
                 runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/contextAndExtensionSameName.kt");
             }
 
+            @Test
+            @TestMetadata("contextualFunctionConversion.kt")
+            public void testContextualFunctionConversion() throws Exception {
+                runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/contextualFunctionConversion.kt");
+            }
+
             @Test
             @TestMetadata("delegatedPropertiesOperators.kt")
             public void testDelegatedPropertiesOperators() throws Exception {
@@ -16235,6 +16241,12 @@ public class FirBlackBoxCodegenTestGenerated extends AbstractFirBlackBoxCodegenT
                 runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/overloading.kt");
             }
 
+            @Test
+            @TestMetadata("passingLambdaToContextualParam.kt")
+            public void testPassingLambdaToContextualParam() throws Exception {
+                runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/passingLambdaToContextualParam.kt");
+            }
+
             @Test
             @TestMetadata("plusAssign.kt")
             public void testPlusAssign() throws Exception {

compiler/fir/fir2ir/tests-gen/org/jetbrains/kotlin/test/runners/ir/Fir2IrTextTestGenerated.java

@@ -775,6 +775,12 @@ public class Fir2IrTextTestGenerated extends AbstractFir2IrTextTest {
                 runTest("compiler/testData/ir/irText/declarations/contextReceivers/compoundAssignmentOperators.kt");
             }
 
+            @Test
+            @TestMetadata("contextualFunctionConversion.kt")
+            public void testContextualFunctionConversion() throws Exception {
+                runTest("compiler/testData/ir/irText/declarations/contextReceivers/contextualFunctionConversion.kt");
+            }
+
             @Test
             @TestMetadata("contextualPrimaryConstructorWithParams.kt")
             public void testContextualPrimaryConstructorWithParams() throws Exception {
@@ -835,6 +841,12 @@ public class Fir2IrTextTestGenerated extends AbstractFir2IrTextTest {
                 runTest("compiler/testData/ir/irText/declarations/contextReceivers/overloading.kt");
             }
 
+            @Test
+            @TestMetadata("passingLambdaToContextualParam.kt")
+            public void testPassingLambdaToContextualParam() throws Exception {
+                runTest("compiler/testData/ir/irText/declarations/contextReceivers/passingLambdaToContextualParam.kt");
+            }
+
             @Test
             @TestMetadata("plusMatrix.kt")
             public void testPlusMatrix() throws Exception {

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

@@ -0,0 +1,12 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// IGNORE_BACKEND_FIR: JVM_IR
+
+fun withContext(f: context(String) () -> String) = f("OK")
+
+fun callWithContext(f: (String) -> String) = withContext(f)
+
+fun box(): String {
+    return callWithContext { s -> s }
+}
+

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

@@ -0,0 +1,11 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// IGNORE_BACKEND_FIR: JVM_IR
+
+class C {
+    val result = "OK"
+}
+
+fun contextual(f: context(C) () -> String) = f(C())
+
+fun box(): String = contextual { result }

compiler/testData/ir/irText/declarations/contextReceivers/contextualFunctionConversion.ir.txt

@@ -0,0 +1,24 @@
+FILE fqName:<root> fileName:/contextualFunctionConversion.kt
+  FUN name:withContext visibility:public modality:FINAL <> (f:@[ContextFunctionTypeParams(count = '1')] kotlin.Function1<kotlin.String, kotlin.String>) returnType:kotlin.String
+    VALUE_PARAMETER name:f index:0 type:@[ContextFunctionTypeParams(count = '1')] kotlin.Function1<kotlin.String, kotlin.String>
+    BLOCK_BODY
+      RETURN type=kotlin.Nothing from='public final fun withContext (f: @[ContextFunctionTypeParams(count = '1')] kotlin.Function1<kotlin.String, kotlin.String>): kotlin.String declared in <root>'
+        CALL 'public abstract fun invoke (p1: P1 of kotlin.Function1): R of kotlin.Function1 [operator] declared in kotlin.Function1' type=kotlin.String origin=INVOKE
+          $this: GET_VAR 'f: @[ContextFunctionTypeParams(count = '1')] kotlin.Function1<kotlin.String, kotlin.String> declared in <root>.withContext' type=@[ContextFunctionTypeParams(count = '1')] kotlin.Function1<kotlin.String, kotlin.String> origin=VARIABLE_AS_FUNCTION
+          p1: CONST String type=kotlin.String value="OK"
+  FUN name:callWithContext visibility:public modality:FINAL <> (f:kotlin.Function1<kotlin.String, kotlin.String>) returnType:kotlin.String
+    VALUE_PARAMETER name:f index:0 type:kotlin.Function1<kotlin.String, kotlin.String>
+    BLOCK_BODY
+      RETURN type=kotlin.Nothing from='public final fun callWithContext (f: kotlin.Function1<kotlin.String, kotlin.String>): kotlin.String declared in <root>'
+        CALL 'public final fun withContext (f: @[ContextFunctionTypeParams(count = '1')] kotlin.Function1<kotlin.String, kotlin.String>): kotlin.String declared in <root>' type=kotlin.String origin=null
+          f: GET_VAR 'f: kotlin.Function1<kotlin.String, kotlin.String> declared in <root>.callWithContext' type=kotlin.Function1<kotlin.String, kotlin.String> origin=null
+  FUN name:box visibility:public modality:FINAL <> () returnType:kotlin.String
+    BLOCK_BODY
+      RETURN type=kotlin.Nothing from='public final fun box (): kotlin.String declared in <root>'
+        CALL 'public final fun callWithContext (f: kotlin.Function1<kotlin.String, kotlin.String>): kotlin.String declared in <root>' type=kotlin.String origin=null
+          f: FUN_EXPR type=kotlin.Function1<kotlin.String, kotlin.String> origin=LAMBDA
+            FUN LOCAL_FUNCTION_FOR_LAMBDA name:<anonymous> visibility:local modality:FINAL <> (s:kotlin.String) returnType:kotlin.String
+              VALUE_PARAMETER name:s index:0 type:kotlin.String
+              BLOCK_BODY
+                RETURN type=kotlin.Nothing from='local final fun <anonymous> (s: kotlin.String): kotlin.String declared in <root>.box'
+                  GET_VAR 's: kotlin.String declared in <root>.box.<anonymous>' type=kotlin.String origin=null

compiler/testData/ir/irText/declarations/contextReceivers/contextualFunctionConversion.kt

@@ -0,0 +1,12 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// IGNORE_BACKEND_FIR: JVM_IR
+
+fun withContext(f: context(String) () -> String) = f("OK")
+
+fun callWithContext(f: (String) -> String) = withContext(f)
+
+fun box(): String {
+    return callWithContext { s -> s }
+}
+

compiler/testData/ir/irText/declarations/contextReceivers/contextualFunctionConversion.kt.txt

@@ -0,0 +1,14 @@
+fun withContext(f: @ContextFunctionTypeParams(count = 1) Function1<String, String>): String {
+  return f.invoke(p1 = "OK")
+}
+
+fun callWithContext(f: Function1<String, String>): String {
+  return withContext(f = f)
+}
+
+fun box(): String {
+  return callWithContext(f = local fun <anonymous>(s: String): String {
+    return s
+  }
+)
+}

compiler/testData/ir/irText/declarations/contextReceivers/passingLambdaToContextualParam.ir.txt

@@ -0,0 +1,50 @@
+FILE fqName:<root> fileName:/passingLambdaToContextualParam.kt
+  CLASS CLASS name:C modality:FINAL visibility:public superTypes:[kotlin.Any]
+    $this: VALUE_PARAMETER INSTANCE_RECEIVER name:<this> type:<root>.C
+    CONSTRUCTOR visibility:public <> () returnType:<root>.C [primary]
+      BLOCK_BODY
+        DELEGATING_CONSTRUCTOR_CALL 'public constructor <init> () [primary] declared in kotlin.Any'
+        INSTANCE_INITIALIZER_CALL classDescriptor='CLASS CLASS name:C modality:FINAL visibility:public superTypes:[kotlin.Any]'
+    PROPERTY name:result visibility:public modality:FINAL [val]
+      FIELD PROPERTY_BACKING_FIELD name:result type:kotlin.String visibility:private [final]
+        EXPRESSION_BODY
+          CONST String type=kotlin.String value="OK"
+      FUN DEFAULT_PROPERTY_ACCESSOR name:<get-result> visibility:public modality:FINAL <> ($this:<root>.C) returnType:kotlin.String
+        correspondingProperty: PROPERTY name:result visibility:public modality:FINAL [val]
+        $this: VALUE_PARAMETER name:<this> type:<root>.C
+        BLOCK_BODY
+          RETURN type=kotlin.Nothing from='public final fun <get-result> (): kotlin.String declared in <root>.C'
+            GET_FIELD 'FIELD PROPERTY_BACKING_FIELD name:result type:kotlin.String visibility:private [final]' type=kotlin.String origin=null
+              receiver: GET_VAR '<this>: <root>.C declared in <root>.C.<get-result>' type=<root>.C origin=null
+    FUN FAKE_OVERRIDE name:equals visibility:public modality:OPEN <> ($this:kotlin.Any, other:kotlin.Any?) returnType:kotlin.Boolean [fake_override,operator]
+      overridden:
+        public open fun equals (other: kotlin.Any?): kotlin.Boolean [operator] declared in kotlin.Any
+      $this: VALUE_PARAMETER name:<this> type:kotlin.Any
+      VALUE_PARAMETER name:other index:0 type:kotlin.Any?
+    FUN FAKE_OVERRIDE name:hashCode visibility:public modality:OPEN <> ($this:kotlin.Any) returnType:kotlin.Int [fake_override]
+      overridden:
+        public open fun hashCode (): kotlin.Int declared in kotlin.Any
+      $this: VALUE_PARAMETER name:<this> type:kotlin.Any
+    FUN FAKE_OVERRIDE name:toString visibility:public modality:OPEN <> ($this:kotlin.Any) returnType:kotlin.String [fake_override]
+      overridden:
+        public open fun toString (): kotlin.String declared in kotlin.Any
+      $this: VALUE_PARAMETER name:<this> type:kotlin.Any
+  FUN name:contextual visibility:public modality:FINAL <> (f:@[ContextFunctionTypeParams(count = '1')] kotlin.Function1<<root>.C, kotlin.String>) returnType:kotlin.String
+    VALUE_PARAMETER name:f index:0 type:@[ContextFunctionTypeParams(count = '1')] kotlin.Function1<<root>.C, kotlin.String>
+    BLOCK_BODY
+      RETURN type=kotlin.Nothing from='public final fun contextual (f: @[ContextFunctionTypeParams(count = '1')] kotlin.Function1<<root>.C, kotlin.String>): kotlin.String declared in <root>'
+        CALL 'public abstract fun invoke (p1: P1 of kotlin.Function1): R of kotlin.Function1 [operator] declared in kotlin.Function1' type=kotlin.String origin=INVOKE
+          $this: GET_VAR 'f: @[ContextFunctionTypeParams(count = '1')] kotlin.Function1<<root>.C, kotlin.String> declared in <root>.contextual' type=@[ContextFunctionTypeParams(count = '1')] kotlin.Function1<<root>.C, kotlin.String> origin=VARIABLE_AS_FUNCTION
+          p1: CONSTRUCTOR_CALL 'public constructor <init> () [primary] declared in <root>.C' type=<root>.C origin=null
+  FUN name:box visibility:public modality:FINAL <> () returnType:kotlin.String
+    BLOCK_BODY
+      RETURN type=kotlin.Nothing from='public final fun box (): kotlin.String declared in <root>'
+        CALL 'public final fun contextual (f: @[ContextFunctionTypeParams(count = '1')] kotlin.Function1<<root>.C, kotlin.String>): kotlin.String declared in <root>' type=kotlin.String origin=null
+          f: FUN_EXPR type=@[ContextFunctionTypeParams(count = '1')] kotlin.Function1<<root>.C, kotlin.String> origin=LAMBDA
+            FUN LOCAL_FUNCTION_FOR_LAMBDA name:<anonymous> visibility:local modality:FINAL <> (<this>:<root>.C) returnType:kotlin.String
+              contextReceiverParametersCount: 1
+              VALUE_PARAMETER name:<this> index:0 type:<root>.C
+              BLOCK_BODY
+                RETURN type=kotlin.Nothing from='local final fun <anonymous> (<this>: <root>.C): kotlin.String declared in <root>.box'
+                  CALL 'public final fun <get-result> (): kotlin.String declared in <root>.C' type=kotlin.String origin=GET_PROPERTY
+                    $this: GET_VAR '<this>: <root>.C declared in <root>.box.<anonymous>' type=<root>.C origin=null

compiler/testData/ir/irText/declarations/contextReceivers/passingLambdaToContextualParam.kt

@@ -0,0 +1,11 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// IGNORE_BACKEND_FIR: JVM_IR
+
+class C {
+    val result = "OK"
+}
+
+fun contextual(f: context(C) () -> String) = f(C())
+
+fun box(): String = contextual { result }

compiler/testData/ir/irText/declarations/contextReceivers/passingLambdaToContextualParam.kt.txt

@@ -0,0 +1,23 @@
+class C {
+  constructor() /* primary */ {
+    super/*Any*/()
+    /* <init>() */
+
+  }
+
+  val result: String
+    field = "OK"
+    get
+
+}
+
+fun contextual(f: @ContextFunctionTypeParams(count = 1) Function1<C, String>): String {
+  return f.invoke(p1 = C())
+}
+
+fun box(): String {
+  return contextual(f = local fun <anonymous>(<this>: C): String {
+    return <this>.<get-result>()
+  }
+)
+}

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/codegen/IrBlackBoxCodegenTestGenerated.java

@@ -16199,6 +16199,12 @@ public class IrBlackBoxCodegenTestGenerated extends AbstractIrBlackBoxCodegenTes
                 runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/contextAndExtensionSameName.kt");
             }
 
+            @Test
+            @TestMetadata("contextualFunctionConversion.kt")
+            public void testContextualFunctionConversion() throws Exception {
+                runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/contextualFunctionConversion.kt");
+            }
+
             @Test
             @TestMetadata("delegatedPropertiesOperators.kt")
             public void testDelegatedPropertiesOperators() throws Exception {
@@ -16235,6 +16241,12 @@ public class IrBlackBoxCodegenTestGenerated extends AbstractIrBlackBoxCodegenTes
                 runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/overloading.kt");
             }
 
+            @Test
+            @TestMetadata("passingLambdaToContextualParam.kt")
+            public void testPassingLambdaToContextualParam() throws Exception {
+                runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/passingLambdaToContextualParam.kt");
+            }
+
             @Test
             @TestMetadata("plusAssign.kt")
             public void testPlusAssign() throws Exception {

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/ir/IrTextTestGenerated.java

@@ -775,6 +775,12 @@ public class IrTextTestGenerated extends AbstractIrTextTest {
                 runTest("compiler/testData/ir/irText/declarations/contextReceivers/compoundAssignmentOperators.kt");
             }
 
+            @Test
+            @TestMetadata("contextualFunctionConversion.kt")
+            public void testContextualFunctionConversion() throws Exception {
+                runTest("compiler/testData/ir/irText/declarations/contextReceivers/contextualFunctionConversion.kt");
+            }
+
             @Test
             @TestMetadata("contextualPrimaryConstructorWithParams.kt")
             public void testContextualPrimaryConstructorWithParams() throws Exception {
@@ -835,6 +841,12 @@ public class IrTextTestGenerated extends AbstractIrTextTest {
                 runTest("compiler/testData/ir/irText/declarations/contextReceivers/overloading.kt");
             }
 
+            @Test
+            @TestMetadata("passingLambdaToContextualParam.kt")
+            public void testPassingLambdaToContextualParam() throws Exception {
+                runTest("compiler/testData/ir/irText/declarations/contextReceivers/passingLambdaToContextualParam.kt");
+            }
+
             @Test
             @TestMetadata("plusMatrix.kt")
             public void testPlusMatrix() throws Exception {

native/native.tests/tests-gen/org/jetbrains/kotlin/konan/blackboxtest/NativeExtBlackBoxTestGenerated.java

@@ -16386,6 +16386,12 @@ public class NativeExtBlackBoxTestGenerated extends AbstractNativeBlackBoxTest {
                     runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/contextAndExtensionSameName.kt");
                 }
 
+                @Test
+                @TestMetadata("contextualFunctionConversion.kt")
+                public void testContextualFunctionConversion() throws Exception {
+                    runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/contextualFunctionConversion.kt");
+                }
+
                 @Test
                 @TestMetadata("delegatedPropertiesOperators.kt")
                 public void testDelegatedPropertiesOperators() throws Exception {
@@ -16422,6 +16428,12 @@ public class NativeExtBlackBoxTestGenerated extends AbstractNativeBlackBoxTest {
                     runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/overloading.kt");
                 }
 
+                @Test
+                @TestMetadata("passingLambdaToContextualParam.kt")
+                public void testPassingLambdaToContextualParam() throws Exception {
+                    runTest("compiler/testData/codegen/box/extensionFunctions/contextReceivers/passingLambdaToContextualParam.kt");
+                }
+
                 @Test
                 @TestMetadata("plusAssign.kt")
                 public void testPlusAssign() throws Exception {