71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
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)
|