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 index a51de95..6f8239e 100644 --- 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) = 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): PackageVisitor? { val n = names.mapKeys { nsMap[it.key] ?: it.key } if (n.isEmpty()) return null