[FIR] Add getCorrespondingPlatformClass, getCorrespondingKotlinClass methods to FirPlatformClassMapper

This commit is contained in:
Ivan Kochurkin
2021-09-16 20:56:18 +03:00
committed by TeamCityServer
parent 291bc74676
commit 4ca757446a
4 changed files with 30 additions and 3 deletions
@@ -11,11 +11,23 @@ import org.jetbrains.kotlin.fir.NoMutableState
import org.jetbrains.kotlin.fir.declarations.FirClassLikeDeclaration
import org.jetbrains.kotlin.fir.declarations.FirRegularClass
import org.jetbrains.kotlin.fir.resolve.providers.symbolProvider
import org.jetbrains.kotlin.fir.scopes.FirPlatformClassMapper
import org.jetbrains.kotlin.name.ClassId
@NoMutableState
class FirJavaClassMapper(private val session: FirSession) : FirPlatformClassMapper() {
override fun getCorrespondingPlatformClass(declaration: FirClassLikeDeclaration): FirRegularClass? {
val javaClassId = JavaToKotlinClassMap.mapKotlinToJava(declaration.symbol.classId.asSingleFqName().toUnsafe())
val javaClassId = getCorrespondingPlatformClass(declaration.symbol.classId)
return javaClassId?.let { session.symbolProvider.getClassLikeSymbolByClassId(it)?.fir } as? FirRegularClass
}
override fun getCorrespondingPlatformClass(classId: ClassId?): ClassId? {
if (classId == null) return null
return JavaToKotlinClassMap.mapKotlinToJava(classId.asSingleFqName().toUnsafe())
}
override fun getCorrespondingKotlinClass(classId: ClassId?): ClassId? {
if (classId == null) return null
return JavaToKotlinClassMap.mapJavaToKotlin(classId.asSingleFqName())
}
}
@@ -17,6 +17,7 @@ import org.jetbrains.kotlin.fir.scopes.FirContainingNamesAwareScope
import org.jetbrains.kotlin.fir.scopes.FirTypeScope
import org.jetbrains.kotlin.fir.scopes.impl.FirClassSubstitutionScope
import org.jetbrains.kotlin.fir.scopes.jvm.JvmMappedScope
import org.jetbrains.kotlin.fir.scopes.platformClassMapper
import org.jetbrains.kotlin.fir.scopes.unsubstitutedScope
import org.jetbrains.kotlin.fir.symbols.impl.FirClassSymbol