src/main/kotlin/platform/mixin/expression/MEExpressionInjector.kt diff

← → edited src/main/kotlin/platform/mixin/expression/MEExpressionInjector.kt

Ignore whitespaces Open in IDE

⁠ joe: Revert "More debug printing for MEExpressionInjector"

This reverts commit 0bd63b3e7c473ebd9571f3f7c637cbbba6ede118.

Copy Copy

/*
* 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.expression

import com.demonwav.mcdev.platform.mixin.util.MixinConstants
import com.demonwav.mcdev.util.findContainingModifierList
import com.demonwav.mcdev.util.findContainingNameValuePair
import com.demonwav.mcdev.util.parseArray
import com.intellij.lang.injection.InjectedLanguageManager
import com.intellij.lang.injection.MultiHostInjector
import com.intellij.lang.injection.MultiHostRegistrar
import com.intellij.openapi.util.Key
import com.intellij.openapi.util.TextRange
import com.intellij.openapi.util.component1
import com.intellij.openapi.util.component2
import com.intellij.psi.ElementManipulators
import com.intellij.psi.JavaTokenType
import com.intellij.psi.PsiAnnotation
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiLanguageInjectionHost
import com.intellij.psi.PsiLiteralExpression
import com.intellij.psi.PsiParenthesizedExpression
import com.intellij.psi.PsiPolyadicExpression
import com.intellij.psi.impl.source.tree.injected.JavaConcatenationToInjectorAdapter
import com.intellij.psi.util.PsiLiteralUtil
import com.intellij.psi.util.PsiModificationTracker
import com.intellij.psi.util.PsiUtil
import com.intellij.psi.util.parentOfType
import com.intellij.util.SmartList
import org.intellij.plugins.intelliLang.inject.InjectorUtils

class MEExpressionInjector : MultiHostInjector {
companion object {
private val ELEMENTS = listOf(PsiLiteralExpression::class.java)
private val ME_EXPRESSION_INJECTION = Key.create<MEExpressionInjection>("mcdev.meExpressionInjection")

private val CLASS_INJECTION_RESULT =
Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionResult")
private val CLASS_INJECTION_REGISTRAR_IMPL =
Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionRegistrarImpl")
@JvmStatic
private val METHOD_ADD_TO_RESULTS =
CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("addToResults", CLASS_INJECTION_RESULT)
.also { it.isAccessible = true }
@JvmStatic
private val METHOD_GET_INJECTED_RESULT =
CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("getInjectedResult")
.also { it.isAccessible = true }
}

private data class MEExpressionInjection(val modCount: Long, val injectionResult: Any)

private fun shouldInjectIn(anchor: PsiElement): Boolean {
val nameValuePair = anchor.findContainingNameValuePair() ?: return false
return when (nameValuePair.name) {
"value", null -> nameValuePair.parentOfType<PsiAnnotation>()
?.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION) == true
"id" -> nameValuePair.parentOfType<PsiAnnotation>()
?.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION) == true
else -> false
}
}

override fun getLanguagesToInject(registrar: MultiHostRegistrar, context: PsiElement) {
if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 1")
val project = context.project
val (anchor, _) = JavaConcatenationToInjectorAdapter(project).computeAnchorAndOperands(context)

if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 2 ${anchor.text}")
if (!shouldInjectIn(anchor)) {
return
}
if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 3")
if (MEExpressionCompletionUtil.debugCompletionUnitTest) Thread.dumpStack()

val modifierList = anchor.findContainingModifierList() ?: return
if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 4")

val modCount = PsiModificationTracker.getInstance(project).modificationCount
val primaryElement = modifierList.getUserData(ME_EXPRESSION_INJECTION)
if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 5, $modCount")
if (primaryElement != null && primaryElement.modCount == modCount) {
METHOD_ADD_TO_RESULTS.invoke(registrar, primaryElement.injectionResult)
return
}
if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 6")

// A Frankenstein injection is an injection where we don't know the entire contents, and therefore errors should
// not be reported.
var isFrankenstein = false
registrar.startInjecting(MEExpressionLanguage)

for (annotation in modifierList.annotations) {
if (annotation.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION)) {
val idExpr = annotation.findDeclaredAttributeValue("id") ?: continue
val isType = annotation.findDeclaredAttributeValue("type") != null
var needsPrefix = true
iterateConcatenation(idExpr) { op ->
if (op is PsiLanguageInjectionHost) {
for (textRange in getTextRanges(op)) {
val prefix = "\nclass $isType ".takeIf { needsPrefix }
needsPrefix = false
registrar.addPlace(prefix, null, op, textRange)
}
} else {
isFrankenstein = true
}
}
} else if (annotation.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION)) {
for (valueExpr in annotation.findDeclaredAttributeValue("value")?.parseArray { it }.orEmpty()) {
val places = mutableListOf<Pair<PsiLanguageInjectionHost, TextRange>>()
iterateConcatenation(valueExpr) { op ->
if (op is PsiLanguageInjectionHost) {
for (textRange in getTextRanges(op)) {
places += op to textRange
}
} else {
isFrankenstein = true
}
}
if (places.isNotEmpty()) {
for ((i, place) in places.withIndex()) {
val (host, range) = place
val prefix = "\ndo { ".takeIf { i == 0 }
val suffix = " }".takeIf { i == places.size - 1 }
registrar.addPlace(prefix, suffix, host, range)
}
}
}
}
}

registrar.doneInjecting()

modifierList.putUserData(
ME_EXPRESSION_INJECTION,
MEExpressionInjection(modCount, METHOD_GET_INJECTED_RESULT.invoke(registrar))
)

if (isFrankenstein) {
InjectorUtils.putInjectedFileUserData(
context,
MEExpressionLanguage,
InjectedLanguageManager.FRANKENSTEIN_INJECTION,
true
)
}
}

private fun iterateConcatenation(element: PsiElement, consumer: (PsiElement) -> Unit) {
when (element) {
is PsiParenthesizedExpression -> {
val inner = PsiUtil.skipParenthesizedExprDown(element) ?: return
iterateConcatenation(inner, consumer)
}
is PsiPolyadicExpression -> {
if (element.operationTokenType == JavaTokenType.PLUS) {
for (operand in element.operands) {
iterateConcatenation(operand, consumer)
}
} else {
consumer(element)
}
}
else -> consumer(element)
}
}

private fun getTextRanges(host: PsiLanguageInjectionHost): List<TextRange> {
if (host is PsiLiteralExpression && host.isTextBlock) {
val textRange = ElementManipulators.getValueTextRange(host)
val indent = PsiLiteralUtil.getTextBlockIndent(host)
if (indent <= 0) {
return listOf(textRange)
}

val text = (host as PsiElement).text
var startOffset = textRange.startOffset + indent
var endOffset = text.indexOf('\n', startOffset)
val result = SmartList<TextRange>()
while (endOffset > 0) {
endOffset++
result.add(TextRange(startOffset, endOffset))
startOffset = endOffset + indent
endOffset = text.indexOf('\n', startOffset)
}
endOffset = textRange.endOffset
if (startOffset < endOffset) {
result.add(TextRange(startOffset, endOffset))
}
return result
} else {
return listOf(ElementManipulators.getValueTextRange(host))
}
}

override fun elementsToInjectIn() = ELEMENTS
}

/*
* 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.expression

import com.demonwav.mcdev.platform.mixin.util.MixinConstants
import com.demonwav.mcdev.util.findContainingModifierList
import com.demonwav.mcdev.util.findContainingNameValuePair
import com.demonwav.mcdev.util.parseArray
import com.intellij.lang.injection.InjectedLanguageManager
import com.intellij.lang.injection.MultiHostInjector
import com.intellij.lang.injection.MultiHostRegistrar
import com.intellij.openapi.util.Key
import com.intellij.openapi.util.TextRange
import com.intellij.openapi.util.component1
import com.intellij.openapi.util.component2
import com.intellij.psi.ElementManipulators
import com.intellij.psi.JavaTokenType
import com.intellij.psi.PsiAnnotation
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiLanguageInjectionHost
import com.intellij.psi.PsiLiteralExpression
import com.intellij.psi.PsiParenthesizedExpression
import com.intellij.psi.PsiPolyadicExpression
import com.intellij.psi.impl.source.tree.injected.JavaConcatenationToInjectorAdapter
import com.intellij.psi.util.PsiLiteralUtil
import com.intellij.psi.util.PsiModificationTracker
import com.intellij.psi.util.PsiUtil
import com.intellij.psi.util.parentOfType
import com.intellij.util.SmartList
import org.intellij.plugins.intelliLang.inject.InjectorUtils

class MEExpressionInjector : MultiHostInjector {
companion object {
private val ELEMENTS = listOf(PsiLiteralExpression::class.java)
private val ME_EXPRESSION_INJECTION = Key.create<MEExpressionInjection>("mcdev.meExpressionInjection")

private val CLASS_INJECTION_RESULT =
Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionResult")
private val CLASS_INJECTION_REGISTRAR_IMPL =
Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionRegistrarImpl")
@JvmStatic
private val METHOD_ADD_TO_RESULTS =
CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("addToResults", CLASS_INJECTION_RESULT)
.also { it.isAccessible = true }
@JvmStatic
private val METHOD_GET_INJECTED_RESULT =
CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("getInjectedResult")
.also { it.isAccessible = true }
}

private data class MEExpressionInjection(val modCount: Long, val injectionResult: Any)

private fun shouldInjectIn(anchor: PsiElement): Boolean {
val nameValuePair = anchor.findContainingNameValuePair() ?: return false
return when (nameValuePair.name) {
"value", null -> nameValuePair.parentOfType<PsiAnnotation>()
?.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION) == true
"id" -> nameValuePair.parentOfType<PsiAnnotation>()
?.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION) == true
else -> false
}
}

override fun getLanguagesToInject(registrar: MultiHostRegistrar, context: PsiElement) {
val project = context.project
val (anchor, _) = JavaConcatenationToInjectorAdapter(project).computeAnchorAndOperands(context)

if (!shouldInjectIn(anchor)) {
return
}

val modifierList = anchor.findContainingModifierList() ?: return

val modCount = PsiModificationTracker.getInstance(project).modificationCount
val primaryElement = modifierList.getUserData(ME_EXPRESSION_INJECTION)
if (primaryElement != null && primaryElement.modCount == modCount) {
METHOD_ADD_TO_RESULTS.invoke(registrar, primaryElement.injectionResult)
return
}

// A Frankenstein injection is an injection where we don't know the entire contents, and therefore errors should
// not be reported.
var isFrankenstein = false
registrar.startInjecting(MEExpressionLanguage)

for (annotation in modifierList.annotations) {
if (annotation.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION)) {
val idExpr = annotation.findDeclaredAttributeValue("id") ?: continue
val isType = annotation.findDeclaredAttributeValue("type") != null
var needsPrefix = true
iterateConcatenation(idExpr) { op ->
if (op is PsiLanguageInjectionHost) {
for (textRange in getTextRanges(op)) {
val prefix = "\nclass $isType ".takeIf { needsPrefix }
needsPrefix = false
registrar.addPlace(prefix, null, op, textRange)
}
} else {
isFrankenstein = true
}
}
} else if (annotation.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION)) {
for (valueExpr in annotation.findDeclaredAttributeValue("value")?.parseArray { it }.orEmpty()) {
val places = mutableListOf<Pair<PsiLanguageInjectionHost, TextRange>>()
iterateConcatenation(valueExpr) { op ->
if (op is PsiLanguageInjectionHost) {
for (textRange in getTextRanges(op)) {
places += op to textRange
}
} else {
isFrankenstein = true
}
}
if (places.isNotEmpty()) {
for ((i, place) in places.withIndex()) {
val (host, range) = place
val prefix = "\ndo { ".takeIf { i == 0 }
val suffix = " }".takeIf { i == places.size - 1 }
registrar.addPlace(prefix, suffix, host, range)
}
}
}
}
}

registrar.doneInjecting()

modifierList.putUserData(
ME_EXPRESSION_INJECTION,
MEExpressionInjection(modCount, METHOD_GET_INJECTED_RESULT.invoke(registrar))
)

if (isFrankenstein) {
InjectorUtils.putInjectedFileUserData(
context,
MEExpressionLanguage,
InjectedLanguageManager.FRANKENSTEIN_INJECTION,
true
)
}
}

private fun iterateConcatenation(element: PsiElement, consumer: (PsiElement) -> Unit) {
when (element) {
is PsiParenthesizedExpression -> {
val inner = PsiUtil.skipParenthesizedExprDown(element) ?: return
iterateConcatenation(inner, consumer)
}
is PsiPolyadicExpression -> {
if (element.operationTokenType == JavaTokenType.PLUS) {
for (operand in element.operands) {
iterateConcatenation(operand, consumer)
}
} else {
consumer(element)
}
}
else -> consumer(element)
}
}

private fun getTextRanges(host: PsiLanguageInjectionHost): List<TextRange> {
if (host is PsiLiteralExpression && host.isTextBlock) {
val textRange = ElementManipulators.getValueTextRange(host)
val indent = PsiLiteralUtil.getTextBlockIndent(host)
if (indent <= 0) {
return listOf(textRange)
}

val text = (host as PsiElement).text
var startOffset = textRange.startOffset + indent
var endOffset = text.indexOf('\n', startOffset)
val result = SmartList<TextRange>()
while (endOffset > 0) {
endOffset++
result.add(TextRange(startOffset, endOffset))
startOffset = endOffset + indent
endOffset = text.indexOf('\n', startOffset)
}
endOffset = textRange.endOffset
if (startOffset < endOffset) {
result.add(TextRange(startOffset, endOffset))
}
return result
} else {
return listOf(ElementManipulators.getValueTextRange(host))
}
}

override fun elementsToInjectIn() = ELEMENTS
}

/* * 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.expression import com.demonwav.mcdev.platform.mixin.util.MixinConstants import com.demonwav.mcdev.util.findContainingModifierList import com.demonwav.mcdev.util.findContainingNameValuePair import com.demonwav.mcdev.util.parseArray import com.intellij.lang.injection.InjectedLanguageManager import com.intellij.lang.injection.MultiHostInjector import com.intellij.lang.injection.MultiHostRegistrar import com.intellij.openapi.util.Key import com.intellij.openapi.util.TextRange import com.intellij.openapi.util.component1 import com.intellij.openapi.util.component2 import com.intellij.psi.ElementManipulators import com.intellij.psi.JavaTokenType import com.intellij.psi.PsiAnnotation import com.intellij.psi.PsiElement import com.intellij.psi.PsiLanguageInjectionHost import com.intellij.psi.PsiLiteralExpression import com.intellij.psi.PsiParenthesizedExpression import com.intellij.psi.PsiPolyadicExpression import com.intellij.psi.impl.source.tree.injected.JavaConcatenationToInjectorAdapter import com.intellij.psi.util.PsiLiteralUtil import com.intellij.psi.util.PsiModificationTracker import com.intellij.psi.util.PsiUtil import com.intellij.psi.util.parentOfType import com.intellij.util.SmartList import org.intellij.plugins.intelliLang.inject.InjectorUtils class MEExpressionInjector : MultiHostInjector { companion object { private val ELEMENTS = listOf(PsiLiteralExpression::class.java) private val ME_EXPRESSION_INJECTION = Key.create<MEExpressionInjection>("mcdev.meExpressionInjection") private val CLASS_INJECTION_RESULT = Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionResult") private val CLASS_INJECTION_REGISTRAR_IMPL = Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionRegistrarImpl") @JvmStatic private val METHOD_ADD_TO_RESULTS = CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("addToResults", CLASS_INJECTION_RESULT) .also { it.isAccessible = true } @JvmStatic private val METHOD_GET_INJECTED_RESULT = CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("getInjectedResult") .also { it.isAccessible = true } } private data class MEExpressionInjection(val modCount: Long, val injectionResult: Any) private fun shouldInjectIn(anchor: PsiElement): Boolean { val nameValuePair = anchor.findContainingNameValuePair() ?: return false return when (nameValuePair.name) { "value", null -> nameValuePair.parentOfType<PsiAnnotation>() ?.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION) == true "id" -> nameValuePair.parentOfType<PsiAnnotation>() ?.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION) == true else -> false } } override fun getLanguagesToInject(registrar: MultiHostRegistrar, context: PsiElement) { if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 1") val project = context.project val (anchor, _) = JavaConcatenationToInjectorAdapter(project).computeAnchorAndOperands(context) if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 2 ${anchor.text}") if (!shouldInjectIn(anchor)) { return } if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 3") if (MEExpressionCompletionUtil.debugCompletionUnitTest) Thread.dumpStack() val modifierList = anchor.findContainingModifierList() ?: return if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 4") val modCount = PsiModificationTracker.getInstance(project).modificationCount val primaryElement = modifierList.getUserData(ME_EXPRESSION_INJECTION) if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 5, $modCount") if (primaryElement != null && primaryElement.modCount == modCount) { METHOD_ADD_TO_RESULTS.invoke(registrar, primaryElement.injectionResult) return } if (MEExpressionCompletionUtil.debugCompletionUnitTest) System.err.println("Injector 6") // A Frankenstein injection is an injection where we don't know the entire contents, and therefore errors should // not be reported. var isFrankenstein = false registrar.startInjecting(MEExpressionLanguage) for (annotation in modifierList.annotations) { if (annotation.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION)) { val idExpr = annotation.findDeclaredAttributeValue("id") ?: continue val isType = annotation.findDeclaredAttributeValue("type") != null var needsPrefix = true iterateConcatenation(idExpr) { op -> if (op is PsiLanguageInjectionHost) { for (textRange in getTextRanges(op)) { val prefix = "\nclass $isType ".takeIf { needsPrefix } needsPrefix = false registrar.addPlace(prefix, null, op, textRange) } } else { isFrankenstein = true } } } else if (annotation.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION)) { for (valueExpr in annotation.findDeclaredAttributeValue("value")?.parseArray { it }.orEmpty()) { val places = mutableListOf<Pair<PsiLanguageInjectionHost, TextRange>>() iterateConcatenation(valueExpr) { op -> if (op is PsiLanguageInjectionHost) { for (textRange in getTextRanges(op)) { places += op to textRange } } else { isFrankenstein = true } } if (places.isNotEmpty()) { for ((i, place) in places.withIndex()) { val (host, range) = place val prefix = "\ndo { ".takeIf { i == 0 } val suffix = " }".takeIf { i == places.size - 1 } registrar.addPlace(prefix, suffix, host, range) } } } } } registrar.doneInjecting() modifierList.putUserData( ME_EXPRESSION_INJECTION, MEExpressionInjection(modCount, METHOD_GET_INJECTED_RESULT.invoke(registrar)) ) if (isFrankenstein) { InjectorUtils.putInjectedFileUserData( context, MEExpressionLanguage, InjectedLanguageManager.FRANKENSTEIN_INJECTION, true ) } } private fun iterateConcatenation(element: PsiElement, consumer: (PsiElement) -> Unit) { when (element) { is PsiParenthesizedExpression -> { val inner = PsiUtil.skipParenthesizedExprDown(element) ?: return iterateConcatenation(inner, consumer) } is PsiPolyadicExpression -> { if (element.operationTokenType == JavaTokenType.PLUS) { for (operand in element.operands) { iterateConcatenation(operand, consumer) } } else { consumer(element) } } else -> consumer(element) } } private fun getTextRanges(host: PsiLanguageInjectionHost): List<TextRange> { if (host is PsiLiteralExpression && host.isTextBlock) { val textRange = ElementManipulators.getValueTextRange(host) val indent = PsiLiteralUtil.getTextBlockIndent(host) if (indent <= 0) { return listOf(textRange) } val text = (host as PsiElement).text var startOffset = textRange.startOffset + indent var endOffset = text.indexOf('\n', startOffset) val result = SmartList<TextRange>() while (endOffset > 0) { endOffset++ result.add(TextRange(startOffset, endOffset)) startOffset = endOffset + indent endOffset = text.indexOf('\n', startOffset) } endOffset = textRange.endOffset if (startOffset < endOffset) { result.add(TextRange(startOffset, endOffset)) } return result } else { return listOf(ElementManipulators.getValueTextRange(host)) } } override fun elementsToInjectIn() = ELEMENTS } /* * 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.expression import com.demonwav.mcdev.platform.mixin.util.MixinConstants import com.demonwav.mcdev.util.findContainingModifierList import com.demonwav.mcdev.util.findContainingNameValuePair import com.demonwav.mcdev.util.parseArray import com.intellij.lang.injection.InjectedLanguageManager import com.intellij.lang.injection.MultiHostInjector import com.intellij.lang.injection.MultiHostRegistrar import com.intellij.openapi.util.Key import com.intellij.openapi.util.TextRange import com.intellij.openapi.util.component1 import com.intellij.openapi.util.component2 import com.intellij.psi.ElementManipulators import com.intellij.psi.JavaTokenType import com.intellij.psi.PsiAnnotation import com.intellij.psi.PsiElement import com.intellij.psi.PsiLanguageInjectionHost import com.intellij.psi.PsiLiteralExpression import com.intellij.psi.PsiParenthesizedExpression import com.intellij.psi.PsiPolyadicExpression import com.intellij.psi.impl.source.tree.injected.JavaConcatenationToInjectorAdapter import com.intellij.psi.util.PsiLiteralUtil import com.intellij.psi.util.PsiModificationTracker import com.intellij.psi.util.PsiUtil import com.intellij.psi.util.parentOfType import com.intellij.util.SmartList import org.intellij.plugins.intelliLang.inject.InjectorUtils class MEExpressionInjector : MultiHostInjector { companion object { private val ELEMENTS = listOf(PsiLiteralExpression::class.java) private val ME_EXPRESSION_INJECTION = Key.create<MEExpressionInjection>("mcdev.meExpressionInjection") private val CLASS_INJECTION_RESULT = Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionResult") private val CLASS_INJECTION_REGISTRAR_IMPL = Class.forName("com.intellij.psi.impl.source.tree.injected.InjectionRegistrarImpl") @JvmStatic private val METHOD_ADD_TO_RESULTS = CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("addToResults", CLASS_INJECTION_RESULT) .also { it.isAccessible = true } @JvmStatic private val METHOD_GET_INJECTED_RESULT = CLASS_INJECTION_REGISTRAR_IMPL.getDeclaredMethod("getInjectedResult") .also { it.isAccessible = true } } private data class MEExpressionInjection(val modCount: Long, val injectionResult: Any) private fun shouldInjectIn(anchor: PsiElement): Boolean { val nameValuePair = anchor.findContainingNameValuePair() ?: return false return when (nameValuePair.name) { "value", null -> nameValuePair.parentOfType<PsiAnnotation>() ?.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION) == true "id" -> nameValuePair.parentOfType<PsiAnnotation>() ?.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION) == true else -> false } } override fun getLanguagesToInject(registrar: MultiHostRegistrar, context: PsiElement) { val project = context.project val (anchor, _) = JavaConcatenationToInjectorAdapter(project).computeAnchorAndOperands(context) if (!shouldInjectIn(anchor)) { return } val modifierList = anchor.findContainingModifierList() ?: return val modCount = PsiModificationTracker.getInstance(project).modificationCount val primaryElement = modifierList.getUserData(ME_EXPRESSION_INJECTION) if (primaryElement != null && primaryElement.modCount == modCount) { METHOD_ADD_TO_RESULTS.invoke(registrar, primaryElement.injectionResult) return } // A Frankenstein injection is an injection where we don't know the entire contents, and therefore errors should // not be reported. var isFrankenstein = false registrar.startInjecting(MEExpressionLanguage) for (annotation in modifierList.annotations) { if (annotation.hasQualifiedName(MixinConstants.MixinExtras.DEFINITION)) { val idExpr = annotation.findDeclaredAttributeValue("id") ?: continue val isType = annotation.findDeclaredAttributeValue("type") != null var needsPrefix = true iterateConcatenation(idExpr) { op -> if (op is PsiLanguageInjectionHost) { for (textRange in getTextRanges(op)) { val prefix = "\nclass $isType ".takeIf { needsPrefix } needsPrefix = false registrar.addPlace(prefix, null, op, textRange) } } else { isFrankenstein = true } } } else if (annotation.hasQualifiedName(MixinConstants.MixinExtras.EXPRESSION)) { for (valueExpr in annotation.findDeclaredAttributeValue("value")?.parseArray { it }.orEmpty()) { val places = mutableListOf<Pair<PsiLanguageInjectionHost, TextRange>>() iterateConcatenation(valueExpr) { op -> if (op is PsiLanguageInjectionHost) { for (textRange in getTextRanges(op)) { places += op to textRange } } else { isFrankenstein = true } } if (places.isNotEmpty()) { for ((i, place) in places.withIndex()) { val (host, range) = place val prefix = "\ndo { ".takeIf { i == 0 } val suffix = " }".takeIf { i == places.size - 1 } registrar.addPlace(prefix, suffix, host, range) } } } } } registrar.doneInjecting() modifierList.putUserData( ME_EXPRESSION_INJECTION, MEExpressionInjection(modCount, METHOD_GET_INJECTED_RESULT.invoke(registrar)) ) if (isFrankenstein) { InjectorUtils.putInjectedFileUserData( context, MEExpressionLanguage, InjectedLanguageManager.FRANKENSTEIN_INJECTION, true ) } } private fun iterateConcatenation(element: PsiElement, consumer: (PsiElement) -> Unit) { when (element) { is PsiParenthesizedExpression -> { val inner = PsiUtil.skipParenthesizedExprDown(element) ?: return iterateConcatenation(inner, consumer) } is PsiPolyadicExpression -> { if (element.operationTokenType == JavaTokenType.PLUS) { for (operand in element.operands) { iterateConcatenation(operand, consumer) } } else { consumer(element) } } else -> consumer(element) } } private fun getTextRanges(host: PsiLanguageInjectionHost): List<TextRange> { if (host is PsiLiteralExpression && host.isTextBlock) { val textRange = ElementManipulators.getValueTextRange(host) val indent = PsiLiteralUtil.getTextBlockIndent(host) if (indent <= 0) { return listOf(textRange) } val text = (host as PsiElement).text var startOffset = textRange.startOffset + indent var endOffset = text.indexOf('\n', startOffset) val result = SmartList<TextRange>() while (endOffset > 0) { endOffset++ result.add(TextRange(startOffset, endOffset)) startOffset = endOffset + indent endOffset = text.indexOf('\n', startOffset) } endOffset = textRange.endOffset if (startOffset < endOffset) { result.add(TextRange(startOffset, endOffset)) } return result } else { return listOf(ElementManipulators.getValueTextRange(host)) } } override fun elementsToInjectIn() = ELEMENTS }

Add build comment

Mute test

Server communication failure

Run Custom Build

TODO

Responsibility

Edit tags

Loading...

Are you sure?