From 4849a7be5827ddf35409a93e94c41b72fd19bc3b Mon Sep 17 00:00:00 2001
From: Waterpicker <waterpickerenternity@gmail.com>
Date: Mon, 27 Jun 2022 03:14:41 -0500
Subject: [PATCH] Screen stuff.

---
 src/main/java/net/montoyo/wd/net/Messages.java              | 1 +
 .../java/net/montoyo/wd/net/client/CMessageOpenGui.java     | 6 +++---
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/main/java/net/montoyo/wd/net/Messages.java b/src/main/java/net/montoyo/wd/net/Messages.java
index e035a36..e249e9d 100644
--- a/src/main/java/net/montoyo/wd/net/Messages.java
+++ b/src/main/java/net/montoyo/wd/net/Messages.java
@@ -43,5 +43,6 @@ public class Messages {
         INSTANCE.registerMessage(index++, SMessageRequestTEData.class, SMessageRequestTEData::encode, SMessageRequestTEData::decode, SMessageRequestTEData::handle);
         INSTANCE.registerMessage(index++, SMessageScreenCtrl.class, SMessageScreenCtrl::encode, SMessageScreenCtrl::decode, SMessageScreenCtrl::handle);
         INSTANCE.registerMessage(index++, SMessagePadCtrl.class, SMessagePadCtrl::encode, SMessagePadCtrl::decode, SMessagePadCtrl::handle);
+        INSTANCE.registerMessage(index++, CMessageOpenGui.class, CMessageOpenGui::encode, CMessageOpenGui::decode, CMessageOpenGui::handle);
     }
 }
diff --git a/src/main/java/net/montoyo/wd/net/client/CMessageOpenGui.java b/src/main/java/net/montoyo/wd/net/client/CMessageOpenGui.java
index dc82c29..faef23e 100644
--- a/src/main/java/net/montoyo/wd/net/client/CMessageOpenGui.java
+++ b/src/main/java/net/montoyo/wd/net/client/CMessageOpenGui.java
@@ -21,16 +21,16 @@ public class CMessageOpenGui implements Runnable {
         this.data = data;
     }
 
-    public void decode(FriendlyByteBuf buf) {
+    public static CMessageOpenGui decode(FriendlyByteBuf buf) {
         String name = buf.readUtf();
         Class<? extends GuiData> cls = GuiData.classOf(name);
 
         if(cls == null) {
             Log.error("Could not create GuiData of type %s because it doesn't exist!", name);
-            return;
+            return null;
         }
 
-        data = (GuiData) Util.unserialize(buf, cls);
+        return new CMessageOpenGui((GuiData) Util.unserialize(buf, cls));
     }
 
     public void encode(FriendlyByteBuf buf) {