org.jetbrains.kotlin.fir.java.declarations.FirJavaExternalAnnotation.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler-embeddable Show documentation
the Kotlin compiler embeddable
There is a newer version: 2.1.0-RC
Show newest version
/*
 * Copyright 2010-2023 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.java.declarations

import org.jetbrains.kotlin.KtSourceElement
import org.jetbrains.kotlin.descriptors.annotations.AnnotationUseSiteTarget
import org.jetbrains.kotlin.fir.FirImplementationDetail
import org.jetbrains.kotlin.fir.builder.FirBuilderDsl
import org.jetbrains.kotlin.fir.expressions.FirAnnotation
import org.jetbrains.kotlin.fir.expressions.FirAnnotationArgumentMapping
import org.jetbrains.kotlin.fir.expressions.UnresolvedExpressionTypeAccess
import org.jetbrains.kotlin.fir.types.ConeKotlinType
import org.jetbrains.kotlin.fir.types.FirTypeProjection
import org.jetbrains.kotlin.fir.types.FirTypeRef
import org.jetbrains.kotlin.fir.types.coneTypeOrNull
import org.jetbrains.kotlin.fir.visitors.FirTransformer
import org.jetbrains.kotlin.fir.visitors.FirVisitor
import kotlin.contracts.ExperimentalContracts
import kotlin.contracts.InvocationKind
import kotlin.contracts.contract

class FirJavaExternalAnnotation @FirImplementationDetail constructor(
    override var annotationTypeRef: FirTypeRef,
    override var argumentMapping: FirAnnotationArgumentMapping,
) : FirAnnotation() {
    override val useSiteTarget: AnnotationUseSiteTarget? get() = null
    override val source: KtSourceElement? get() = null

    override val typeArguments: List
        get() = emptyList()

    @OptIn(UnresolvedExpressionTypeAccess::class)
    override val coneTypeOrNull: ConeKotlinType?
        get() = annotationTypeRef.coneTypeOrNull

    override val annotations: List
        get() = emptyList()

    override fun  acceptChildren(visitor: FirVisitor, data: D) {
        annotationTypeRef.accept(visitor, data)
        argumentMapping.accept(visitor, data)
    }

    override fun  transformChildren(transformer: FirTransformer, data: D): FirJavaExternalAnnotation {
        transformAnnotationTypeRef(transformer, data)
        argumentMapping = argumentMapping.transform(transformer, data)
        return this
    }

    override fun  transformAnnotations(transformer: FirTransformer, data: D): FirJavaExternalAnnotation {
        return this
    }

    override fun  transformAnnotationTypeRef(transformer: FirTransformer, data: D): FirJavaExternalAnnotation {
        annotationTypeRef = annotationTypeRef.transform(transformer, data)
        return this
    }

    override fun  transformTypeArguments(transformer: FirTransformer, data: D): FirJavaExternalAnnotation {
        return this
    }

    override fun replaceConeTypeOrNull(newConeTypeOrNull: ConeKotlinType?) {}

    override fun replaceAnnotations(newAnnotations: List) {}

    override fun replaceUseSiteTarget(newUseSiteTarget: AnnotationUseSiteTarget?) {}

    override fun replaceAnnotationTypeRef(newAnnotationTypeRef: FirTypeRef) {
        annotationTypeRef = newAnnotationTypeRef
    }

    override fun replaceArgumentMapping(newArgumentMapping: FirAnnotationArgumentMapping) {
        argumentMapping = newArgumentMapping
    }

    override fun replaceTypeArguments(newTypeArguments: List) {}
}

@FirBuilderDsl
class FirJavaExternalAnnotationBuilder {
    lateinit var annotationTypeRef: FirTypeRef
    lateinit var argumentMapping: FirAnnotationArgumentMapping

    @OptIn(FirImplementationDetail::class)
    fun build(): FirJavaExternalAnnotation = FirJavaExternalAnnotation(
        annotationTypeRef,
        argumentMapping,
    )
}

@OptIn(ExperimentalContracts::class)
inline fun buildJavaExternalAnnotation(init: FirJavaExternalAnnotationBuilder.() -> Unit): FirJavaExternalAnnotation {
    contract {
        callsInPlace(init, InvocationKind.EXACTLY_ONCE)
    }

    return FirJavaExternalAnnotationBuilder().apply(init).build()
}