feat: introduce Extension#afterServerStarted method which will be c…

…alled after the server is started

CHANGELOG.md

@@ -36,6 +36,7 @@ Unless otherwise specified, any version comparison below is the comparison of se
 - Introduced [sentry](https://www.sentry.io) to capture exception and upload them to sentry server automatically, which
   helps us to track and fix bug more efficiently. Sentry is only enabled in non-dev version.
 - Server version will also be uploaded to bStats now.
+- Introduced `Extension#afterServerStarted` method which will be called after the server is started.
 
 ### Changed
 

server/src/main/java/org/allaymc/server/Allay.java

@@ -60,7 +60,7 @@
 public final class Allay {
 
     public static final DynamicURLClassLoader EXTRA_RESOURCE_CLASS_LOADER = new DynamicURLClassLoader(Allay.class.getClassLoader());
-    private static final ExtensionManager EXTENSION_MANAGER = new ExtensionManager(Path.of("extensions"));
+    public static final ExtensionManager EXTENSION_MANAGER = new ExtensionManager(Path.of("extensions"));
 
     public static Dashboard DASHBOARD;
 

server/src/main/java/org/allaymc/server/AllayServer.java

@@ -218,8 +218,9 @@ public void start(long initialTime) {
 
         Metrics.AllayMetrics.startMetrics();
         if (SETTINGS.genericSettings().enableGui()) {
-            Allay.DASHBOARD.serverStarted();
+            Allay.DASHBOARD.afterServerStarted();
         }
+        Allay.EXTENSION_MANAGER.afterServerStarted();
         this.gameLoop.startLoop();
     }
 

server/src/main/java/org/allaymc/server/extension/Extension.java

@@ -5,4 +5,8 @@
  */
 public abstract class Extension {
     public abstract void main(String[] args);
+
+    public void afterServerStarted() {
+        // noop
+    }
 }

server/src/main/java/org/allaymc/server/extension/ExtensionManager.java

@@ -24,6 +24,7 @@ public final class ExtensionManager {
     private static final PathMatcher PATH_MATCHER = FileSystems.getDefault().getPathMatcher("glob:**.jar");
 
     private final Path source;
+    private final List<Extension> extensionInstances = new ArrayList<>();
 
     public ExtensionManager(Path source) {
         this.source = source;
@@ -35,16 +36,19 @@ public void loadExtensions(String[] args) {
             Files.createDirectory(source);
         }
 
-        List<Runnable> entrances = new ArrayList<>();
         try (var stream = Files.walk(source)) {
             stream.filter(path -> PATH_MATCHER.matches(path) && Files.isRegularFile(path))
-                    .forEach(extensionPath -> loadExtension(extensionPath, args, entrances));
+                    .forEach(extensionPath -> loadExtension(extensionPath, args));
         }
 
-        entrances.forEach(Runnable::run);
+        this.extensionInstances.forEach(extension -> extension.main(args));
     }
 
-    private void loadExtension(Path extensionPath, String[] args, List<Runnable> entrances) {
+    public void afterServerStarted() {
+        this.extensionInstances.forEach(Extension::afterServerStarted);
+    }
+
+    private void loadExtension(Path extensionPath, String[] args) {
         log.info(I18n.get().tr(TrKeys.A_EXTENSION_LOADING, extensionPath));
         Allay.EXTRA_RESOURCE_CLASS_LOADER.addJar(extensionPath);
 
@@ -62,11 +66,12 @@ private void loadExtension(Path extensionPath, String[] args, List<Runnable> ent
         Extension extensionInstance;
         try {
             extensionInstance = (Extension) mainClass.getConstructor().newInstance();
+            extensionInstance.main(args);
         } catch (Exception e) {
             throw new ExtensionException(I18n.get().tr(TrKeys.A_EXTENSION_CONSTRUCT_INSTANCE_ERROR, extensionPath, e));
         }
 
-        entrances.add(() -> extensionInstance.main(args));
+        extensionInstances.add(extensionInstance);
     }
 
     @SneakyThrows

server/src/main/java/org/allaymc/server/gui/Dashboard.java

@@ -248,7 +248,7 @@ public void write(byte @NonNull [] b) throws IOException {
         };
     }
 
-    public void serverStarted() {
+    public void afterServerStarted() {
         Runnable guiUpdateTask = () -> {
             // Update ram graph
             final long freeMemory = Runtime.getRuntime().freeMemory();