Add ProjectService to listen for the project lifecycle

3 files changed, 37 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 a9795bc..9faa0d0 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -49,6 +49,7 @@ import cuchaz.enigma.analysis.StructureTreeOptions;
 import cuchaz.enigma.api.DataInvalidationEvent;
 import cuchaz.enigma.api.DataInvalidationListener;
 import cuchaz.enigma.api.service.ObfuscationTestService;
+import cuchaz.enigma.api.service.ProjectService;
 import cuchaz.enigma.api.view.GuiView;
 import cuchaz.enigma.api.view.entry.EntryReferenceView;
 import cuchaz.enigma.classhandle.ClassHandle;
@@ -133,11 +134,15 @@ public class GuiController implements ClientPacketHandler, GuiView, DataInvalida
                 this.gui.onStartOpenJar();
 
                 return ProgressDialog.runOffThread(gui.getFrame(), progress -> {
-                        project = enigma.openJars(jarPaths, new ClasspathClassProvider(), progress);
+                        project = enigma.openJars(jarPaths, new ClasspathClassProvider(), progress, false);
                         project.addDataInvalidationListener(this);
                         indexTreeBuilder = new IndexTreeBuilder(project.getJarIndex());
                         chp = new ClassHandleProvider(project, UiConfig.getDecompiler().service);
                         SwingUtilities.invokeLater(() -> {
+                                for (ProjectService projectService : enigma.getServices().get(ProjectService.TYPE)) {
+                                        projectService.onProjectOpen(project);
+                                }
+
                                 gui.onFinishOpenJar(getFileNames(jarPaths));
                                 refreshClasses();
                         });
@@ -152,6 +157,10 @@ public class GuiController implements ClientPacketHandler, GuiView, DataInvalida
         }
 
         public void closeJar() {
+                for (ProjectService projectService : enigma.getServices().get(ProjectService.TYPE)) {
+                        projectService.onProjectClose(project);
+                }
+
                 this.chp.destroy();
                 this.chp = null;
                 this.project = null;
diff --git a/enigma/src/main/java/cuchaz/enigma/Enigma.java b/enigma/src/main/java/cuchaz/enigma/Enigma.java
index 729a114..13640bc 100644
--- a/enigma/src/main/java/cuchaz/enigma/Enigma.java
+++ b/enigma/src/main/java/cuchaz/enigma/Enigma.java
@@ -36,6 +36,7 @@ import cuchaz.enigma.api.service.EnigmaService;
 import cuchaz.enigma.api.service.EnigmaServiceFactory;
 import cuchaz.enigma.api.service.EnigmaServiceType;
 import cuchaz.enigma.api.service.JarIndexerService;
+import cuchaz.enigma.api.service.ProjectService;
 import cuchaz.enigma.classprovider.CachingClassProvider;
 import cuchaz.enigma.classprovider.ClassProvider;
 import cuchaz.enigma.classprovider.CombiningClassProvider;
@@ -71,6 +72,10 @@ public class Enigma {
         }
 
         public EnigmaProject openJars(List<Path> paths, ClassProvider libraryClassProvider, ProgressListener progress) throws IOException {
+                return openJars(paths, libraryClassProvider, progress, true);
+        }
+
+        public EnigmaProject openJars(List<Path> paths, ClassProvider libraryClassProvider, ProgressListener progress, boolean callServices) throws IOException {
                 ClassProvider jarClassProvider = getJarClassProvider(paths);
                 ClassProvider classProvider = new CachingClassProvider(new CombiningClassProvider(jarClassProvider, libraryClassProvider));
                 Set<String> scope = Set.copyOf(jarClassProvider.getClassNames());
@@ -79,7 +84,15 @@ public class Enigma {
                 ClassProvider classProviderWithFrames = index.indexJar(scope, classProvider, progress);
                 services.get(JarIndexerService.TYPE).forEach(indexer -> indexer.acceptJar(scope, classProviderWithFrames, index));
 
-                return new EnigmaProject(this, paths, classProvider, index, Utils.zipSha1(paths.toArray(new Path[0])));
+                EnigmaProject project = new EnigmaProject(this, paths, classProvider, index, Utils.zipSha1(paths.toArray(new Path[0])));
+
+                if (callServices) {
+                        for (ProjectService projectService : services.get(ProjectService.TYPE)) {
+                                projectService.onProjectOpen(project);
+                        }
+                }
+
+                return project;
         }
 
         private ClassProvider getJarClassProvider(List<Path> jars) throws IOException {
diff --git a/enigma/src/main/java/cuchaz/enigma/api/service/ProjectService.java b/enigma/src/main/java/cuchaz/enigma/api/service/ProjectService.java
new file mode 100644
index 0000000..a9deefc
--- /dev/null
+++ b/enigma/src/main/java/cuchaz/enigma/api/service/ProjectService.java
@@ -0,0 +1,13 @@
+package cuchaz.enigma.api.service;
+
+import cuchaz.enigma.api.view.ProjectView;
+
+public interface ProjectService extends EnigmaService {
+        EnigmaServiceType<ProjectService> TYPE = EnigmaServiceType.create("project");
+
+        default void onProjectOpen(ProjectView project) {
+        }
+
+        default void onProjectClose(ProjectView project) {
+        }
+}