Fix getting a server address

dima-dencep · Jul 5, 2023 · 475cb05 · 475cb05
1 parent cb4e1a1
commit 475cb05
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/common/src/main/java/com/github/dima_dencep/mods/online_emotes/utils/EmotePacketWrapper.java b/common/src/main/java/com/github/dima_dencep/mods/online_emotes/utils/EmotePacketWrapper.java
@@ -4,6 +4,7 @@
 import com.google.gson.Gson;
 import com.google.gson.GsonBuilder;
 import io.netty.buffer.Unpooled;
+import io.netty.channel.local.LocalAddress;
 import io.netty.handler.codec.http.websocketx.BinaryWebSocketFrame;
 import net.minecraft.client.network.ClientPlayerEntity;
 import org.jetbrains.annotations.Nullable;
@@ -46,6 +47,13 @@ public BinaryWebSocketFrame toWebSocketFrame() {
     }
 
     private static String getIP(SocketAddress address) {
-        return ((InetSocketAddress) address).getAddress().getHostAddress();
+        if (address instanceof LocalAddress) {
+            return "localhost";
+
+        } else if (address instanceof InetSocketAddress inetSocketAddress) {
+            return inetSocketAddress.getAddress().getHostAddress();
+        }
+
+        return address.toString();
     }
 }