src/main/kotlin/platform/fabric/reference/ResourceFileReference.kt diff

← → edited src/main/kotlin/platform/fabric/reference/ResourceFileReference.kt

Ignore whitespaces Open in IDE

⁠ rednesto: Fix #2362 CME in fabric.mod.json

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.fabric.reference

import com.demonwav.mcdev.facet.MinecraftFacet
import com.demonwav.mcdev.util.SourceType
import com.demonwav.mcdev.util.findModule
import com.demonwav.mcdev.util.manipulator
import com.demonwav.mcdev.util.mapFirstNotNull
import com.demonwav.mcdev.util.reference.InspectionReference
import com.intellij.json.psi.JsonStringLiteral
import com.intellij.openapi.application.runReadAction
import com.intellij.openapi.module.Module
import com.intellij.openapi.module.ModuleManager
import com.intellij.openapi.module.ModuleUtilCore
import com.intellij.openapi.project.rootManager
import com.intellij.openapi.roots.ProjectRootManager
import com.intellij.openapi.vfs.findPsiFile
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiFile
import com.intellij.psi.PsiManager
import com.intellij.psi.PsiReference
import com.intellij.psi.PsiReferenceBase
import com.intellij.psi.PsiReferenceProvider
import com.intellij.util.IncorrectOperationException
import com.intellij.util.ProcessingContext
import org.jetbrains.jps.model.java.JavaResourceRootType

class ResourceFileReference(
private val description: String,
private val filenamePattern: Regex? = null
) : PsiReferenceProvider() {
override fun getReferencesByElement(element: PsiElement, context: ProcessingContext): Array<PsiReference> {
return arrayOf(Reference(description, element as JsonStringLiteral))
}

private inner class Reference(desc: String, element: JsonStringLiteral) :
PsiReferenceBase<JsonStringLiteral>(element),
InspectionReference {

val isInTestSourceSet: Boolean = run {
val containingVFile = element.containingFile.originalFile.virtualFile
val inTestSourceContent =
ProjectRootManager.getInstance(element.project).fileIndex.isInTestSourceContent(containingVFile)
inTestSourceContent
}

override val description = desc
override val unresolved = resolve() == null

override fun resolve(): PsiElement? {
fun findFileIn(module: Module): PsiFile? {
val facet = MinecraftFacet.getInstance(module) ?: return null
var virtualFile = facet.findFile(element.value, SourceType.RESOURCE)
if (virtualFile == null && isInTestSourceSet) {
virtualFile = facet.findFile(element.value, SourceType.TEST_RESOURCE)
}

if (virtualFile == null) {
return null
}

return PsiManager.getInstance(element.project).findFile(virtualFile)
}

return getRelevantModules().mapFirstNotNull(::findFileIn)
}

override fun bindToElement(newTarget: PsiElement): PsiElement? {
if (newTarget !is PsiFile) {
throw IncorrectOperationException("Cannot target $newTarget")
}
val manipulator = element.manipulator ?: return null
return manipulator.handleContentChange(element, manipulator.getRangeInElement(element), newTarget.name)
}

override fun getVariants(): Array<out Any?> {
if (filenamePattern == null) {
return emptyArray()
}

val variants = mutableListOf<Any>()
runReadAction {
val relevantModules = getRelevantModules()

val relevantRootTypes = mutableSetOf(JavaResourceRootType.RESOURCE)
if (isInTestSourceSet) {
relevantRootTypes.add(JavaResourceRootType.TEST_RESOURCE)
}

val relevantRoots = relevantModules.flatMap {
it.rootManager.getSourceRoots(relevantRootTypes)
}

for (roots in relevantRoots) {
for (child in roots.children) {
val relativePath = child.path.removePrefix(roots.path)
val testRelativePath = "/$relativePath"
if (testRelativePath.matches(filenamePattern)) {
variants.add(child.findPsiFile(element.project) ?: relativePath)
}
}
}
}
return variants.toTypedArray()
}
}

private fun Reference.getRelevantModules(): Set<Module> {
val module = element.findModule() ?: return emptySet()
val relevantModules = mutableSetOf<Module>()
val moduleManager = ModuleManager.getInstance(element.project)
ModuleUtilCore.getDependencies(module, relevantModules)
relevantModules.flatMapTo(relevantModules) {
moduleManager.getModuleDependentModules(it)
}
return relevantModules
}
}

