Allow exporting mappings as SRG or Enigma

1 files changed, 82 insertions, 0 deletions

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 0000000..64da709
--- /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;
+    }
+}