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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation
the Kotlin compiler
There is a newer version: 2.1.0-RC
/*
 * 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.descriptors.Modality
import org.jetbrains.kotlin.descriptors.Visibility
import org.jetbrains.kotlin.fir.FirSession
import org.jetbrains.kotlin.fir.declarations.*
import org.jetbrains.kotlin.fir.declarations.impl.FirDeclarationStatusImpl
import org.jetbrains.kotlin.fir.symbols.impl.FirClassLikeSymbol
import kotlin.reflect.KClass

abstract class FirStatusTransformerExtension(session: FirSession) : FirExtension(session) {
    companion object {
        val NAME = FirExtensionPointName("StatusTransformer")
    }

    final override val name: FirExtensionPointName
        get() = NAME

    final override val extensionType: KClass = FirStatusTransformerExtension::class

    abstract fun needTransformStatus(declaration: FirDeclaration): Boolean

    protected open fun transformStatus(
        status: FirDeclarationStatus,
        declaration: FirDeclaration
    ): FirDeclarationStatus {
        return status
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        property: FirProperty,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, property)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        function: FirSimpleFunction,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, function)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        regularClass: FirRegularClass,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, regularClass)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        typeAlias: FirTypeAlias,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, typeAlias)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        propertyAccessor: FirPropertyAccessor,
        containingClass: FirClassLikeSymbol<*>?,
        containingProperty: FirProperty?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, propertyAccessor)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        constructor: FirConstructor,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, constructor)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        field: FirField,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, field)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        backingField: FirBackingField,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, backingField)
    }

    open fun transformStatus(
        status: FirDeclarationStatus,
        enumEntry: FirEnumEntry,
        containingClass: FirClassLikeSymbol<*>?,
        isLocal: Boolean
    ): FirDeclarationStatus {
        return transformStatus(status, enumEntry)
    }

    fun interface Factory : FirExtension.Factory
}

val FirExtensionService.statusTransformerExtensions: List by FirExtensionService.registeredExtensions()

inline fun FirDeclarationStatus.transform(
    visibility: Visibility = this.visibility,
    modality: Modality? = this.modality,
    init: FirDeclarationStatusImpl.() -> Unit = {}
): FirDeclarationStatus {
    @Suppress("DuplicatedCode")
    return FirDeclarationStatusImpl(visibility, modality).apply {
        isExpect = [email protected]
        isActual = [email protected]
        isOverride = [email protected]
        isOperator = [email protected]
        isInfix = [email protected]
        isInline = [email protected]
        isTailRec = [email protected]
        isExternal = [email protected]
        isConst = [email protected]
        isLateInit = [email protected]
        isInner = [email protected]
        isCompanion = [email protected]
        isData = [email protected]
        isSuspend = [email protected]
        isStatic = [email protected]
        isFromSealedClass = [email protected]
        isFromEnumClass = [email protected]
    }.apply(init)
}