Allow exporting mappings as SRG or Enigma

5 files changed, 220 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/Gui.java b/src/main/java/cuchaz/enigma/gui/Gui.java
index d93aa9fc..5b0b599f 100644
--- a/src/main/java/cuchaz/enigma/gui/Gui.java
+++ b/src/main/java/cuchaz/enigma/gui/Gui.java
@@ -319,6 +319,8 @@ public class Gui {
         this.menuBar.openMappingsMenu.setEnabled(true);
         this.menuBar.saveMappingsMenu.setEnabled(false);
         this.menuBar.saveMappingsAsMenu.setEnabled(true);
+        this.menuBar.saveMappingsOldMenu.setEnabled(true);
+        this.menuBar.saveMappingsSrgMenu.setEnabled(true);
         this.menuBar.closeMappingsMenu.setEnabled(true);
         this.menuBar.exportSourceMenu.setEnabled(true);
         this.menuBar.exportJarMenu.setEnabled(true);
@@ -340,6 +342,8 @@ public class Gui {
         this.menuBar.openMappingsMenu.setEnabled(false);
         this.menuBar.saveMappingsMenu.setEnabled(false);
         this.menuBar.saveMappingsAsMenu.setEnabled(false);
+        this.menuBar.saveMappingsOldMenu.setEnabled(false);
+        this.menuBar.saveMappingsSrgMenu.setEnabled(false);
         this.menuBar.closeMappingsMenu.setEnabled(false);
         this.menuBar.exportSourceMenu.setEnabled(false);
         this.menuBar.exportJarMenu.setEnabled(false);
diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java
index c3015948..fe7d0978 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);
diff --git a/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java b/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
index e8703340..e79476bb 100644
--- a/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
+++ b/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
@@ -22,6 +22,8 @@ public class MenuBar extends JMenuBar {
 
     public final JMenuItem saveMappingsMenu;
     public final JMenuItem saveMappingsAsMenu;
+    public final JMenuItem saveMappingsOldMenu;
+    public final JMenuItem saveMappingsSrgMenu;
     public final JMenuItem closeMappingsMenu;
 
 
@@ -121,6 +123,38 @@ public class MenuBar extends JMenuBar {
                 this.saveMappingsAsMenu = item;
             }
             {
+                JMenuItem item = new JMenuItem("Save Mappings as Enigma");
+                menu.add(item);
+                item.addActionListener(event -> {
+                    if (this.gui.mappingsFileChooser.showSaveDialog(this.gui.getFrame()) == JFileChooser.APPROVE_OPTION) {
+                        try {
+                            this.gui.getController().saveOldMappings(this.gui.mappingsFileChooser.getSelectedFile());
+                            this.saveMappingsMenu.setEnabled(true);
+                        } catch (IOException ex) {
+                            throw new Error(ex);
+                        }
+                    }
+                });
+                item.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_S, InputEvent.CTRL_DOWN_MASK | InputEvent.SHIFT_DOWN_MASK));
+                this.saveMappingsOldMenu = item;
+            }
+            {
+                JMenuItem item = new JMenuItem("Save Mappings as SRG");
+                menu.add(item);
+                item.addActionListener(event -> {
+                    if (this.gui.mappingsFileChooser.showSaveDialog(this.gui.getFrame()) == JFileChooser.APPROVE_OPTION) {
+                        try {
+                            this.gui.getController().saveSRGMappings(this.gui.mappingsFileChooser.getSelectedFile());
+                            this.saveMappingsMenu.setEnabled(true);
+                        } catch (IOException ex) {
+                            throw new Error(ex);
+                        }
+                    }
+                });
+                item.setAccelerator(KeyStroke.getKeyStroke(KeyEvent.VK_S, InputEvent.CTRL_DOWN_MASK | InputEvent.SHIFT_DOWN_MASK));
+                this.saveMappingsSrgMenu = item;
+            }
+            {
                 JMenuItem item = new JMenuItem("Close Mappings");
                 menu.add(item);
                 item.addActionListener(event -> this.gui.getController().closeMappings());
diff --git a/src/main/java/cuchaz/enigma/mapping/MappingsOldWriter.java b/src/main/java/cuchaz/enigma/mapping/MappingsOldWriter.java
new file mode 100644
index 00000000..8b766a31
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/mapping/MappingsOldWriter.java
@@ -0,0 +1,88 @@
+/*******************************************************************************
+ * Copyright (c) 2015 Jeff Martin.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the GNU Lesser General Public
+ * License v3.0 which accompanies this distribution, and is available at
+ * http://www.gnu.org/licenses/lgpl.html
+ * 
+ * Contributors:
+ *     Jeff Martin - initial API and implementation
+ ******************************************************************************/
+package cuchaz.enigma.mapping;
+
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.io.Writer;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+public class MappingsOldWriter {
+        
+        public void write(Writer out, Mappings mappings) throws IOException {
+                write(new PrintWriter(out), mappings);
+        }
+        
+        public void write(PrintWriter out, Mappings mappings) throws IOException {
+                for (ClassMapping classMapping : sorted(mappings.classes())) {
+                        write(out, classMapping, 0);
+                }
+        }
+        
+        private void write(PrintWriter out, ClassMapping classMapping, int depth) throws IOException {
+                if (classMapping.getDeobfName() == null) {
+                        out.format("%sCLASS %s\n", getIndent(depth), classMapping.getObfFullName());
+                } else {
+                        out.format("%sCLASS %s %s\n", getIndent(depth), classMapping.getObfFullName(), classMapping.getDeobfName());
+                }
+                
+                for (ClassMapping innerClassMapping : sorted(classMapping.innerClasses())) {
+                        write(out, innerClassMapping, depth + 1);
+                }
+                
+                for (FieldMapping fieldMapping : sorted(classMapping.fields())) {
+                        write(out, fieldMapping, depth + 1);
+                }
+                
+                for (MethodMapping methodMapping : sorted(classMapping.methods())) {
+                        write(out, methodMapping, depth + 1);
+                }
+        }
+        
+        private void write(PrintWriter out, FieldMapping fieldMapping, int depth) throws IOException {
+                out.format("%sFIELD %s %s %s\n", getIndent(depth), fieldMapping.getObfName(), fieldMapping.getDeobfName(), fieldMapping.getObfType().toString());
+        }
+        
+        private void write(PrintWriter out, MethodMapping methodMapping, int depth) throws IOException {
+                if (methodMapping.getDeobfName() == null) {
+                        out.format("%sMETHOD %s %s\n", getIndent(depth), methodMapping.getObfName(), methodMapping.getObfSignature());
+                } else {
+                        out.format("%sMETHOD %s %s %s\n", getIndent(depth), methodMapping.getObfName(), methodMapping.getDeobfName(), methodMapping.getObfSignature());
+                }
+                
+                for (ArgumentMapping argumentMapping : sorted(methodMapping.arguments())) {
+                        write(out, argumentMapping, depth + 1);
+                }
+        }
+        
+        private void write(PrintWriter out, ArgumentMapping argumentMapping, int depth) throws IOException {
+                out.format("%sARG %d %s\n", getIndent(depth), argumentMapping.getIndex(), argumentMapping.getName());
+        }
+        
+        private <T extends Comparable<T>> List<T> sorted(Iterable<T> classes) {
+                List<T> out = new ArrayList<T>();
+                for (T t : classes) {
+                        out.add(t);
+                }
+                Collections.sort(out);
+                return out;
+        }
+        
+        private String getIndent(int depth) {
+                StringBuilder buf = new StringBuilder();
+                for (int i = 0; i < depth; i++) {
+                        buf.append("\t");
+                }
+                return buf.toString();
+        }
+}
diff --git a/src/main/java/cuchaz/enigma/mapping/MappingsSRGWriter.java b/src/main/java/cuchaz/enigma/mapping/MappingsSRGWriter.java
new file mode 100644
index 00000000..64da7091
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/mapping/MappingsSRGWriter.java
@@ -0,0 +1,82 @@
+package cuchaz.enigma.mapping;
+
+import cuchaz.enigma.Deobfuscator;
+
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+/**
+ * Created by Mark on 11/08/2016.
+ */
+public class MappingsSRGWriter {
+
+    public void write(File file, Deobfuscator deobfuscator) throws IOException {
+        if(file.exists()){
+           file.delete();
+        }
+        file.createNewFile();
+
+        Mappings mappings = deobfuscator.getMappings();
+
+        FileWriter writer = new FileWriter(file);
+        List<String> fieldMappings = new ArrayList<>();
+        List<String> methodMappings = new ArrayList<>();
+        for (ClassMapping classMapping : sorted(mappings.classes())) {
+            if(classMapping.getDeobfName() == null || classMapping.getObfSimpleName() == null || classMapping.getDeobfName() == null){
+                continue;
+            }
+            writer.write("CL: " + classMapping.getObfSimpleName() + " " + classMapping.getDeobfName());
+            writer.write(System.lineSeparator());
+            for (ClassMapping innerClassMapping : sorted(classMapping.innerClasses())) {
+                if(innerClassMapping.getDeobfName() == null || innerClassMapping.getObfSimpleName() == null || innerClassMapping.getDeobfName() == null){
+                    continue;
+                }
+                String innerClassName = classMapping.getObfSimpleName() + "$" + innerClassMapping.getObfSimpleName().replace("none/", "");
+                String innerDebofClassName = classMapping.getDeobfName() + "$" + innerClassMapping.getDeobfName().replace("none/", "");
+                writer.write("CL: " + innerClassName + " " + classMapping.getDeobfName() + "$" + innerClassMapping.getDeobfName());
+                writer.write(System.lineSeparator());
+                for (FieldMapping fieldMapping : sorted(innerClassMapping.fields())) {
+                    fieldMappings.add("FD: " + innerClassName + "/" + fieldMapping.getObfName() + " " + innerDebofClassName + "/" + fieldMapping.getDeobfName());
+                }
+
+                for (MethodMapping methodMapping : sorted(innerClassMapping.methods())) {
+                    methodMappings.add("MD: " + innerClassName + "/" + methodMapping.getObfName() + " " + methodMapping.getObfSignature().toString().replace("none/", "") + " " + innerDebofClassName + "/" + methodMapping.getDeobfName() + " " + deobfuscator.getTranslator(TranslationDirection.Deobfuscating).translateSignature(methodMapping.getObfSignature()));
+                }
+            }
+
+            for (FieldMapping fieldMapping : sorted(classMapping.fields())) {
+                fieldMappings.add("FD: " + classMapping.getObfFullName().replace("none/", "") + "/" + fieldMapping.getObfName() + " " + classMapping.getDeobfName() + "/" + fieldMapping.getDeobfName());
+            }
+
+            for (MethodMapping methodMapping : sorted(classMapping.methods())) {
+                methodMappings.add("MD: " + classMapping.getObfFullName().replace("none/", "") + "/" + methodMapping.getObfName() + " " + methodMapping.getObfSignature().toString().replace("none/", "") + " " + classMapping.getDeobfName() + "/" + methodMapping.getDeobfName() + " " + deobfuscator.getTranslator(TranslationDirection.Deobfuscating).translateSignature(methodMapping.getObfSignature()));
+            }
+        }
+        for(String fd : fieldMappings){
+            writer.write(fd);
+            writer.write(System.lineSeparator());
+        }
+
+        for(String md : methodMappings){
+            writer.write(md);
+            writer.write(System.lineSeparator());
+        }
+
+
+        writer.close();
+    }
+
+
+    private <T extends Comparable<T>> List<T> sorted(Iterable<T> classes) {
+        List<T> out = new ArrayList<T>();
+        for (T t : classes) {
+            out.add(t);
+        }
+        Collections.sort(out);
+        return out;
+    }
+}