/*
* 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.fabric.reference

import com.demonwav.mcdev.facet.MinecraftFacet
import com.demonwav.mcdev.util.SourceType
import com.demonwav.mcdev.util.findModule
import com.demonwav.mcdev.util.manipulator
import com.demonwav.mcdev.util.mapFirstNotNull
import com.demonwav.mcdev.util.reference.InspectionReference
import com.intellij.json.psi.JsonStringLiteral
import com.intellij.openapi.application.runReadAction
import com.intellij.openapi.module.Module
import com.intellij.openapi.module.ModuleManager
import com.intellij.openapi.module.ModuleUtilCore
import com.intellij.openapi.project.rootManager
import com.intellij.openapi.roots.ProjectRootManager
import com.intellij.openapi.vfs.findPsiFile
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiFile
import com.intellij.psi.PsiManager
import com.intellij.psi.PsiReference
import com.intellij.psi.PsiReferenceBase
import com.intellij.psi.PsiReferenceProvider
import com.intellij.util.IncorrectOperationException
import com.intellij.util.ProcessingContext
import org.jetbrains.jps.model.java.JavaResourceRootType

class ResourceFileReference(
private val description: String,
private val filenamePattern: Regex? = null
) : PsiReferenceProvider() {
override fun getReferencesByElement(element: PsiElement, context: ProcessingContext): Array<PsiReference> {
return arrayOf(Reference(description, element as JsonStringLiteral))
}

private inner class Reference(desc: String, element: JsonStringLiteral) :
PsiReferenceBase<JsonStringLiteral>(element),
InspectionReference {

val isInTestSourceSet: Boolean = run {
val containingVFile = element.containingFile.originalFile.virtualFile
val inTestSourceContent =
ProjectRootManager.getInstance(element.project).fileIndex.isInTestSourceContent(containingVFile)
inTestSourceContent
}

override val description = desc
override val unresolved = resolve() == null

override fun resolve(): PsiElement? {
fun findFileIn(module: Module): PsiFile? {
val facet = MinecraftFacet.getInstance(module) ?: return null
var virtualFile = facet.findFile(element.value, SourceType.RESOURCE)
if (virtualFile == null && isInTestSourceSet) {
virtualFile = facet.findFile(element.value, SourceType.TEST_RESOURCE)
}

if (virtualFile == null) {
return null
}

return PsiManager.getInstance(element.project).findFile(virtualFile)
}

return getRelevantModules().mapFirstNotNull(::findFileIn)
}

override fun bindToElement(newTarget: PsiElement): PsiElement? {
if (newTarget !is PsiFile) {
throw IncorrectOperationException("Cannot target $newTarget")
}
val manipulator = element.manipulator ?: return null
return manipulator.handleContentChange(element, manipulator.getRangeInElement(element), newTarget.name)
}

override fun getVariants(): Array<out Any?> {
if (filenamePattern == null) {
return emptyArray()
}

val variants = mutableListOf<Any>()
runReadAction {
val relevantModules = getRelevantModules()

val relevantRootTypes = mutableSetOf(JavaResourceRootType.RESOURCE)
if (isInTestSourceSet) {
relevantRootTypes.add(JavaResourceRootType.TEST_RESOURCE)
}

val relevantRoots = relevantModules.flatMap {
it.rootManager.getSourceRoots(relevantRootTypes)
}

for (roots in relevantRoots) {
for (child in roots.children) {
val relativePath = child.path.removePrefix(roots.path)
val testRelativePath = "/$relativePath"
if (testRelativePath.matches(filenamePattern)) {
variants.add(child.findPsiFile(element.project) ?: relativePath)
}
}
}
}
return variants.toTypedArray()
}
}

private fun Reference.getRelevantModules(): Set<Module> {
val module = element.findModule() ?: return emptySet()
val relevantModules = mutableSetOf<Module>()
val moduleManager = ModuleManager.getInstance(element.project)
ModuleUtilCore.getDependencies(module, relevantModules)
relevantModules.toList().flatMapTo(relevantModules) {
moduleManager.getModuleDependentModules(it)
}
return relevantModules
}
}

/* * 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.fabric.reference import com.demonwav.mcdev.facet.MinecraftFacet import com.demonwav.mcdev.util.SourceType import com.demonwav.mcdev.util.findModule import com.demonwav.mcdev.util.manipulator import com.demonwav.mcdev.util.mapFirstNotNull import com.demonwav.mcdev.util.reference.InspectionReference import com.intellij.json.psi.JsonStringLiteral import com.intellij.openapi.application.runReadAction import com.intellij.openapi.module.Module import com.intellij.openapi.module.ModuleManager import com.intellij.openapi.module.ModuleUtilCore import com.intellij.openapi.project.rootManager import com.intellij.openapi.roots.ProjectRootManager import com.intellij.openapi.vfs.findPsiFile import com.intellij.psi.PsiElement import com.intellij.psi.PsiFile import com.intellij.psi.PsiManager import com.intellij.psi.PsiReference import com.intellij.psi.PsiReferenceBase import com.intellij.psi.PsiReferenceProvider import com.intellij.util.IncorrectOperationException import com.intellij.util.ProcessingContext import org.jetbrains.jps.model.java.JavaResourceRootType class ResourceFileReference( private val description: String, private val filenamePattern: Regex? = null ) : PsiReferenceProvider() { override fun getReferencesByElement(element: PsiElement, context: ProcessingContext): Array<PsiReference> { return arrayOf(Reference(description, element as JsonStringLiteral)) } private inner class Reference(desc: String, element: JsonStringLiteral) : PsiReferenceBase<JsonStringLiteral>(element), InspectionReference { val isInTestSourceSet: Boolean = run { val containingVFile = element.containingFile.originalFile.virtualFile val inTestSourceContent = ProjectRootManager.getInstance(element.project).fileIndex.isInTestSourceContent(containingVFile) inTestSourceContent } override val description = desc override val unresolved = resolve() == null override fun resolve(): PsiElement? { fun findFileIn(module: Module): PsiFile? { val facet = MinecraftFacet.getInstance(module) ?: return null var virtualFile = facet.findFile(element.value, SourceType.RESOURCE) if (virtualFile == null && isInTestSourceSet) { virtualFile = facet.findFile(element.value, SourceType.TEST_RESOURCE) } if (virtualFile == null) { return null } return PsiManager.getInstance(element.project).findFile(virtualFile) } return getRelevantModules().mapFirstNotNull(::findFileIn) } override fun bindToElement(newTarget: PsiElement): PsiElement? { if (newTarget !is PsiFile) { throw IncorrectOperationException("Cannot target $newTarget") } val manipulator = element.manipulator ?: return null return manipulator.handleContentChange(element, manipulator.getRangeInElement(element), newTarget.name) } override fun getVariants(): Array<out Any?> { if (filenamePattern == null) { return emptyArray() } val variants = mutableListOf<Any>() runReadAction { val relevantModules = getRelevantModules() val relevantRootTypes = mutableSetOf(JavaResourceRootType.RESOURCE) if (isInTestSourceSet) { relevantRootTypes.add(JavaResourceRootType.TEST_RESOURCE) } val relevantRoots = relevantModules.flatMap { it.rootManager.getSourceRoots(relevantRootTypes) } for (roots in relevantRoots) { for (child in roots.children) { val relativePath = child.path.removePrefix(roots.path) val testRelativePath = "/$relativePath" if (testRelativePath.matches(filenamePattern)) { variants.add(child.findPsiFile(element.project) ?: relativePath) } } } } return variants.toTypedArray() } } private fun Reference.getRelevantModules(): Set<Module> { val module = element.findModule() ?: return emptySet() val relevantModules = mutableSetOf<Module>() val moduleManager = ModuleManager.getInstance(element.project) ModuleUtilCore.getDependencies(module, relevantModules) relevantModules.flatMapTo(relevantModules) { moduleManager.getModuleDependentModules(it) } return relevantModules } } /* * 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.fabric.reference import com.demonwav.mcdev.facet.MinecraftFacet import com.demonwav.mcdev.util.SourceType import com.demonwav.mcdev.util.findModule import com.demonwav.mcdev.util.manipulator import com.demonwav.mcdev.util.mapFirstNotNull import com.demonwav.mcdev.util.reference.InspectionReference import com.intellij.json.psi.JsonStringLiteral import com.intellij.openapi.application.runReadAction import com.intellij.openapi.module.Module import com.intellij.openapi.module.ModuleManager import com.intellij.openapi.module.ModuleUtilCore import com.intellij.openapi.project.rootManager import com.intellij.openapi.roots.ProjectRootManager import com.intellij.openapi.vfs.findPsiFile import com.intellij.psi.PsiElement import com.intellij.psi.PsiFile import com.intellij.psi.PsiManager import com.intellij.psi.PsiReference import com.intellij.psi.PsiReferenceBase import com.intellij.psi.PsiReferenceProvider import com.intellij.util.IncorrectOperationException import com.intellij.util.ProcessingContext import org.jetbrains.jps.model.java.JavaResourceRootType class ResourceFileReference( private val description: String, private val filenamePattern: Regex? = null ) : PsiReferenceProvider() { override fun getReferencesByElement(element: PsiElement, context: ProcessingContext): Array<PsiReference> { return arrayOf(Reference(description, element as JsonStringLiteral)) } private inner class Reference(desc: String, element: JsonStringLiteral) : PsiReferenceBase<JsonStringLiteral>(element), InspectionReference { val isInTestSourceSet: Boolean = run { val containingVFile = element.containingFile.originalFile.virtualFile val inTestSourceContent = ProjectRootManager.getInstance(element.project).fileIndex.isInTestSourceContent(containingVFile) inTestSourceContent } override val description = desc override val unresolved = resolve() == null override fun resolve(): PsiElement? { fun findFileIn(module: Module): PsiFile? { val facet = MinecraftFacet.getInstance(module) ?: return null var virtualFile = facet.findFile(element.value, SourceType.RESOURCE) if (virtualFile == null && isInTestSourceSet) { virtualFile = facet.findFile(element.value, SourceType.TEST_RESOURCE) } if (virtualFile == null) { return null } return PsiManager.getInstance(element.project).findFile(virtualFile) } return getRelevantModules().mapFirstNotNull(::findFileIn) } override fun bindToElement(newTarget: PsiElement): PsiElement? { if (newTarget !is PsiFile) { throw IncorrectOperationException("Cannot target $newTarget") } val manipulator = element.manipulator ?: return null return manipulator.handleContentChange(element, manipulator.getRangeInElement(element), newTarget.name) } override fun getVariants(): Array<out Any?> { if (filenamePattern == null) { return emptyArray() } val variants = mutableListOf<Any>() runReadAction { val relevantModules = getRelevantModules() val relevantRootTypes = mutableSetOf(JavaResourceRootType.RESOURCE) if (isInTestSourceSet) { relevantRootTypes.add(JavaResourceRootType.TEST_RESOURCE) } val relevantRoots = relevantModules.flatMap { it.rootManager.getSourceRoots(relevantRootTypes) } for (roots in relevantRoots) { for (child in roots.children) { val relativePath = child.path.removePrefix(roots.path) val testRelativePath = "/$relativePath" if (testRelativePath.matches(filenamePattern)) { variants.add(child.findPsiFile(element.project) ?: relativePath) } } } } return variants.toTypedArray() } } private fun Reference.getRelevantModules(): Set<Module> { val module = element.findModule() ?: return emptySet() val relevantModules = mutableSetOf<Module>() val moduleManager = ModuleManager.getInstance(element.project) ModuleUtilCore.getDependencies(module, relevantModules) relevantModules.toList().flatMapTo(relevantModules) { moduleManager.getModuleDependentModules(it) } return relevantModules } }

Add build comment

Mute test

Server communication failure

Run Custom Build

TODO

Responsibility

Edit tags

Loading...

Are you sure?