MapleIR plugin is a go!

Author: rcx

mapleir/src/IRDecompiler.java

@@ -1,5 +1,8 @@
+package org.mapleir.jda;
+
 import club.bytecode.the.jda.FileContainer;
 import club.bytecode.the.jda.decompilers.JDADecompiler;
+import org.mapleir.ir.algorithms.BoissinotDestructor;
 import org.mapleir.ir.cfg.ControlFlowGraph;
 import org.mapleir.ir.printer.ClassPrinter;
 import org.mapleir.ir.printer.FieldNodePrinter;
@@ -16,20 +19,21 @@ public String decompileClassNode(FileContainer container, ClassNode cn) {
         TabbedStringWriter sw = new TabbedStringWriter();
         sw.setTabString("  ");
         IPropertyDictionary settings = PropertyHelper.createDictionary();
-        //			settings.put(new BooleanProperty(ASMPrinter.PROP_ACCESS_FLAG_SAFE, true));
-        ClassPrinter cp = new ClassPrinter(sw, settings,
-                new FieldNodePrinter(sw, settings),
-                new MethodNodePrinter(sw, settings) {
-                    @Override
-                    protected ControlFlowGraph getCfg(MethodNode mn) {
-                        return MaplePlugin.cxt.getIRCache().getFor(mn);
-                    }
-
-                });
+        final FieldNodePrinter fieldPrinter = new FieldNodePrinter(sw, settings);
+        final MethodNodePrinter methodPrinter = new MethodNodePrinter(sw, settings) {
+            @Override
+            protected ControlFlowGraph getCfg(MethodNode mn) {
+                ControlFlowGraph cfg = MaplePlugin.cxts.get(container).getIRCache().getFor(mn);
+                BoissinotDestructor.leaveSSA(cfg);
+                cfg.getLocals().realloc(cfg);
+                return cfg;
+            }
+        };
+        ClassPrinter cp = new ClassPrinter(sw, settings, fieldPrinter, methodPrinter);
         cp.print(cn);
         return sw.toString();
     }
-
+    
     @Override
     public void decompileToZip(String zipName) {
 

mapleir/src/MaplePlugin.java

@@ -0,0 +1,54 @@
+package org.mapleir.jda;
+
+import club.bytecode.the.jda.decompilers.bytecode.*;
+import org.mapleir.ir.algorithms.BoissinotDestructor;
+import org.mapleir.ir.cfg.ControlFlowGraph;
+import org.mapleir.ir.cfg.builder.ControlFlowGraphBuilder;
+import org.objectweb.asm.tree.ClassNode;
+import org.objectweb.asm.tree.MethodNode;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Iterator;
+
+public class IRDecompiler extends BytecodeDecompiler {
+    @Override
+    protected MethodNodeDecompiler getMethodNodeDecompiler(PrefixedStringBuilder sb, ClassNode cn, Iterator<MethodNode> it) {
+        return new IRMethodDecompiler(this, sb, it.next(), cn);
+    }
+
+    @Override
+    public void decompileToZip(String zipName) {
+    }
+
+    @Override
+    public String getName() {
+        return "MapleIR";
+    }
+}
+
+class IRMethodDecompiler extends MethodNodeDecompiler {
+    public IRMethodDecompiler(BytecodeDecompiler parent, PrefixedStringBuilder sb, MethodNode mn, ClassNode cn) {
+        super(parent, sb, mn, cn);
+    }
+
+    @Override
+    protected InstructionPrinter getInstructionPrinter(MethodNode m, TypeAndName[] args) {
+        return new IRInstructionPrinter(this, m, args);
+    }
+}
+
+class IRInstructionPrinter extends InstructionPrinter {
+    public IRInstructionPrinter(MethodNodeDecompiler parent, MethodNode m, TypeAndName[] args) {
+        super(parent, m, args);
+    }
+
+    @Override
+    public ArrayList<String> createPrint() {
+        ControlFlowGraph cfg = ControlFlowGraphBuilder.build(mNode);
+        BoissinotDestructor.leaveSSA(cfg);
+        cfg.getLocals().realloc(cfg);
+        String result = cfg.toString();
+        return new ArrayList<>(Arrays.asList(result.split("\n")));
+    }
+}