FIR: Minor. Drop old FirSupertypeResolverTransformer
This commit is contained in:
Denis Zharkov
-228
@@ -1,228 +0,0 @@
|
||||
/*
|
||||
* Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package org.jetbrains.kotlin.fir.resolve.transformers
|
||||
|
||||
import org.jetbrains.kotlin.fir.FirSession
|
||||
import org.jetbrains.kotlin.fir.declarations.*
|
||||
import org.jetbrains.kotlin.fir.diagnostics.DiagnosticKind
|
||||
import org.jetbrains.kotlin.fir.diagnostics.FirSimpleDiagnostic
|
||||
import org.jetbrains.kotlin.fir.expressions.FirStatement
|
||||
import org.jetbrains.kotlin.fir.render
|
||||
import org.jetbrains.kotlin.fir.resolve.ScopeSession
|
||||
import org.jetbrains.kotlin.fir.resolve.firProvider
|
||||
import org.jetbrains.kotlin.fir.resolve.firSymbolProvider
|
||||
import org.jetbrains.kotlin.fir.scopes.addImportingScopes
|
||||
import org.jetbrains.kotlin.fir.types.*
|
||||
import org.jetbrains.kotlin.fir.types.impl.FirErrorTypeRefImpl
|
||||
import org.jetbrains.kotlin.fir.visitors.CompositeTransformResult
|
||||
import org.jetbrains.kotlin.fir.visitors.compose
|
||||
import org.jetbrains.kotlin.name.ClassId
|
||||
|
||||
class FirSupertypeResolverTransformer : FirAbstractTreeTransformer<Nothing?>(phase = FirResolvePhase.SUPER_TYPES) {
|
||||
override lateinit var session: FirSession
|
||||
private lateinit var file: FirFile
|
||||
|
||||
override fun transformFile(file: FirFile, data: Nothing?): CompositeTransformResult<FirFile> {
|
||||
initFromFile(file)
|
||||
return transformDeclaration(file, data) as CompositeTransformResult<FirFile>
|
||||
}
|
||||
|
||||
fun initFromFile(file: FirFile) {
|
||||
session = file.session
|
||||
this.file = file
|
||||
}
|
||||
|
||||
override fun transformRegularClass(regularClass: FirRegularClass, data: Nothing?): CompositeTransformResult<FirStatement> {
|
||||
val transformedClass = resolveSupertypesOrExpansions(regularClass) as? FirRegularClass ?: regularClass
|
||||
|
||||
// resolve supertypes for nested classes
|
||||
return transformDeclaration(transformedClass, data) as CompositeTransformResult<FirStatement>
|
||||
}
|
||||
|
||||
override fun transformEnumEntry(enumEntry: FirEnumEntry, data: Nothing?): CompositeTransformResult<FirStatement> {
|
||||
return transformRegularClass(enumEntry, data)
|
||||
}
|
||||
|
||||
override fun transformTypeAlias(typeAlias: FirTypeAlias, data: Nothing?): CompositeTransformResult<FirDeclaration> {
|
||||
val result = resolveSupertypesOrExpansions(typeAlias)
|
||||
return result.compose()
|
||||
}
|
||||
|
||||
override fun transformAnonymousInitializer(
|
||||
anonymousInitializer: FirAnonymousInitializer,
|
||||
data: Nothing?
|
||||
): CompositeTransformResult<FirDeclaration> {
|
||||
return anonymousInitializer.compose()
|
||||
}
|
||||
|
||||
override fun transformConstructor(constructor: FirConstructor, data: Nothing?): CompositeTransformResult<FirDeclaration> {
|
||||
return constructor.compose()
|
||||
}
|
||||
|
||||
override fun transformSimpleFunction(simpleFunction: FirSimpleFunction, data: Nothing?): CompositeTransformResult<FirDeclaration> {
|
||||
return simpleFunction.compose()
|
||||
}
|
||||
|
||||
override fun transformProperty(property: FirProperty, data: Nothing?): CompositeTransformResult<FirDeclaration> {
|
||||
return property.compose()
|
||||
}
|
||||
|
||||
override fun <F : FirFunction<F>> transformFunction(function: FirFunction<F>, data: Nothing?): CompositeTransformResult<FirStatement> {
|
||||
return function.compose()
|
||||
}
|
||||
|
||||
private fun resolveSupertypesOrExpansions(classLikeDeclaration: FirClassLikeDeclaration<*>): FirDeclaration {
|
||||
if (classLikeDeclaration.areSupertypesComputed() || classLikeDeclaration.areSupertypesComputing()) {
|
||||
return classLikeDeclaration
|
||||
}
|
||||
|
||||
val classId = classLikeDeclaration.symbol.classId
|
||||
|
||||
val visitor = ResolveSuperTypesTask(session, classId, file, classLikeDeclaration)
|
||||
file.accept(visitor, null).single
|
||||
|
||||
return visitor.resultingClass
|
||||
}
|
||||
|
||||
private class ResolveSuperTypesTask(
|
||||
override val session: FirSession,
|
||||
private val requestedClassId: ClassId,
|
||||
file: FirFile,
|
||||
private val knownFirClassLikeDeclaration: FirClassLikeDeclaration<*>? = null
|
||||
) : FirAbstractTreeTransformerWithSuperTypes(phase = FirResolvePhase.SUPER_TYPES, reversedScopePriority = true) {
|
||||
|
||||
lateinit var resultingClass: FirDeclaration
|
||||
|
||||
init {
|
||||
val scopeSession = ScopeSession()
|
||||
towerScope.addImportingScopes(file, session, scopeSession)
|
||||
}
|
||||
|
||||
override fun transformEnumEntry(enumEntry: FirEnumEntry, data: Nothing?): CompositeTransformResult<FirStatement> {
|
||||
return transformRegularClass(enumEntry, data)
|
||||
}
|
||||
|
||||
override fun transformRegularClass(regularClass: FirRegularClass, data: Nothing?): CompositeTransformResult<FirStatement> {
|
||||
val classId = regularClass.classId
|
||||
if (!isOuterClass(classId, requestedClassId)) return regularClass.compose()
|
||||
val transformedClass = withScopeCleanup {
|
||||
if (regularClass.areSupertypesComputed() || regularClass.areSupertypesComputing()) return@withScopeCleanup regularClass
|
||||
|
||||
regularClass.addTypeParametersScope()
|
||||
|
||||
val transformer = FirSpecificTypeResolverTransformer(towerScope, session)
|
||||
val resolvedTypesRefs = regularClass.superTypeRefs.map {
|
||||
transformer.transformTypeRef(it, data).single
|
||||
}
|
||||
|
||||
val resultingTypeRefs = resolveLoops(regularClass, classId, resolvedTypesRefs)
|
||||
regularClass.replaceSuperTypeRefs(resultingTypeRefs)
|
||||
regularClass
|
||||
}
|
||||
|
||||
if (regularClass.matchesRequestedDeclaration()) {
|
||||
resultingClass = transformedClass
|
||||
return transformedClass.compose()
|
||||
}
|
||||
|
||||
return resolveNestedClassesSupertypes(transformedClass, data)
|
||||
}
|
||||
|
||||
private fun FirClassLikeDeclaration<*>.matchesRequestedDeclaration(): Boolean {
|
||||
if (knownFirClassLikeDeclaration != null) return knownFirClassLikeDeclaration == this
|
||||
return symbol.classId == requestedClassId
|
||||
}
|
||||
|
||||
override fun transformTypeAlias(typeAlias: FirTypeAlias, data: Nothing?): CompositeTransformResult<FirDeclaration> {
|
||||
val classId = typeAlias.symbol.classId
|
||||
// nested type aliases
|
||||
if (typeAlias.areSupertypesComputed() || !typeAlias.matchesRequestedDeclaration()) return typeAlias.compose()
|
||||
|
||||
return withScopeCleanup {
|
||||
typeAlias.addTypeParametersScope()
|
||||
|
||||
val transformer = FirSpecificTypeResolverTransformer(towerScope, session)
|
||||
val resolvedTypesRef = transformer.transformTypeRef(typeAlias.expandedTypeRef, data).single
|
||||
val resultingTypeRef = resolveLoops(typeAlias, classId, listOf(resolvedTypesRef)).firstOrNull()
|
||||
|
||||
typeAlias.replaceExpandedTypeRef(resultingTypeRef ?: resolvedTypesRef)
|
||||
resultingClass = typeAlias
|
||||
typeAlias
|
||||
}.compose()
|
||||
}
|
||||
|
||||
private fun resolveLoops(
|
||||
firClassLikeDeclaration: FirClassLikeDeclaration<*>,
|
||||
classId: ClassId,
|
||||
resolvedTypesRefs: List<FirTypeRef>
|
||||
): List<FirTypeRef> {
|
||||
firClassLikeDeclaration.replaceSupertypesComputationStatus(SupertypesComputationStatus.COMPUTING)
|
||||
|
||||
val resultingTypeRefs = mutableListOf<FirTypeRef>()
|
||||
for (superTypeRef in resolvedTypesRefs) {
|
||||
val coneType = (superTypeRef as FirResolvedTypeRef).type
|
||||
if (coneType is ConeTypeParameterType) {
|
||||
resultingTypeRefs.add(
|
||||
FirErrorTypeRefImpl(superTypeRef.source, FirSimpleDiagnostic("Type parameter cannot be a super-type: ${coneType.render()}", DiagnosticKind.TypeParameterAsSupertype))
|
||||
)
|
||||
continue
|
||||
}
|
||||
val resolvedType = coneType as ConeClassLikeType
|
||||
val superTypeClassId = resolvedType.lookupTag.classId
|
||||
|
||||
if (superTypeClassId.outerClasses().any { it.areSupertypesComputing() }) {
|
||||
resultingTypeRefs.add(
|
||||
FirErrorTypeRefImpl(superTypeRef.source, FirSimpleDiagnostic("Recursion detected: ${superTypeRef.render()}", DiagnosticKind.RecursionInSupertypes))
|
||||
)
|
||||
|
||||
continue
|
||||
}
|
||||
|
||||
val sessionForSupertype = session.firSymbolProvider.getSessionForClass(superTypeClassId) ?: continue
|
||||
|
||||
val firClassForSupertype =
|
||||
sessionForSupertype
|
||||
.firSymbolProvider
|
||||
.getClassLikeSymbolByFqName(superTypeClassId)
|
||||
?.fir
|
||||
|
||||
// TODO: this if is a temporary hack for built-in types (because we can't load file for them)
|
||||
if (firClassForSupertype == null ||
|
||||
(firClassForSupertype is FirClass &&
|
||||
firClassForSupertype.superTypeRefs.any { it !is FirResolvedTypeRef })
|
||||
) {
|
||||
val provider = sessionForSupertype.firProvider
|
||||
val firForSuperClassFile = provider.getFirClassifierContainerFile(superTypeClassId)
|
||||
|
||||
ResolveSuperTypesTask(
|
||||
sessionForSupertype, superTypeClassId, firForSuperClassFile
|
||||
).transformFile(firForSuperClassFile, null)
|
||||
}
|
||||
|
||||
resultingTypeRefs.add(superTypeRef)
|
||||
}
|
||||
|
||||
firClassLikeDeclaration.replaceSupertypesComputationStatus(SupertypesComputationStatus.COMPUTED)
|
||||
return resultingTypeRefs
|
||||
}
|
||||
|
||||
private fun ClassId.areSupertypesComputing(): Boolean {
|
||||
val fir = session.firSymbolProvider.getClassLikeSymbolByFqName(this)?.fir ?: return false
|
||||
return fir.areSupertypesComputing()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun isOuterClass(outerCandidate: ClassId, innerCandidate: ClassId) =
|
||||
innerCandidate.outerClasses().any { outerCandidate == it }
|
||||
|
||||
private fun ClassId.outerClasses() = generateSequence(this, ClassId::getOuterClassId)
|
||||
|
||||
private fun FirClassLikeDeclaration<*>.areSupertypesComputed() =
|
||||
supertypesComputationStatus == SupertypesComputationStatus.COMPUTED
|
||||
|
||||
private fun FirClassLikeDeclaration<*>.areSupertypesComputing() =
|
||||
supertypesComputationStatus == SupertypesComputationStatus.COMPUTING
|
||||
Reference in New Issue
Block a user