From 589c84117e1742d5963c3967e3f3f54d66cfb152 Mon Sep 17 00:00:00 2001
From: Alex <alex@roese.dev>
Date: Mon, 27 Dec 2021 23:36:48 +0100
Subject: [PATCH] auto restart system

---
 .../proxy/listener/SocketClientMessageListener.java        | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java b/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java
index 5f264ac..1399e6c 100644
--- a/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java
+++ b/src/main/java/net/krakatoa/proxy/listener/SocketClientMessageListener.java
@@ -73,6 +73,10 @@ public class SocketClientMessageListener implements KraSocketClientEventInterfac
             ServiceTask serviceTask = CloudNetDriver.getInstance().getServiceTaskProvider()
                 .getServiceTask("Lobby");
 
+            if (!playersAutoRestartServerSystem.containsKey("lobby")) {
+              playersAutoRestartServerSystem.put("lobby", null);
+            }
+
             if (serviceTask != null) {
               ServiceInfoSnapshot serviceInfoSnapshot = ServiceConfiguration.builder(serviceTask)
                   .build().createNewService();
@@ -110,7 +114,8 @@ public class SocketClientMessageListener implements KraSocketClientEventInterfac
       case 59739: { // send players back to server - auto restart system
         String serverName = kraProtocolMessage.getArgs().toLowerCase();
 
-        if (serverName.startsWith("lobby")) { // close old lobbies
+        if (serverName.startsWith("lobby") && playersAutoRestartServerSystem.containsKey(
+            "lobby")) { // close old lobbies
           Collection<ServiceInfoSnapshot> serviceInfoSnapshots = CloudNetDriver.getInstance()
               .getCloudServiceProvider()
               .getCloudServices("Lobby");