137 lines
5.0 KiB
Python
137 lines
5.0 KiB
Python
from bot import loaded_modules_dict
|
|
from bot.constants import TELNET_TIMEOUT_NORMAL
|
|
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=None):
|
|
timeout = TELNET_TIMEOUT_NORMAL
|
|
timeout_start = time()
|
|
event_data[1]["action_identifier"] = action_name
|
|
|
|
if not module.telnet.add_telnet_command_to_queue("listents"):
|
|
module.callback_fail(callback_fail, module, event_data, dispatchers_steamid)
|
|
return
|
|
|
|
poll_is_finished = False
|
|
# Modern format - matches both empty and populated entity lists
|
|
regex = (
|
|
r"Executing\scommand\s\'listents\'\sby\sTelnet\sfrom\s(?P<called_by>.*?)\r?\n"
|
|
r"(?P<raw_entity_data>[\s\S]*?)"
|
|
r"Total\sof\s(?P<entity_count>\d{1,3})\sin\sthe\sgame"
|
|
)
|
|
|
|
while not poll_is_finished and (time() < timeout_start + timeout):
|
|
sleep(0.25)
|
|
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):
|
|
raw_entity_data = match.group("raw_entity_data").lstrip()
|
|
if len(raw_entity_data) >= 1:
|
|
regex = (
|
|
r"\d{1,2}. id=(?P<id>\d+), \["
|
|
r"type=(?P<type>.+), "
|
|
r"name=(?P<name>.*), "
|
|
r"id=(\d+)"
|
|
r"\], "
|
|
r"pos=\((?P<pos_x>.?\d+.\d), (?P<pos_y>.?\d+.\d), (?P<pos_z>.?\d+.\d)\), "
|
|
r"rot=\((?P<rot_x>.?\d+.\d), (?P<rot_y>.?\d+.\d), (?P<rot_z>.?\d+.\d)\), "
|
|
r"lifetime=(?P<lifetime>.+), "
|
|
r"remote=(?P<remote>.+), "
|
|
r"dead=(?P<dead>.+), "
|
|
r"health=(?P<health>\d+)"
|
|
r"\r\n"
|
|
)
|
|
entities_to_update_dict = {}
|
|
|
|
active_dataset = module.dom.data.get("module_game_environment", {}).get("active_dataset", None)
|
|
if active_dataset is None:
|
|
return
|
|
|
|
for m in re.finditer(regex, raw_entity_data):
|
|
last_seen_gametime_string = module.game_environment.get_last_recorded_gametime_string()
|
|
entity_dict = {
|
|
"id": m.group("id"),
|
|
"owner": m.group("id"),
|
|
"identifier": m.group("id"),
|
|
"type": str(m.group("type")),
|
|
"name": str(m.group("name")),
|
|
"pos": {
|
|
"x": int(float(m.group("pos_x"))),
|
|
"y": int(float(m.group("pos_y"))),
|
|
"z": int(float(m.group("pos_z"))),
|
|
},
|
|
"rot": {
|
|
"x": int(float(m.group("rot_x"))),
|
|
"y": int(float(m.group("rot_y"))),
|
|
"z": int(float(m.group("rot_z"))),
|
|
},
|
|
"lifetime": str(m.group("lifetime")),
|
|
"remote": bool(m.group("remote")),
|
|
"dead": bool(m.group("dead")),
|
|
"health": int(m.group("health")),
|
|
"dataset": active_dataset,
|
|
"last_seen_gametime": last_seen_gametime_string
|
|
}
|
|
entities_to_update_dict[m.group("id")] = entity_dict
|
|
|
|
if len(entities_to_update_dict) >= 1:
|
|
module.dom.data.upsert({
|
|
module.get_module_identifier(): {
|
|
"elements": {
|
|
active_dataset: entities_to_update_dict
|
|
}
|
|
}
|
|
})
|
|
|
|
stuff_to_delete = []
|
|
for path, dom_element_key, dom_element in module.dom.get_dom_element_by_query(
|
|
target_module=module.get_module_identifier(),
|
|
query="id"
|
|
):
|
|
# Delete entities that are no longer in the update or have health <= 0 (dead)
|
|
entity_data = entities_to_update_dict.get(dom_element, {})
|
|
health = entity_data.get("health", 0)
|
|
if dom_element not in entities_to_update_dict or health <= 0:
|
|
stuff_to_delete.append([module.get_module_identifier(), "elements"] + path)
|
|
|
|
for dom_element_to_delete in stuff_to_delete:
|
|
module.dom.data.remove_key_by_path(
|
|
dom_element_to_delete,
|
|
dispatchers_steamid=dispatchers_steamid
|
|
)
|
|
|
|
|
|
def callback_fail(module, event_data, dispatchers_steamid):
|
|
pass
|
|
|
|
|
|
def skip_it(module, event_data, dispatchers_steamid=None):
|
|
pass
|
|
|
|
|
|
action_meta = {
|
|
"description": "get game entities",
|
|
"main_function": main_function,
|
|
"callback_success": callback_success,
|
|
"callback_fail": callback_fail,
|
|
"skip_it": skip_it,
|
|
"requires_telnet_connection": True,
|
|
"enabled": True
|
|
}
|
|
|
|
loaded_modules_dict["module_" + module_name].register_action(action_name, action_meta)
|