Add readonly bytecode view. (#443)

5 files changed, 94 insertions, 0 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/analysis/BuiltinPlugin.java b/enigma/src/main/java/cuchaz/enigma/analysis/BuiltinPlugin.java
index 989464d..013c52f 100644
--- a/enigma/src/main/java/cuchaz/enigma/analysis/BuiltinPlugin.java
+++ b/enigma/src/main/java/cuchaz/enigma/analysis/BuiltinPlugin.java
@@ -57,6 +57,7 @@ public final class BuiltinPlugin implements EnigmaPlugin {
         private void registerDecompilerServices(EnigmaPluginContext ctx) {
                 ctx.registerService("enigma:procyon", DecompilerService.TYPE, ctx1 -> Decompilers.PROCYON);
                 ctx.registerService("enigma:cfr", DecompilerService.TYPE, ctx1 -> Decompilers.CFR);
+                ctx.registerService("enigma:bytecode", DecompilerService.TYPE, ctx1 -> Decompilers.BYTECODE);
         }
 
         private static final class EnumFieldNameFindingVisitor extends ClassVisitor {
diff --git a/enigma/src/main/java/cuchaz/enigma/source/Decompilers.java b/enigma/src/main/java/cuchaz/enigma/source/Decompilers.java
index 7d154a6..643ea7a 100644
--- a/enigma/src/main/java/cuchaz/enigma/source/Decompilers.java
+++ b/enigma/src/main/java/cuchaz/enigma/source/Decompilers.java
@@ -1,9 +1,11 @@
 package cuchaz.enigma.source;
 
+import cuchaz.enigma.source.bytecode.BytecodeDecompiler;
 import cuchaz.enigma.source.cfr.CfrDecompiler;
 import cuchaz.enigma.source.procyon.ProcyonDecompiler;
 
 public class Decompilers {
     public static final DecompilerService PROCYON = ProcyonDecompiler::new;
     public static final DecompilerService CFR = CfrDecompiler::new;
+    public static final DecompilerService BYTECODE = BytecodeDecompiler::new;
 }
diff --git a/enigma/src/main/java/cuchaz/enigma/source/bytecode/BytecodeDecompiler.java b/enigma/src/main/java/cuchaz/enigma/source/bytecode/BytecodeDecompiler.java
new file mode 100644
index 0000000..97d2969
--- /dev/null
+++ b/enigma/src/main/java/cuchaz/enigma/source/bytecode/BytecodeDecompiler.java
@@ -0,0 +1,21 @@
+package cuchaz.enigma.source.bytecode;
+
+import cuchaz.enigma.classprovider.ClassProvider;
+import cuchaz.enigma.source.Decompiler;
+import cuchaz.enigma.source.Source;
+import cuchaz.enigma.source.SourceSettings;
+import cuchaz.enigma.translation.mapping.EntryRemapper;
+import org.checkerframework.checker.nullness.qual.Nullable;
+
+public class BytecodeDecompiler implements Decompiler {
+    private final ClassProvider classProvider;
+
+    public BytecodeDecompiler(ClassProvider classProvider, SourceSettings settings) {
+        this.classProvider = classProvider;
+    }
+
+    @Override
+    public Source getSource(String className, @Nullable EntryRemapper remapper) {
+        return new BytecodeSource(classProvider.get(className), remapper);
+    }
+}
diff --git a/enigma/src/main/java/cuchaz/enigma/source/bytecode/BytecodeSource.java b/enigma/src/main/java/cuchaz/enigma/source/bytecode/BytecodeSource.java
new file mode 100644
index 0000000..4364b40
--- /dev/null
+++ b/enigma/src/main/java/cuchaz/enigma/source/bytecode/BytecodeSource.java
@@ -0,0 +1,56 @@
+package cuchaz.enigma.source.bytecode;
+
+import cuchaz.enigma.Enigma;
+import cuchaz.enigma.bytecode.translators.TranslationClassVisitor;
+import cuchaz.enigma.source.Source;
+import cuchaz.enigma.source.SourceIndex;
+import cuchaz.enigma.translation.mapping.EntryRemapper;
+import org.objectweb.asm.tree.ClassNode;
+import org.objectweb.asm.util.TraceClassVisitor;
+
+import java.io.PrintWriter;
+import java.io.StringWriter;
+
+public class BytecodeSource implements Source {
+    private final ClassNode classNode;
+    private final EntryRemapper remapper;
+
+    public BytecodeSource(ClassNode classNode, EntryRemapper remapper) {
+        this.classNode = classNode;
+        this.remapper = remapper;
+    }
+
+    @Override
+    public String asString() {
+        return index().getSource();
+    }
+
+    @Override
+    public Source withJavadocs(EntryRemapper remapper) {
+        return new BytecodeSource(classNode, remapper);
+    }
+
+    @Override
+    public SourceIndex index() {
+        SourceIndex index = new SourceIndex();
+
+        EnigmaTextifier textifier = new EnigmaTextifier(index);
+        StringWriter out = new StringWriter();
+        PrintWriter writer = new PrintWriter(out);
+
+        TraceClassVisitor traceClassVisitor = new TraceClassVisitor(null, textifier, writer);
+
+        ClassNode node = this.classNode;
+
+        if (remapper != null) {
+            ClassNode translatedNode = new ClassNode();
+            node.accept(new TranslationClassVisitor(remapper.getDeobfuscator(), Enigma.ASM_VERSION, translatedNode));
+            node = translatedNode;
+        }
+
+        node.accept(traceClassVisitor);
+        index.setSource(out.toString());
+
+        return index;
+    }
+}
diff --git a/enigma/src/main/java/cuchaz/enigma/source/bytecode/EnigmaTextifier.java b/enigma/src/main/java/cuchaz/enigma/source/bytecode/EnigmaTextifier.java
new file mode 100644
index 0000000..2f3fcf2
--- /dev/null
+++ b/enigma/src/main/java/cuchaz/enigma/source/bytecode/EnigmaTextifier.java
@@ -0,0 +1,14 @@
+package cuchaz.enigma.source.bytecode;
+
+import cuchaz.enigma.Enigma;
+import cuchaz.enigma.source.SourceIndex;
+import org.objectweb.asm.util.Textifier;
+
+public class EnigmaTextifier extends Textifier {
+    private final SourceIndex sourceIndex;
+
+    public EnigmaTextifier(SourceIndex sourceIndex) {
+        super(Enigma.ASM_VERSION);
+        this.sourceIndex = sourceIndex;
+    }
+}