add: onApply teleport context callback

alpine-network · Dec 10, 2024 · 76db6ce · 76db6ce
1 parent 1e27127
commit 76db6ce
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 0 deletions.
diff --git a/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportCallbacks.java b/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportCallbacks.java
@@ -14,6 +14,7 @@
 final class TeleportCallbacks {
     private static final Consumer<TeleportContext> NO_OP = ctx -> {};
     private Consumer<TeleportContext> onInit = NO_OP;
+    private Consumer<TeleportContext> onApply = NO_OP;
     private Consumer<TeleportContext> onMove = NO_OP;
     private Consumer<TeleportContext> onCountdown = NO_OP;
     private Consumer<TeleportContext> onTeleport = NO_OP;

diff --git a/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportHandler.java b/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportHandler.java
@@ -18,6 +18,14 @@ public interface TeleportHandler {
      */
     void onInit(@NotNull TeleportContext context);
 
+
+    /**
+     * Applies the teleport process with the given context.
+     *
+     * @param context the context containing information about the teleportation
+     */
+    void onApply(@NotNull TeleportContext context);
+
     /**
      * Handles the countdown phase of the teleport process.
      *

diff --git a/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportManager.java b/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportManager.java
@@ -70,8 +70,16 @@ public void registerHandler(@NotNull TeleportHandler handler) {
     public void initiateTeleport(@NotNull TeleportTask task) {
         TeleportHandler handler = this.getTeleportHandler();
         TeleportContext context = task.createContext(false);
+
+        // Initialize the teleport process (i.e., mark as instantaneous or cancel)
         task.getCallbacks().getOnInit().accept(context);
         handler.onInit(context);
+
+        // Apply the teleportation
+        task.getCallbacks().getOnApply().accept(context);
+        handler.onApply(context);
+
+        // Copy to the task
         task.apply(context);
 
         if (!context.isCancelled()) {

diff --git a/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportTask.java b/src/main/java/co/crystaldev/alpinecore/framework/teleport/TeleportTask.java
@@ -122,6 +122,12 @@ public static final class Builder {
             return this;
         }
 
+        public @NotNull Builder onApply(@NotNull Consumer<TeleportContext> contextConsumer) {
+            Validate.notNull(contextConsumer, "contextConsumer cannot be null");
+            this.callbacks.setOnApply(contextConsumer);
+            return this;
+        }
+
         public @NotNull Builder onMove(@NotNull Consumer<TeleportContext> contextConsumer) {
             Validate.notNull(contextConsumer, "contextConsumer cannot be null");
             this.callbacks.setOnMove(contextConsumer);