FIR IDE: make smartcast API return unstable smartcasts

analysis/analysis-api/src/org/jetbrains/kotlin/analysis/api/components/KtSmartCastProvider.kt

@@ -10,13 +10,18 @@ import org.jetbrains.kotlin.analysis.api.types.KtType
 import org.jetbrains.kotlin.psi.KtExpression
 
 public abstract class KtSmartCastProvider : KtAnalysisSessionComponent() {
-    public abstract fun getSmartCastedToType(expression: KtExpression): KtType?
+    public abstract fun getSmartCastedInfo(expression: KtExpression): SmartCastInfo?
     public abstract fun getImplicitReceiverSmartCast(expression: KtExpression): Collection<ImplicitReceiverSmartCast>
 }
 
+public class SmartCastInfo(public val smartCastType: KtType, public val isStable: Boolean)
+
 public interface KtSmartCastProviderMixIn : KtAnalysisSessionMixIn {
-    public fun KtExpression.getSmartCast(): KtType? =
-        analysisSession.smartCastProvider.getSmartCastedToType(this)
+    /**
+     * Gets the smart-cast information of the given expression or null if the expression is not smart casted.
+     */
+    public fun KtExpression.getSmartCastInfo(): SmartCastInfo? =
+        analysisSession.smartCastProvider.getSmartCastedInfo(this)
 
     public fun KtExpression.getImplicitReceiverSmartCast(): Collection<ImplicitReceiverSmartCast> =
         analysisSession.smartCastProvider.getImplicitReceiverSmartCast(this)

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcastAsReceiver_stable.descriptors.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: null
+smartCastType: null

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcastAsReceiver_stable.kt

@@ -0,0 +1,10 @@
+interface A
+interface B
+
+fun <T> T.foo() where T: A, T: B {}
+
+fun test(a: Any) {
+    if (a is A && a is B) {
+        <expr>a</expr>.foo()
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcastAsReceiver_stable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: true
+smartCastType: (A&B)

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcastAsReceiver_unstable.descriptors.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: null
+smartCastType: null

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcastAsReceiver_unstable.kt

@@ -0,0 +1,11 @@
+interface A
+interface B
+
+fun <T> T.foo() where T: A, T: B {}
+
+var a: Any = 1
+fun test() {
+    if (a is A && a is B) {
+        <expr>a</expr>.foo()
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcastAsReceiver_unstable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: false
+smartCastType: (A&B)

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcast_stable.descriptors.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: null
+smartCastType: null

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcast_stable.kt

@@ -0,0 +1,8 @@
+interface A
+interface B
+
+fun test(a: Any) {
+    if (a is A && a is B) {
+        <expr>a</expr>
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcast_stable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: true
+smartCastType: (A&B)

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcast_unstable.descriptors.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: null
+smartCastType: null

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcast_unstable.kt

@@ -0,0 +1,9 @@
+interface A
+interface B
+
+var a: Any = 1
+fun test() {
+    if (a is A && a is B) {
+        <expr>a</expr>
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/multiSmartcast_unstable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: false
+smartCastType: (A&B)

analysis/analysis-api/testData/components/smartCastInfo/smartcastAsReceiver_stable.kt

@@ -0,0 +1,5 @@
+fun test(a: Any) {
+    if (a is String) {
+        <expr>a</expr>.length
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/smartcastAsReceiver_stable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: true
+smartCastType: kotlin.String

analysis/analysis-api/testData/components/smartCastInfo/smartcastAsReceiver_unstable.kt

@@ -0,0 +1,6 @@
+var a: Any = 1
+fun test() {
+    if (a is String) {
+        <expr>a</expr>.length
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/smartcastAsReceiver_unstable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: null
+smartCastType: null

analysis/analysis-api/testData/components/smartCastInfo/smartcast_stable.descriptors.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: null
+smartCastType: null

analysis/analysis-api/testData/components/smartCastInfo/smartcast_stable.kt

@@ -0,0 +1,5 @@
+fun test(a: Any) {
+    if (a is String) {
+        <expr>a</expr>
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/smartcast_stable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: true
+smartCastType: kotlin.String

analysis/analysis-api/testData/components/smartCastInfo/smartcast_unstable.descriptors.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: null
+smartCastType: null

analysis/analysis-api/testData/components/smartCastInfo/smartcast_unstable.kt

@@ -0,0 +1,6 @@
+var a: Any = 1
+fun test() {
+    if (a is String) {
+        <expr>a</expr>
+    }
+}

analysis/analysis-api/testData/components/smartCastInfo/smartcast_unstable.txt

@@ -0,0 +1,3 @@
+expression: a
+isStable: false
+smartCastType: kotlin.String