shutdown on disconnect

nothub · Apr 6, 2024 · 0fb5943 · 0fb5943
1 parent 4d711cd
commit 0fb5943
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/main/java/lol/hub/headlessbot/Mod.java b/src/main/java/lol/hub/headlessbot/Mod.java
@@ -59,22 +59,23 @@ public void onInitializeClient() {
         }
 
         ScreenEvents.AFTER_INIT.register((client, screen, scaledWidth, scaledHeight) -> {
-            if (screen instanceof AccessibilityOnboardingScreen sc) {
+            if (screen instanceof AccessibilityOnboardingScreen) {
                 Log.info("Closing AccessibilityOnboardingScreen");
                 screen.close();
-            } else if (screen instanceof MultiplayerWarningScreen sc) {
+            } else if (screen instanceof MultiplayerWarningScreen) {
                 Log.info("Closing MultiplayerWarningScreen");
                 screen.close();
             } else if (screen instanceof ConnectScreen) {
                 Log.info("Connecting ...");
             } else if (screen instanceof DisconnectedScreen) {
-                Log.info("Disconnecting ...");
+                Log.info("Disconnected, shutting down ...");
+                System.exit(0);
             } else if (screen instanceof DownloadingTerrainScreen) {
                 Log.info("Downloading terrain ...");
             } else if (screen instanceof ProgressScreen) {
                 Log.info("Awaiting progress ...");
-            } else if (screen instanceof MessageScreen sc) {
-                Log.info("Showing mssage: %s", sc.getTitle().getString());
+            } else if (screen instanceof MessageScreen) {
+                Log.info("Showing mssage: %s", screen.getTitle().getString());
             } else {
                 Log.info("Showing screen: %s", screen.getClass().getSimpleName());
             }