[FIR] FirCallableSymbol: introduce calculateReturnType

It will help to avoid copy-pasted logic there it is needed

^KT-60728
This commit is contained in:
Dmitrii Gridin
2023-08-02 17:17:49 +02:00
committed by Space Team
parent 35b0cd65e9
commit f549712912
2 changed files with 11 additions and 13 deletions
@@ -26,7 +26,6 @@ import org.jetbrains.kotlin.fir.references.FirSuperReference
import org.jetbrains.kotlin.fir.resolve.constructFunctionType
import org.jetbrains.kotlin.fir.resolve.substitution.ConeSubstitutor
import org.jetbrains.kotlin.fir.symbols.impl.FirClassSymbol
import org.jetbrains.kotlin.fir.symbols.lazyResolveToPhase
import org.jetbrains.kotlin.fir.types.*
import org.jetbrains.kotlin.lexer.KtTokens
import org.jetbrains.kotlin.psi
@@ -179,10 +178,7 @@ internal class KtFirExpressionTypeProvider(
override fun getFunctionalTypeForKtFunction(declaration: KtFunction): KtType {
val firFunction = declaration.resolveToFirSymbol(firResolveSession, FirResolvePhase.TYPES).fir as FirFunction
if (firFunction.returnTypeRef is FirImplicitTypeRef) {
firFunction.lazyResolveToPhase(FirResolvePhase.IMPLICIT_TYPES_BODY_RESOLVE)
}
firFunction.symbol.calculateReturnType()
return firFunction.constructFunctionType(firFunction.specialFunctionTypeKind(firResolveSession.useSiteFirSession)).asKtType()
}