refactor: rework hook registration

bukkit/src/main/java/net/william278/husktowns/BukkitHuskTowns.java

@@ -42,7 +42,10 @@
 import net.william278.husktowns.database.Database;
 import net.william278.husktowns.events.BukkitEventDispatcher;
 import net.william278.husktowns.hook.*;
-import net.william278.husktowns.hook.WorldGuardHook;
+import net.william278.husktowns.hook.maphooks.BlueMapHook;
+import net.william278.husktowns.hook.maphooks.DynmapHook;
+import net.william278.husktowns.hook.maphooks.Pl3xMapHook;
+import net.william278.husktowns.hook.maphooks.PlanHook;
 import net.william278.husktowns.listener.BukkitListener;
 import net.william278.husktowns.manager.Manager;
 import net.william278.husktowns.network.Broker;
@@ -125,8 +128,8 @@ public class BukkitHuskTowns extends JavaPlugin implements HuskTowns, BukkitTask
     @Nullable
     @Getter(AccessLevel.NONE)
     private Advancement advancements;
-
-    WorldGuardHook worldGuardHook;
+    @Setter
+    private HookManager hookManager;
 
     @TestOnly
     @SuppressWarnings("unused")
@@ -136,62 +139,69 @@ private BukkitHuskTowns(@NotNull JavaPluginLoader loader, @NotNull PluginDescrip
     }
 
     @Override
