diff --git a/_conf_schema.json b/_conf_schema.json index 447205a..e8bcb09 100644 --- a/_conf_schema.json +++ b/_conf_schema.json @@ -10,5 +10,23 @@ "type": "list", "hint": "填写需要屏蔽的用户ID，多个ID用英文逗号分隔", "default": [] + }, + "server": { + "description": "后端服务器相关配置", + "type": "object", + "items": { + "url": { + "description": "后端服务器地址", + "type": "string", + "hint": "填写后端服务器的地址，用于收到消息时从服务端获取响应", + "default": "http://127.0.0.1:8080" + }, + "token": { + "description": "后端服务器请求token，请手动从后端服务器生成token并配置", + "type": "string", + "hint": "填写后端服务器的token", + "default": "" + } + } } } diff --git a/main.py b/main.py index 71add91..4905914 100644 --- a/main.py +++ b/main.py @@ -4,26 +4,12 @@ from astrbot.api.star import Context, Star, register from astrbot.api import logger from astrbot.core.platform import AstrBotMessage from astrbot.core.message.components import At - -def check_at_bot(message_obj:AstrBotMessage) -> bool: - """ - 检测群聊消息是否@了当前机器人。 - 依据: plugin.md 中 AstrBotMessage 及消息链定义: - - message_obj.message 为 BaseMessageComponent 列表 - - At(qq=...) 为 @ 消息段，message_obj.self_id 为机器人的标识 ID - 当消息链中存在 At 且其 qq 等于 self_id 时，判定为@了机器人。 - 忽略 AtAll(@全体)。 - """ - self_id = message_obj.self_id - segments = message_obj.message - for segment in segments: - if isinstance(segment, At) and str(segment.qq) == self_id: - logger.info(f"[check_at_bot] 消息 {message_obj.message_id} @了机器人 {self_id}") - return True - logger.info(f"[check_at_bot] 消息 {message_obj.message_id} 未@机器人 {self_id}") - return False +import astrbot.api.message_components as Comp +import aiohttp +import json +from typing import Dict, Any, Optional @register("qqfight-bot", "hxuanyu", "消息中转插件", "1.0.0") @@ -34,23 +20,154 @@ class RelayPlugin(Star): # 读取群聊白名单和私聊黑名单，全部转为字符串集合，避免类型不一致 self.group_whitelist = set(str(x) for x in self.config.get("group_whitelist", [])) self.private_blacklist = set(str(x) for x in self.config.get("private_blacklist", [])) + self.server_url = self.config.get("server", {}).get("url", "http://127.0.0.1:8080") + self.server_token = self.config.get("server", {}).get("token", "") logger.info(f"[RelayPlugin] 群聊白名单: {self.group_whitelist}") logger.info(f"[RelayPlugin] 私聊黑名单: {self.private_blacklist}") + logger.info(f"[RelayPlugin] 服务器地址: {self.server_url}") + + async def send_to_server(self, data: Dict[Any, Any]) -> Optional[Dict[str, Any]]: + """ + 发送JSON数据到服务器的/api/bot接口 + + Args: + data: 要发送的JSON数据字典 + + Returns: + 服务器响应的JSON字典，格式为 {"code": "200", "msg": "", "data": ""} + 如果请求失败则返回 None + """ + url = f"{self.server_url.rstrip('/')}/api/bot" + + headers = { + "Content-Type": "application/json", + "Authorization": f"Bearer {self.server_token}" + } + + try: + async with aiohttp.ClientSession() as session: + async with session.post(url, + json=data, + headers=headers, + timeout=aiohttp.ClientTimeout(total=10)) as response: + + logger.info(f"[RelayPlugin] 发送请求到 {url}, 状态码: {response.status}") + + if response.status == 200: + response_json = await response.json() + logger.info(f"[RelayPlugin] 服务器响应: {response_json}") + return response_json + else: + logger.error(f"[RelayPlugin] HTTP请求失败, 状态码: {response.status}") + response_text = await response.text() + logger.error(f"[RelayPlugin] 错误响应内容: {response_text}") + return None + + except aiohttp.ClientError as e: + logger.error(f"[RelayPlugin] 网络请求异常: {e}") + return None + except json.JSONDecodeError as e: + logger.error(f"[RelayPlugin] JSON解析异常: {e}") + return None + except Exception as e: + logger.error(f"[RelayPlugin] 未知异常: {e}") + return None + + async def send_message_to_server(self, message_type: str, message_obj: AstrBotMessage, + extra_data: Dict[str, Any] = None) -> Optional[str]: + """ + 向服务器发送消息的封装方法 + + Args: + message_type: 消息类型，"group" 或 "private" + message_obj: AstrBotMessage 消息对象 + extra_data: 额外的数据字典 + + Returns: + 服务器响应的 data 字段内容，如果失败则返回 None + """ + # 构建基础消息数据 + message_data = { + "type": message_type, + "message": str(message_obj.message_str), + "message_id": message_obj.message_id, + "timestamp": getattr(message_obj, 'timestamp', None), + "sender_nickname": str(message_obj.sender.nickname), + } + + # 根据消息类型添加特定字段 + if message_type == "group": + message_data.update({ + "group_id": str(message_obj.group_id), + "sender_id": str(message_obj.sender.user_id) + }) + elif message_type == "private": + message_data.update({ + "sender_id": str(message_obj.sender.user_id) + }) + + # 合并额外数据 + if extra_data: + message_data.update(extra_data) + + logger.info(f"[RelayPlugin] 准备发送{message_type}消息到服务器: {message_data}") + + # 发送请求 + response = await self.send_to_server(message_data) + + if response and isinstance(response, dict): + # 提取 data 字段 + data = response.get("data", "") + logger.info(f"[RelayPlugin] 服务器返回数据: {data}") + return data + else: + logger.error(f"[RelayPlugin] 服务器请求失败或返回格式错误") + return None + + def check_at_bot(self, message_obj: AstrBotMessage) -> bool: + """ + 检测群聊消息是否@了当前机器人。 + 依据: plugin.md 中 AstrBotMessage 及消息链定义: + - message_obj.message 为 BaseMessageComponent 列表 + - At(qq=...) 为 @ 消息段，message_obj.self_id 为机器人的标识 ID + 当消息链中存在 At 且其 qq 等于 self_id 时，判定为@了机器人。 + 忽略 AtAll(@全体)。 + """ + self_id = message_obj.self_id + segments = message_obj.message + for segment in segments: + if isinstance(segment, At) and str(segment.qq) == self_id: + logger.info(f"[check_at_bot] 消息 {message_obj.message_id} @了机器人 {self_id}") + return True + logger.info(f"[check_at_bot] 消息 {message_obj.message_id} 未@机器人 {self_id}") + return False @event_message_type(EventMessageType.GROUP_MESSAGE) async def on_group_message(self, event: AstrMessageEvent): - group_id = str(event.message_obj.group_id) - at_bot = check_at_bot(event.message_obj) - + at_bot = self.check_at_bot(event.message_obj) + if group_id not in self.group_whitelist: logger.info(f"[RelayPlugin] 群聊 {group_id} 不在白名单，忽略消息") return if not at_bot: logger.info(f"[RelayPlugin] 群聊 {group_id} 消息未@机器人，忽略消息") return + logger.info(f"[RelayPlugin] 群聊 {group_id} 消息链: {event.message_obj.message}") - # TODO: 处理函数留空 + + # 发送群聊消息到服务器 + result_data = await self.send_message_to_server("group", event.message_obj) + logger.info(f"收到服务端响应：{result_data}") + if result_data: + message_chan = [ + Comp.At(qq=event.get_sender_id()), + Comp.Plain(str(result_data)) + ] + yield event.chain_result(message_chan) + else: + logger.error(f"[RelayPlugin] 处理群聊消息失败") + yield event.plain_result("处理消息时发生错误，请稍后重试") @event_message_type(EventMessageType.PRIVATE_MESSAGE) async def on_private_message(self, event: AstrMessageEvent): @@ -59,8 +176,19 @@ class RelayPlugin(Star): sender_id = str(sender.id) else: sender_id = str(sender) - if sender_id in set(str(x) for x in self.private_blacklist): + + if sender_id in self.private_blacklist: logger.info(f"[RelayPlugin] 私聊 {sender_id} 在黑名单，忽略消息") return + logger.info(f"[RelayPlugin] 私聊 {sender_id} 消息链: {event.message_obj.message}") - # TODO: 处理函数留空 \ No newline at end of file + + # 发送私聊消息到服务器 + if event.message_obj.message_str: + result_data = await self.send_message_to_server("private", event.message_obj) + logger.info(f"收到服务端响应：{result_data}") + if result_data: + yield event.plain_result(str(result_data)) + else: + logger.error(f"[RelayPlugin] 处理私聊消息失败") + yield event.plain_result("处理消息时发生错误，请稍后重试")