import org.jetbrains.kotlin.analysis.api.platform.KotlinPlatformComponent

public abstract class KotlinLifetimeTokenProvider : KotlinPlatformComponent {
    public abstract fun getLifetimeTokenFactory(): KaLifetimeTokenFactory

    public companion object {
        public fun getService(project: Project): KotlinLifetimeTokenProvider =
            project.getService(KotlinLifetimeTokenProvider::class.java)
    }

        KotlinAlwaysAccessibleLifetimeToken(project)
}

public class KotlinAlwaysAccessibleLifetimeTokenProvider : KotlinLifetimeTokenProvider() {
    override fun getLifetimeTokenFactory(): KaLifetimeTokenFactory {
        return KotlinAlwaysAccessibleLifetimeTokenFactory
    }

    override val tokenFactory: KtLifetimeTokenFactory by lazy(LazyThreadSafetyMode.PUBLICATION) {
        KotlinLifetimeTokenProvider.getService(project).getLifetimeTokenFactory()
    }

    override fun beforeEnteringAnalysis(session: KtAnalysisSession, useSiteElement: KtElement) {
        // Catch issues with analysis on invalid PSI as early as possible.

        KotlinReadActionConfinementLifetimeToken(project, modificationTracker)
}

public class KotlinReadActionConfinementLifetimeTokenProvider : KotlinLifetimeTokenProvider() {
    override fun getLifetimeTokenFactory(): KaLifetimeTokenFactory {
        return KotlinReadActionConfinementLifetimeTokenFactory
    }