5 files changed, 32 insertions, 2 deletions

diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
index 9faa0d0..1c4549d 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -27,6 +27,7 @@ import java.util.concurrent.ExecutionException;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
+import javax.swing.JFrame;
 import javax.swing.JOptionPane;
 import javax.swing.SwingUtilities;
 
@@ -126,6 +127,11 @@ public class GuiController implements ClientPacketHandler, GuiView, DataInvalida
                 return project;
         }
 
+        @Override
+        public JFrame getFrame() {
+                return gui.getFrame();
+        }
+
         public boolean isDirty() {
                 return project != null && project.getMapper().isDirty();
         }
diff --git a/enigma/src/main/java/cuchaz/enigma/Enigma.java b/enigma/src/main/java/cuchaz/enigma/Enigma.java
index 13640bc..4483ab7 100644
--- a/enigma/src/main/java/cuchaz/enigma/Enigma.java
+++ b/enigma/src/main/java/cuchaz/enigma/Enigma.java
@@ -84,7 +84,7 @@ public class Enigma {
                 ClassProvider classProviderWithFrames = index.indexJar(scope, classProvider, progress);
                 services.get(JarIndexerService.TYPE).forEach(indexer -> indexer.acceptJar(scope, classProviderWithFrames, index));
 
-                EnigmaProject project = new EnigmaProject(this, paths, classProvider, index, Utils.zipSha1(paths.toArray(new Path[0])));
+                EnigmaProject project = new EnigmaProject(this, paths, classProvider, scope, index, Utils.zipSha1(paths.toArray(new Path[0])));
 
                 if (callServices) {
                         for (ProjectService projectService : services.get(ProjectService.TYPE)) {
diff --git a/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java b/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java
index 1e4afa2..3f4a067 100644
--- a/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java
+++ b/enigma/src/main/java/cuchaz/enigma/EnigmaProject.java
@@ -11,6 +11,7 @@ import java.util.Collection;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
+import java.util.Set;
 import java.util.concurrent.atomic.AtomicInteger;
 import java.util.function.Function;
 import java.util.jar.JarEntry;
@@ -58,12 +59,13 @@ public class EnigmaProject implements ProjectView {
         private final ClassProvider classProvider;
         private final JarIndex jarIndex;
         private final byte[] jarChecksum;
+        private final Set<String> projectClasses;
 
         private EntryRemapper mapper;
 
         private final List<DataInvalidationListener> dataInvalidationListeners = new ArrayList<>();
 
-        public EnigmaProject(Enigma enigma, List<Path> jarPaths, ClassProvider classProvider, JarIndex jarIndex, byte[] jarChecksum) {
+        public EnigmaProject(Enigma enigma, List<Path> jarPaths, ClassProvider classProvider, Set<String> projectClasses, JarIndex jarIndex, byte[] jarChecksum) {
                 if (jarChecksum.length != 20) {
                         throw new IllegalArgumentException();
                 }
@@ -73,6 +75,7 @@ public class EnigmaProject implements ProjectView {
                 this.classProvider = classProvider;
                 this.jarIndex = jarIndex;
                 this.jarChecksum = jarChecksum;
+                this.projectClasses = projectClasses;
 
                 this.mapper = EntryRemapper.empty(jarIndex);
         }
@@ -341,6 +344,17 @@ public class EnigmaProject implements ProjectView {
         }
 
         @Override
+        public Collection<String> getProjectClasses() {
+                return projectClasses;
+        }
+
+        @Override
+        @Nullable
+        public ClassNode getBytecode(String className) {
+                return classProvider.get(className);
+        }
+
+        @Override
         public void addDataInvalidationListener(DataInvalidationListener listener) {
                 dataInvalidationListeners.add(listener);
         }
diff --git a/enigma/src/main/java/cuchaz/enigma/api/view/GuiView.java b/enigma/src/main/java/cuchaz/enigma/api/view/GuiView.java
index 15c2dc6..519a8ce 100644
--- a/enigma/src/main/java/cuchaz/enigma/api/view/GuiView.java
+++ b/enigma/src/main/java/cuchaz/enigma/api/view/GuiView.java
@@ -1,5 +1,7 @@
 package cuchaz.enigma.api.view;
 
+import javax.swing.JFrame;
+
 import org.jetbrains.annotations.Nullable;
 
 import cuchaz.enigma.api.view.entry.EntryReferenceView;
@@ -10,4 +12,6 @@ public interface GuiView {
 
         @Nullable
         EntryReferenceView getCursorReference();
+
+        JFrame getFrame();
 }
diff --git a/enigma/src/main/java/cuchaz/enigma/api/view/ProjectView.java b/enigma/src/main/java/cuchaz/enigma/api/view/ProjectView.java
index eae776b..b8760e2 100644
--- a/enigma/src/main/java/cuchaz/enigma/api/view/ProjectView.java
+++ b/enigma/src/main/java/cuchaz/enigma/api/view/ProjectView.java
@@ -4,6 +4,7 @@ import java.util.Collection;
 import java.util.List;
 
 import org.jetbrains.annotations.Nullable;
+import org.objectweb.asm.tree.ClassNode;
 
 import cuchaz.enigma.api.DataInvalidationEvent;
 import cuchaz.enigma.api.DataInvalidationListener;
@@ -12,6 +13,11 @@ import cuchaz.enigma.api.view.entry.EntryView;
 public interface ProjectView {
         <T extends EntryView> T deobfuscate(T entry);
 
+        Collection<String> getProjectClasses();
+
+        @Nullable
+        ClassNode getBytecode(String className);
+
         void addDataInvalidationListener(DataInvalidationListener listener);
 
         default void invalidateData(DataInvalidationEvent.InvalidationType type) {