1 files changed, 13 insertions, 15 deletions

diff --git a/src/main/java/cuchaz/enigma/mapping/MappingsEnigmaWriter.java b/src/main/java/cuchaz/enigma/mapping/MappingsEnigmaWriter.java
index a3b0616e..ba0720a0 100644
--- a/src/main/java/cuchaz/enigma/mapping/MappingsEnigmaWriter.java
+++ b/src/main/java/cuchaz/enigma/mapping/MappingsEnigmaWriter.java
@@ -30,10 +30,23 @@ public class MappingsEnigmaWriter {
                         writeAsDirectory(out, mappings);
         }
 
+        private void deleteDir(File file) {
+                File[] contents = file.listFiles();
+                if (contents != null) {
+                        for (File f : contents) {
+                                deleteDir(f);
+                        }
+                }
+                file.delete();
+        }
+
         public void writeAsDirectory(File target, Mappings mappings) throws IOException {
+                //TODO: Know what have changes during write to not rewrite all the things
+                deleteDir(target);
                 if (!target.exists() && !target.mkdirs())
                         throw  new IOException("Cannot create mapping directory!");
 
+
                 for (ClassMapping classMapping : sorted(mappings.classes())) {
                         File obFile = new File(target, classMapping.getObfFullName() + ".mapping");
                         File result;
@@ -55,21 +68,6 @@ public class MappingsEnigmaWriter {
                         outputWriter.close();
                 }
 
-                // Remove empty directories
-                File[] fileList = target.listFiles();
-                if (fileList != null)
-                {
-                        for (File file : fileList)
-                        {
-                                if (file.isDirectory())
-                                {
-                                        File[] childFiles = file.listFiles();
-                                        if (childFiles != null && childFiles.length == 0)
-                                                file.delete();
-                                }
-                        }
-                }
-
         }
         
         public void write(PrintWriter out, Mappings mappings) throws IOException {