azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

FE1.0 Analysis API: Fix kind in implicit smart-casts info

Browse Source
This commit is contained in:
Simon Ogorodnik
2022-08-09 22:16:11 +02:00
committed by teamcity
parent 513af2dfbc
commit 6dd9115ee9
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/analysis-api-fe10/src/org/jetbrains/kotlin/analysis/api/descriptors/components/KtFe10SmartCastProvider.kt
+12 -4
View File
@@ -18,6 +18,7 @@ import org.jetbrains.kotlin.psi.KtExpression
import org.jetbrains.kotlin.resolve.BindingContext import org.jetbrains.kotlin.resolve.BindingContext
import org.jetbrains.kotlin.resolve.calls.smartcasts.ExplicitSmartCasts import org.jetbrains.kotlin.resolve.calls.smartcasts.ExplicitSmartCasts
import org.jetbrains.kotlin.resolve.calls.smartcasts.MultipleSmartCasts import org.jetbrains.kotlin.resolve.calls.smartcasts.MultipleSmartCasts
import org.jetbrains.kotlin.types.KotlinType
import org.jetbrains.kotlin.types.checker.intersectWrappedTypes import org.jetbrains.kotlin.types.checker.intersectWrappedTypes
internal class KtFe10SmartCastProvider( internal class KtFe10SmartCastProvider(
@@ -51,12 +52,19 @@ internal class KtFe10SmartCastProvider(
return defaultType?.toKtType(analysisContext) return defaultType?.toKtType(analysisContext)
} }
private fun smartCastedImplicitReceiver(type: KotlinType?, kind: KtImplicitReceiverSmartCastKind): KtImplicitReceiverSmartCast? {
if (type == null) return null
return KtImplicitReceiverSmartCast(type.toKtType(analysisContext), kind, token)
}
override fun getImplicitReceiverSmartCast(expression: KtExpression): Collection<KtImplicitReceiverSmartCast> { override fun getImplicitReceiverSmartCast(expression: KtExpression): Collection<KtImplicitReceiverSmartCast> {
val bindingContext = analysisContext.analyze(expression) val bindingContext = analysisContext.analyze(expression)
val smartCasts = bindingContext[BindingContext.IMPLICIT_RECEIVER_SMARTCAST, expression] ?: return emptyList() val smartCasts = bindingContext[BindingContext.IMPLICIT_RECEIVER_SMARTCAST, expression] ?: return emptyList()
return smartCasts.receiverTypes.map { (_, type) -> val call = bindingContext[BindingContext.CALL, expression] ?: return emptyList()
val kind = KtImplicitReceiverSmartCastKind.DISPATCH // TODO provide precise kind val resolvedCall = bindingContext[BindingContext.RESOLVED_CALL, call] ?: return emptyList()
KtImplicitReceiverSmartCast(type.toKtType(analysisContext), kind, token) return listOfNotNull(
} smartCastedImplicitReceiver(smartCasts.receiverTypes[resolvedCall.dispatchReceiver], KtImplicitReceiverSmartCastKind.DISPATCH),
smartCastedImplicitReceiver(smartCasts.receiverTypes[resolvedCall.extensionReceiver], KtImplicitReceiverSmartCastKind.EXTENSION)
)
} }
} }