AA: use nameFilter of KtFileScope#get*Symbols

This commit is contained in:
Jinseong Jeon
2023-05-06 23:15:38 -07:00
committed by Ilya Kirillov
parent f87fdb43fa
commit b42d2e56a4
2 changed files with 6 additions and 7 deletions
@@ -44,7 +44,7 @@ internal class KtFe10FileScope(
for (declaration in ktFile.declarations) { for (declaration in ktFile.declarations) {
if (declaration is KtCallableDeclaration) { if (declaration is KtCallableDeclaration) {
val descriptor = context[BindingContext.DECLARATION_TO_DESCRIPTOR, declaration] as? CallableDescriptor ?: continue val descriptor = context[BindingContext.DECLARATION_TO_DESCRIPTOR, declaration] as? CallableDescriptor ?: continue
val ktSymbol = descriptor.toKtCallableSymbol(analysisContext) ?: continue val ktSymbol = descriptor.takeIf { nameFilter(it.name) }?.toKtCallableSymbol(analysisContext) ?: continue
yield(ktSymbol) yield(ktSymbol)
} }
} }
@@ -58,7 +58,7 @@ internal class KtFe10FileScope(
for (declaration in ktFile.declarations) { for (declaration in ktFile.declarations) {
if (declaration is KtClassLikeDeclaration) { if (declaration is KtClassLikeDeclaration) {
val descriptor = context[BindingContext.DECLARATION_TO_DESCRIPTOR, declaration] as? ClassifierDescriptor ?: continue val descriptor = context[BindingContext.DECLARATION_TO_DESCRIPTOR, declaration] as? ClassifierDescriptor ?: continue
val ktSymbol = descriptor.toKtClassifierSymbol(analysisContext) ?: continue val ktSymbol = descriptor.takeIf { nameFilter(it.name) }?.toKtClassifierSymbol(analysisContext) ?: continue
yield(ktSymbol) yield(ktSymbol)
} }
} }
@@ -82,10 +82,10 @@ internal class KtFirFileScope(
sequence { sequence {
owner.firSymbol.fir.declarations.forEach { firDeclaration -> owner.firSymbol.fir.declarations.forEach { firDeclaration ->
val classLikeDeclaration = when (firDeclaration) { val classLikeDeclaration = when (firDeclaration) {
is FirTypeAlias -> if (nameFilter(firDeclaration.name)) firDeclaration else null is FirTypeAlias -> firDeclaration.takeIf { nameFilter(it.name) }
is FirRegularClass -> if (nameFilter(firDeclaration.name)) firDeclaration else null is FirRegularClass -> firDeclaration.takeIf { nameFilter(it.name) }
else -> null else -> null
} }
if (classLikeDeclaration != null) { if (classLikeDeclaration != null) {
yield(builder.classifierBuilder.buildClassLikeSymbol(classLikeDeclaration.symbol)) yield(builder.classifierBuilder.buildClassLikeSymbol(classLikeDeclaration.symbol))
} }
@@ -93,7 +93,6 @@ internal class KtFirFileScope(
} }
} }
override fun getConstructors(): Sequence<KtConstructorSymbol> = withValidityAssertion { emptySequence() } override fun getConstructors(): Sequence<KtConstructorSymbol> = withValidityAssertion { emptySequence() }
override fun getPackageSymbols(nameFilter: KtScopeNameFilter): Sequence<KtPackageSymbol> = withValidityAssertion { override fun getPackageSymbols(nameFilter: KtScopeNameFilter): Sequence<KtPackageSymbol> = withValidityAssertion {