Allow exporting mappings as SRG or Enigma

1 files changed, 12 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java
index c301594..fe7d097 100644
--- a/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -17,6 +17,7 @@ import com.strobel.decompiler.languages.java.ast.CompilationUnit;
 
 import java.io.File;
 import java.io.FileReader;
+import java.io.FileWriter;
 import java.io.IOException;
 import java.util.Collection;
 import java.util.Deque;
@@ -87,6 +88,17 @@ public class GuiController {
         this.isDirty = false;
     }
 
+    public void saveOldMappings(File file) throws IOException {
+        FileWriter out = new FileWriter(file);
+        new MappingsOldWriter().write(out, this.deobfuscator.getMappings());
+        this.isDirty = false;
+    }
+
+    public void saveSRGMappings(File file) throws IOException {
+        new MappingsSRGWriter().write(file, this.deobfuscator);
+        this.isDirty = false;
+    }
+
     public void closeMappings() {
         this.deobfuscator.setMappings(null);
         this.gui.setMappingsFile(null);