import org.jetbrains.kotlin.analysis.api.symbols.KaConstructorSymbol
import org.jetbrains.kotlin.analysis.api.symbols.KaPackageSymbol
import org.jetbrains.kotlin.fir.scopes.FirScope
import org.jetbrains.kotlin.name.Name

internal abstract class KaFirBasedScope<S : FirScope>(
    internal val firScope: S,
    protected val builder: KaSymbolByFirBuilder,
) : KaScope {
    final override val token: KaLifetimeToken get() = builder.token

        firScope.getCallableNames()
    }

    override fun getPossibleClassifierNames(): Set<Name> = withValidityAssertion {
        firScope.getClassifierNames()
    }

    override fun getCallableSignatures(nameFilter: KaScopeNameFilter): Sequence<KaCallableSignature<*>> = withValidityAssertion {
        firScope.getCallableSignatures(getPossibleCallableNames().filter(nameFilter), builder)

        return firScopes.map { (index, firScope) ->
            KaScopeWithKind(convertToKtScope(firScope), getScopeKind(firScope, index), token)
        }
    }

    private fun flattenFirScope(firScope: FirScope): List<FirScope> = when (firScope) {
        is FirCompositeScope -> firScope.scopes.flatMap { flattenFirScope(it) }

import org.jetbrains.kotlin.fir.scopes.FirContainingNamesAwareScope
import org.jetbrains.kotlin.name.Name

internal open class KaFirDelegatingNamesAwareScope(
    firScope: FirContainingNamesAwareScope,
    builder: KaSymbolByFirBuilder,
) : KaFirBasedScope<FirContainingNamesAwareScope>(firScope, builder) {
    private val allNamesCached by cached {
        getPossibleCallableNames() + getPossibleClassifierNames()
    }

internal class KaFirDefaultStarImportingScope(
    firScope: FirDefaultStarImportingScope,
    analysisSession: KaFirSession,
) : KaFirBasedScope<FirDefaultStarImportingScope>(firScope, analysisSession.firSymbolBuilder) {

    private val firstWrappedScope = KaFirStarImportingScope(firScope.first, analysisSession)
    private val secondWrappedScope = KaFirStarImportingScope(firScope.second, analysisSession)

import org.jetbrains.kotlin.analysis.api.symbols.KaConstructorSymbol
import org.jetbrains.kotlin.fir.scopes.FirScope
import org.jetbrains.kotlin.fir.scopes.processClassifiersByName
import org.jetbrains.kotlin.name.Name


internal fun FirScope.getCallableSymbols(
    callableNames: Collection<Name>,
    builder: KaSymbolByFirBuilder
): Sequence<KaCallableSymbol> = sequence {

import org.jetbrains.kotlin.name.Name

internal class KaFirStarImportingScope(
    firScope: FirAbstractStarImportingScope,
    private val analysisSession: KaFirSession,
) : KaFirBasedScope<FirAbstractStarImportingScope>(firScope, analysisSession.firSymbolBuilder) {

    private val imports: List<StarImport> by cached {
        firScope.starImports.map { import ->
            StarImport(
                import.packageFqName,

        firScope.getCallableSymbols(getPossibleCallableNames().filter(nameFilter), analysisSession.firSymbolBuilder)
    }

    override fun getCallableSymbols(names: Collection<Name>): Sequence<KaCallableSymbol> = withValidityAssertion {
        firScope.getCallableSymbols(names, analysisSession.firSymbolBuilder)
    }

import org.jetbrains.kotlin.name.ClassId
import org.jetbrains.kotlin.name.Name

internal inline fun <reified D : FirCallableDeclaration> FirScope.findDeclarationWithSignature(
    signature: FirCallableSignature,
    processor: FirScope.((FirBasedSymbol<*>) -> Unit) -> Unit,
): D? {
    var foundSymbol: D? = null
    processor { symbol ->
        val declaration = symbol.fir

import org.jetbrains.kotlin.name.Name

internal class KaFirNonStarImportingScope(
    firScope: FirAbstractSimpleImportingScope,
    builder: KaSymbolByFirBuilder,
) : KaFirBasedScope<FirAbstractSimpleImportingScope>(firScope, builder) {

    private val imports: List<NonStarImport> by cached {
        buildList {
            firScope.simpleImports.values.forEach { imports ->
                imports.forEach { import ->