org.jetbrains.kotlin.fir.extensions.FirExtensionService.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010-2020 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.extensions

import org.jetbrains.kotlin.fir.FirSession
import org.jetbrains.kotlin.fir.FirSessionComponent
import org.jetbrains.kotlin.fir.NoMutableState
import org.jetbrains.kotlin.fir.utils.ArrayMapAccessor
import org.jetbrains.kotlin.fir.utils.ComponentArrayOwner
import org.jetbrains.kotlin.fir.utils.Protected
import org.jetbrains.kotlin.fir.utils.TypeRegistry
import kotlin.reflect.KClass

@RequiresOptIn
annotation class PluginServicesInitialization

@NoMutableState
class FirExtensionService(val session: FirSession) : ComponentArrayOwner>(), FirSessionComponent {
    companion object : TypeRegistry>() {
        inline fun > registeredExtensions(): ArrayMapAccessor, V> {
            return generateAccessor(P::class)
        }

        fun > registeredExtensions(kClass: KClass): ArrayMapAccessor, V> {
            return generateAccessor(kClass)
        }
    }

    override val typeRegistry: TypeRegistry>
        get() = Companion

    var registeredExtensionsSize: Int = 0
        private set

    var registeredPredicateBasedExtensionsSize: Int = 0
        private set

    @PluginServicesInitialization
    fun registerExtensions(extensionClass: KClass, extensionFactories: List>) {
        registeredExtensionsSize += extensionFactories.size
        val extensions = extensionFactories.map { it.create(session) }
        registeredPredicateBasedExtensionsSize += extensions.count { it is FirPredicateBasedExtension }
        registerComponent(
            extensionClass,
            extensions
        )
    }

    @PluginServicesInitialization
    @OptIn(Protected::class)
    fun getAllExtensions(): List {
        return arrayMap.flatten()
    }
}

val FirSession.extensionService: FirExtensionService by FirSession.sessionComponentAccessor()

val FirExtensionService.hasExtensions: Boolean
    get() = registeredExtensionsSize > 0

val FirExtensionService.hasPredicateBasedExtensions: Boolean
    get() = registeredPredicateBasedExtensionsSize > 0