JetBrains · jsjeon · Dec 4, 2025
diff --git a/...xpressionTypeProvider/Fe10IdeNormalAnalysisSourceModuleHLExpressionTypeTestGenerated.java b/...xpressionTypeProvider/Fe10IdeNormalAnalysisSourceModuleHLExpressionTypeTestGenerated.java
diff --git a/...ressionTypeProvider/FirIdeDependentAnalysisSourceModuleHLExpressionTypeTestGenerated.java b/...ressionTypeProvider/FirIdeDependentAnalysisSourceModuleHLExpressionTypeTestGenerated.java
diff --git a/...expressionTypeProvider/FirIdeNormalAnalysisSourceModuleHLExpressionTypeTestGenerated.java b/...expressionTypeProvider/FirIdeNormalAnalysisSourceModuleHLExpressionTypeTestGenerated.java
diff --git a/...ionTypeProvider/FirStandaloneNormalAnalysisSourceModuleHLExpressionTypeTestGenerated.java b/...ionTypeProvider/FirStandaloneNormalAnalysisSourceModuleHLExpressionTypeTestGenerated.java
@@ -0,0 +1,32 @@
+// WITH_STDLIB
+import kotlin.contracts.ExperimentalContracts
+import kotlin.contracts.InvocationKind
+import kotlin.contracts.contract
+
+interface MyDeferred<T> {
+    suspend fun await(): T
+}
+
+abstract class MyException : Exception() {
+    abstract fun isInternal(): Boolean
+}
+
+@OptIn(ExperimentalContracts::class)
+suspend fun <T> MyDeferred<T>.safeAwait(
+    fallbackOnAbort: suspend () -> T,
+    onCancelled: suspend (MyException) -> T = { throw it },
+) : T {
+    contract <expr>{
+        callsInPlace(fallbackOnAbort, InvocationKind.AT_MOST_ONCE)
+        callsInPlace(onCancelled, InvocationKind.AT_MOST_ONCE)
+    }</expr>
+    return try {
+        await()
+    } catch (e: MyException) {
+        if (e.isInternal()) {
+            fallbackOnAbort()
+        } else {
+            onCancelled(e)
+        }
+    }
+}
@@ -0,0 +1,5 @@
+expression: {
+        callsInPlace(fallbackOnAbort, InvocationKind.AT_MOST_ONCE)
+        callsInPlace(onCancelled, InvocationKind.AT_MOST_ONCE)
+    }
+type: kotlin.contracts.ContractBuilder.() -> kotlin.Unit