/*
 * Minecraft Development for IntelliJ
 *
 * https://mcdev.io/
 *
 * Copyright (C) 2026 minecraft-dev
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, version 3.0 only.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
package com.demonwav.mcdev.platform.mixin.handlers.injectionPoint
 
import com.demonwav.mcdev.platform.mixin.handlers.MixinAnnotationHandler
import com.demonwav.mcdev.platform.mixin.reference.MixinSelector
import com.demonwav.mcdev.platform.mixin.util.MethodTargetMember
import com.demonwav.mcdev.util.constantValue
import com.demonwav.mcdev.util.createLiteralExpression
import com.demonwav.mcdev.util.descriptor
import com.demonwav.mcdev.util.enumValueOfOrNull
import com.demonwav.mcdev.util.ifNotBlank
import com.demonwav.mcdev.util.mapToArray
import com.demonwav.mcdev.util.toTypedArray
import com.intellij.codeInsight.lookup.LookupElementBuilder
import com.intellij.openapi.editor.Editor
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.text.StringUtil
import com.intellij.psi.CommonClassNames
import com.intellij.psi.JavaPsiFacade
import com.intellij.psi.JavaTokenType
import com.intellij.psi.PsiAnnotation
import com.intellij.psi.PsiArrayType
import com.intellij.psi.PsiBinaryExpression
import com.intellij.psi.PsiClass
import com.intellij.psi.PsiClassObjectAccessExpression
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiExpression
import com.intellij.psi.PsiForeachStatement
import com.intellij.psi.PsiLiteral
import com.intellij.psi.PsiLiteralExpression
import com.intellij.psi.PsiSwitchLabelStatementBase
import com.intellij.psi.util.PsiUtil
import com.intellij.util.ArrayUtilRt
import java.util.Locale
import org.objectweb.asm.Opcodes
import org.objectweb.asm.Type
import org.objectweb.asm.tree.AbstractInsnNode
import org.objectweb.asm.tree.ClassNode
import org.objectweb.asm.tree.FrameNode
import org.objectweb.asm.tree.InsnNode
import org.objectweb.asm.tree.IntInsnNode
import org.objectweb.asm.tree.JumpInsnNode
import org.objectweb.asm.tree.LabelNode
import org.objectweb.asm.tree.LdcInsnNode
import org.objectweb.asm.tree.MethodNode
import org.objectweb.asm.tree.TypeInsnNode
 
class ConstantInjectionPoint : InjectionPoint<PsiElement>() {
    private object Const {
        val ARGS_KEYS = arrayOf(
            "nullValue=true",
            "intValue",
            "floatValue",
            "longValue",
            "doubleValue",
            "stringValue",
            "classValue",
            "expandZeroConditions"
        )
 
        val COMMA_LIST_DELIMITER = ",".toRegex()
    }
 
    override fun onCompleted(editor: Editor, reference: PsiLiteral) {
        completeExtraStringAtAttribute(editor, reference, "args")
    }
 
    override fun getArgsKeys(at: PsiAnnotation) = Const.ARGS_KEYS
 
    override fun getArgsValues(at: PsiAnnotation, key: String): Array<out Any> {
        fun collectTargets(constantToCompletion: (Any) -> Any?): Array<Any> {
            val injectorAnnotation = AtResolver.findInjectorAnnotation(at) ?: return ArrayUtilRt.EMPTY_OBJECT_ARRAY
            val expandConditions = parseExpandConditions(AtResolver.getArgs(at))
 
            return MixinAnnotationHandler.resolveTarget(injectorAnnotation)
                .asSequence()
                .filterIsInstance<MethodTargetMember>()
                .flatMap { target ->
                    target.classAndMethod.method.instructions?.let { insns ->
                        Iterable { insns.iterator() }.asSequence()
                            .mapNotNull { it.computeConstantValue(expandConditions) }
                            .mapNotNull(constantToCompletion)
                    } ?: emptySequence()
                }
                .toTypedArray()
        }
 
        return when (key) {
            "expandZeroConditions" -> ExpandCondition.entries.toTypedArray().mapToArray { it.name.lowercase(Locale.ROOT) }
            "intValue" -> collectTargets { cst -> cst.takeIf { it is Int } }
            "floatValue" -> collectTargets { cst -> cst.takeIf { it is Float } }
            "longValue" -> collectTargets { cst -> cst.takeIf { it is Long } }
            "doubleValue" -> collectTargets { cst -> cst.takeIf { it is Double } }
            "stringValue" -> collectTargets { cst ->
                (cst as? String)?.let { str ->
                    val escapedStr = StringUtil.escapeStringCharacters(str)
                    when {
                        str.isEmpty() -> null
                        escapedStr.trim() != escapedStr -> LookupElementBuilder.create(escapedStr)
                            .withPresentableText("\"${escapedStr}\"")
                        else -> escapedStr
                    }
                }
            }
            "classValue" -> collectTargets { cst -> (cst as? Type)?.internalName }
            else -> ArrayUtilRt.EMPTY_OBJECT_ARRAY
        }
    }
 
    override fun getArgValueListDelimiter(at: PsiAnnotation, key: String) =
        Const.COMMA_LIST_DELIMITER.takeIf { key == "expandZeroConditions" }
 
    fun getConstantInfo(at: PsiAnnotation): ConstantInfo? {
        val args = AtResolver.getArgs(at)
        val nullValue = args["nullValue"]?.let(java.lang.Boolean::parseBoolean) ?: false
        val intValue = args["intValue"]?.toIntOrNull()
        val floatValue = args["floatValue"]?.toFloatOrNull()
        val longValue = args["longValue"]?.toLongOrNull()
        val doubleValue = args["doubleValue"]?.toDoubleOrNull()
        val stringValue = args["stringValue"]
        val classValue = args["classValue"]?.ifNotBlank { Type.getObjectType(it.replace('.', '/')) }
        val count =
            nullValue.toInt() +
                (intValue != null).toInt() +
                (floatValue != null).toInt() +
                (longValue != null).toInt() +
                (doubleValue != null).toInt() +
                (stringValue != null).toInt() +
                (classValue != null).toInt()
        if (count != 1) {
            return null
        }
 
        val constant = if (nullValue) {
            null
        } else {
            intValue ?: floatValue ?: longValue ?: doubleValue ?: stringValue ?: classValue!!
        }
 
        return ConstantInfo(constant, parseExpandConditions(args))
    }
 
    private fun parseExpandConditions(args: Map<String, String>): Set<ExpandCondition> {
        return args["expandZeroConditions"]
            ?.replace(" ", "")
            ?.split(',')
            ?.mapNotNull { enumValueOfOrNull<ExpandCondition>(it.uppercase(Locale.ROOT)) }
            ?.toSet() ?: emptySet()
    }
 
    private fun Boolean.toInt(): Int {
        return if (this) 1 else 0
    }
 
    override fun createNavigationVisitor(
        at: PsiAnnotation,
        target: MixinSelector?,
        targetClass: PsiClass,
    ): NavigationVisitor {
        return MyNavigationVisitor(getConstantInfo(at))
    }
 
    override fun doCreateCollectVisitor(
        at: PsiAnnotation,
        target: MixinSelector?,
        targetClass: ClassNode,
        mode: CollectVisitor.Mode,
    ): CollectVisitor<PsiElement> {
        return MyCollectVisitor(at.project, mode, getConstantInfo(at))
    }
 
    override fun createLookup(
        targetClass: ClassNode,
        result: CollectVisitor.Result<PsiElement>,
    ): LookupElementBuilder? {
        return null
    }
 
    class ConstantInfo(val constant: Any?, val expandConditions: Set<ExpandCondition>)
 
    enum class ExpandCondition(vararg val opcodes: Int) {
        LESS_THAN_ZERO(Opcodes.IFLT, Opcodes.IFGE),
        LESS_THAN_OR_EQUAL_TO_ZERO(Opcodes.IFLE, Opcodes.IFGT),
        GREATER_THAN_ZERO(Opcodes.IFLE, Opcodes.IFGT),
        GREATER_THAN_OR_EQUAL_TO_ZERO(Opcodes.IFLT, Opcodes.IFGE),
    }
 
    class MyNavigationVisitor(
        private val constantInfo: ConstantInfo?,
        private val expectedType: Type? = null,
    ) : NavigationVisitor() {
        override fun visitForeachStatement(statement: PsiForeachStatement) {
            if (statement.iteratedValue?.type is PsiArrayType) {
                // index initialized to 0
                visitConstant(statement, 0)
            }
            super.visitForeachStatement(statement)
        }
 
        override fun visitClassObjectAccessExpression(expression: PsiClassObjectAccessExpression) {
            visitConstant(expression, Type.getType(expression.operand.type.descriptor))
        }
 
        override fun visitLiteralExpression(expression: PsiLiteralExpression) {
            if (expression.textMatches("null")) {
                visitConstant(expression, null)
            } else {
                super.visitLiteralExpression(expression)
            }
        }
 
        override fun visitExpression(expression: PsiExpression) {
            if (PsiUtil.isConstantExpression(expression)) {
                val value = expression.constantValue
                if (value != null) {
                    visitConstant(expression, value)
                    return
                }
            }
            super.visitExpression(expression)
        }
 
        private fun visitConstant(element: PsiElement, value: Any?) {
            if (constantInfo != null && value != constantInfo.constant) {
                return
            }
 
            if (expectedType != null && value != null) {
                // First check if we expect any String literal
                if (value is String &&
                    (expectedType.sort != Type.OBJECT || expectedType.className != CommonClassNames.JAVA_LANG_STRING)
                ) {
                    return
                }
 
                // then check if we expect any class literal
                if (value is Type && (
                    expectedType.sort != Type.ARRAY && expectedType.sort != Type.OBJECT ||
                        expectedType.className != CommonClassNames.JAVA_LANG_CLASS
                    )
                ) {
                    return
                }
 
                // otherwise we expect a primitive literal
                if (expectedType.sort in Type.BOOLEAN..Type.DOUBLE &&
                    value::class.javaPrimitiveType?.let(Type::getType) != expectedType
                ) {
                    return
                }
            }
 
            val parent = PsiUtil.skipParenthesizedExprUp(element.parent)
 
            // check for expandZeroConditions
            if (value == null || value == 0) {
                if (parent is PsiBinaryExpression) {
                    val operation = parent.operationTokenType
                    if (operation == JavaTokenType.EQEQ || operation == JavaTokenType.NE) {
                        return
                    }
                    val opcode = when (operation) {
                        JavaTokenType.LT -> Opcodes.IFLT
                        JavaTokenType.LE -> Opcodes.IFLE
                        JavaTokenType.GT -> Opcodes.IFGT
                        JavaTokenType.GE -> Opcodes.IFGE
                        else -> null
                    }
                    if (opcode != null && (
                        constantInfo == null ||
                            !constantInfo.expandConditions.any { opcode in it.opcodes }
                        )
                    ) {
                        return
                    }
                }
            }
 
            // check for switch statement (compiles to tableswitch or lookupswitch which aren't targeted)
            if (parent is PsiSwitchLabelStatementBase) {
                return
            }
 
            addResult(element)
        }
    }
 
    class MyCollectVisitor(
        private val project: Project,
        mode: Mode,
        private val constantInfo: ConstantInfo?,
        private val expectedType: Type? = null,
    ) : CollectVisitor<PsiElement>(mode) {
        override fun accept(methodNode: MethodNode) = sequence {
            for (insn in methodNode.instructions ?: emptyList()) {
                val constant = (
                    insn.computeConstantValue(constantInfo?.expandConditions ?: emptySet())
                        ?: continue
                    ).let { if (it is NullSentinel) null else it }
 
                if (constantInfo != null && constant != constantInfo.constant) {
                    continue
                }
 
                if (expectedType != null && constant != null) {
                    // First check if we expect any String literal
                    if (constant is String && (
                        expectedType.sort != Type.OBJECT ||
                            expectedType.className != CommonClassNames.JAVA_LANG_STRING
                        )
                    ) {
                        continue
                    }
 
                    // then check if we expect any class literal
                    if (constant is Type && (
                        expectedType.sort != Type.ARRAY && expectedType.sort != Type.OBJECT ||
                            expectedType.className != CommonClassNames.JAVA_LANG_CLASS
                        )
                    ) {
                        continue
                    }
 
                    // otherwise we expect a primitive literal
                    if (expectedType.sort in Type.BOOLEAN..Type.DOUBLE &&
                        constant::class.javaPrimitiveType?.let(Type::getType) != expectedType
                    ) {
                        continue
                    }
                }
 
                val elementFactory = JavaPsiFacade.getElementFactory(project)
                val literal = if (constant is Type) {
                    elementFactory.createExpressionFromText("${constant.className}.class", null)
                } else {
                    elementFactory.createLiteralExpression(constant)
                }
                addResult(insn, literal)
            }
        }
    }
}
 
private object NullSentinel
 
private fun AbstractInsnNode.computeConstantValue(expandConditions: Set<ConstantInjectionPoint.ExpandCondition>): Any? {
    return when (this) {
        is InsnNode -> when (opcode) {
            in Opcodes.ICONST_M1..Opcodes.ICONST_5 -> opcode - Opcodes.ICONST_0
            Opcodes.LCONST_0 -> 0L
            Opcodes.LCONST_1 -> 1L
            Opcodes.FCONST_0 -> 0.0f
            Opcodes.FCONST_1 -> 1.0f
            Opcodes.FCONST_2 -> 2.0f
            Opcodes.DCONST_0 -> 0.0
            Opcodes.DCONST_1 -> 1.0
            Opcodes.ACONST_NULL -> NullSentinel
            else -> null
        }
 
        is IntInsnNode -> when (opcode) {
            Opcodes.BIPUSH, Opcodes.SIPUSH -> operand
            else -> null
        }
 
        is LdcInsnNode -> cst
        is JumpInsnNode -> {
            if (expandConditions.none { opcode in it.opcodes }) {
                return null
            }
            var lastInsn = previous
            while (lastInsn != null && (lastInsn is LabelNode || lastInsn is FrameNode)) {
                lastInsn = lastInsn.previous
            }
            if (lastInsn != null) {
                val lastOpcode = lastInsn.opcode
                if (lastOpcode == Opcodes.LCMP ||
                    lastOpcode == Opcodes.FCMPL ||
                    lastOpcode == Opcodes.FCMPG ||
                    lastOpcode == Opcodes.DCMPL ||
                    lastOpcode == Opcodes.DCMPG
                ) {
                    return null
                }
            }
            0
        }
 
        is TypeInsnNode -> {
            if (opcode < Opcodes.CHECKCAST) {
                // Don't treat NEW and ANEWARRAY as constants
                // Matches Mixin's handling
                return null
            }
            Type.getObjectType(desc)
        }
 
        else -> null
    }
}

      

        
/*
 * Minecraft Development for IntelliJ
 *
 * https://mcdev.io/
 *
 * Copyright (C) 2026 minecraft-dev
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, version 3.0 only.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
package com.demonwav.mcdev.platform.mixin.handlers.injectionPoint
 
import com.demonwav.mcdev.platform.mixin.handlers.MixinAnnotationHandler
import com.demonwav.mcdev.platform.mixin.reference.MixinSelector
import com.demonwav.mcdev.platform.mixin.util.MethodTargetMember
import com.demonwav.mcdev.util.constantValue
import com.demonwav.mcdev.util.createLiteralExpression
import com.demonwav.mcdev.util.descriptor
import com.demonwav.mcdev.util.enumValueOfOrNull
import com.demonwav.mcdev.util.ifNotBlank
import com.demonwav.mcdev.util.mapToArray
import com.demonwav.mcdev.util.toTypedArray
import com.intellij.codeInsight.lookup.LookupElementBuilder
import com.intellij.openapi.editor.Editor
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.text.StringUtil
import com.intellij.psi.CommonClassNames
import com.intellij.psi.JavaPsiFacade
import com.intellij.psi.JavaTokenType
import com.intellij.psi.PsiAnnotation
import com.intellij.psi.PsiArrayType
import com.intellij.psi.PsiBinaryExpression
import com.intellij.psi.PsiClass
import com.intellij.psi.PsiClassObjectAccessExpression
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiExpression
import com.intellij.psi.PsiForeachStatement
import com.intellij.psi.PsiLiteral
import com.intellij.psi.PsiLiteralExpression
import com.intellij.psi.PsiSwitchLabelStatementBase
import com.intellij.psi.util.PsiUtil
import com.intellij.util.ArrayUtilRt
import java.util.EnumSet
import java.util.Locale
import org.objectweb.asm.Opcodes
import org.objectweb.asm.Type
import org.objectweb.asm.tree.AbstractInsnNode
import org.objectweb.asm.tree.ClassNode
import org.objectweb.asm.tree.FrameNode
import org.objectweb.asm.tree.InsnNode
import org.objectweb.asm.tree.IntInsnNode
import org.objectweb.asm.tree.JumpInsnNode
import org.objectweb.asm.tree.LabelNode
import org.objectweb.asm.tree.LdcInsnNode
import org.objectweb.asm.tree.MethodNode
import org.objectweb.asm.tree.TypeInsnNode
 
class ConstantInjectionPoint : InjectionPoint<PsiElement>() {
    private object Const {
        val ARGS_KEYS = arrayOf(
            "nullValue=true",
            "intValue",
            "floatValue",
            "longValue",
            "doubleValue",
            "stringValue",
            "classValue",
            "expandZeroConditions"
        )
 
        val COMMA_LIST_DELIMITER = ",".toRegex()
    }
 
    override fun onCompleted(editor: Editor, reference: PsiLiteral) {
        completeExtraStringAtAttribute(editor, reference, "args")
    }
 
    override fun getArgsKeys(at: PsiAnnotation) = Const.ARGS_KEYS
 
    override fun getArgsValues(at: PsiAnnotation, key: String): Array<out Any> {
        fun collectTargets(constantToCompletion: (Any) -> Any?): Array<Any> {
            val injectorAnnotation = AtResolver.findInjectorAnnotation(at) ?: return ArrayUtilRt.EMPTY_OBJECT_ARRAY
            val expandConditions = parseExpandConditions(AtResolver.getArgs(at))
 
            return MixinAnnotationHandler.resolveTarget(injectorAnnotation)
                .asSequence()
                .filterIsInstance<MethodTargetMember>()
                .flatMap { target ->
                    target.classAndMethod.method.instructions?.let { insns ->
                        Iterable { insns.iterator() }.asSequence()
                            .mapNotNull { it.computeConstantValue(expandConditions) }
                            .mapNotNull(constantToCompletion)
                    } ?: emptySequence()
                }
                .toTypedArray()
        }
 
        return when (key) {
            "expandZeroConditions" -> ExpandCondition.entries.toTypedArray().mapToArray { it.name.lowercase(Locale.ROOT) }
            "intValue" -> collectTargets { cst -> cst.takeIf { it is Int } }
            "floatValue" -> collectTargets { cst -> cst.takeIf { it is Float } }
            "longValue" -> collectTargets { cst -> cst.takeIf { it is Long } }
            "doubleValue" -> collectTargets { cst -> cst.takeIf { it is Double } }
            "stringValue" -> collectTargets { cst ->
                (cst as? String)?.let { str ->
                    val escapedStr = StringUtil.escapeStringCharacters(str)
                    when {
                        str.isEmpty() -> null
                        escapedStr.trim() != escapedStr -> LookupElementBuilder.create(escapedStr)
                            .withPresentableText("\"${escapedStr}\"")
                        else -> escapedStr
                    }
                }
            }
            "classValue" -> collectTargets { cst -> (cst as? Type)?.internalName }
            else -> ArrayUtilRt.EMPTY_OBJECT_ARRAY
        }
    }
 
    override fun getArgValueListDelimiter(at: PsiAnnotation, key: String) =
        Const.COMMA_LIST_DELIMITER.takeIf { key == "expandZeroConditions" }
 
    fun getConstantInfo(at: PsiAnnotation): ConstantInfo? {
        val args = AtResolver.getArgs(at)
        val nullValue = args["nullValue"]?.let(java.lang.Boolean::parseBoolean) ?: false
        val intValue = args["intValue"]?.toIntOrNull()
        val floatValue = args["floatValue"]?.toFloatOrNull()
        val longValue = args["longValue"]?.toLongOrNull()
        val doubleValue = args["doubleValue"]?.toDoubleOrNull()
        val stringValue = args["stringValue"]
        val classValue = args["classValue"]?.ifNotBlank { Type.getObjectType(it.replace('.', '/')) }
        val count =
            nullValue.toInt() +
                (intValue != null).toInt() +
                (floatValue != null).toInt() +
                (longValue != null).toInt() +
                (doubleValue != null).toInt() +
                (stringValue != null).toInt() +
                (classValue != null).toInt()
        if (count != 1) {
            return null
        }
 
        val constant = if (nullValue) {
            null
        } else {
            intValue ?: floatValue ?: longValue ?: doubleValue ?: stringValue ?: classValue!!
        }
 
        return ConstantInfo(constant, parseExpandConditions(args))
    }
 
    private fun parseExpandConditions(args: Map<String, String>): Set<ExpandCondition> {
        return args["expandZeroConditions"]
            ?.replace(" ", "")
            ?.split(',')
            ?.mapNotNull { enumValueOfOrNull<ExpandCondition>(it.uppercase(Locale.ROOT)) }
            ?.toSet() ?: emptySet()
    }
 
    private fun Boolean.toInt(): Int {
        return if (this) 1 else 0
    }
 
    override fun createNavigationVisitor(
        at: PsiAnnotation,
        target: MixinSelector?,
        targetClass: PsiClass,
    ): NavigationVisitor {
        return MyNavigationVisitor(getConstantInfo(at))
    }
 
    override fun doCreateCollectVisitor(
        at: PsiAnnotation,
        target: MixinSelector?,
        targetClass: ClassNode,
        mode: CollectVisitor.Mode,
    ): CollectVisitor<PsiElement> {
        return MyCollectVisitor(at.project, mode, getConstantInfo(at))
    }
 
    override fun createLookup(
        targetClass: ClassNode,
        result: CollectVisitor.Result<PsiElement>,
    ): LookupElementBuilder? {
        return null
    }
 
    fun getTargetedConstant(insn: AbstractInsnNode) =
        insn.computeConstantValue(EnumSet.allOf(ExpandCondition::class.java))
            ?.let { if (it === NullSentinel) null else it }
 
    class ConstantInfo(val constant: Any?, val expandConditions: Set<ExpandCondition>)
 
    enum class ExpandCondition(vararg val opcodes: Int) {
        LESS_THAN_ZERO(Opcodes.IFLT, Opcodes.IFGE),
        LESS_THAN_OR_EQUAL_TO_ZERO(Opcodes.IFLE, Opcodes.IFGT),
        GREATER_THAN_ZERO(Opcodes.IFLE, Opcodes.IFGT),
        GREATER_THAN_OR_EQUAL_TO_ZERO(Opcodes.IFLT, Opcodes.IFGE),
    }
 
    class MyNavigationVisitor(
        private val constantInfo: ConstantInfo?,
        private val expectedType: Type? = null,
    ) : NavigationVisitor() {
        override fun visitForeachStatement(statement: PsiForeachStatement) {
            if (statement.iteratedValue?.type is PsiArrayType) {
                // index initialized to 0
                visitConstant(statement, 0)
            }
            super.visitForeachStatement(statement)
        }
 
        override fun visitClassObjectAccessExpression(expression: PsiClassObjectAccessExpression) {
            visitConstant(expression, Type.getType(expression.operand.type.descriptor))
        }
 
        override fun visitLiteralExpression(expression: PsiLiteralExpression) {
            if (expression.textMatches("null")) {
                visitConstant(expression, null)
            } else {
                super.visitLiteralExpression(expression)
            }
        }
 
        override fun visitExpression(expression: PsiExpression) {
            if (PsiUtil.isConstantExpression(expression)) {
                val value = expression.constantValue
                if (value != null) {
                    visitConstant(expression, value)
                    return
                }
            }
            super.visitExpression(expression)
        }
 
        private fun visitConstant(element: PsiElement, value: Any?) {
            if (constantInfo != null && value != constantInfo.constant) {
                return
            }
 
            if (expectedType != null && value != null) {
                // First check if we expect any String literal
                if (value is String &&
                    (expectedType.sort != Type.OBJECT || expectedType.className != CommonClassNames.JAVA_LANG_STRING)
                ) {
                    return
                }
 
                // then check if we expect any class literal
                if (value is Type && (
                    expectedType.sort != Type.ARRAY && expectedType.sort != Type.OBJECT ||
                        expectedType.className != CommonClassNames.JAVA_LANG_CLASS
                    )
                ) {
                    return
                }
 
                // otherwise we expect a primitive literal
                if (expectedType.sort in Type.BOOLEAN..Type.DOUBLE &&
                    value::class.javaPrimitiveType?.let(Type::getType) != expectedType
                ) {
                    return
                }
            }
 
            val parent = PsiUtil.skipParenthesizedExprUp(element.parent)
 
            // check for expandZeroConditions
            if (value == null || value == 0) {
                if (parent is PsiBinaryExpression) {
                    val operation = parent.operationTokenType
                    if (operation == JavaTokenType.EQEQ || operation == JavaTokenType.NE) {
                        return
                    }
                    val opcode = when (operation) {
                        JavaTokenType.LT -> Opcodes.IFLT
                        JavaTokenType.LE -> Opcodes.IFLE
                        JavaTokenType.GT -> Opcodes.IFGT
                        JavaTokenType.GE -> Opcodes.IFGE
                        else -> null
                    }
                    if (opcode != null && (
                        constantInfo == null ||
                            !constantInfo.expandConditions.any { opcode in it.opcodes }
                        )
                    ) {
                        return
                    }
                }
            }
 
            // check for switch statement (compiles to tableswitch or lookupswitch which aren't targeted)
            if (parent is PsiSwitchLabelStatementBase) {
                return
            }
 
            addResult(element)
        }
    }
 
    class MyCollectVisitor(
        private val project: Project,
        mode: Mode,
        private val constantInfo: ConstantInfo?,
        private val expectedType: Type? = null,
    ) : CollectVisitor<PsiElement>(mode) {
        override fun accept(methodNode: MethodNode) = sequence {
            for (insn in methodNode.instructions ?: emptyList()) {
                val constant = (
                    insn.computeConstantValue(constantInfo?.expandConditions ?: emptySet())
                        ?: continue
                    ).let { if (it is NullSentinel) null else it }
 
                if (constantInfo != null && constant != constantInfo.constant) {
                    continue
                }
 
                if (expectedType != null && constant != null) {
                    // First check if we expect any String literal
                    if (constant is String && (
                        expectedType.sort != Type.OBJECT ||
                            expectedType.className != CommonClassNames.JAVA_LANG_STRING
                        )
                    ) {
                        continue
                    }
 
                    // then check if we expect any class literal
                    if (constant is Type && (
                        expectedType.sort != Type.ARRAY && expectedType.sort != Type.OBJECT ||
                            expectedType.className != CommonClassNames.JAVA_LANG_CLASS
                        )
                    ) {
                        continue
                    }
 
                    // otherwise we expect a primitive literal
                    if (expectedType.sort in Type.BOOLEAN..Type.DOUBLE &&
                        constant::class.javaPrimitiveType?.let(Type::getType) != expectedType
                    ) {
                        continue
                    }
                }
 
                val elementFactory = JavaPsiFacade.getElementFactory(project)
                val literal = if (constant is Type) {
                    elementFactory.createExpressionFromText("${constant.className}.class", null)
                } else {
                    elementFactory.createLiteralExpression(constant)
                }
                addResult(insn, literal)
            }
        }
    }
}
 
private object NullSentinel
 
private fun AbstractInsnNode.computeConstantValue(expandConditions: Set<ConstantInjectionPoint.ExpandCondition>): Any? {
    return when (this) {
        is InsnNode -> when (opcode) {
            in Opcodes.ICONST_M1..Opcodes.ICONST_5 -> opcode - Opcodes.ICONST_0
            Opcodes.LCONST_0 -> 0L
            Opcodes.LCONST_1 -> 1L
            Opcodes.FCONST_0 -> 0.0f
            Opcodes.FCONST_1 -> 1.0f
            Opcodes.FCONST_2 -> 2.0f
            Opcodes.DCONST_0 -> 0.0
            Opcodes.DCONST_1 -> 1.0
            Opcodes.ACONST_NULL -> NullSentinel
            else -> null
        }
 
        is IntInsnNode -> when (opcode) {
            Opcodes.BIPUSH, Opcodes.SIPUSH -> operand
            else -> null
        }
 
        is LdcInsnNode -> cst
        is JumpInsnNode -> {
            if (expandConditions.none { opcode in it.opcodes }) {
                return null
            }
            var lastInsn = previous
            while (lastInsn != null && (lastInsn is LabelNode || lastInsn is FrameNode)) {
                lastInsn = lastInsn.previous
            }
            if (lastInsn != null) {
                val lastOpcode = lastInsn.opcode
                if (lastOpcode == Opcodes.LCMP ||
                    lastOpcode == Opcodes.FCMPL ||
                    lastOpcode == Opcodes.FCMPG ||
                    lastOpcode == Opcodes.DCMPL ||
                    lastOpcode == Opcodes.DCMPG
                ) {
                    return null
                }
            }
            0
        }
 
        is TypeInsnNode -> {
            if (opcode < Opcodes.CHECKCAST) {
                // Don't treat NEW and ANEWARRAY as constants
                // Matches Mixin's handling
                return null
            }
            Type.getObjectType(desc)
        }
 
        else -> null
    }
}