mirror of
https://github.com/kuitoi/kuitoi-Server.git
synced 2026-04-03 06:26:02 +00:00
santaspeen
1e95a7519b
[+] ev.unregister_by_id [+] Completes for PluginsLoader [~] asyncio.to_thread [~] console.legacy_mode
225 lines
8.8 KiB
Python
225 lines
8.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Developed by KuiToi Dev
|
|
# File modules.EventsSystem
|
|
# Written by: SantaSpeen
|
|
# Version 1.0
|
|
# Licence: FPA
|
|
# (c) kuitoi.su 2023
|
|
import asyncio
|
|
import builtins
|
|
import inspect
|
|
|
|
from core import get_logger
|
|
|
|
|
|
# noinspection PyShadowingBuiltins
|
|
class EventsSystem:
|
|
|
|
def __init__(self):
|
|
# TODO: default events
|
|
self.log = get_logger("EventsSystem")
|
|
self.loop = asyncio.get_event_loop()
|
|
self.as_tasks = []
|
|
self.__events = {
|
|
"onServerStarted": [], # No handler
|
|
"onPlayerSentKey": [], # Only sync, no handler
|
|
"onPlayerAuthenticated": [], # (!) Only sync, With handler
|
|
"onPlayerJoin": [], # (!) With handler
|
|
"onPlayerReady": [], # No handler
|
|
"onChatReceive": [], # (!) With handler
|
|
"onCarSpawn": [], # (!) With handler
|
|
"onCarDelete": [], # (!) With handler (admin allow)
|
|
"onCarEdited": [], # (!) With handler
|
|
"onCarReset": [], # No handler
|
|
"onCarChanged": [], # No handler
|
|
"onCarFocusMove": [], # No handler
|
|
"onSentPing": [], # Only sync, no handler
|
|
"onChangePosition": [], # Only sync, no handler
|
|
"onPlayerDisconnect": [], # No handler
|
|
"onServerStopped": [], # No handler
|
|
"serverTick": [],
|
|
"serverTick_0.5s": [],
|
|
"serverTick_1s": [],
|
|
"serverTick_2s": [],
|
|
"serverTick_3s": [],
|
|
"serverTick_4s": [],
|
|
"serverTick_5s": [],
|
|
"serverTick_10s": [],
|
|
"serverTick_30s": [],
|
|
"serverTick_60s": [],
|
|
}
|
|
self.__async_events = {
|
|
"onServerStarted": [],
|
|
"onPlayerJoin": [],
|
|
"onPlayerReady": [],
|
|
"onChatReceive": [],
|
|
"onCarSpawn": [],
|
|
"onCarDelete": [],
|
|
"onCarEdited": [],
|
|
"onCarReset": [],
|
|
"onCarChanged": [],
|
|
"onCarFocusMove": [],
|
|
"onPlayerDisconnect": [],
|
|
"onServerStopped": [],
|
|
"serverTick": [],
|
|
"serverTick_0.5s": [],
|
|
"serverTick_1s": [],
|
|
"serverTick_2s": [],
|
|
"serverTick_3s": [],
|
|
"serverTick_4s": [],
|
|
"serverTick_5s": [],
|
|
"serverTick_10s": [],
|
|
"serverTick_30s": [],
|
|
"serverTick_60s": [],
|
|
}
|
|
|
|
self.__lua_events = {
|
|
"onInit": [], # onServerStarted
|
|
"onShutdown": [], # onServerStopped
|
|
"onPlayerAuth": [], # onPlayerAuthenticated
|
|
"onPlayerConnecting": [], # No
|
|
"onPlayerJoining": [], # No
|
|
"onPlayerJoin": [], # onPlayerJoin
|
|
"onPlayerDisconnect": [], # onPlayerDisconnect
|
|
"onChatMessage": [], # onChatReceive
|
|
"onVehicleSpawn": [], # onCarSpawn
|
|
"onVehicleEdited": [], # onCarEdited
|
|
"onVehicleDeleted": [], # onCarDelete
|
|
"onVehicleReset": [], # onCarReset
|
|
"onFileChanged": [], # TODO lua onFileChanged
|
|
"onConsoleInput": [], # kt.add_command
|
|
}
|
|
self.register_event = self.register
|
|
|
|
def builtins_hook(self):
|
|
self.log.debug("used builtins_hook")
|
|
builtins.ev = self
|
|
|
|
def unregister_by_id(self, _id):
|
|
self.log.debug(f"unregister_by_id '{_id}'")
|
|
if not isinstance(_id, int):
|
|
return
|
|
s = a = 0
|
|
for k, funcs in self.__events.items():
|
|
for f in funcs:
|
|
if id(f) == _id:
|
|
s += 1
|
|
self.__events[k].remove(f)
|
|
for k, funcs in self.__async_events.items():
|
|
for f in funcs:
|
|
if id(f) == _id:
|
|
a += 1
|
|
self.__async_events[k].remove(f)
|
|
self.log.debug(f"unregister in {s + a} events; S:{s}; A:{a};")
|
|
|
|
def unregister(self, func):
|
|
self.log.debug(f"unregister '{func.__name__}' id: {id(func)}")
|
|
s = a = 0
|
|
for k, funcs in self.__events.items():
|
|
for f in funcs:
|
|
if f == func:
|
|
s += 1
|
|
self.__events[k].remove(func)
|
|
for k, funcs in self.__async_events.items():
|
|
for f in funcs:
|
|
if f == func:
|
|
a += 1
|
|
self.__async_events[k].remove(func)
|
|
self.log.debug(f"unregister in {s + a} events; S:{s}; A:{a};")
|
|
|
|
def is_event(self, event_name):
|
|
return (event_name in self.__async_events.keys() or
|
|
event_name in self.__events.keys() or
|
|
event_name in self.__lua_events.keys())
|
|
|
|
def register(self, event_name, event_func, async_event=False, lua=None, return_id=True):
|
|
self.log.debug(f"register(event_name='{event_name}', event_func='{event_func.__name__}'(id: {id(event_func)}), "
|
|
f"async_event={async_event}, lua_event={lua}):")
|
|
if lua:
|
|
if event_name not in self.__lua_events:
|
|
self.__lua_events.update({str(event_name): [{"func_name": event_func, "lua": lua}]})
|
|
else:
|
|
self.__lua_events[event_name].append({"func_name": event_func, "lua": lua})
|
|
self.log.debug("Register ok")
|
|
return
|
|
|
|
if not callable(event_func):
|
|
self.log.error(i18n.events_not_callable.format(event_name, f"kt.add_event(\"{event_name}\", function)"))
|
|
return
|
|
if async_event or inspect.iscoroutinefunction(event_func):
|
|
if event_name not in self.__async_events:
|
|
self.__async_events[event_name] = []
|
|
self.__async_events[event_name].append(event_func)
|
|
self.log.debug("Register ok")
|
|
else:
|
|
if event_name not in self.__events:
|
|
self.__events[event_name] = []
|
|
self.__events[event_name].append(event_func)
|
|
self.log.debug("Register ok")
|
|
if return_id:
|
|
return id(event_func)
|
|
|
|
async def call_as_events(self, *args, **kwargs):
|
|
return await self.call_async_event(*args, **kwargs) + self.call_event(*args, **kwargs)
|
|
|
|
async def call_async_event(self, event_name, *args, **kwargs):
|
|
if not event_name.startswith("serverTick"):
|
|
self.log.debug(f"Calling async event: '{event_name}'")
|
|
funcs_data = []
|
|
if event_name in self.__async_events.keys():
|
|
for func in self.__async_events[event_name]:
|
|
try:
|
|
event_data = {"event_name": event_name, "args": args, "kwargs": kwargs}
|
|
data = await func(event_data)
|
|
funcs_data.append(data)
|
|
except Exception as e:
|
|
self.log.error(i18n.events_calling_error.format(event_name, func.__name__))
|
|
self.log.exception(e)
|
|
elif not self.is_event(event_name):
|
|
self.log.warning(i18n.events_not_found.format(event_name, "kt.call_event()"))
|
|
|
|
return funcs_data
|
|
|
|
def call_event(self, event_name: str, *args, **kwargs):
|
|
if event_name not in (
|
|
"onChangePosition", "onSentPing", # UDP events
|
|
"_get_player"
|
|
) and not event_name.startswith("serverTick"):
|
|
self.log.debug(f"Calling sync event: '{event_name}'")
|
|
funcs_data = []
|
|
|
|
if event_name in self.__events.keys():
|
|
for func in self.__events[event_name]:
|
|
try:
|
|
event_data = {"event_name": event_name, "args": args, "kwargs": kwargs}
|
|
funcs_data.append(func(event_data))
|
|
except Exception as e:
|
|
self.log.error(i18n.events_calling_error.format(event_name, func.__name__))
|
|
self.log.exception(e)
|
|
elif not self.is_event(event_name):
|
|
self.log.warning(i18n.events_not_found.format(event_name, "kt.call_async_event()"))
|
|
|
|
return funcs_data
|
|
|
|
def call_lua_event(self, event_name, *args):
|
|
self.log.debug(f"Calling lua event: '{event_name}{args}'")
|
|
funcs_data = []
|
|
if event_name in self.__lua_events.keys():
|
|
for data in self.__lua_events[event_name]:
|
|
lua = data['lua']
|
|
func_name = data["func_name"]
|
|
try:
|
|
func = lua.globals()[func_name]
|
|
if not func:
|
|
self.log.warning(i18n.events_lua_function_not_found.format("", func_name))
|
|
continue
|
|
fd = func(*args)
|
|
funcs_data.append(fd)
|
|
except Exception as e:
|
|
self.log.error(i18n.events_lua_calling_error.format(f"{e}", event_name, func_name, f"{args}"))
|
|
elif not self.is_event(event_name):
|
|
self.log.warning(i18n.events_not_found.format(event_name, "ev.call_lua_event(), MP.Trigger<>Event()"))
|
|
|
|
return funcs_data
|