import json from typing import Optional, Union, List, Tuple, Any from app.core.context import MediaInfo, Context from app.log import logger from app.modules import _ModuleBase, _MessageBase from app.schemas import MessageChannel, CommingMessage, Notification from app.schemas.types import ModuleType try: from app.modules.discord.discord import Discord except Exception as err: # ImportError or other load issues Discord = None logger.error(f"Discord 模块未加载，缺少依赖或初始化错误：{err}") class DiscordModule(_ModuleBase, _MessageBase[Discord]): def init_module(self) -> None: """ 初始化模块 """ if not Discord: logger.error("Discord 依赖未就绪（需要安装 discord.py==2.6.4），模块未启动") return self.stop() super().init_service(service_name=Discord.__name__.lower(), service_type=Discord) self._channel = MessageChannel.Discord @staticmethod def get_name() -> str: return "Discord" @staticmethod def get_type() -> ModuleType: """ 获取模块类型 """ return ModuleType.Notification @staticmethod def get_subtype() -> MessageChannel: """ 获取模块子类型 """ return MessageChannel.Discord @staticmethod def get_priority() -> int: """ 获取模块优先级，数字越小优先级越高，只有同一接口下优先级才生效 """ return 4 def stop(self): """ 停止模块 """ for client in self.get_instances().values(): client.stop() def test(self) -> Optional[Tuple[bool, str]]: """ 测试模块连接性 """ if not self.get_instances(): return None for name, client in self.get_instances().items(): state = client.get_state() if not state: return False, f"Discord {name} Bot 未就绪" return True, "" def init_setting(self) -> Tuple[str, Union[str, bool]]: pass def message_parser(self, source: str, body: Any, form: Any, args: Any) -> Optional[CommingMessage]: """ 解析消息内容，返回字典，注意以下约定值： userid: 用户ID username: 用户名 text: 内容 :param source: 消息来源 :param body: 请求体 :param form: 表单 :param args: 参数 :return: 渠道、消息体 """ client_config = self.get_config(source) if not client_config: return None try: msg_json: dict = json.loads(body) except Exception as e: logger.debug(f"解析 Discord 消息失败：{str(e)}") return None if not msg_json: return None msg_type = msg_json.get("type") userid = msg_json.get("userid") username = msg_json.get("username") if msg_type == "interaction": callback_data = msg_json.get("callback_data") message_id = msg_json.get("message_id") chat_id = msg_json.get("chat_id") if callback_data and userid: logger.info(f"收到来自 {client_config.name} 的 Discord 按钮回调：" f"userid={userid}, username={username}, callback_data={callback_data}") return CommingMessage( channel=MessageChannel.Discord, source=client_config.name, userid=userid, username=username, text=f"CALLBACK:{callback_data}", is_callback=True, callback_data=callback_data, message_id=message_id, chat_id=str(chat_id) if chat_id else None ) return None if msg_type == "message": text = msg_json.get("text") chat_id = msg_json.get("chat_id") if text and userid: logger.info(f"收到来自 {client_config.name} 的 Discord 消息：" f"userid={userid}, username={username}, text={text}") return CommingMessage(channel=MessageChannel.Discord, source=client_config.name, userid=userid, username=username, text=text, chat_id=str(chat_id) if chat_id else None) return None def post_message(self, message: Notification, **kwargs) -> None: """ 发送通知消息 :param message: 消息通知对象 """ # DEBUG: Log entry and configs configs = self.get_configs() logger.debug(f"[Discord] post_message 被调用，message.source={message.source}, " f"message.userid={message.userid}, message.channel={message.channel}") logger.debug(f"[Discord] 当前配置数量: {len(configs)}, 配置名称: {list(configs.keys())}") logger.debug(f"[Discord] 当前实例数量: {len(self.get_instances())}, 实例名称: {list(self.get_instances().keys())}") if not configs: logger.warning("[Discord] get_configs() 返回空，没有可用的 Discord 配置") return for conf in configs.values(): logger.debug(f"[Discord] 检查配置: name={conf.name}, type={conf.type}, enabled={conf.enabled}") if not self.check_message(message, conf.name): logger.debug(f"[Discord] check_message 返回 False，跳过配置: {conf.name}") continue logger.debug(f"[Discord] check_message 通过，准备发送到: {conf.name}") targets = message.targets userid = message.userid if not userid and targets is not None: userid = targets.get('discord_userid') if not userid: logger.warn("用户没有指定 Discord 用户ID，消息无法发送") return client: Discord = self.get_instance(conf.name) logger.debug(f"[Discord] get_instance('{conf.name}') 返回: {client is not None}") if client: logger.debug(f"[Discord] 调用 client.send_msg, userid={userid}, title={message.title[:50] if message.title else None}...") result = client.send_msg(title=message.title, text=message.text, image=message.image, userid=userid, link=message.link, buttons=message.buttons, original_message_id=message.original_message_id, original_chat_id=message.original_chat_id, mtype=message.mtype) logger.debug(f"[Discord] send_msg 返回结果: {result}") else: logger.warning(f"[Discord] 未找到配置 '{conf.name}' 对应的 Discord 客户端实例") def post_medias_message(self, message: Notification, medias: List[MediaInfo]) -> None: """ 发送媒体信息选择列表 :param message: 消息体 :param medias: 媒体信息 :return: 成功或失败 """ for conf in self.get_configs().values(): if not self.check_message(message, conf.name): continue client: Discord = self.get_instance(conf.name) if client: client.send_medias_msg(title=message.title, medias=medias, userid=message.userid, buttons=message.buttons, original_message_id=message.original_message_id, original_chat_id=message.original_chat_id) def post_torrents_message(self, message: Notification, torrents: List[Context]) -> None: """ 发送种子信息选择列表 :param message: 消息体 :param torrents: 种子信息 :return: 成功或失败 """ for conf in self.get_configs().values(): if not self.check_message(message, conf.name): continue client: Discord = self.get_instance(conf.name) if client: client.send_torrents_msg(title=message.title, torrents=torrents, userid=message.userid, buttons=message.buttons, original_message_id=message.original_message_id, original_chat_id=message.original_chat_id) def delete_message(self, channel: MessageChannel, source: str, message_id: str, chat_id: Optional[str] = None) -> bool: """ 删除消息 :param channel: 消息渠道 :param source: 指定的消息源 :param message_id: 消息ID（Slack中为时间戳） :param chat_id: 聊天ID（频道ID） :return: 删除是否成功 """ success = False for conf in self.get_configs().values(): if channel != self._channel: break if source != conf.name: continue client: Discord = self.get_instance(conf.name) if client: result = client.delete_msg(message_id=message_id, chat_id=chat_id) if result: success = True return success