Defer existing Enigma dir deletion to writer visit pass start (#67)

FabricMC · Oct 16, 2023 · a301899 · a301899
1 parent 6e7de77
commit a301899
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/main/java/net/fabricmc/mappingio/format/enigma/EnigmaDirWriter.java b/src/main/java/net/fabricmc/mappingio/format/enigma/EnigmaDirWriter.java
@@ -35,7 +35,11 @@ public final class EnigmaDirWriter extends EnigmaWriterBase {
 	public EnigmaDirWriter(Path dir, boolean deleteExistingFiles) throws IOException {
 		super(null);
 		this.dir = dir.toAbsolutePath().normalize();
+		this.deleteExistingFiles = deleteExistingFiles;
+	}
 
+	@Override
+	public boolean visitHeader() throws IOException {
 		if (deleteExistingFiles && Files.exists(dir)) {
 			Files.walkFileTree(dir, new SimpleFileVisitor<Path>() {
 				@Override
@@ -59,6 +63,8 @@ public FileVisitResult postVisitDirectory(Path file, IOException exc) throws IOE
 				}
 			});
 		}
+
+		return super.visitHeader();
 	}
 
 	@Override
@@ -143,4 +149,5 @@ public boolean visitElementContent(MappedElementKind targetKind) throws IOExcept
 	}
 
 	private final Path dir;
+	private final boolean deleteExistingFiles;
 }
diff --git a/src/main/java/net/fabricmc/mappingio/format/enigma/EnigmaWriterBase.java b/src/main/java/net/fabricmc/mappingio/format/enigma/EnigmaWriterBase.java
@@ -41,6 +41,11 @@ public Set<MappingFlag> getFlags() {
 		return flags;
 	}
 
+	@Override
+	public boolean visitHeader() throws IOException {
+		return false;
+	}
+
 	@Override
 	public void visitNamespaces(String srcNamespace, List<String> dstNamespaces) { }