From 6903a0b04efea17c449bb022e2fa0be29cdcb43f Mon Sep 17 00:00:00 2001
From: yoyosource <yoyosource@nidido.de>
Date: Sun, 26 Dec 2021 15:18:30 +0100
Subject: [PATCH] Fix ProtocolAPI

Signed-off-by: yoyosource <yoyosource@nidido.de>
---
 .../steamwar/bausystem/utils/ProtocolAPI.java | 21 +++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/BauSystem_Main/src/de/steamwar/bausystem/utils/ProtocolAPI.java b/BauSystem_Main/src/de/steamwar/bausystem/utils/ProtocolAPI.java
index bfa5bda4..d286f021 100644
--- a/BauSystem_Main/src/de/steamwar/bausystem/utils/ProtocolAPI.java
+++ b/BauSystem_Main/src/de/steamwar/bausystem/utils/ProtocolAPI.java
@@ -36,7 +36,23 @@ public class ProtocolAPI {
 
     public static final TinyProtocol tinyProtocol = TinyProtocol.instance;
 
-    public static void setOutgoingHandler(Class<?> packetClass, BiFunction<Player, Object, Object> handler) {
+    static {
+        TinyProtocol.instance.setOutFilter((receiver, channel, packet) -> {
+            BiFunction<Player, Object, Object> handler = outgoingHandler.get(packet.getClass());
+            if (handler == null)
+                return packet;
+            return handler.apply(receiver, packet);
+        });
+        TinyProtocol.instance.setInFilter((sender, channel, packet) -> {
+            BiFunction<Player, Object, Object> handler = incomingHandler.get(packet.getClass());
+            if (handler == null)
+                return packet;
+            return handler.apply(sender, packet);
+        });
+    }
+
+    public static void setOutgoingHandler
+            (Class<?> packetClass, BiFunction<Player, Object, Object> handler) {
         outgoingHandler.put(packetClass, handler);
     }
 
@@ -44,7 +60,8 @@ public class ProtocolAPI {
         outgoingHandler.remove(packetClass);
     }
 
-    public static void setIncomingHandler(Class<?> packetClass, BiFunction<Player, Object, Object> handler) {
+    public static void setIncomingHandler
+            (Class<?> packetClass, BiFunction<Player, Object, Object> handler) {
         incomingHandler.put(packetClass, handler);
     }