Release 0.9.0

bot/modules/game_environment/actions/shutdown.py

@@ -0,0 +1,70 @@
+from bot import loaded_modules_dict
+from os import path, pardir
+from time import sleep, time
+import re
+
+module_name = path.basename(path.normpath(path.join(path.abspath(__file__), pardir, pardir)))
+action_name = path.basename(path.abspath(__file__))[:-3]
+
+
+def main_function(module, event_data, dispatchers_steamid):
+    timeout = 10
+
+    if not module.telnet.add_telnet_command_to_queue("shutdown"):
+        module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
+        return
+
+    poll_is_finished = False
+    # Modern format - no datetime/stardate prefix, just look for "Disconnect"
+    regex = r"Disconnect.*"
+
+    timeout_start = time()
+    while not poll_is_finished and (time() < timeout_start + timeout):
+        sleep(0.5)
+        match = False
+        for match in re.finditer(regex, module.telnet.telnet_buffer):
+            poll_is_finished = True
+
+        if match:
+            module.callback_success(callback_success, module, event_data, dispatchers_steamid, match)
+            return
+
+    module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
+
+
+def callback_success(module, event_data, dispatchers_steamid, match=None):
+    active_dataset = module.dom.data.get("module_game_environment", {}).get("active_dataset", None)
+    module.dom.data.upsert({
+        "module_game_environment": {
+            active_dataset: {
+                "cancel_shutdown": False,
+                "shutdown_in_seconds": None,
+                "force_shutdown": False
+            }
+        }
+    })
+
+
+def callback_fail(module, event_data, dispatchers_steamid):
+    active_dataset = module.dom.data.get("module_game_environment", {}).get("active_dataset", None)
+    module.dom.data.upsert({
+        "module_game_environment": {
+            active_dataset: {
+                "cancel_shutdown": False,
+                "shutdown_in_seconds": None,
+                "force_shutdown": False
+            }
+        }
+    })
+
+
+action_meta = {
+    "description": "Cleanly shuts down the server",
+    "main_function": main_function,
+    "callback_success": callback_success,
+    "callback_fail": callback_fail,
+    "requires_telnet_connection": True,
+    "enabled": True
+}
+
+loaded_modules_dict["module_" + module_name].register_action(action_name, action_meta)