/*
 * 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.platform.fabric.util.FabricConstants
import com.demonwav.mcdev.platform.mcp.fabricloom.FabricLoomData
import com.intellij.openapi.module.ModuleManager
import com.intellij.openapi.module.ModuleUtilCore
import com.intellij.openapi.project.Project
import com.intellij.openapi.roots.ProjectRootManager
import com.intellij.openapi.vfs.VirtualFile
import com.intellij.psi.ResolveScopeEnlarger
import com.intellij.psi.search.GlobalSearchScope
import com.intellij.psi.search.SearchScope
import org.jetbrains.plugins.gradle.util.GradleUtil
 
class FabricModJsonResolveScopeEnlarger : ResolveScopeEnlarger() {
 
    override fun getAdditionalResolveScope(file: VirtualFile, project: Project): SearchScope? {
        if (file.name != FabricConstants.FABRIC_MOD_JSON) {
            return null
        }
 
        val module = ModuleUtilCore.findModuleForFile(file, project)
            ?: return null
        val loomData = GradleUtil.findGradleModuleData(module)?.children
            ?.find { it.key == FabricLoomData.KEY }?.data as? FabricLoomData
            ?: return null
        val modSourceSets = loomData.modSourceSets
            ?: return null
 
        val moduleScopes = mutableListOf<GlobalSearchScope>()
        val moduleManager = ModuleManager.getInstance(project)
        val parentPath = module.name.substringBeforeLast('.')
        val rootType = ProjectRootManager.getInstance(project).fileIndex.getContainingSourceRootType(file)
            ?: return null
        for ((_, sourceSets) in modSourceSets) {
            for (sourceSet in sourceSets) {
                val childModule = moduleManager.findModuleByName("$parentPath.$sourceSet")
                if (childModule != null) {
                    moduleScopes.add(childModule.getModuleScope(rootType.isForTests))
                }
            }
        }
 
        if (moduleScopes.isEmpty()) {
            return null
        }
 
        return GlobalSearchScope.union(moduleScopes)
    }
}

      

        
/*
 * 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.platform.fabric.util.FabricConstants
import com.intellij.openapi.module.ModuleUtilCore
import com.intellij.openapi.project.Project
import com.intellij.openapi.vfs.VirtualFile
import com.intellij.psi.ResolveScopeEnlarger
import com.intellij.psi.search.SearchScope
 
class FabricModJsonResolveScopeEnlarger : ResolveScopeEnlarger() {
 
    override fun getAdditionalResolveScope(file: VirtualFile, project: Project): SearchScope? {
        if (file.name != FabricConstants.FABRIC_MOD_JSON) {
            return null
        }
 
        val module = ModuleUtilCore.findModuleForFile(file, project)
            ?: return null
        return module.moduleWithDependentsScope.union(module.moduleTestsWithDependentsScope)
    }
}