Don't hardcode namespaces for CT files; don't assume LoomExtension has a mappingsFile property

Author: Earthcomputer

src/gradle-tooling-extension/groovy/com/demonwav/mcdev/platform/mcp/gradle/tooling/fabricloom/FabricLoomModelBuilderImpl.groovy

@@ -61,7 +61,7 @@ class FabricLoomModelBuilderImpl extends AbstractModelBuilderService {
 
     FabricLoomModel build(Project project, Object loomExtension) {
         def minecraftVersion = loomExtension.minecraftProvider.minecraftVersion()
-        def tinyMappings = loomExtension.mappingsFile
+        def tinyMappings = loomExtension.hasProperty("mappingsFile") ? loomExtension.mappingsFile : null
         def splitMinecraftJar = loomExtension.areEnvironmentSourceSetsSplit()
 
         def decompilers = [:]

src/gradle-tooling-extension/groovy/com/demonwav/mcdev/platform/mcp/gradle/tooling/fabricloom/FabricLoomModelImpl.groovy

@@ -21,10 +21,12 @@
 package com.demonwav.mcdev.platform.mcp.gradle.tooling.fabricloom
 
 import groovy.transform.Immutable
+import org.jetbrains.annotations.Nullable
 
 @Immutable(knownImmutableClasses = [File])
 class FabricLoomModelImpl implements FabricLoomModel, Serializable {
     String minecraftVersion
+    @Nullable
     File tinyMappings
     Map<String, List<DecompilerModel>> decompilers
     boolean splitMinecraftJar

src/gradle-tooling-extension/java/com/demonwav/mcdev/platform/mcp/gradle/tooling/fabricloom/FabricLoomModel.java

@@ -20,6 +20,8 @@
 
 package com.demonwav.mcdev.platform.mcp.gradle.tooling.fabricloom;
 
+import org.jetbrains.annotations.Nullable;
+
 import java.io.File;
 import java.util.List;
 import java.util.Map;
@@ -28,6 +30,7 @@ public interface FabricLoomModel {
 
     String getMinecraftVersion();
 
+    @Nullable
     File getTinyMappings();
 
     Map<String, List<DecompilerModel>> getDecompilers();

src/main/grammars/CtLexer.flex

@@ -52,7 +52,7 @@ import static com.intellij.psi.TokenType.*;
 
 HEADER_NAME=accessWidener|classTweaker
 HEADER_VERSION_ELEMENT=v\d+
-HEADER_NAMESPACE_ELEMENT=named|intermediary
+HEADER_NAMESPACE_ELEMENT=\w+
 PRIMITIVE=[ZBCSIFDJV]
 CLASS_VALUE=L[^;\n]+;
 SIGNATURE_CLASS_VALUE_START=L[^;<\n]+

src/main/grammars/CtParser.bnf

@@ -43,6 +43,9 @@ private end_line ::= crlf | <<eof>>
 header ::= HEADER_NAME HEADER_VERSION_ELEMENT HEADER_NAMESPACE_ELEMENT {
     mixin="com.demonwav.mcdev.platform.mcp.ct.psi.mixins.impl.CtHeaderImplMixin"
     implements="com.demonwav.mcdev.platform.mcp.ct.psi.mixins.CtHeaderMixin"
+    methods=[
+        namespaceElement="HEADER_NAMESPACE_ELEMENT"
+    ]
     pin = 1
 }
 

src/main/kotlin/platform/fabric/FabricModule.kt

@@ -56,6 +56,10 @@ class FabricModule internal constructor(facet: MinecraftFacet) : AbstractModule(
     override val namedToMojangManager: MappingsManager?
         get() = namedToMojangManagerField
 
+    private var mappingNamespacesField = emptyList<String>()
+    val mappingNamespaces: List<String>
+        get() = mappingNamespacesField
+
     override val moduleType = FabricModuleType
     override val type = PlatformType.FABRIC
     override val icon = PlatformAssets.FABRIC_ICON
@@ -79,60 +83,74 @@ class FabricModule internal constructor(facet: MinecraftFacet) : AbstractModule(
     }
 
     override fun refresh() {
-        namedToMojangManagerField = if (detectYarn()) {
+        val mappingDetection = detectMappings()
+
+        namedToMojangManagerField = if (mappingDetection.yarnDetected) {
             MappingsManager.Immediate(HardcodedYarnToMojmap.createMappings())
         } else {
             null
         }
+
+        mappingNamespacesField = mappingDetection.namespaces
     }
 
-    private fun detectYarn(): Boolean {
-        val gradleData = GradleUtil.findGradleModuleData(facet.module) ?: return false
-        val loomData =
-            gradleData.children.find { it.key == FabricLoomData.KEY }?.data as? FabricLoomData ?: return false
-        val mappingsFile = loomData.tinyMappings ?: return false
+    private fun detectMappings(): MappingDetectionResult {
+        val gradleData = GradleUtil.findGradleModuleData(facet.module) ?: return MappingDetectionResult.DEFAULT
+        val loomData = gradleData.children.find { it.key == FabricLoomData.KEY }?.data as? FabricLoomData
+            ?: return MappingDetectionResult.DEFAULT
+        val mappingsFile = loomData.tinyMappings ?: return MappingDetectionResult.DEFAULT
 
-        var yarnDetected = false
-        val visitor = object : MappingVisitor {
-            private var namedIndex = -1
+        val result = MappingDetectionResult()
 
-            override fun visitNamespaces(srcNamespace: String?, dstNamespaces: List<String>) {
-                namedIndex = dstNamespaces.indexOf("named")
-            }
+        try {
+            MappingReader.read(mappingsFile.toPath(), result)
+        } catch (_: IOException) {
+            return MappingDetectionResult.DEFAULT
+        }
 
-            override fun visitContent() = namedIndex >= 0
+        return result
+    }
 
-            override fun visitClass(srcName: String) = true
+    override fun dispose() {
+        super.dispose()
+        fabricJson = null
+    }
 
-            override fun visitField(srcName: String?, srcDesc: String?) = false
+    private class MappingDetectionResult : MappingVisitor {
+        var yarnDetected = false
+        var namespaces = emptyList<String>()
+        private var namedIndex = -1
 
-            override fun visitMethod(srcName: String?, srcDesc: String?) = false
+        override fun visitNamespaces(srcNamespace: String?, dstNamespaces: List<String>) {
+            namedIndex = dstNamespaces.indexOf("named")
+            namespaces = listOfNotNull(srcNamespace) + dstNamespaces
+        }
 
-            override fun visitMethodArg(argPosition: Int, lvIndex: Int, srcName: String?) = false
+        override fun visitContent() = namedIndex >= 0
 
-            override fun visitMethodVar(lvtRowIndex: Int, lvIndex: Int, startOpIdx: Int, srcName: String?) = false
+        override fun visitClass(srcName: String) = true
 
-            override fun visitDstName(targetKind: MappedElementKind?, namespace: Int, name: String) {
-                if (namespace == namedIndex && name == "net/minecraft/client/MinecraftClient") {
-                    yarnDetected = true
-                }
-            }
+        override fun visitField(srcName: String?, srcDesc: String?) = false
+
+        override fun visitMethod(srcName: String?, srcDesc: String?) = false
+
+        override fun visitMethodArg(argPosition: Int, lvIndex: Int, srcName: String?) = false
+
+        override fun visitMethodVar(lvtRowIndex: Int, lvIndex: Int, startOpIdx: Int, srcName: String?) = false
 
-            override fun visitComment(targetKind: MappedElementKind?, comment: String?) {
+        override fun visitDstName(targetKind: MappedElementKind?, namespace: Int, name: String) {
+            if (namespace == namedIndex && name == "net/minecraft/client/MinecraftClient") {
+                yarnDetected = true
             }
         }
 
-        try {
-            MappingReader.read(mappingsFile.toPath(), visitor)
-        } catch (e: IOException) {
-            return false
+        override fun visitComment(targetKind: MappedElementKind?, comment: String?) {
         }
 
-        return yarnDetected
-    }
-
-    override fun dispose() {
-        super.dispose()
-        fabricJson = null
+        companion object {
+            val DEFAULT = MappingDetectionResult().apply {
+                namespaces = listOf("official")
+            }
+        }
     }
 }

src/main/kotlin/platform/mcp/ct/CtCompletionContributor.kt

@@ -20,7 +20,10 @@
 
 package com.demonwav.mcdev.platform.mcp.ct
 
+import com.demonwav.mcdev.facet.MinecraftFacet
+import com.demonwav.mcdev.platform.fabric.FabricModuleType
 import com.demonwav.mcdev.platform.mcp.ct.gen.psi.CtTypes
+import com.demonwav.mcdev.util.findModule
 import com.intellij.codeInsight.completion.CodeCompletionHandlerBase
 import com.intellij.codeInsight.completion.CompletionContributor
 import com.intellij.codeInsight.completion.CompletionParameters
@@ -86,7 +89,11 @@ object CtNamespaceCompletionProvider : CompletionProvider<CompletionParameters>(
         parameters: CompletionParameters,
         context: ProcessingContext,
         result: CompletionResultSet,
-    ) = result.addAllElements(listOf("named", "intermediary").map(LookupElementBuilder::create))
+    ) {
+        val module = parameters.originalFile.findModule() ?: return
+        val fabricModule = MinecraftFacet.getInstance(module, FabricModuleType) ?: return
+        result.addAllElements(fabricModule.mappingNamespaces.map(LookupElementBuilder::create))
+    }
 }
 
 object CtEntryStartCompletionProvider : CompletionProvider<CompletionParameters>() {

src/main/kotlin/platform/mcp/ct/CtSyntaxHighlighter.kt

@@ -55,7 +55,7 @@ class CtSyntaxHighlighter : SyntaxHighlighterBase() {
         val HEADER_NAME =
             TextAttributesKey.createTextAttributesKey("CT_HEADER_NAME", DefaultLanguageHighlighterColors.KEYWORD)
         val HEADER_NAMESPACE =
-            TextAttributesKey.createTextAttributesKey("CT_HEADER_NAMESPACE", DefaultLanguageHighlighterColors.KEYWORD)
+            TextAttributesKey.createTextAttributesKey("CT_HEADER_NAMESPACE", DefaultLanguageHighlighterColors.CLASS_REFERENCE)
         val ACCESS =
             TextAttributesKey.createTextAttributesKey("CT_ACCESS", DefaultLanguageHighlighterColors.KEYWORD)
         val INJECT_INTERFACE =

src/main/kotlin/platform/mcp/ct/inspections/UnknownCtNamespaceInspection.kt

@@ -0,0 +1,51 @@
+/*
+ * 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.mcp.ct.inspections
+
+import com.demonwav.mcdev.facet.MinecraftFacet
+import com.demonwav.mcdev.platform.fabric.FabricModuleType
+import com.demonwav.mcdev.platform.mcp.ct.gen.psi.CtHeader
+import com.demonwav.mcdev.platform.mcp.ct.gen.psi.CtVisitor
+import com.demonwav.mcdev.platform.mcp.ct.psi.mixins.CtHeaderMixin
+import com.demonwav.mcdev.util.findModule
+import com.intellij.codeInspection.LocalInspectionTool
+import com.intellij.codeInspection.ProblemsHolder
+
+class UnknownCtNamespaceInspection : LocalInspectionTool() {
+    override fun getDisplayName() = "Unknown CT namespace"
+    override fun getStaticDescription() = "Reports an unknown namespace in a ClassTweaker header"
+
+    override fun buildVisitor(holder: ProblemsHolder, isOnTheFly: Boolean) = object : CtVisitor() {
+        override fun visitHeaderMixin(header: CtHeaderMixin) {
+            val namespace = header.namespaceString ?: return
+
+            val module = header.findModule() ?: return
+            val fabricModule = MinecraftFacet.getInstance(module, FabricModuleType) ?: return
+
+            if (namespace !in fabricModule.mappingNamespaces) {
+                holder.registerProblem(
+                    (header as CtHeader).namespaceElement ?: header,
+                    "Unrecognized namespace"
+                )
+            }
+        }
+    }
+}

src/main/resources/META-INF/plugin.xml

@@ -1027,6 +1027,13 @@
                          level="WARNING"
                          hasStaticDescription="true"
                          implementationClass="com.demonwav.mcdev.platform.mcp.ct.inspections.DuplicateAwEntryInspection"/>
+        <localInspection displayName="Unknown CT namespace"
+                         groupName="MCP"
+                         language="Class Tweaker"
+                         enabledByDefault="true"
+                         level="WARNING"
+                         hasStaticDescription="true"
+                         implementationClass="com.demonwav.mcdev.platform.mcp.ct.inspections.UnknownCtNamespaceInspection"/>
         <!--endregion-->
 
         <!--region MIXIN INSPECTIONS-->