From 9e7ecf310ea0f0e3cf650eda036187a2983d6857 Mon Sep 17 00:00:00 2001
From: Felipe Paschoal Bergamo <64669985+felipepasc@users.noreply.github.com>
Date: Thu, 3 Apr 2025 00:03:40 -0300
Subject: [PATCH] feat(api): add setPacketDispatcher method to EntityLibAPI

---
 api/src/main/java/me/tofaa/entitylib/EntityLibAPI.java    | 2 ++
 .../me/tofaa/entitylib/common/AbstractEntityLibAPI.java   | 8 +++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/api/src/main/java/me/tofaa/entitylib/EntityLibAPI.java b/api/src/main/java/me/tofaa/entitylib/EntityLibAPI.java
index 4c3c66e..44614fc 100644
--- a/api/src/main/java/me/tofaa/entitylib/EntityLibAPI.java
+++ b/api/src/main/java/me/tofaa/entitylib/EntityLibAPI.java
@@ -60,6 +60,8 @@ public interface EntityLibAPI<T> {
 
     @NotNull BiConsumer<UUID, PacketWrapper<?>> getPacketDispatcher();
 
+    void setPacketDispatcher(@NotNull BiConsumer<UUID, PacketWrapper<?>> packetDispatcher);
+
     @NotNull EntityContainer getDefaultContainer();
 
 }
diff --git a/common/src/main/java/me/tofaa/entitylib/common/AbstractEntityLibAPI.java b/common/src/main/java/me/tofaa/entitylib/common/AbstractEntityLibAPI.java
index ba98c95..d8577c9 100644
--- a/common/src/main/java/me/tofaa/entitylib/common/AbstractEntityLibAPI.java
+++ b/common/src/main/java/me/tofaa/entitylib/common/AbstractEntityLibAPI.java
@@ -25,7 +25,8 @@ public abstract class AbstractEntityLibAPI<P, T> implements EntityLibAPI<T> {
     protected final APIConfig settings;
     protected final Collection<TickContainer<?, T>> tickContainers;
     protected final EntityContainer defaultEntityContainer = EntityContainer.basic();
-    protected final BiConsumer<UUID, PacketWrapper<?>> packetDispatcher;
+
+    protected BiConsumer<UUID, PacketWrapper<?>> packetDispatcher;
 
     protected AbstractEntityLibAPI(Platform<P> platform, APIConfig settings) {
         this.platform = platform;
@@ -81,6 +82,11 @@ public abstract class AbstractEntityLibAPI<P, T> implements EntityLibAPI<T> {
         return packetDispatcher;
     }
 
+    @Override
+    public void setPacketDispatcher(@NotNull BiConsumer<UUID, PacketWrapper<?>> packetDispatcher) {
+        this.packetDispatcher = packetDispatcher;
+    }
+
     @Override
     public @NotNull Collection<TickContainer<?, T>> getTickContainers() {
         return tickContainers;