org.jetbrains.kotlin.fir.scopes.impl.FirAbstractStarImportingScope.kt Maven / Gradle / Ivy
/*
* Copyright 2010-2021 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.scopes.impl
import org.jetbrains.kotlin.fir.FirSession
import org.jetbrains.kotlin.fir.declarations.FirResolvedImport
import org.jetbrains.kotlin.fir.resolve.ScopeSession
import org.jetbrains.kotlin.fir.resolve.substitution.ConeSubstitutor
import org.jetbrains.kotlin.fir.symbols.impl.FirClassifierSymbol
import org.jetbrains.kotlin.fir.symbols.impl.FirNamedFunctionSymbol
import org.jetbrains.kotlin.fir.symbols.impl.FirVariableSymbol
import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.name.Name
abstract class FirAbstractStarImportingScope(
session: FirSession,
scopeSession: ScopeSession,
lookupInFir: Boolean,
val excludedImportNames: Set
) : FirAbstractImportingScope(session, scopeSession, lookupInFir) {
// TODO try to hide this
abstract val starImports: List
override fun isExcluded(import: FirResolvedImport, name: Name): Boolean {
if (excludedImportNames.isNotEmpty()) {
return import.importedFqName!!.child(name) in excludedImportNames
}
return false
}
private val absentClassifierNames = mutableSetOf()
override fun processClassifiersByNameWithSubstitution(name: Name, processor: (FirClassifierSymbol<*>, ConeSubstitutor) -> Unit) {
if ((!name.isSpecial && name.identifier.isEmpty()) || starImports.isEmpty() || name in absentClassifierNames) {
return
}
var foundAny = false
processImportsByName(name, starImports) { symbol ->
foundAny = true
processor(symbol, ConeSubstitutor.Empty)
}
if (!foundAny) {
absentClassifierNames += name
}
}
override fun processFunctionsByName(name: Name, processor: (FirNamedFunctionSymbol) -> Unit) =
processFunctionsByName(name, starImports, processor)
override fun processPropertiesByName(name: Name, processor: (FirVariableSymbol<*>) -> Unit) =
processPropertiesByName(name, starImports, processor)
}