package dev.openrs2.deob.transform

import com.github.michaelbull.logging.InlineLogger
import dev.openrs2.asm.classpath.ClassPath
import dev.openrs2.asm.classpath.Library
import dev.openrs2.asm.nextReal
import dev.openrs2.asm.transform.Transformer
import org.objectweb.asm.Opcodes
import org.objectweb.asm.tree.ClassNode
import org.objectweb.asm.tree.InsnNode
import org.objectweb.asm.tree.MethodNode

class FernflowerExceptionTransformer : Transformer() {
    private var nopsInserted = 0

    override fun preTransform(classPath: ClassPath) {
        nopsInserted = 0
    }

    override fun transformCode(classPath: ClassPath, library: Library, clazz: ClassNode, method: MethodNode): Boolean {
        if (method.tryCatchBlocks.any { it.end.nextReal == null }) {
            method.instructions.add(InsnNode(Opcodes.NOP))
            nopsInserted++
        }

        return false
    }

    override fun postTransform(classPath: ClassPath) {
        logger.info { "Inserted $nopsInserted NOPs to correct Fernflower's exception generation" }
    }

    private companion object {
        private val logger = InlineLogger()
    }
}