Fix Enigma dir writer behaving incorrectly after first write when usi…

…ng MIO
FabricMC · Aug 18, 2024 · 3a29b65 · 3a29b65
1 parent 6fd510c
commit 3a29b65
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/enigma/src/main/java/cuchaz/enigma/translation/mapping/serde/MappingFormat.java b/enigma/src/main/java/cuchaz/enigma/translation/mapping/serde/MappingFormat.java
@@ -80,6 +80,7 @@ public void write(EntryTree<EntryMapping> mappings, MappingDelta<EntryMapping> d
 		try {
 			if (this == ENIGMA_DIRECTORY) { // TODO: Remove once MIO supports deltas
 				EnigmaMappingsWriter.DIRECTORY.write(mappings, lastUsedMappingIoWriter ? MappingDelta.added(mappings) : delta, path, progressListener, saveParameters, true);
+				lastUsedMappingIoWriter = false;
 			} else {
 				if (this == PROGUARD) {
 					mappings = MappingOperations.invert(mappings);
@@ -91,9 +92,9 @@ public void write(EntryTree<EntryMapping> mappings, MappingDelta<EntryMapping> d
 
 				tree.accept(MappingWriter.create(path, mappingIoCounterpart), VisitOrder.createByName());
 				progressListener.step(1, I18n.translate("progress.done"));
+				lastUsedMappingIoWriter = true;
 			}
 
-			lastUsedMappingIoWriter = true;
 		} catch (IOException e) {
 			throw new UncheckedIOException(e);
 		}