-    public void onEnable() {
-        // Initialize PaperLib and Adventure
-        this.paperLib = new MorePaperLib(this);
-        this.audiences = BukkitAudiences.create(this);
-
+    public void onLoad() {
         // Load configuration and subsystems
         this.loadConfig();
         if (this.settings.getGeneral().isDoAdvancements()) {
             loadAdvancements();
         }
 
-        // Prepare the database and networking system
-        this.database = this.loadDatabase();
-        if (!database.hasLoaded()) {
-            log(Level.SEVERE, "Failed to load database! Please check your credentials! Disabling plugin...");
-            Bukkit.getPluginManager().disablePlugin(this);
-            return;
-        }
-
-        // Load manager and broker
-        this.manager = new Manager(this);
-        this.broker = this.loadBroker();
-
         // Register hooks
+        this.hookManager = new BukkitHookManager(this);
         final PluginManager plugins = Bukkit.getPluginManager();
         if (settings.getGeneral().isEconomyHook()) {
             if (plugins.getPlugin("Vault") != null) {
-                this.registerHook(new VaultEconomyHook(this));
+                hookManager.registerHook(new VaultEconomyHook(this));
             }
         }
         if (settings.getGeneral().getWebMapHook().isEnabled()) {
             if (plugins.getPlugin("BlueMap") != null) {
-                this.registerHook(new BlueMapHook(this));
+                hookManager.registerHook(new BlueMapHook(this));
             } else if (plugins.getPlugin("dynmap") != null) {
-                this.registerHook(new DynmapHook(this));
+                hookManager.registerHook(new DynmapHook(this));
             } else if (plugins.getPlugin("Pl3xMap") != null) {
-                this.registerHook(new Pl3xMapHook(this));
+                hookManager.registerHook(new Pl3xMapHook(this));
             }
         }
         if (settings.getGeneral().isLuckpermsContextsHook() && plugins.getPlugin("LuckPerms") != null) {
-            this.registerHook(new LuckPermsHook(this));
+            hookManager.registerHook(new LuckPermsHook(this));
         }
         if (settings.getGeneral().isPlaceholderapiHook() && plugins.getPlugin("PlaceholderAPI") != null) {
-            this.registerHook(new PlaceholderAPIHook(this));
+            hookManager.registerHook(new PlaceholderAPIHook(this));
         }
         if (settings.getGeneral().isHuskhomesHook() && plugins.getPlugin("HuskHomes") != null) {
-            this.registerHook(new HuskHomesHook(this));
+            hookManager.registerHook(new HuskHomesHook(this));
         }
         if (settings.getGeneral().isPlanHook() && plugins.getPlugin("Plan") != null) {
-            this.registerHook(new PlanHook(this));
+            hookManager.registerHook(new PlanHook(this));
         }
         if (settings.getGeneral().isWorldGuardHook() && plugins.getPlugin("WorldGuard") != null) {
-            worldGuardHook = new BukkitWorldGuardHook(this);
-            this.registerHook(worldGuardHook);
+            hookManager.registerHook(new BukkitWorldGuardHook(this));
         }
 
+        hookManager.registerOnLoad();
+    }
+
+    @Override
+    public void onEnable() {
+        // Initialize PaperLib and Adventure
+        this.paperLib = new MorePaperLib(this);
+        this.audiences = BukkitAudiences.create(this);
+
+        // Prepare the database and networking system
+        this.database = this.loadDatabase();
+        if (!database.hasLoaded()) {
+            log(Level.SEVERE, "Failed to load database! Please check your credentials! Disabling plugin...");
+            Bukkit.getPluginManager().disablePlugin(this);
+            return;
+        }
+
+        // Load manager and broker
+        this.manager = new Manager(this);
+        this.broker = this.loadBroker();
+
+        hookManager.registerOnEnable();
+
         // Load towns and claim worlds
         this.loadData();
 
@@ -208,6 +218,8 @@ public void onEnable() {
         initializeMetrics();
         log(Level.INFO, "Enabled HuskTowns v" + getVersion());
         checkForUpdates();
+
+        runSyncDelayed(hookManager::registerDelayed, 20L);
     }
 
     @Override
@@ -243,8 +255,8 @@ public Optional<Broker> getMessageBroker() {
     }
 
     @Override
-    public WorldGuardHook getWorldGuardHook() {
-        return worldGuardHook;
+    public @NotNull HookManager getHookManager() {
+        return hookManager;
     }
 
     @Override
@@ -364,7 +376,7 @@ private void initializeMetrics() {
             metrics.addCustomChart(new SimplePie("using_map",
                     () -> getMapHook().isPresent() ? "true" : "false"));
             getMapHook().ifPresent(hook -> metrics.addCustomChart(new SimplePie("map_type",
-                    () -> hook.getName().toLowerCase())));
+                    () -> hook.getHookInfo().id().toLowerCase())));
             getMessageBroker().ifPresent(broker -> metrics.addCustomChart(new SimplePie("messenger_type",
                     () -> settings.getCrossServer().getBrokerType().name().toLowerCase())));
         } catch (Exception e) {

bukkit/src/main/java/net/william278/husktowns/hook/BukkitHookManager.java

@@ -0,0 +1,71 @@
+/*
+ * This file is part of HuskTowns, licensed under the Apache License 2.0.
+ *
+ *  Copyright (c) William278 <will27528@gmail.com>
+ *  Copyright (c) contributors
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+
+package net.william278.husktowns.hook;
+
+import net.william278.husktowns.HuskTowns;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.logging.Level;
+
+public class BukkitHookManager extends HookManager {
+    public BukkitHookManager(@NotNull HuskTowns plugin) {
+        super(plugin);
+    }
+
+    @Override
+    public void registerOnLoad() {
+        plugin.log(Level.INFO, "Loading early hooks...");
+        AtomicInteger loaded = new AtomicInteger();
+        registeredHooks.stream().filter(Hook::isDisabled)
+                .filter((hook) -> hook.getHookInfo().register() == PluginHook.Register.ON_LOAD)
+                .forEach((hook) -> {
+                    hook.enable();
+                    loaded.getAndIncrement();
+                });
+        plugin.log(Level.INFO, "Successfully loaded %s hooks".formatted(loaded.get()));
+    }
+
+    @Override
+    public void registerOnEnable() {
+        plugin.log(Level.INFO, "Loading hooks...");
+        AtomicInteger loaded = new AtomicInteger();
+        registeredHooks.stream().filter(Hook::isDisabled)
+                .filter((hook) -> hook.getHookInfo().register() == PluginHook.Register.ON_ENABLE)
+                .forEach((hook) -> {
+                    hook.enable();
+                    loaded.getAndIncrement();
+                });
+        plugin.log(Level.INFO, "Successfully loaded %s hooks".formatted(loaded.get()));
+    }
+
+    @Override
+    public void registerDelayed() {
+        plugin.log(Level.INFO, "Loading late hooks...");
+        AtomicInteger loaded = new AtomicInteger();
+        registeredHooks.stream().filter(Hook::isDisabled)
+                .filter((hook) -> hook.getHookInfo().register() == PluginHook.Register.DELAYED)
+                .forEach((hook) -> {
+                    hook.enable();
+                    loaded.getAndIncrement();
+                });
+        plugin.log(Level.INFO, "Successfully loaded %s hooks".formatted(loaded.get()));
+    }
+}

bukkit/src/main/java/net/william278/husktowns/hook/BukkitWorldGuardHook.java

@@ -34,6 +34,7 @@
 import org.jetbrains.annotations.NotNull;
 
 public class BukkitWorldGuardHook extends WorldGuardHook {
+    @PluginHook(id = "WorldGuard", register = PluginHook.Register.ON_LOAD, platform = "bukkit")
     public BukkitWorldGuardHook(@NotNull HuskTowns plugin) {
         super(plugin);
     }

bukkit/src/main/java/net/william278/husktowns/hook/HuskHomesHook.java

@@ -44,9 +44,9 @@
 public class HuskHomesHook extends TeleportationHook implements Listener {
     @Nullable
     private HuskHomesAPI api;
-
+    @PluginHook(id = "HuskHomes", register = PluginHook.Register.ON_ENABLE, platform = "bukkit")
     public HuskHomesHook(@NotNull HuskTowns plugin) {
-        super(plugin, "HuskHomes");
+        super(plugin);
     }
 
     @Override

bukkit/src/main/java/net/william278/husktowns/hook/LuckPermsHook.java

@@ -46,12 +46,12 @@
 import java.util.logging.Level;
 
 public class LuckPermsHook extends Hook {
-
     private ContextManager contexts;
     private final List<ContextCalculator<Player>> calculators = new ArrayList<>();
 
+    @PluginHook(id = "LuckPerms", register = PluginHook.Register.ON_ENABLE, platform = "bukkit")
     public LuckPermsHook(@NotNull HuskTowns plugin) {
-        super(plugin, "LuckPerms");
+        super(plugin);
     }
 
     @Override

bukkit/src/main/java/net/william278/husktowns/hook/PlaceholderAPIHook.java

@@ -41,8 +41,9 @@
 import java.util.stream.Collectors;
 
 public class PlaceholderAPIHook extends Hook {
+    @PluginHook(id = "PlaceholderAPI", register = PluginHook.Register.ON_ENABLE, platform = "bukkit")
     public PlaceholderAPIHook(@NotNull HuskTowns plugin) {
-        super(plugin, "PlaceholderAPI");
+        super(plugin);
     }
 
     @Override

bukkit/src/main/java/net/william278/husktowns/hook/VaultEconomyHook.java

@@ -32,11 +32,11 @@
 import java.util.logging.Level;
 
 public class VaultEconomyHook extends EconomyHook {
-
     protected Economy economy;
 
+    @PluginHook(id = "Vault", register = PluginHook.Register.ON_ENABLE, platform = "bukkit")
     public VaultEconomyHook(@NotNull HuskTowns plugin) {
-        super(plugin, "Vault");
+        super(plugin);
     }
 
     @Override

bukkit/src/test/java/net/william278/husktowns/BukkitPluginTests.java

@@ -25,14 +25,12 @@
 import net.william278.husktowns.audit.Action;
 import net.william278.husktowns.audit.Log;
 import net.william278.husktowns.claim.*;
-import net.william278.husktowns.hook.BukkitWorldGuardHook;
 import net.william278.husktowns.map.MapSquare;
 import net.william278.husktowns.town.Town;
 import net.william278.husktowns.user.BukkitUser;
 import net.william278.husktowns.user.OnlineUser;
 import net.william278.husktowns.user.Preferences;
 import org.bukkit.Location;
-import org.bukkit.entity.Husk;
 import org.bukkit.entity.Player;
 import org.jetbrains.annotations.NotNull;
 import org.junit.jupiter.api.*;

common/src/main/java/net/william278/husktowns/HuskTowns.java

@@ -35,7 +35,10 @@
 import net.william278.husktowns.database.MySqlDatabase;
 import net.william278.husktowns.database.SqLiteDatabase;
 import net.william278.husktowns.events.EventDispatcher;
-import net.william278.husktowns.hook.*;
+import net.william278.husktowns.hook.EconomyHook;
+import net.william278.husktowns.hook.HookManager;
+import net.william278.husktowns.hook.MapHook;
+import net.william278.husktowns.hook.TeleportationHook;
 import net.william278.husktowns.listener.OperationHandler;
 import net.william278.husktowns.listener.UserListener;
 import net.william278.husktowns.manager.Manager;
@@ -86,7 +89,8 @@ public interface HuskTowns extends Task.Supplier, ConfigProvider, EventDispatche
     @NotNull
     Map<UUID, Deque<Invite>> getInvites();
 
-    WorldGuardHook getWorldGuardHook();
+    @NotNull
+    HookManager getHookManager();
 
     default void addInvite(@NotNull UUID recipient, @NotNull Invite invite) {
         if (!getInvites().containsKey(recipient)) {
@@ -192,7 +196,6 @@ default void loadData() {
                 log(Level.INFO, String.format("Loaded data in %s seconds.",
                         (ChronoUnit.MILLIS.between(startTime, LocalTime.now()) / 1000d)));
                 setLoaded(true);
-                loadHooks();
             } catch (IllegalStateException e) {
                 setLoaded(false);
                 log(Level.SEVERE, String.format("Failed to load data (after %s seconds). Interaction will be disabled!",
@@ -424,27 +427,8 @@ default void teleportUser(@NotNull OnlineUser user, @NotNull Position position,
 
     double getHighestYAt(double x, double z, @NotNull World world);
 
-    @NotNull
-    Set<Hook> getHooks();
-
-    default void registerHook(@NotNull Hook hook) {
-        getHooks().add(hook);
-    }
-
-    default void loadHooks() {
-        getHooks().stream().filter(Hook::isDisabled).forEach(Hook::enable);
-        log(Level.INFO, "Successfully loaded " + getHooks().size() + " hooks");
-    }
-
-    default <T extends Hook> Optional<T> getHook(@NotNull Class<T> hookClass) {
-        return getHooks().stream()
-                .filter(hook -> hookClass.isAssignableFrom(hook.getClass()))
-                .map(hookClass::cast)
-                .findFirst();
-    }
-
     default Optional<EconomyHook> getEconomyHook() {
-        return getHook(EconomyHook.class);
+        return getHookManager(). getHook(EconomyHook.class);
     }
 
     @NotNull
@@ -457,11 +441,11 @@ default String formatMoney(@NotNull BigDecimal amount) {
     }
 
     default Optional<MapHook> getMapHook() {
-        return getHook(MapHook.class);
+        return getHookManager().getHook(MapHook.class);
     }
 
     default Optional<TeleportationHook> getTeleportationHook() {
-        return getHook(TeleportationHook.class);
+        return getHookManager().getHook(TeleportationHook.class);
     }
 
     void dispatchCommand(@NotNull String command);

common/src/main/java/net/william278/husktowns/hook/EconomyHook.java

@@ -28,8 +28,8 @@
 
 public abstract class EconomyHook extends Hook {
 
-    protected EconomyHook(@NotNull HuskTowns plugin, @NotNull String name) {
-        super(plugin, name);
+    protected EconomyHook(@NotNull HuskTowns plugin) {
+        super(plugin);
     }
 
     /**