1 files changed, 101 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/mapping/MappingsJsonWriter.java b/src/main/java/cuchaz/enigma/mapping/MappingsJsonWriter.java
new file mode 100644
index 0000000..db95322
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/mapping/MappingsJsonWriter.java
@@ -0,0 +1,101 @@
+/*******************************************************************************
+ * 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
+ * <p>
+ * Contributors:
+ * Jeff Martin - initial API and implementation
+ ******************************************************************************/
+package cuchaz.enigma.mapping;
+
+import com.google.common.base.Charsets;
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+
+import java.io.*;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+import cuchaz.enigma.json.*;
+
+public class MappingsJsonWriter {
+
+    public void write(File file, Mappings mappings) throws IOException {
+        if (!file.isDirectory()) {
+            return;
+        }
+
+        String[] entries = file.list();
+        for (String s : entries) {
+            File currentFile = new File(file.getPath(), s);
+            deleteDirectory(currentFile);
+        }
+
+        Gson gson = new GsonBuilder().setPrettyPrinting().create();
+        for (ClassMapping classMapping : sorted(mappings.classes())) {
+            if (classMapping.getDeobfName() != null && !classMapping.getDeobfName().equalsIgnoreCase("") && !classMapping.getDeobfName().equalsIgnoreCase("null")) {
+                JsonClass jsonClass = new JsonClass(classMapping.getObfSimpleName(), classMapping.getDeobfName());
+                write(jsonClass, classMapping);
+
+                File f = new File(file, jsonClass.getName() + ".json");
+                f.getParentFile().mkdirs();
+                f.createNewFile();
+                PrintWriter writer = new PrintWriter(new OutputStreamWriter(new FileOutputStream(f), Charsets.UTF_8));
+                writer.write(gson.toJson(jsonClass));
+                writer.close();
+            }
+        }
+    }
+
+    private void write(JsonClass jsonClass, ClassMapping classMapping) {
+        for (ClassMapping innerClassMapping : sorted(classMapping.innerClasses())) {
+            JsonClass innerClass = new JsonClass(classMapping.getObfSimpleName() + "$" + innerClassMapping.getObfSimpleName().replace("nome/", ""), innerClassMapping.getDeobfName());
+            write(innerClass, innerClassMapping);
+            jsonClass.addInnerClass(innerClass);
+        }
+
+        for (FieldMapping fieldMapping : sorted(classMapping.fields())) {
+            jsonClass.addField(new JsonField(fieldMapping.getObfName(), fieldMapping.getDeobfName(), fieldMapping.getObfType().toString()));
+        }
+
+        for (MethodMapping methodMapping : sorted(classMapping.methods())) {
+            List<JsonArgument> args = new ArrayList<>();
+            for (ArgumentMapping argumentMapping : sorted(methodMapping.arguments())) {
+                args.add(new JsonArgument(argumentMapping.getIndex(), argumentMapping.getName()));
+            }
+            if (methodMapping.getObfName().contains("<init>") || methodMapping.getObfName().contains("<clinit>")) {
+                jsonClass.addConstructor(new JsonConstructor(methodMapping.getObfSignature().toString(), args, methodMapping.getObfName().contains("<clinit>")));
+            } else {
+                jsonClass.addMethod(new JsonMethod(methodMapping.getObfName(), methodMapping.getDeobfName(), methodMapping.getObfSignature().toString(), args));
+            }
+        }
+    }
+
+    private <T extends Comparable<T>> List<T> sorted(Iterable<T> classes) {
+        List<T> out = new ArrayList<>();
+        for (T t : classes) {
+            out.add(t);
+        }
+        Collections.sort(out);
+        return out;
+    }
+
+    public static boolean deleteDirectory(File directory) {
+        if (directory.exists()) {
+            File[] files = directory.listFiles();
+            if (null != files) {
+                for (int i = 0; i < files.length; i++) {
+                    if (files[i].isDirectory()) {
+                        deleteDirectory(files[i]);
+                    } else {
+                        files[i].delete();
+                    }
+                }
+            }
+        }
+        return (directory.delete());
+    }
+}