main.seskar.compiler.common.backend.JsAnnotations.kt Maven / Gradle / Ivy

Go to download
package seskar.compiler.common.backend

import org.jetbrains.kotlin.backend.common.extensions.IrPluginContext
import org.jetbrains.kotlin.ir.UNDEFINED_OFFSET
import org.jetbrains.kotlin.ir.expressions.IrConstructorCall
import org.jetbrains.kotlin.ir.expressions.impl.IrConstructorCallImpl
import org.jetbrains.kotlin.ir.types.defaultType
import org.jetbrains.kotlin.name.ClassId
import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.name.Name

private val JS_FILE_NAME = ClassId(
    FqName("kotlin.js"),
    Name.identifier("JsFileName"),
)

private val JS_QUALIFIER = ClassId(
    FqName("kotlin.js"),
    Name.identifier("JsQualifier"),
)

private val JS_NAME = ClassId(
    FqName("kotlin.js"),
    Name.identifier("JsName"),
)

internal fun JsFileName(
    context: IrPluginContext,
    value: String,
): IrConstructorCall =
    annotation(
        context = context,
        classId = JS_FILE_NAME,
        parameters = arrayOf(value),
    )

internal fun JsQualifier(
    context: IrPluginContext,
    value: String,
): IrConstructorCall =
    annotation(
        context = context,
        classId = JS_QUALIFIER,
        parameters = arrayOf(value),
    )

internal fun JsName(
    context: IrPluginContext,
    value: String,
): IrConstructorCall =
    annotation(
        context = context,
        classId = JS_NAME,
        parameters = arrayOf(value),
    )

internal fun annotation(
    context: IrPluginContext,
    classId: ClassId,
    vararg parameters: String,
): IrConstructorCall {
    val type = context.referenceClass(classId)!!.defaultType
    val symbol = context.referenceConstructors(classId).single()
    val annotation = IrConstructorCallImpl(
        startOffset = UNDEFINED_OFFSET,
        endOffset = UNDEFINED_OFFSET,
        type = type,
        symbol = symbol,
        typeArgumentsCount = 0,
        constructorTypeArgumentsCount = 0,
        origin = null,
    )

    for ((index, value) in parameters.withIndex()) {
        annotation.putValueArgument(index, stringConst(context, value))
    }

    return annotation
}