Oops

AsoDesu · Jul 6, 2023 · cdeb107 · cdeb107
1 parent 5b95679
commit cdeb107
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/src/main/java/net/asodev/islandutils/mixins/discord/ConnectionMixin.java b/src/main/java/net/asodev/islandutils/mixins/discord/ConnectionMixin.java
@@ -21,13 +21,16 @@ public abstract class ConnectionMixin {
 
     @Inject(method = "disconnect", at = @At("HEAD"))
     private void disconnect(Component component, CallbackInfo ci) {
-        if (getRemoteAddress() == null) return;
-        String hostName = ((InetSocketAddress) getRemoteAddress()).getHostName();
-        if (hostName == null) return;
-        if (hostName.contains("mccisland.net")) {
-            DiscordPresenceUpdator.started = null;
-            DiscordPresenceUpdator.clear();
-            MccIslandState.setGame(GAME.HUB);
+        SocketAddress remoteAddress = getRemoteAddress();
+        if (remoteAddress == null) return;
+        if (remoteAddress instanceof InetSocketAddress socketAddress) {
+            String hostName = socketAddress.getHostName();
+            if (hostName == null) return;
+            if (hostName.contains("mccisland.net")) {
+                DiscordPresenceUpdator.started = null;
+                DiscordPresenceUpdator.clear();
+                MccIslandState.setGame(GAME.HUB);
+            }
         }
     }