ElementaToolsAssignmentTransformer.kt

class ElementaToolsAssignmentTransformer(
    private val context: IrPluginContext
) : IrElementTransformerVoidWithContext(), FileLoweringPass {
    private lateinit var file: IrFile
    private lateinit var fileSource: String

    override fun lower(irFile: IrFile) {
        file = irFile
        fileSource = File(irFile.path).readText()

        irFile.transformChildrenVoid()
    }

    override fun visitSetField(expression: IrSetField): IrExpression {
        val rhs = expression.value
        if (rhs !is IrConstructorCall)
            return super.visitSetField(expression)
        val constructorType = rhs.symbol.owner

        return super.visitSetField(expression)
    }

    override fun visitSetVariable(expression: IrSetVariable): IrExpression {
        val rhs = expression.value
        if (rhs !is IrConstructorCall)
            return super.visitSetVariable(expression)
        val constructorType = rhs.symbol.owner

        return super.visitSetVariable(expression)
    }
}

ElementaToolsIrGenerationExtension.kt

class ElementaToolsIrGenerationExtension : IrGenerationExtension {
    override fun generate(moduleFragment: IrModuleFragment, pluginContext: IrPluginContext) {
        for (file in moduleFragment.files) {
            ElementaToolsAssignmentTransformer(pluginContext).lower(file)
        }
    }
}

test.kt

import club.sk1er.elementa.UIBlock
import kotlin.test.assertEquals
fun main() {
  val myVarName = UIBlock()
  assertEquals("myVarName", myVarName.name)
}