228 lines
7.3 KiB
Python
228 lines
7.3 KiB
Python
import traceback
|
||
from threading import Thread, Event
|
||
from typing import Any, Union
|
||
|
||
from app.chain import ChainBase
|
||
from app.chain.cookiecloud import CookieCloudChain
|
||
from app.chain.douban import DoubanChain
|
||
from app.chain.download import DownloadChain
|
||
from app.chain.site import SiteChain
|
||
from app.chain.subscribe import SubscribeChain
|
||
from app.chain.transfer import TransferChain
|
||
from app.core.event import eventmanager, EventManager
|
||
from app.core.plugin import PluginManager
|
||
from app.core.event import Event as ManagerEvent
|
||
from app.log import logger
|
||
from app.utils.object import ObjectUtils
|
||
from app.utils.singleton import Singleton
|
||
from app.schemas.types import EventType
|
||
|
||
|
||
class CommandChian(ChainBase):
|
||
"""
|
||
插件处理链
|
||
"""
|
||
|
||
def process(self, *args, **kwargs):
|
||
pass
|
||
|
||
|
||
class Command(metaclass=Singleton):
|
||
"""
|
||
全局命令管理，消费事件
|
||
"""
|
||
# 内建命令
|
||
_commands = {}
|
||
|
||
# 退出事件
|
||
_event = Event()
|
||
|
||
def __init__(self):
|
||
# 事件管理器
|
||
self.eventmanager = EventManager()
|
||
# 插件管理器
|
||
self.pluginmanager = PluginManager()
|
||
# 内置命令
|
||
self._commands = {
|
||
"/cookiecloud": {
|
||
"func": CookieCloudChain().remote_sync,
|
||
"description": "同步站点",
|
||
"data": {}
|
||
},
|
||
"/sites": {
|
||
"func": SiteChain().remote_list,
|
||
"description": "查询站点",
|
||
"data": {}
|
||
},
|
||
"/site_cookie": {
|
||
"func": SiteChain().remote_cookie,
|
||
"description": "更新站点Cookie",
|
||
"data": {}
|
||
},
|
||
"/site_enable": {
|
||
"func": SiteChain().remote_enable,
|
||
"description": "启用站点",
|
||
"data": {}
|
||
},
|
||
"/site_disable": {
|
||
"func": SiteChain().remote_disable,
|
||
"description": "禁用站点",
|
||
"data": {}
|
||
},
|
||
"/douban_sync": {
|
||
"func": DoubanChain().remote_sync,
|
||
"description": "同步豆瓣想看",
|
||
"data": {}
|
||
},
|
||
"/subscribes": {
|
||
"func": SubscribeChain().remote_list,
|
||
"description": "查询订阅",
|
||
"data": {}
|
||
},
|
||
"/subscribe_refresh": {
|
||
"func": SubscribeChain().remote_refresh,
|
||
"description": "刷新订阅",
|
||
"data": {}
|
||
},
|
||
"/subscribe_search": {
|
||
"func": SubscribeChain().remote_search,
|
||
"description": "搜索订阅",
|
||
"data": {}
|
||
},
|
||
"/subscribe_delete": {
|
||
"func": SubscribeChain().remote_delete,
|
||
"description": "删除订阅",
|
||
"data": {}
|
||
},
|
||
"/downloading": {
|
||
"func": DownloadChain().remote_downloading,
|
||
"description": "正在下载",
|
||
"data": {}
|
||
},
|
||
"/transfer": {
|
||
"func": TransferChain().process,
|
||
"description": "下载文件整理",
|
||
"data": {}
|
||
}
|
||
}
|
||
# 汇总插件命令
|
||
plugin_commands = self.pluginmanager.get_plugin_commands()
|
||
for command in plugin_commands:
|
||
self.register(
|
||
cmd=command.get('cmd'),
|
||
func=Command.send_plugin_event,
|
||
desc=command.get('desc'),
|
||
data={
|
||
'etype': command.get('event'),
|
||
'data': command.get('data')
|
||
}
|
||
)
|
||
# 处理链
|
||
self.chain = CommandChian()
|
||
# 广播注册命令菜单
|
||
self.chain.register_commands(commands=self.get_commands())
|
||
# 消息处理线程
|
||
self._thread = Thread(target=self.__run)
|
||
# 启动事件处理线程
|
||
self._thread.start()
|
||
|
||
def __run(self):
|
||
"""
|
||
事件处理线程
|
||
"""
|
||
while not self._event.is_set():
|
||
event, handlers = self.eventmanager.get_event()
|
||
if event:
|
||
logger.info(f"处理事件：{event.event_type} - {handlers}")
|
||
for handler in handlers:
|
||
try:
|
||
names = handler.__qualname__.split(".")
|
||
if names[0] == "Command":
|
||
self.command_event(event)
|
||
else:
|
||
self.pluginmanager.run_plugin_method(names[0], names[1], event)
|
||
except Exception as e:
|
||
logger.error(f"事件处理出错：{str(e)} - {traceback.format_exc()}")
|
||
|
||
def stop(self):
|
||
"""
|
||
停止事件处理线程
|
||
"""
|
||
self._event.set()
|
||
self._thread.join()
|
||
|
||
def get_commands(self):
|
||
"""
|
||
获取命令列表
|
||
"""
|
||
return self._commands
|
||
|
||
def register(self, cmd: str, func: Any, data: dict = None, desc: str = None) -> None:
|
||
"""
|
||
注册命令
|
||
"""
|
||
self._commands[cmd] = {
|
||
"func": func,
|
||
"description": desc,
|
||
"data": data or {}
|
||
}
|
||
|
||
def get(self, cmd: str) -> Any:
|
||
"""
|
||
获取命令
|
||
"""
|
||
return self._commands.get(cmd, {})
|
||
|
||
def execute(self, cmd: str, data_str: str = "", userid: Union[str, int] = None) -> None:
|
||
"""
|
||
执行命令
|
||
"""
|
||
command = self.get(cmd)
|
||
if command:
|
||
try:
|
||
logger.info(f"用户 {userid} 开始执行：{command.get('description')} ...")
|
||
cmd_data = command['data'] if command.get('data') else {}
|
||
args_num = ObjectUtils.has_arguments(command['func'])
|
||
if args_num > 0:
|
||
if cmd_data:
|
||
# 有内置参数直接使用内置参数
|
||
command['func'](**cmd_data)
|
||
elif args_num == 1:
|
||
# 没有用户输入参数
|
||
command['func'](userid)
|
||
else:
|
||
# 多个输入参数：用户输入、用户ID
|
||
command['func'](data_str, userid)
|
||
else:
|
||
# 没有参数
|
||
command['func']()
|
||
logger.info(f"用户 {userid} {command.get('description')} 执行完成")
|
||
except Exception as err:
|
||
logger.error(f"执行命令 {cmd} 出错：{str(err)}")
|
||
traceback.print_exc()
|
||
|
||
@staticmethod
|
||
def send_plugin_event(etype: EventType, data: dict) -> None:
|
||
"""
|
||
发送插件命令
|
||
"""
|
||
EventManager().send_event(etype, data)
|
||
|
||
@eventmanager.register(EventType.CommandExcute)
|
||
def command_event(self, event: ManagerEvent) -> None:
|
||
"""
|
||
注册命令执行事件
|
||
event_data: {
|
||
"cmd": "/xxx args"
|
||
}
|
||
"""
|
||
# 命令参数
|
||
event_str = event.event_data.get('cmd')
|
||
# 消息用户
|
||
event_user = event.event_data.get('user')
|
||
if event_str:
|
||
cmd = event_str.split()[0]
|
||
args = " ".join(event_str.split()[1:])
|
||
if self.get(cmd):
|
||
self.execute(cmd, args, event_user)
|