src/main/kotlin/platform/mixin/handlers/injectionPoint/LoadInjectionPoint.kt diff

← → edited src/main/kotlin/platform/mixin/handlers/injectionPoint/LoadInjectionPoint.kt

Ignore whitespaces Open in IDE

⁠ joe: 2025

Copy Copy

/*
* Minecraft Development for IntelliJ
*
* https://mcdev.io/
*
* Copyright (C) 2024 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.reference.MixinSelector
import com.demonwav.mcdev.platform.mixin.util.AsmDfaUtil
import com.demonwav.mcdev.platform.mixin.util.LocalInfo
import com.demonwav.mcdev.platform.mixin.util.LocalVariables
import com.demonwav.mcdev.platform.mixin.util.MixinConstants.Annotations.MODIFY_VARIABLE
import com.demonwav.mcdev.util.constantValue
import com.demonwav.mcdev.util.findContainingMethod
import com.demonwav.mcdev.util.findModule
import com.intellij.codeInsight.lookup.LookupElementBuilder
import com.intellij.openapi.module.Module
import com.intellij.psi.JavaPsiFacade
import com.intellij.psi.JavaTokenType
import com.intellij.psi.PsiAnnotation
import com.intellij.psi.PsiAssignmentExpression
import com.intellij.psi.PsiClass
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiExpressionStatement
import com.intellij.psi.PsiForeachStatement
import com.intellij.psi.PsiParameter
import com.intellij.psi.PsiPrimitiveType
import com.intellij.psi.PsiReferenceExpression
import com.intellij.psi.PsiThisExpression
import com.intellij.psi.PsiTypes
import com.intellij.psi.PsiUnaryExpression
import com.intellij.psi.PsiVariable
import com.intellij.psi.util.PsiUtil
import com.intellij.psi.util.parentOfType
import org.objectweb.asm.Opcodes
import org.objectweb.asm.tree.ClassNode
import org.objectweb.asm.tree.MethodNode
import org.objectweb.asm.tree.VarInsnNode

abstract class AbstractLoadInjectionPoint(private val store: Boolean) : InjectionPoint<PsiElement>() {
private fun getModifyVariableInfo(at: PsiAnnotation, mode: CollectVisitor.Mode?): LocalInfo? {
val modifyVariable = at.parentOfType<PsiAnnotation>() ?: return null
if (!modifyVariable.hasQualifiedName(MODIFY_VARIABLE)) {
return null
}

val method = modifyVariable.findContainingMethod() ?: return null
val localType = method.parameterList.getParameter(0)?.type
if (localType == null && mode != CollectVisitor.Mode.COMPLETION) {
return null
}
return LocalInfo.fromAnnotation(localType, modifyVariable)
}

override fun isShiftDiscouraged(shift: Int): Boolean {
if (shift == 0) {
return false
}
if (store) {
// allow shift before the STORE
if (shift == -1) {
return false
}
} else {
// allow shift after the LOAD
if (shift == 1) {
return false
}
}
return true
}

override fun createNavigationVisitor(
at: PsiAnnotation,
target: MixinSelector?,
targetClass: PsiClass,
): NavigationVisitor? {
val info = getModifyVariableInfo(at, null) ?: return null
return MyNavigationVisitor(info, store)
}

override fun doCreateCollectVisitor(
at: PsiAnnotation,
target: MixinSelector?,
targetClass: ClassNode,
mode: CollectVisitor.Mode,
): CollectVisitor<PsiElement>? {
val module = at.findModule() ?: return null
val info = getModifyVariableInfo(at, mode) ?: return null
return MyCollectVisitor(module, targetClass, mode, info, store)
}

override fun createLookup(
targetClass: ClassNode,
result: CollectVisitor.Result<PsiElement>,
): LookupElementBuilder? {
return null
}

override fun addOrdinalFilter(at: PsiAnnotation, targetClass: ClassNode, collectVisitor: CollectVisitor<*>) {
val ordinal = at.findDeclaredAttributeValue("ordinal")?.constantValue as? Int ?: return
if (ordinal < 0) return

// Replace the ordinal filter with one that takes into account the type of the local variable being modified.
// Fixes otherwise incorrect results for completion.
val project = at.project
val ordinals = mutableMapOf<String, Int>()
collectVisitor.addResultFilter("ordinal") { result, method ->
// store returns the instruction after the variable
val varInsn = (if (store) result.originalInsn.previous ?: result.originalInsn else result.originalInsn)
as? VarInsnNode ?: throw IllegalStateException("AbstractLoadInjectionPoint returned non-var insn")
val localType = AsmDfaUtil.getLocalVariableType(
project,
targetClass,
method,
result.originalInsn,
varInsn.`var`,
) ?: return@addResultFilter true
val desc = localType.descriptor
val ord = ordinals[desc] ?: 0
ordinals[desc] = ord + 1
ord == ordinal
}
}

private class MyNavigationVisitor(
private val info: LocalInfo,
private val store: Boolean,
) : NavigationVisitor() {
override fun visitThisExpression(expression: PsiThisExpression) {
super.visitThisExpression(expression)
if (!store && expression.qualifier == null) {
addLocalUsage(expression, "this")
}
}

override fun visitVariable(variable: PsiVariable) {
super.visitVariable(variable)
if (store && variable.initializer != null) {
val name = variable.name
if (name != null) {
addLocalUsage(variable, name)
}
}
}

override fun visitReferenceExpression(expression: PsiReferenceExpression) {
super.visitReferenceExpression(expression)
val referenceName = expression.referenceName ?: return
if (expression.qualifierExpression == null) {
val isCorrectAccessType = if (store) {
PsiUtil.isAccessedForWriting(expression)
} else {
PsiUtil.isAccessedForReading(expression)
}
if (!isCorrectAccessType) {
return
}
val resolved = expression.resolve() as? PsiVariable ?: return
val type = resolved.type
if (type is PsiPrimitiveType &&
type != PsiTypes.floatType() &&
type != PsiTypes.doubleType() &&
type != PsiTypes.longType() &&
type != PsiTypes.booleanType()
) {
// ModifyVariable currently cannot handle iinc
val parentExpr = PsiUtil.skipParenthesizedExprUp(expression.parent)
val isIincUnary = parentExpr is PsiUnaryExpression &&
(
parentExpr.operationTokenType == JavaTokenType.PLUSPLUS ||
parentExpr.operationTokenType == JavaTokenType.MINUSMINUS
)
val isIincAssignment = parentExpr is PsiAssignmentExpression &&
(
parentExpr.operationTokenType == JavaTokenType.PLUSEQ ||
parentExpr.operationTokenType == JavaTokenType.MINUSEQ
) &&
PsiUtil.isConstantExpression(parentExpr.rExpression) &&
(parentExpr.rExpression?.constantValue as? Number)?.toInt()
?.let { it >= Short.MIN_VALUE && it <= Short.MAX_VALUE } == true
val isIinc = isIincUnary || isIincAssignment
if (isIinc) {
if (store) {
return
}
val parentParent = PsiUtil.skipParenthesizedExprUp(parentExpr.parent)
if (parentParent is PsiExpressionStatement) {
return
}
}
}
if (!info.argsOnly || resolved is PsiParameter) {
addLocalUsage(expression, referenceName)
}
}
}

override fun visitForeachStatement(statement: PsiForeachStatement) {
checkImplicitLocalsPre(statement)
if (store) {
addLocalUsage(statement.iterationParameter, statement.iterationParameter.name)
}
super.visitForeachStatement(statement)
checkImplicitLocalsPost(statement)
}

private fun checkImplicitLocalsPre(location: PsiElement) {
checkImplicitLocals(location, true)
}

private fun checkImplicitLocalsPost(location: PsiElement) {
checkImplicitLocals(location, false)
}

private fun checkImplicitLocals(location: PsiElement, isPre: Boolean) {
val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, isPre)
val localIndex = LocalVariables.guessLocalVariableIndex(location) ?: return
val localCount = LocalVariables.getLocalVariableSize(location)
for (i in localIndex until (localIndex + localCount)) {
val local = localsHere.firstOrNull { it.index == i } ?: continue
if (store) {
repeat(local.implicitStoreCountAfter) {
addLocalUsage(location, local.name, localsHere)
}
} else {
repeat(local.implicitLoadCountAfter) {
addLocalUsage(location, local.name, localsHere)
}
}
}
}

private fun addLocalUsage(location: PsiElement, name: String) {
val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, !store)
addLocalUsage(location, name, localsHere)
}

private fun addLocalUsage(
location: PsiElement,
name: String,
localsHere: List<LocalVariables.SourceLocalVariable>,
) {
for (local in info.matchSourceLocals(localsHere)) {
if (name == local.name) {
addResult(location)
}
}
}
}

private class MyCollectVisitor(
private val module: Module,
private val targetClass: ClassNode,
mode: Mode,
private val info: LocalInfo,
private val store: Boolean,
) : CollectVisitor<PsiElement>(mode) {
override fun accept(methodNode: MethodNode) {
var opcode = when (info.type) {
null -> null
!is PsiPrimitiveType -> Opcodes.ALOAD
PsiTypes.longType() -> Opcodes.LLOAD
PsiTypes.floatType() -> Opcodes.FLOAD
PsiTypes.doubleType() -> Opcodes.DLOAD
else -> Opcodes.ILOAD
}
if (store && opcode != null) {
opcode += (Opcodes.ISTORE - Opcodes.ILOAD)
}
for (insn in methodNode.instructions) {
if (insn !is VarInsnNode) {
continue
}
if (opcode != null) {
if (opcode != insn.opcode) {
continue
}
} else {
if (store) {
if (insn.opcode < Opcodes.ISTORE || insn.opcode > Opcodes.ASTORE) {
continue
}
} else {
if (insn.opcode < Opcodes.ILOAD || insn.opcode > Opcodes.ALOAD) {
continue
}
}
}

val shiftedInsn = if (store) insn.next ?: insn else insn
val locals = info.getLocals(module, targetClass, methodNode, shiftedInsn) ?: continue

val elementFactory = JavaPsiFacade.getElementFactory(module.project)

for (result in info.matchLocals(locals, mode)) {
addResult(shiftedInsn, elementFactory.createExpressionFromText(result.name, null))
}
}
}
}
}

class LoadInjectionPoint : AbstractLoadInjectionPoint(false)
class StoreInjectionPoint : AbstractLoadInjectionPoint(true)

/*
* Minecraft Development for IntelliJ
*
* https://mcdev.io/
*
* Copyright (C) 2025 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.reference.MixinSelector
import com.demonwav.mcdev.platform.mixin.util.AsmDfaUtil
import com.demonwav.mcdev.platform.mixin.util.LocalInfo
import com.demonwav.mcdev.platform.mixin.util.LocalVariables
import com.demonwav.mcdev.platform.mixin.util.MixinConstants.Annotations.MODIFY_VARIABLE
import com.demonwav.mcdev.util.constantValue
import com.demonwav.mcdev.util.findContainingMethod
import com.demonwav.mcdev.util.findModule
import com.intellij.codeInsight.lookup.LookupElementBuilder
import com.intellij.openapi.module.Module
import com.intellij.psi.JavaPsiFacade
import com.intellij.psi.JavaTokenType
import com.intellij.psi.PsiAnnotation
import com.intellij.psi.PsiAssignmentExpression
import com.intellij.psi.PsiClass
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiExpressionStatement
import com.intellij.psi.PsiForeachStatement
import com.intellij.psi.PsiParameter
import com.intellij.psi.PsiPrimitiveType
import com.intellij.psi.PsiReferenceExpression
import com.intellij.psi.PsiThisExpression
import com.intellij.psi.PsiTypes
import com.intellij.psi.PsiUnaryExpression
import com.intellij.psi.PsiVariable
import com.intellij.psi.util.PsiUtil
import com.intellij.psi.util.parentOfType
import org.objectweb.asm.Opcodes
import org.objectweb.asm.tree.ClassNode
import org.objectweb.asm.tree.MethodNode
import org.objectweb.asm.tree.VarInsnNode

abstract class AbstractLoadInjectionPoint(private val store: Boolean) : InjectionPoint<PsiElement>() {
private fun getModifyVariableInfo(at: PsiAnnotation, mode: CollectVisitor.Mode?): LocalInfo? {
val modifyVariable = at.parentOfType<PsiAnnotation>() ?: return null
if (!modifyVariable.hasQualifiedName(MODIFY_VARIABLE)) {
return null
}

val method = modifyVariable.findContainingMethod() ?: return null
val localType = method.parameterList.getParameter(0)?.type
if (localType == null && mode != CollectVisitor.Mode.COMPLETION) {
return null
}
return LocalInfo.fromAnnotation(localType, modifyVariable)
}

override fun isShiftDiscouraged(shift: Int): Boolean {
if (shift == 0) {
return false
}
if (store) {
// allow shift before the STORE
if (shift == -1) {
return false
}
} else {
// allow shift after the LOAD
if (shift == 1) {
return false
}
}
return true
}

override fun createNavigationVisitor(
at: PsiAnnotation,
target: MixinSelector?,
targetClass: PsiClass,
): NavigationVisitor? {
val info = getModifyVariableInfo(at, null) ?: return null
return MyNavigationVisitor(info, store)
}

override fun doCreateCollectVisitor(
at: PsiAnnotation,
target: MixinSelector?,
targetClass: ClassNode,
mode: CollectVisitor.Mode,
): CollectVisitor<PsiElement>? {
val module = at.findModule() ?: return null
val info = getModifyVariableInfo(at, mode) ?: return null
return MyCollectVisitor(module, targetClass, mode, info, store)
}

override fun createLookup(
targetClass: ClassNode,
result: CollectVisitor.Result<PsiElement>,
): LookupElementBuilder? {
return null
}

override fun addOrdinalFilter(at: PsiAnnotation, targetClass: ClassNode, collectVisitor: CollectVisitor<*>) {
val ordinal = at.findDeclaredAttributeValue("ordinal")?.constantValue as? Int ?: return
if (ordinal < 0) return

// Replace the ordinal filter with one that takes into account the type of the local variable being modified.
// Fixes otherwise incorrect results for completion.
val project = at.project
val ordinals = mutableMapOf<String, Int>()
collectVisitor.addResultFilter("ordinal") { result, method ->
// store returns the instruction after the variable
val varInsn = (if (store) result.originalInsn.previous ?: result.originalInsn else result.originalInsn)
as? VarInsnNode ?: throw IllegalStateException("AbstractLoadInjectionPoint returned non-var insn")
val localType = AsmDfaUtil.getLocalVariableType(
project,
targetClass,
method,
result.originalInsn,
varInsn.`var`,
) ?: return@addResultFilter true
val desc = localType.descriptor
val ord = ordinals[desc] ?: 0
ordinals[desc] = ord + 1
ord == ordinal
}
}

private class MyNavigationVisitor(
private val info: LocalInfo,
private val store: Boolean,
) : NavigationVisitor() {
override fun visitThisExpression(expression: PsiThisExpression) {
super.visitThisExpression(expression)
if (!store && expression.qualifier == null) {
addLocalUsage(expression, "this")
}
}

override fun visitVariable(variable: PsiVariable) {
super.visitVariable(variable)
if (store && variable.initializer != null) {
val name = variable.name
if (name != null) {
addLocalUsage(variable, name)
}
}
}

override fun visitReferenceExpression(expression: PsiReferenceExpression) {
super.visitReferenceExpression(expression)
val referenceName = expression.referenceName ?: return
if (expression.qualifierExpression == null) {
val isCorrectAccessType = if (store) {
PsiUtil.isAccessedForWriting(expression)
} else {
PsiUtil.isAccessedForReading(expression)
}
if (!isCorrectAccessType) {
return
}
val resolved = expression.resolve() as? PsiVariable ?: return
val type = resolved.type
if (type is PsiPrimitiveType &&
type != PsiTypes.floatType() &&
type != PsiTypes.doubleType() &&
type != PsiTypes.longType() &&
type != PsiTypes.booleanType()
) {
// ModifyVariable currently cannot handle iinc
val parentExpr = PsiUtil.skipParenthesizedExprUp(expression.parent)
val isIincUnary = parentExpr is PsiUnaryExpression &&
(
parentExpr.operationTokenType == JavaTokenType.PLUSPLUS ||
parentExpr.operationTokenType == JavaTokenType.MINUSMINUS
)
val isIincAssignment = parentExpr is PsiAssignmentExpression &&
(
parentExpr.operationTokenType == JavaTokenType.PLUSEQ ||
parentExpr.operationTokenType == JavaTokenType.MINUSEQ
) &&
PsiUtil.isConstantExpression(parentExpr.rExpression) &&
(parentExpr.rExpression?.constantValue as? Number)?.toInt()
?.let { it >= Short.MIN_VALUE && it <= Short.MAX_VALUE } == true
val isIinc = isIincUnary || isIincAssignment
if (isIinc) {
if (store) {
return
}
val parentParent = PsiUtil.skipParenthesizedExprUp(parentExpr.parent)
if (parentParent is PsiExpressionStatement) {
return
}
}
}
if (!info.argsOnly || resolved is PsiParameter) {
addLocalUsage(expression, referenceName)
}
}
}

override fun visitForeachStatement(statement: PsiForeachStatement) {
checkImplicitLocalsPre(statement)
if (store) {
addLocalUsage(statement.iterationParameter, statement.iterationParameter.name)
}
super.visitForeachStatement(statement)
checkImplicitLocalsPost(statement)
}

private fun checkImplicitLocalsPre(location: PsiElement) {
checkImplicitLocals(location, true)
}

private fun checkImplicitLocalsPost(location: PsiElement) {
checkImplicitLocals(location, false)
}

private fun checkImplicitLocals(location: PsiElement, isPre: Boolean) {
val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, isPre)
val localIndex = LocalVariables.guessLocalVariableIndex(location) ?: return
val localCount = LocalVariables.getLocalVariableSize(location)
for (i in localIndex until (localIndex + localCount)) {
val local = localsHere.firstOrNull { it.index == i } ?: continue
if (store) {
repeat(local.implicitStoreCountAfter) {
addLocalUsage(location, local.name, localsHere)
}
} else {
repeat(local.implicitLoadCountAfter) {
addLocalUsage(location, local.name, localsHere)
}
}
}
}

private fun addLocalUsage(location: PsiElement, name: String) {
val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, !store)
addLocalUsage(location, name, localsHere)
}

private fun addLocalUsage(
location: PsiElement,
name: String,
localsHere: List<LocalVariables.SourceLocalVariable>,
) {
for (local in info.matchSourceLocals(localsHere)) {
if (name == local.name) {
addResult(location)
}
}
}
}

private class MyCollectVisitor(
private val module: Module,
private val targetClass: ClassNode,
mode: Mode,
private val info: LocalInfo,
private val store: Boolean,
) : CollectVisitor<PsiElement>(mode) {
override fun accept(methodNode: MethodNode) {
var opcode = when (info.type) {
null -> null
!is PsiPrimitiveType -> Opcodes.ALOAD
PsiTypes.longType() -> Opcodes.LLOAD
PsiTypes.floatType() -> Opcodes.FLOAD
PsiTypes.doubleType() -> Opcodes.DLOAD
else -> Opcodes.ILOAD
}
if (store && opcode != null) {
opcode += (Opcodes.ISTORE - Opcodes.ILOAD)
}
for (insn in methodNode.instructions) {
if (insn !is VarInsnNode) {
continue
}
if (opcode != null) {
if (opcode != insn.opcode) {
continue
}
} else {
if (store) {
if (insn.opcode < Opcodes.ISTORE || insn.opcode > Opcodes.ASTORE) {
continue
}
} else {
if (insn.opcode < Opcodes.ILOAD || insn.opcode > Opcodes.ALOAD) {
continue
}
}
}

val shiftedInsn = if (store) insn.next ?: insn else insn
val locals = info.getLocals(module, targetClass, methodNode, shiftedInsn) ?: continue

val elementFactory = JavaPsiFacade.getElementFactory(module.project)

for (result in info.matchLocals(locals, mode)) {
addResult(shiftedInsn, elementFactory.createExpressionFromText(result.name, null))
}
}
}
}
}

class LoadInjectionPoint : AbstractLoadInjectionPoint(false)
class StoreInjectionPoint : AbstractLoadInjectionPoint(true)

/* * Minecraft Development for IntelliJ * * https://mcdev.io/ * * Copyright (C) 2024 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.reference.MixinSelector import com.demonwav.mcdev.platform.mixin.util.AsmDfaUtil import com.demonwav.mcdev.platform.mixin.util.LocalInfo import com.demonwav.mcdev.platform.mixin.util.LocalVariables import com.demonwav.mcdev.platform.mixin.util.MixinConstants.Annotations.MODIFY_VARIABLE import com.demonwav.mcdev.util.constantValue import com.demonwav.mcdev.util.findContainingMethod import com.demonwav.mcdev.util.findModule import com.intellij.codeInsight.lookup.LookupElementBuilder import com.intellij.openapi.module.Module import com.intellij.psi.JavaPsiFacade import com.intellij.psi.JavaTokenType import com.intellij.psi.PsiAnnotation import com.intellij.psi.PsiAssignmentExpression import com.intellij.psi.PsiClass import com.intellij.psi.PsiElement import com.intellij.psi.PsiExpressionStatement import com.intellij.psi.PsiForeachStatement import com.intellij.psi.PsiParameter import com.intellij.psi.PsiPrimitiveType import com.intellij.psi.PsiReferenceExpression import com.intellij.psi.PsiThisExpression import com.intellij.psi.PsiTypes import com.intellij.psi.PsiUnaryExpression import com.intellij.psi.PsiVariable import com.intellij.psi.util.PsiUtil import com.intellij.psi.util.parentOfType import org.objectweb.asm.Opcodes import org.objectweb.asm.tree.ClassNode import org.objectweb.asm.tree.MethodNode import org.objectweb.asm.tree.VarInsnNode abstract class AbstractLoadInjectionPoint(private val store: Boolean) : InjectionPoint<PsiElement>() { private fun getModifyVariableInfo(at: PsiAnnotation, mode: CollectVisitor.Mode?): LocalInfo? { val modifyVariable = at.parentOfType<PsiAnnotation>() ?: return null if (!modifyVariable.hasQualifiedName(MODIFY_VARIABLE)) { return null } val method = modifyVariable.findContainingMethod() ?: return null val localType = method.parameterList.getParameter(0)?.type if (localType == null && mode != CollectVisitor.Mode.COMPLETION) { return null } return LocalInfo.fromAnnotation(localType, modifyVariable) } override fun isShiftDiscouraged(shift: Int): Boolean { if (shift == 0) { return false } if (store) { // allow shift before the STORE if (shift == -1) { return false } } else { // allow shift after the LOAD if (shift == 1) { return false } } return true } override fun createNavigationVisitor( at: PsiAnnotation, target: MixinSelector?, targetClass: PsiClass, ): NavigationVisitor? { val info = getModifyVariableInfo(at, null) ?: return null return MyNavigationVisitor(info, store) } override fun doCreateCollectVisitor( at: PsiAnnotation, target: MixinSelector?, targetClass: ClassNode, mode: CollectVisitor.Mode, ): CollectVisitor<PsiElement>? { val module = at.findModule() ?: return null val info = getModifyVariableInfo(at, mode) ?: return null return MyCollectVisitor(module, targetClass, mode, info, store) } override fun createLookup( targetClass: ClassNode, result: CollectVisitor.Result<PsiElement>, ): LookupElementBuilder? { return null } override fun addOrdinalFilter(at: PsiAnnotation, targetClass: ClassNode, collectVisitor: CollectVisitor<*>) { val ordinal = at.findDeclaredAttributeValue("ordinal")?.constantValue as? Int ?: return if (ordinal < 0) return // Replace the ordinal filter with one that takes into account the type of the local variable being modified. // Fixes otherwise incorrect results for completion. val project = at.project val ordinals = mutableMapOf<String, Int>() collectVisitor.addResultFilter("ordinal") { result, method -> // store returns the instruction after the variable val varInsn = (if (store) result.originalInsn.previous ?: result.originalInsn else result.originalInsn) as? VarInsnNode ?: throw IllegalStateException("AbstractLoadInjectionPoint returned non-var insn") val localType = AsmDfaUtil.getLocalVariableType( project, targetClass, method, result.originalInsn, varInsn.`var`, ) ?: return@addResultFilter true val desc = localType.descriptor val ord = ordinals[desc] ?: 0 ordinals[desc] = ord + 1 ord == ordinal } } private class MyNavigationVisitor( private val info: LocalInfo, private val store: Boolean, ) : NavigationVisitor() { override fun visitThisExpression(expression: PsiThisExpression) { super.visitThisExpression(expression) if (!store && expression.qualifier == null) { addLocalUsage(expression, "this") } } override fun visitVariable(variable: PsiVariable) { super.visitVariable(variable) if (store && variable.initializer != null) { val name = variable.name if (name != null) { addLocalUsage(variable, name) } } } override fun visitReferenceExpression(expression: PsiReferenceExpression) { super.visitReferenceExpression(expression) val referenceName = expression.referenceName ?: return if (expression.qualifierExpression == null) { val isCorrectAccessType = if (store) { PsiUtil.isAccessedForWriting(expression) } else { PsiUtil.isAccessedForReading(expression) } if (!isCorrectAccessType) { return } val resolved = expression.resolve() as? PsiVariable ?: return val type = resolved.type if (type is PsiPrimitiveType && type != PsiTypes.floatType() && type != PsiTypes.doubleType() && type != PsiTypes.longType() && type != PsiTypes.booleanType() ) { // ModifyVariable currently cannot handle iinc val parentExpr = PsiUtil.skipParenthesizedExprUp(expression.parent) val isIincUnary = parentExpr is PsiUnaryExpression && ( parentExpr.operationTokenType == JavaTokenType.PLUSPLUS || parentExpr.operationTokenType == JavaTokenType.MINUSMINUS ) val isIincAssignment = parentExpr is PsiAssignmentExpression && ( parentExpr.operationTokenType == JavaTokenType.PLUSEQ || parentExpr.operationTokenType == JavaTokenType.MINUSEQ ) && PsiUtil.isConstantExpression(parentExpr.rExpression) && (parentExpr.rExpression?.constantValue as? Number)?.toInt() ?.let { it >= Short.MIN_VALUE && it <= Short.MAX_VALUE } == true val isIinc = isIincUnary || isIincAssignment if (isIinc) { if (store) { return } val parentParent = PsiUtil.skipParenthesizedExprUp(parentExpr.parent) if (parentParent is PsiExpressionStatement) { return } } } if (!info.argsOnly || resolved is PsiParameter) { addLocalUsage(expression, referenceName) } } } override fun visitForeachStatement(statement: PsiForeachStatement) { checkImplicitLocalsPre(statement) if (store) { addLocalUsage(statement.iterationParameter, statement.iterationParameter.name) } super.visitForeachStatement(statement) checkImplicitLocalsPost(statement) } private fun checkImplicitLocalsPre(location: PsiElement) { checkImplicitLocals(location, true) } private fun checkImplicitLocalsPost(location: PsiElement) { checkImplicitLocals(location, false) } private fun checkImplicitLocals(location: PsiElement, isPre: Boolean) { val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, isPre) val localIndex = LocalVariables.guessLocalVariableIndex(location) ?: return val localCount = LocalVariables.getLocalVariableSize(location) for (i in localIndex until (localIndex + localCount)) { val local = localsHere.firstOrNull { it.index == i } ?: continue if (store) { repeat(local.implicitStoreCountAfter) { addLocalUsage(location, local.name, localsHere) } } else { repeat(local.implicitLoadCountAfter) { addLocalUsage(location, local.name, localsHere) } } } } private fun addLocalUsage(location: PsiElement, name: String) { val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, !store) addLocalUsage(location, name, localsHere) } private fun addLocalUsage( location: PsiElement, name: String, localsHere: List<LocalVariables.SourceLocalVariable>, ) { for (local in info.matchSourceLocals(localsHere)) { if (name == local.name) { addResult(location) } } } } private class MyCollectVisitor( private val module: Module, private val targetClass: ClassNode, mode: Mode, private val info: LocalInfo, private val store: Boolean, ) : CollectVisitor<PsiElement>(mode) { override fun accept(methodNode: MethodNode) { var opcode = when (info.type) { null -> null !is PsiPrimitiveType -> Opcodes.ALOAD PsiTypes.longType() -> Opcodes.LLOAD PsiTypes.floatType() -> Opcodes.FLOAD PsiTypes.doubleType() -> Opcodes.DLOAD else -> Opcodes.ILOAD } if (store && opcode != null) { opcode += (Opcodes.ISTORE - Opcodes.ILOAD) } for (insn in methodNode.instructions) { if (insn !is VarInsnNode) { continue } if (opcode != null) { if (opcode != insn.opcode) { continue } } else { if (store) { if (insn.opcode < Opcodes.ISTORE || insn.opcode > Opcodes.ASTORE) { continue } } else { if (insn.opcode < Opcodes.ILOAD || insn.opcode > Opcodes.ALOAD) { continue } } } val shiftedInsn = if (store) insn.next ?: insn else insn val locals = info.getLocals(module, targetClass, methodNode, shiftedInsn) ?: continue val elementFactory = JavaPsiFacade.getElementFactory(module.project) for (result in info.matchLocals(locals, mode)) { addResult(shiftedInsn, elementFactory.createExpressionFromText(result.name, null)) } } } } } class LoadInjectionPoint : AbstractLoadInjectionPoint(false) class StoreInjectionPoint : AbstractLoadInjectionPoint(true) /* * Minecraft Development for IntelliJ * * https://mcdev.io/ * * Copyright (C) 2025 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.reference.MixinSelector import com.demonwav.mcdev.platform.mixin.util.AsmDfaUtil import com.demonwav.mcdev.platform.mixin.util.LocalInfo import com.demonwav.mcdev.platform.mixin.util.LocalVariables import com.demonwav.mcdev.platform.mixin.util.MixinConstants.Annotations.MODIFY_VARIABLE import com.demonwav.mcdev.util.constantValue import com.demonwav.mcdev.util.findContainingMethod import com.demonwav.mcdev.util.findModule import com.intellij.codeInsight.lookup.LookupElementBuilder import com.intellij.openapi.module.Module import com.intellij.psi.JavaPsiFacade import com.intellij.psi.JavaTokenType import com.intellij.psi.PsiAnnotation import com.intellij.psi.PsiAssignmentExpression import com.intellij.psi.PsiClass import com.intellij.psi.PsiElement import com.intellij.psi.PsiExpressionStatement import com.intellij.psi.PsiForeachStatement import com.intellij.psi.PsiParameter import com.intellij.psi.PsiPrimitiveType import com.intellij.psi.PsiReferenceExpression import com.intellij.psi.PsiThisExpression import com.intellij.psi.PsiTypes import com.intellij.psi.PsiUnaryExpression import com.intellij.psi.PsiVariable import com.intellij.psi.util.PsiUtil import com.intellij.psi.util.parentOfType import org.objectweb.asm.Opcodes import org.objectweb.asm.tree.ClassNode import org.objectweb.asm.tree.MethodNode import org.objectweb.asm.tree.VarInsnNode abstract class AbstractLoadInjectionPoint(private val store: Boolean) : InjectionPoint<PsiElement>() { private fun getModifyVariableInfo(at: PsiAnnotation, mode: CollectVisitor.Mode?): LocalInfo? { val modifyVariable = at.parentOfType<PsiAnnotation>() ?: return null if (!modifyVariable.hasQualifiedName(MODIFY_VARIABLE)) { return null } val method = modifyVariable.findContainingMethod() ?: return null val localType = method.parameterList.getParameter(0)?.type if (localType == null && mode != CollectVisitor.Mode.COMPLETION) { return null } return LocalInfo.fromAnnotation(localType, modifyVariable) } override fun isShiftDiscouraged(shift: Int): Boolean { if (shift == 0) { return false } if (store) { // allow shift before the STORE if (shift == -1) { return false } } else { // allow shift after the LOAD if (shift == 1) { return false } } return true } override fun createNavigationVisitor( at: PsiAnnotation, target: MixinSelector?, targetClass: PsiClass, ): NavigationVisitor? { val info = getModifyVariableInfo(at, null) ?: return null return MyNavigationVisitor(info, store) } override fun doCreateCollectVisitor( at: PsiAnnotation, target: MixinSelector?, targetClass: ClassNode, mode: CollectVisitor.Mode, ): CollectVisitor<PsiElement>? { val module = at.findModule() ?: return null val info = getModifyVariableInfo(at, mode) ?: return null return MyCollectVisitor(module, targetClass, mode, info, store) } override fun createLookup( targetClass: ClassNode, result: CollectVisitor.Result<PsiElement>, ): LookupElementBuilder? { return null } override fun addOrdinalFilter(at: PsiAnnotation, targetClass: ClassNode, collectVisitor: CollectVisitor<*>) { val ordinal = at.findDeclaredAttributeValue("ordinal")?.constantValue as? Int ?: return if (ordinal < 0) return // Replace the ordinal filter with one that takes into account the type of the local variable being modified. // Fixes otherwise incorrect results for completion. val project = at.project val ordinals = mutableMapOf<String, Int>() collectVisitor.addResultFilter("ordinal") { result, method -> // store returns the instruction after the variable val varInsn = (if (store) result.originalInsn.previous ?: result.originalInsn else result.originalInsn) as? VarInsnNode ?: throw IllegalStateException("AbstractLoadInjectionPoint returned non-var insn") val localType = AsmDfaUtil.getLocalVariableType( project, targetClass, method, result.originalInsn, varInsn.`var`, ) ?: return@addResultFilter true val desc = localType.descriptor val ord = ordinals[desc] ?: 0 ordinals[desc] = ord + 1 ord == ordinal } } private class MyNavigationVisitor( private val info: LocalInfo, private val store: Boolean, ) : NavigationVisitor() { override fun visitThisExpression(expression: PsiThisExpression) { super.visitThisExpression(expression) if (!store && expression.qualifier == null) { addLocalUsage(expression, "this") } } override fun visitVariable(variable: PsiVariable) { super.visitVariable(variable) if (store && variable.initializer != null) { val name = variable.name if (name != null) { addLocalUsage(variable, name) } } } override fun visitReferenceExpression(expression: PsiReferenceExpression) { super.visitReferenceExpression(expression) val referenceName = expression.referenceName ?: return if (expression.qualifierExpression == null) { val isCorrectAccessType = if (store) { PsiUtil.isAccessedForWriting(expression) } else { PsiUtil.isAccessedForReading(expression) } if (!isCorrectAccessType) { return } val resolved = expression.resolve() as? PsiVariable ?: return val type = resolved.type if (type is PsiPrimitiveType && type != PsiTypes.floatType() && type != PsiTypes.doubleType() && type != PsiTypes.longType() && type != PsiTypes.booleanType() ) { // ModifyVariable currently cannot handle iinc val parentExpr = PsiUtil.skipParenthesizedExprUp(expression.parent) val isIincUnary = parentExpr is PsiUnaryExpression && ( parentExpr.operationTokenType == JavaTokenType.PLUSPLUS || parentExpr.operationTokenType == JavaTokenType.MINUSMINUS ) val isIincAssignment = parentExpr is PsiAssignmentExpression && ( parentExpr.operationTokenType == JavaTokenType.PLUSEQ || parentExpr.operationTokenType == JavaTokenType.MINUSEQ ) && PsiUtil.isConstantExpression(parentExpr.rExpression) && (parentExpr.rExpression?.constantValue as? Number)?.toInt() ?.let { it >= Short.MIN_VALUE && it <= Short.MAX_VALUE } == true val isIinc = isIincUnary || isIincAssignment if (isIinc) { if (store) { return } val parentParent = PsiUtil.skipParenthesizedExprUp(parentExpr.parent) if (parentParent is PsiExpressionStatement) { return } } } if (!info.argsOnly || resolved is PsiParameter) { addLocalUsage(expression, referenceName) } } } override fun visitForeachStatement(statement: PsiForeachStatement) { checkImplicitLocalsPre(statement) if (store) { addLocalUsage(statement.iterationParameter, statement.iterationParameter.name) } super.visitForeachStatement(statement) checkImplicitLocalsPost(statement) } private fun checkImplicitLocalsPre(location: PsiElement) { checkImplicitLocals(location, true) } private fun checkImplicitLocalsPost(location: PsiElement) { checkImplicitLocals(location, false) } private fun checkImplicitLocals(location: PsiElement, isPre: Boolean) { val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, isPre) val localIndex = LocalVariables.guessLocalVariableIndex(location) ?: return val localCount = LocalVariables.getLocalVariableSize(location) for (i in localIndex until (localIndex + localCount)) { val local = localsHere.firstOrNull { it.index == i } ?: continue if (store) { repeat(local.implicitStoreCountAfter) { addLocalUsage(location, local.name, localsHere) } } else { repeat(local.implicitLoadCountAfter) { addLocalUsage(location, local.name, localsHere) } } } } private fun addLocalUsage(location: PsiElement, name: String) { val localsHere = LocalVariables.guessLocalsAt(location, info.argsOnly, !store) addLocalUsage(location, name, localsHere) } private fun addLocalUsage( location: PsiElement, name: String, localsHere: List<LocalVariables.SourceLocalVariable>, ) { for (local in info.matchSourceLocals(localsHere)) { if (name == local.name) { addResult(location) } } } } private class MyCollectVisitor( private val module: Module, private val targetClass: ClassNode, mode: Mode, private val info: LocalInfo, private val store: Boolean, ) : CollectVisitor<PsiElement>(mode) { override fun accept(methodNode: MethodNode) { var opcode = when (info.type) { null -> null !is PsiPrimitiveType -> Opcodes.ALOAD PsiTypes.longType() -> Opcodes.LLOAD PsiTypes.floatType() -> Opcodes.FLOAD PsiTypes.doubleType() -> Opcodes.DLOAD else -> Opcodes.ILOAD } if (store && opcode != null) { opcode += (Opcodes.ISTORE - Opcodes.ILOAD) } for (insn in methodNode.instructions) { if (insn !is VarInsnNode) { continue } if (opcode != null) { if (opcode != insn.opcode) { continue } } else { if (store) { if (insn.opcode < Opcodes.ISTORE || insn.opcode > Opcodes.ASTORE) { continue } } else { if (insn.opcode < Opcodes.ILOAD || insn.opcode > Opcodes.ALOAD) { continue } } } val shiftedInsn = if (store) insn.next ?: insn else insn val locals = info.getLocals(module, targetClass, methodNode, shiftedInsn) ?: continue val elementFactory = JavaPsiFacade.getElementFactory(module.project) for (result in info.matchLocals(locals, mode)) { addResult(shiftedInsn, elementFactory.createExpressionFromText(result.name, null)) } } } } } class LoadInjectionPoint : AbstractLoadInjectionPoint(false) class StoreInjectionPoint : AbstractLoadInjectionPoint(true)

Add build comment

Mute test

Server communication failure

Run Custom Build

TODO

Responsibility

Edit tags

Loading...

Are you sure?