fix mapNs missing visitHeader

unimined · Oct 25, 2024 · 54194e0 · 54194e0
1 parent a7ba45d
commit 54194e0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/commonMain/kotlin/xyz/wagyourtail/unimined/mapping/visitor/delegate/Visitors.kt b/src/commonMain/kotlin/xyz/wagyourtail/unimined/mapping/visitor/delegate/Visitors.kt
@@ -19,6 +19,10 @@ fun MappingVisitor.delegator(delegator: Delegator) = DelegateMappingVisitor(this
 
 fun MappingVisitor.mapNs(nsMap: Map<Namespace, Namespace>) = DelegateMappingVisitor(this, object : Delegator() {
 
+    override fun visitHeader(delegate: MappingVisitor, vararg namespaces: String) {
+        super.visitHeader(delegate, *namespaces.map { nsMap[Namespace(it)]?.name ?: it }.toTypedArray())
+    }
+
     override fun visitPackage(delegate: MappingVisitor, names: Map<Namespace, PackageName>): PackageVisitor? {
         val n = names.mapKeys { nsMap[it.key] ?: it.key }
         if (n.isEmpty()) return null