Merge pull request #135 from gegy1000/proposal-tweak

Plugin rework

11 files changed, 86 insertions, 92 deletions

diff --git a/src/test/java/cuchaz/enigma/PackageVisibilityIndexTest.java b/src/test/java/cuchaz/enigma/PackageVisibilityIndexTest.java
index ae5d6d2c..1dc9748b 100644
--- a/src/test/java/cuchaz/enigma/PackageVisibilityIndexTest.java
+++ b/src/test/java/cuchaz/enigma/PackageVisibilityIndexTest.java
@@ -11,13 +11,13 @@
 
 package cuchaz.enigma;
 
-import cuchaz.enigma.analysis.ParsedJar;
+import cuchaz.enigma.analysis.ClassCache;
 import cuchaz.enigma.analysis.index.JarIndex;
 import cuchaz.enigma.analysis.index.PackageVisibilityIndex;
 import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import org.junit.Test;
 
-import java.util.jar.JarFile;
+import java.nio.file.Paths;
 
 import static cuchaz.enigma.TestEntryFactory.newClass;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -35,10 +35,8 @@ public class PackageVisibilityIndexTest {
         private final JarIndex jarIndex;
 
         public PackageVisibilityIndexTest() throws Exception {
-                jarIndex = JarIndex.empty();
-                ParsedJar jar = new ParsedJar(new JarFile("build/test-obf/packageAccess.jar"));
-                jarIndex.indexJar(jar, s -> {
-                });
+                ClassCache classCache = ClassCache.of(Paths.get("build/test-obf/packageAccess.jar"));
+                jarIndex = classCache.index(ProgressListener.none());
         }
 
         @Test
diff --git a/src/test/java/cuchaz/enigma/TestDeobfed.java b/src/test/java/cuchaz/enigma/TestDeobfed.java
index 14b1418d..3ee86cb7 100644
--- a/src/test/java/cuchaz/enigma/TestDeobfed.java
+++ b/src/test/java/cuchaz/enigma/TestDeobfed.java
@@ -11,12 +11,12 @@
 
 package cuchaz.enigma;
 
-import cuchaz.enigma.analysis.ParsedJar;
+import cuchaz.enigma.analysis.ClassCache;
 import cuchaz.enigma.analysis.index.JarIndex;
 import org.junit.BeforeClass;
 import org.junit.Test;
 
-import java.util.jar.JarFile;
+import java.nio.file.Paths;
 
 import static cuchaz.enigma.TestEntryFactory.newClass;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -24,15 +24,16 @@ import static org.hamcrest.Matchers.containsInAnyOrder;
 
 public class TestDeobfed {
 
-        private static ParsedJar jar;
+        private static Enigma enigma;
+        private static ClassCache classCache;
         private static JarIndex index;
 
         @BeforeClass
-        public static void beforeClass()
-                throws Exception {
-                jar = new ParsedJar(new JarFile("build/test-deobf/translation.jar"));
-                index = JarIndex.empty();
-                index.indexJar(jar, s -> {});
+        public static void beforeClass() throws Exception {
+                enigma = Enigma.create();
+
+                classCache = ClassCache.of(Paths.get("build/test-deobf/translation.jar"));
+                index = classCache.index(ProgressListener.none());
         }
 
         @Test
@@ -65,10 +66,12 @@ public class TestDeobfed {
         }
 
         @Test
-        public void decompile()
-                throws Exception {
-                Deobfuscator deobfuscator = new Deobfuscator(jar);
-                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
+        public void decompile() {
+                EnigmaProject project = new EnigmaProject(enigma, classCache, index);
+
+                CompiledSourceTypeLoader typeLoader = new CompiledSourceTypeLoader(project.getClassCache());
+                SourceProvider sourceProvider = new SourceProvider(SourceProvider.createSettings(), typeLoader);
+
                 sourceProvider.getSources("a");
                 sourceProvider.getSources("b");
                 sourceProvider.getSources("c");
diff --git a/src/test/java/cuchaz/enigma/TestDeobfuscator.java b/src/test/java/cuchaz/enigma/TestDeobfuscator.java
index e070b66f..2a6fec49 100644
--- a/src/test/java/cuchaz/enigma/TestDeobfuscator.java
+++ b/src/test/java/cuchaz/enigma/TestDeobfuscator.java
@@ -11,47 +11,31 @@
 
 package cuchaz.enigma;
 
-import com.google.common.collect.Lists;
-import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import org.junit.Test;
 
 import java.io.IOException;
-import java.util.List;
-import java.util.jar.JarFile;
-
-import static org.junit.Assert.assertEquals;
+import java.nio.file.Paths;
 
 public class TestDeobfuscator {
 
-        private Deobfuscator getDeobfuscator()
-                throws IOException {
-                return new Deobfuscator(new JarFile("build/test-obf/loneClass.jar"));
+        private EnigmaProject openProject() throws IOException {
+                Enigma enigma = Enigma.create();
+                return enigma.openJar(Paths.get("build/test-obf/loneClass.jar"), ProgressListener.none());
         }
 
         @Test
         public void loadJar()
                 throws Exception {
-                getDeobfuscator();
+                openProject();
         }
 
         @Test
-        public void getClasses()
-                throws Exception {
-                Deobfuscator deobfuscator = getDeobfuscator();
-                List<ClassEntry> obfClasses = Lists.newArrayList();
-                List<ClassEntry> deobfClasses = Lists.newArrayList();
-                deobfuscator.getSeparatedClasses(obfClasses, deobfClasses);
-                assertEquals(1, obfClasses.size());
-                assertEquals("a", obfClasses.get(0).getName());
-                assertEquals(1, deobfClasses.size());
-                assertEquals("cuchaz/enigma/inputs/Keep", deobfClasses.get(0).getName());
-        }
+        public void decompileClass() throws Exception {
+                EnigmaProject project = openProject();
+
+                CompiledSourceTypeLoader typeLoader = new CompiledSourceTypeLoader(project.getClassCache());
+                SourceProvider sourceProvider = new SourceProvider(SourceProvider.createSettings(), typeLoader);
 
-        @Test
-        public void decompileClass()
-                throws Exception {
-                Deobfuscator deobfuscator = getDeobfuscator();
-                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
                 sourceProvider.writeSourceToString(sourceProvider.getSources("a"));
         }
 }
diff --git a/src/test/java/cuchaz/enigma/TestInnerClasses.java b/src/test/java/cuchaz/enigma/TestInnerClasses.java
index 8738fd79..18e49369 100644
--- a/src/test/java/cuchaz/enigma/TestInnerClasses.java
+++ b/src/test/java/cuchaz/enigma/TestInnerClasses.java
@@ -11,12 +11,12 @@
 
 package cuchaz.enigma;
 
-import cuchaz.enigma.analysis.ParsedJar;
+import cuchaz.enigma.analysis.ClassCache;
 import cuchaz.enigma.analysis.index.JarIndex;
 import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import org.junit.Test;
 
-import java.util.jar.JarFile;
+import java.nio.file.Paths;
 
 import static cuchaz.enigma.TestEntryFactory.newClass;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -33,14 +33,14 @@ public class TestInnerClasses {
         private static final ClassEntry ClassTreeLevel2 = newClass("f$a$a");
         private static final ClassEntry ClassTreeLevel3 = newClass("f$a$a$a");
         private JarIndex index;
-        private Deobfuscator deobfuscator;
-
-        public TestInnerClasses()
-                throws Exception {
-                index = JarIndex.empty();
-                ParsedJar jar = new ParsedJar(new JarFile("build/test-obf/innerClasses.jar"));
-                index.indexJar(jar, s -> {});
-                deobfuscator = new Deobfuscator(jar);
+        private SourceProvider sourceProvider;
+
+        public TestInnerClasses() throws Exception {
+                ClassCache classCache = ClassCache.of(Paths.get("build/test-obf/innerClasses.jar"));
+                index = classCache.index(ProgressListener.none());
+
+                CompiledSourceTypeLoader typeLoader = new CompiledSourceTypeLoader(classCache);
+                sourceProvider = new SourceProvider(SourceProvider.createSettings(), typeLoader);
         }
 
         @Test
@@ -79,6 +79,6 @@ public class TestInnerClasses {
         }
 
         private void decompile(ClassEntry classEntry) {
-                deobfuscator.getObfSourceProvider().getSources(classEntry.getName());
+                sourceProvider.getSources(classEntry.getName());
         }
 }
diff --git a/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java b/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java
index c3f3b669..48975c82 100644
--- a/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java
+++ b/src/test/java/cuchaz/enigma/TestJarIndexConstructorReferences.java
@@ -11,17 +11,16 @@
 
 package cuchaz.enigma;
 
+import cuchaz.enigma.analysis.ClassCache;
 import cuchaz.enigma.analysis.EntryReference;
-import cuchaz.enigma.analysis.ParsedJar;
 import cuchaz.enigma.analysis.index.JarIndex;
 import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import cuchaz.enigma.translation.representation.entry.MethodDefEntry;
 import cuchaz.enigma.translation.representation.entry.MethodEntry;
 import org.junit.Test;
 
-import java.io.File;
+import java.nio.file.Paths;
 import java.util.Collection;
-import java.util.jar.JarFile;
 
 import static cuchaz.enigma.TestEntryFactory.*;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -37,11 +36,9 @@ public class TestJarIndexConstructorReferences {
         private ClassEntry defaultClass = newClass("c");
         private ClassEntry callerClass = newClass("b");
 
-        public TestJarIndexConstructorReferences()
-                        throws Exception {
-                File jarFile = new File("build/test-obf/constructors.jar");
-                index = JarIndex.empty();
-                index.indexJar(new ParsedJar(new JarFile(jarFile)), s -> {});
+        public TestJarIndexConstructorReferences() throws Exception {
+                ClassCache classCache = ClassCache.of(Paths.get("build/test-obf/constructors.jar"));
+                index = classCache.index(ProgressListener.none());
         }
 
         @Test
diff --git a/src/test/java/cuchaz/enigma/TestJarIndexInheritanceTree.java b/src/test/java/cuchaz/enigma/TestJarIndexInheritanceTree.java
index 36595a3b..76e379c3 100644
--- a/src/test/java/cuchaz/enigma/TestJarIndexInheritanceTree.java
+++ b/src/test/java/cuchaz/enigma/TestJarIndexInheritanceTree.java
@@ -11,8 +11,8 @@
 
 package cuchaz.enigma;
 
+import cuchaz.enigma.analysis.ClassCache;
 import cuchaz.enigma.analysis.EntryReference;
-import cuchaz.enigma.analysis.ParsedJar;
 import cuchaz.enigma.analysis.index.EntryIndex;
 import cuchaz.enigma.analysis.index.InheritanceIndex;
 import cuchaz.enigma.analysis.index.JarIndex;
@@ -26,8 +26,8 @@ import cuchaz.enigma.translation.representation.entry.MethodEntry;
 import org.junit.Test;
 import org.objectweb.asm.Opcodes;
 
+import java.nio.file.Paths;
 import java.util.Collection;
-import java.util.jar.JarFile;
 
 import static cuchaz.enigma.TestEntryFactory.*;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -46,8 +46,8 @@ public class TestJarIndexInheritanceTree {
 
         public TestJarIndexInheritanceTree()
                         throws Exception {
-                index = JarIndex.empty();
-                index.indexJar(new ParsedJar(new JarFile("build/test-obf/inheritanceTree.jar")), s -> {});
+                ClassCache classCache = ClassCache.of(Paths.get("build/test-obf/inheritanceTree.jar"));
+                index = classCache.index(ProgressListener.none());
         }
 
         @Test
diff --git a/src/test/java/cuchaz/enigma/TestJarIndexLoneClass.java b/src/test/java/cuchaz/enigma/TestJarIndexLoneClass.java
index 1299bccc..103c366b 100644
--- a/src/test/java/cuchaz/enigma/TestJarIndexLoneClass.java
+++ b/src/test/java/cuchaz/enigma/TestJarIndexLoneClass.java
@@ -23,9 +23,9 @@ import cuchaz.enigma.translation.representation.entry.MethodDefEntry;
 import cuchaz.enigma.translation.representation.entry.MethodEntry;
 import org.junit.Test;
 
+import java.nio.file.Paths;
 import java.util.Collection;
 import java.util.List;
-import java.util.jar.JarFile;
 
 import static cuchaz.enigma.TestEntryFactory.*;
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -35,10 +35,9 @@ public class TestJarIndexLoneClass {
 
         private JarIndex index;
 
-        public TestJarIndexLoneClass()
-                        throws Exception {
-                index = JarIndex.empty();
-                index.indexJar(new ParsedJar(new JarFile("build/test-obf/loneClass.jar")), s -> {});
+        public TestJarIndexLoneClass() throws Exception {
+                ClassCache classCache = ClassCache.of(Paths.get("build/test-obf/loneClass.jar"));
+                index = classCache.index(ProgressListener.none());
         }
 
         @Test
diff --git a/src/test/java/cuchaz/enigma/TestSourceIndex.java b/src/test/java/cuchaz/enigma/TestSourceIndex.java
index ce5d6316..b1d11822 100644
--- a/src/test/java/cuchaz/enigma/TestSourceIndex.java
+++ b/src/test/java/cuchaz/enigma/TestSourceIndex.java
@@ -13,13 +13,15 @@ package cuchaz.enigma;
 
 import com.google.common.collect.Sets;
 import com.strobel.decompiler.languages.java.ast.CompilationUnit;
+import cuchaz.enigma.analysis.ClassCache;
 import cuchaz.enigma.analysis.SourceIndex;
+import cuchaz.enigma.analysis.index.JarIndex;
 import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import org.junit.Test;
 
-import java.io.File;
+import java.nio.file.Path;
+import java.nio.file.Paths;
 import java.util.Set;
-import java.util.jar.JarFile;
 
 public class TestSourceIndex {
         @Test
@@ -27,31 +29,41 @@ public class TestSourceIndex {
                 throws Exception {
                 // Figure out where Minecraft is...
                 final String mcDir = System.getProperty("enigma.test.minecraftdir");
-                File mcJar = null;
+                Path mcJar = null;
                 if (mcDir == null) {
                         String osname = System.getProperty("os.name").toLowerCase();
                         if (osname.contains("nix") || osname.contains("nux") || osname.contains("solaris")) {
-                                mcJar = new File(System.getProperty("user.home"), ".minecraft/versions/1.8.3/1.8.3.jar");
+                                mcJar = Paths.get(System.getProperty("user.home"), ".minecraft/versions/1.8.3/1.8.3.jar");
                         } else if (osname.contains("mac") || osname.contains("darwin")) {
-                                mcJar = new File(System.getProperty("user.home"), "Library/Application Support/minecraft/versions/1.8.3/1.8.3.jar");
+                                mcJar = Paths.get(System.getProperty("user.home"), "Library/Application Support/minecraft/versions/1.8.3/1.8.3.jar");
                         } else if (osname.contains("win")) {
-                                mcJar = new File(System.getenv("AppData"), ".minecraft/versions/1.8.3/1.8.3.jar");
+                                mcJar = Paths.get(System.getenv("AppData"), ".minecraft/versions/1.8.3/1.8.3.jar");
                         }
                 } else {
-                        mcJar = new File(mcDir, "versions/1.8.3/1.8.3.jar");
+                        mcJar = Paths.get(mcDir, "versions/1.8.3/1.8.3.jar");
                 }
 
-                Deobfuscator deobfuscator = new Deobfuscator(new JarFile(mcJar));
+                if (mcJar == null) {
+                        throw new NullPointerException("Couldn't find jar");
+                }
+
+                Enigma enigma = Enigma.create();
+                EnigmaProject project = enigma.openJar(mcJar, ProgressListener.none());
+
+                ClassCache classCache = project.getClassCache();
+                JarIndex index = project.getJarIndex();
+
+                CompiledSourceTypeLoader typeLoader = new CompiledSourceTypeLoader(classCache);
+                SourceProvider sourceProvider = new SourceProvider(SourceProvider.createSettings(), typeLoader);
 
                 // get all classes that aren't inner classes
                 Set<ClassEntry> classEntries = Sets.newHashSet();
-                for (ClassEntry obfClassEntry : deobfuscator.getJarIndex().getEntryIndex().getClasses()) {
+                for (ClassEntry obfClassEntry : index.getEntryIndex().getClasses()) {
                         if (!obfClassEntry.isInnerClass()) {
                                 classEntries.add(obfClassEntry);
                         }
                 }
 
-                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
                 for (ClassEntry obfClassEntry : classEntries) {
                         try {
                                 CompilationUnit tree = sourceProvider.getSources(obfClassEntry.getName());
diff --git a/src/test/java/cuchaz/enigma/TestTokensConstructors.java b/src/test/java/cuchaz/enigma/TestTokensConstructors.java
index 1ee0bde1..0398de4f 100644
--- a/src/test/java/cuchaz/enigma/TestTokensConstructors.java
+++ b/src/test/java/cuchaz/enigma/TestTokensConstructors.java
@@ -14,7 +14,7 @@ package cuchaz.enigma;
 import cuchaz.enigma.translation.representation.entry.MethodEntry;
 import org.junit.Test;
 
-import java.util.jar.JarFile;
+import java.nio.file.Paths;
 
 import static cuchaz.enigma.TestEntryFactory.newBehaviorReferenceByMethod;
 import static cuchaz.enigma.TestEntryFactory.newMethod;
@@ -25,7 +25,7 @@ public class TestTokensConstructors extends TokenChecker {
 
         public TestTokensConstructors()
                         throws Exception {
-                super(new JarFile("build/test-obf/constructors.jar"));
+                super(Paths.get("build/test-obf/constructors.jar"));
         }
 
         @Test
diff --git a/src/test/java/cuchaz/enigma/TestTranslator.java b/src/test/java/cuchaz/enigma/TestTranslator.java
index b9781297..a420afe1 100644
--- a/src/test/java/cuchaz/enigma/TestTranslator.java
+++ b/src/test/java/cuchaz/enigma/TestTranslator.java
@@ -23,7 +23,7 @@ public class TestTranslator {
         public static void beforeClass()
                 throws Exception {
                 //TODO FIx
-                //deobfuscator = new Deobfuscator(new JarFile("build/test-obf/translation.jar"));
+                //deobfuscator = new Enigma(new JarFile("build/test-obf/translation.jar"));
                 //try (InputStream in = TestTranslator.class.getResourceAsStream("/cuchaz/enigma/resources/translation.mappings")) {
                 //      mappings = new MappingsJsonReader().read(new InputStreamReader(in));
                 //      deobfuscator.setMappings(mappings);
diff --git a/src/test/java/cuchaz/enigma/TokenChecker.java b/src/test/java/cuchaz/enigma/TokenChecker.java
index c4670a20..1dde0349 100644
--- a/src/test/java/cuchaz/enigma/TokenChecker.java
+++ b/src/test/java/cuchaz/enigma/TokenChecker.java
@@ -13,28 +13,30 @@ package cuchaz.enigma;
 
 import com.google.common.collect.Lists;
 import com.strobel.decompiler.languages.java.ast.CompilationUnit;
+import cuchaz.enigma.analysis.ClassCache;
 import cuchaz.enigma.analysis.EntryReference;
 import cuchaz.enigma.analysis.SourceIndex;
 import cuchaz.enigma.analysis.Token;
 import cuchaz.enigma.translation.representation.entry.Entry;
 
 import java.io.IOException;
+import java.nio.file.Path;
 import java.util.Collection;
 import java.util.List;
-import java.util.jar.JarFile;
 
 public class TokenChecker {
 
-        private Deobfuscator deobfuscator;
+        private SourceProvider sourceProvider;
 
-        protected TokenChecker(JarFile jarFile)
-                throws IOException {
-                deobfuscator = new Deobfuscator(jarFile);
+        protected TokenChecker(Path path) throws IOException {
+                ClassCache classCache = ClassCache.of(path);
+
+                CompiledSourceTypeLoader typeLoader = new CompiledSourceTypeLoader(classCache);
+                sourceProvider = new SourceProvider(SourceProvider.createSettings(), typeLoader);
         }
 
         protected String getDeclarationToken(Entry<?> entry) {
                 // decompile the class
-                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
                 CompilationUnit tree = sourceProvider.getSources(entry.getContainingClass().getFullName());
                 // DEBUG
                 // tree.acceptVisitor( new TreeDumpVisitor( new File( "tree." + entry.getClassName().replace( '/', '.' ) + ".txt" ) ), null );
@@ -52,7 +54,6 @@ public class TokenChecker {
         @SuppressWarnings("unchecked")
         protected Collection<String> getReferenceTokens(EntryReference<? extends Entry<?>, ? extends Entry<?>> reference) {
                 // decompile the class
-                SourceProvider sourceProvider = deobfuscator.getObfSourceProvider();
                 CompilationUnit tree = sourceProvider.getSources(reference.context.getContainingClass().getFullName());
                 String source = sourceProvider.writeSourceToString(tree);
                 SourceIndex index = SourceIndex.buildIndex(source, tree, true);