dev.datlag.sekret.transformer.DeobfuscatorTransformer.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of compiler-plugin Show documentation

Compiler plugin for sekret annotations

The newest version!

package dev.datlag.sekret.transformer

import dev.datlag.sekret.Logger
import dev.datlag.sekret.common.isAnyString
import dev.datlag.sekret.generator.DeobfuscatorGenerator
import org.jetbrains.kotlin.backend.common.IrElementTransformerVoidWithContext
import org.jetbrains.kotlin.backend.common.extensions.IrPluginContext
import org.jetbrains.kotlin.backend.common.lower.DeclarationIrBuilder
import org.jetbrains.kotlin.ir.builders.irCall
import org.jetbrains.kotlin.ir.builders.irInt
import org.jetbrains.kotlin.ir.builders.irLong
import org.jetbrains.kotlin.ir.declarations.*
import org.jetbrains.kotlin.ir.expressions.IrConst
import org.jetbrains.kotlin.ir.expressions.IrConstKind
import org.jetbrains.kotlin.ir.expressions.IrExpression
import org.jetbrains.kotlin.ir.util.addChild

class DeobfuscatorTransformer(
    private val logger: Logger,
    private val pluginContext: IrPluginContext
) : IrElementTransformerVoidWithContext() {

    override fun visitConst(expression: IrConst<*>): IrExpression {
        val getMethod = DeobfuscatorGenerator.getFunction
        if (getMethod != null) {
            if (expression.kind == IrConstKind.String || expression.type.isAnyString()) {
                val value = expression.value?.toString()
                if (value != null) {
                    val stringId = DeobfuscatorGenerator.registerString(value)

                    val getCall = DeclarationIrBuilder(pluginContext, pluginContext.symbols.string).irCall(getMethod)
                    getCall.putValueArgument(0, DeclarationIrBuilder(pluginContext, pluginContext.symbols.long).irLong(stringId))
                    return getCall
                }
            }
        }
        return super.visitConst(expression)
    }
}