import org.jetbrains.kotlin.analysis.api.symbols.KaNamedClassOrObjectSymbol

public abstract class KaInheritorsProvider : KaSessionComponent() {
    public abstract fun getInheritorsOfSealedClass(classSymbol: KaNamedClassOrObjectSymbol): List<KaNamedClassOrObjectSymbol>
    public abstract fun getEnumEntries(classSymbol: KaNamedClassOrObjectSymbol): List<KaEnumEntrySymbol>
}

internal class KaFirInheritorsProvider(
    override val analysisSession: KaFirSession,
    override val token: KaLifetimeToken,
) : KaInheritorsProvider(), KaFirSessionComponent {
    override fun getInheritorsOfSealedClass(
        classSymbol: KaNamedClassOrObjectSymbol
    ): List<KaNamedClassOrObjectSymbol> {
        require(classSymbol.modality == Modality.SEALED)
        require(classSymbol is KaFirNamedClassOrObjectSymbol)

    override val analysisSession: KaFe10Session
) : KaInheritorsProvider(), KaFe10SessionComponent {
    override val token: KaLifetimeToken
        get() = analysisSession.token

    override fun getInheritorsOfSealedClass(classSymbol: KaNamedClassOrObjectSymbol): List<KaNamedClassOrObjectSymbol> {
        val classDescriptor = getSymbolDescriptor(classSymbol) as? ClassDescriptor ?: return emptyList()