8bd084ce3f
* fix(discord): 修复 WebSocket 连接检测并增强跨平台文件处理 修复 Discord WebSocket 连接检测逻辑，使用正确的属性检查连接状态 为跨平台消息处理添加文件类型支持，并增加详细的调试日志 优化附件处理逻辑，确保所有文件类型都能正确识别和转发 * feat(跨平台): 优化消息处理并添加纯文本提取功能 添加 extract_text_only 函数过滤非文本标记 修改翻译逻辑仅处理纯文本内容 完善附件处理和消息内容拼接 修复仅包含表情时的消息处理问题 * refactor(discord-cross): 使用模块专用日志记录器替换全局日志记录器 将各模块中的全局日志记录器替换为模块专用日志记录器，以提供更清晰的日志来源标识 同时在适配器中添加会话状态检查和重连机制，提升消息发送的可靠性 * feat(翻译): 改进翻译功能，同时显示原文和译文 修改翻译功能，不再替换原文而是同时显示原文和翻译内容，方便用户对照 更新 DeepSeek API 配置为官方地址和模型 优化 Discord 适配器的重连逻辑，直接关闭 WebSocket 触发重连 修复 Discord 频道 ID 转换逻辑，简化处理流程 * feat(cross-platform): 添加跨平台功能支持及配置优化 - 新增跨平台配置模型和全局配置支持 - 优化 Discord 适配器的连接管理和错误处理 - 添加 watchdog 和 discord.py 依赖 - 创建 DeepSeek API 配置文档 - 移除重复的同步帮助图片代码 - 改进跨平台插件配置加载逻辑 * fix(jrcd): 修正群组ID检查条件 删除不再使用的示例插件文件 * feat: 改进配置加载逻辑并更新项目配置 当配置文件不存在时自动生成示例配置 添加pyproject.toml作为项目构建配置 更新.gitignore忽略更多文件类型 删除不再使用的反向WebSocket示例文件 * docs: 更新架构文档和项目结构说明 添加反向WebSocket连接模式说明 补充核心管理器文档 更新项目结构文件 在文档首页添加特色功能说明 * fix(discord): 修复WebSocket连接检查并添加错误日志 refactor(config): 更新配置文件的网络和认证信息 feat(cross-platform): 为跨平台消息处理添加异常捕获和日志 * fix(discord-cross): 修复跨平台消息处理和附件下载问题 修复QQ群消息处理中的非群消息过滤问题 优化Discord附件下载逻辑，使用aiohttp替代requests 修复Redis订阅任务重复创建问题 调整消息格式化的embed字段处理逻辑 * feat(vectordb): 添加向量数据库支持及集成功能 新增向量数据库管理器模块，支持文本的存储、检索和相似度查询 添加知识库插件和AI聊天插件，利用向量数据库实现记忆功能 优化跨平台翻译模块，集成向量数据库存储历史翻译记录 改进消息处理逻辑，优先使用用户显示名称
564 lines
30 KiB
Python
564 lines
30 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
事件路由与转换器 (Event Router & Converter)
|
||
|
||
此模块负责在不同平台（如 Discord）和 OneBot 业务逻辑之间进行数据转换。
|
||
核心目标是：**让现有的 OneBot 插件（如 bili.py）在不修改任何代码的情况下，能够处理 Discord 消息。**
|
||
|
||
实现原理：
|
||
1. 接收 Discord 消息 (`discord.Message`)。
|
||
2. 将其"伪装"成 OneBot 的 `GroupMessageEvent` 或 `PrivateMessageEvent`。
|
||
3. 拦截插件调用的 `event.reply()` 方法。
|
||
4. 将插件返回的 OneBot `MessageSegment` 转换为 Discord 格式并发送。
|
||
"""
|
||
import asyncio
|
||
from typing import Union, List, Any, Optional, Dict
|
||
|
||
try:
|
||
import discord
|
||
DISCORD_AVAILABLE = True
|
||
except ImportError:
|
||
DISCORD_AVAILABLE = False
|
||
|
||
from models.events.message import GroupMessageEvent, PrivateMessageEvent
|
||
from models.message import MessageSegment as OneBotMessageSegment
|
||
from models.sender import Sender
|
||
from core.utils.logger import ModuleLogger
|
||
|
||
logger = ModuleLogger("EventRouter")
|
||
|
||
class DiscordBotWrapper:
|
||
"""
|
||
包装 DiscordAdapter，提供与 OneBot 相同的发送接口。
|
||
"""
|
||
def __init__(self, adapter: Any):
|
||
self.adapter = adapter
|
||
self.self_id = adapter.user.id if adapter.user else 0
|
||
|
||
async def send_group_msg(self, group_id: int, message: Union[str, OneBotMessageSegment, List[OneBotMessageSegment]], auto_escape: bool = False):
|
||
channel = self.adapter.get_channel(group_id)
|
||
if not channel:
|
||
logger.error(f"Discord channel {group_id} not found")
|
||
return
|
||
await DiscordToOneBotConverter.send_discord_message(channel, message, self.adapter)
|
||
|
||
async def send_private_msg(self, user_id: int, message: Union[str, OneBotMessageSegment, List[OneBotMessageSegment]], auto_escape: bool = False):
|
||
user = self.adapter.get_user(user_id)
|
||
if not user:
|
||
logger.error(f"Discord user {user_id} not found")
|
||
return
|
||
if not user.dm_channel:
|
||
await user.create_dm()
|
||
await DiscordToOneBotConverter.send_discord_message(user.dm_channel, message, self.adapter)
|
||
|
||
async def send(self, event, message, **kwargs):
|
||
if isinstance(event, GroupMessageEvent):
|
||
await self.send_group_msg(event.group_id, message)
|
||
elif isinstance(event, PrivateMessageEvent):
|
||
await self.send_private_msg(event.user_id, message)
|
||
|
||
def build_forward_node(self, user_id: int, nickname: str, message: Union[str, OneBotMessageSegment, List[OneBotMessageSegment]]) -> Dict[str, Any]:
|
||
"""
|
||
构建一个用于合并转发的消息节点 (Node)。
|
||
"""
|
||
processed_message = message
|
||
if isinstance(message, OneBotMessageSegment):
|
||
processed_message = [{"type": message.type, "data": message.data}]
|
||
elif isinstance(message, list):
|
||
processed_message = [{"type": seg.type, "data": seg.data} if isinstance(seg, OneBotMessageSegment) else seg for seg in message]
|
||
|
||
return {
|
||
"type": "node",
|
||
"data": {
|
||
"uin": user_id,
|
||
"name": nickname,
|
||
"content": processed_message
|
||
}
|
||
}
|
||
|
||
async def send_forwarded_messages(self, target, nodes):
|
||
"""
|
||
模拟发送合并转发消息。
|
||
Discord 不支持像 QQ 那样的合并转发，所以我们将其转换为普通消息发送。
|
||
"""
|
||
content = ""
|
||
files = []
|
||
|
||
try:
|
||
for node in nodes:
|
||
if node.get("type") == "node":
|
||
node_data = node.get("data", {})
|
||
node_content = node_data.get("content", [])
|
||
|
||
if isinstance(node_content, str):
|
||
import re
|
||
cq_pattern = r'\[CQ:([^,]+)(?:,([^\]]+))?\]'
|
||
matches = list(re.finditer(cq_pattern, node_content))
|
||
|
||
if not matches:
|
||
content += f"{node_content}\n"
|
||
else:
|
||
last_end = 0
|
||
for match in matches:
|
||
if match.start() > last_end:
|
||
content += node_content[last_end:match.start()]
|
||
|
||
cq_type = match.group(1)
|
||
cq_params_str = match.group(2) or ""
|
||
|
||
params = {}
|
||
if cq_params_str:
|
||
for param in cq_params_str.split(','):
|
||
if '=' in param:
|
||
k, v = param.split('=', 1)
|
||
params[k] = v
|
||
|
||
if cq_type in ("image", "video", "record"):
|
||
file_url = params.get("url") or params.get("file")
|
||
if file_url:
|
||
if str(file_url).startswith("http"):
|
||
content += f"\n{file_url}\n"
|
||
elif str(file_url).startswith("base64://"):
|
||
import base64
|
||
import io
|
||
b64_data = str(file_url)[9:]
|
||
if b64_data.startswith("data:image") or b64_data.startswith("data:audio") or b64_data.startswith("data:video"):
|
||
b64_data = b64_data.split(",", 1)[1]
|
||
try:
|
||
file_bytes = base64.b64decode(b64_data)
|
||
filename = "file.png" if cq_type == "image" else ("file.mp4" if cq_type == "video" else "file.ogg")
|
||
files.append(discord.File(fp=io.BytesIO(file_bytes), filename=filename))
|
||
except Exception as e:
|
||
logger.error(f"解析 Base64 文件失败: {e}")
|
||
else:
|
||
try:
|
||
files.append(discord.File(file_url))
|
||
except Exception as e:
|
||
logger.error(f"无法读取本地文件 {file_url}: {e}")
|
||
elif cq_type == "face":
|
||
# QQ 表情，简单转为文本
|
||
face_id = params.get("id")
|
||
content += f"[表情:{face_id}]"
|
||
elif cq_type == "at":
|
||
qq_id = params.get("qq")
|
||
if qq_id == "all":
|
||
content += "@everyone "
|
||
else:
|
||
content += f"<@{qq_id}> "
|
||
|
||
last_end = match.end()
|
||
|
||
if last_end < len(node_content):
|
||
content += node_content[last_end:]
|
||
content += "\n"
|
||
elif isinstance(node_content, list):
|
||
for seg in node_content:
|
||
if isinstance(seg, dict):
|
||
seg_type = seg.get("type")
|
||
seg_data = seg.get("data", {})
|
||
|
||
if seg_type == "text":
|
||
content += seg_data.get("text", "")
|
||
elif seg_type in ("image", "video", "record"):
|
||
file_url = seg_data.get("url") or seg_data.get("file")
|
||
if file_url:
|
||
if isinstance(file_url, bytes):
|
||
import io
|
||
try:
|
||
filename = "file.png" if seg_type == "image" else ("file.mp4" if seg_type == "video" else "file.ogg")
|
||
files.append(discord.File(fp=io.BytesIO(file_url), filename=filename))
|
||
except Exception as e:
|
||
logger.error(f"解析 bytes 文件失败: {e}")
|
||
elif str(file_url).startswith("http"):
|
||
content += f"\n{file_url}\n"
|
||
elif str(file_url).startswith("base64://") or "data:image" in str(file_url) or "data:audio" in str(file_url) or "data:video" in str(file_url):
|
||
import base64
|
||
import io
|
||
b64_data = str(file_url)
|
||
if b64_data.startswith("base64://"):
|
||
b64_data = b64_data[9:]
|
||
if b64_data.startswith("data:image") or b64_data.startswith("data:audio") or b64_data.startswith("data:video"):
|
||
b64_data = b64_data.split(",", 1)[1]
|
||
try:
|
||
file_bytes = base64.b64decode(b64_data)
|
||
filename = "file.png" if seg_type == "image" else ("file.mp4" if seg_type == "video" else "file.ogg")
|
||
files.append(discord.File(fp=io.BytesIO(file_bytes), filename=filename))
|
||
except Exception as e:
|
||
logger.error(f"解析 Base64 文件失败: {e}")
|
||
else:
|
||
try:
|
||
files.append(discord.File(file_url))
|
||
except Exception as e:
|
||
logger.error(f"无法读取本地文件 {file_url}: {e}")
|
||
elif seg_type == "face":
|
||
face_id = seg_data.get("id")
|
||
content += f"[表情:{face_id}]"
|
||
content += "\n"
|
||
|
||
if content or files:
|
||
# target is usually event, we can use event.bot.send
|
||
if isinstance(target, GroupMessageEvent):
|
||
channel = self.adapter.get_channel(target.group_id)
|
||
if channel:
|
||
await channel.send(content=content, files=files if files else None)
|
||
elif isinstance(target, PrivateMessageEvent):
|
||
user = self.adapter.get_user(target.user_id)
|
||
if user:
|
||
if not user.dm_channel:
|
||
await user.create_dm()
|
||
await user.dm_channel.send(content=content, files=files if files else None)
|
||
except Exception as e:
|
||
logger.error(f"发送 Discord 合并转发消息失败: {e}")
|
||
import traceback
|
||
logger.error(f"异常堆栈: {traceback.format_exc()}")
|
||
|
||
class DiscordToOneBotConverter:
|
||
"""
|
||
将 Discord 消息转换为 OneBot 消息事件的转换器。
|
||
"""
|
||
|
||
@staticmethod
|
||
def create_mock_event(discord_message: 'discord.Message', adapter: Any) -> Union[GroupMessageEvent, PrivateMessageEvent]:
|
||
"""
|
||
将 discord.Message 伪装成 OneBot 的 MessageEvent。
|
||
|
||
Args:
|
||
discord_message: 原始的 Discord 消息对象
|
||
adapter: DiscordAdapter 实例，用于回调发送消息
|
||
|
||
Returns:
|
||
伪装后的 OneBot 事件对象
|
||
"""
|
||
# 在静态方法内部创建模块专用日志记录器
|
||
from core.utils.logger import ModuleLogger
|
||
mod_logger = ModuleLogger("DiscordConverter")
|
||
|
||
# 1. 提取基础信息
|
||
user_id = discord_message.author.id
|
||
message_id = discord_message.id
|
||
|
||
# 处理 Discord 的 raw_message
|
||
# 如果消息是以 @机器人 开头，Discord 的 content 会是 "<@机器人ID> /echo 1"
|
||
# 我们需要把前面的 @ 提及去掉，否则命令匹配器 (matcher) 无法识别以 "/" 开头的命令
|
||
raw_message = discord_message.content
|
||
|
||
# 构造 message 列表 (将文本和附件转换为 MessageSegment)
|
||
message_list = []
|
||
|
||
# 添加文本内容
|
||
if discord_message.content:
|
||
# 处理 Discord 自定义表情 <:name:id> 或 <a:name:id>
|
||
import re
|
||
content = discord_message.content
|
||
|
||
# 查找所有自定义表情
|
||
emoji_pattern = r'<a?:([^:]+):(\d+)>'
|
||
|
||
# 如果有表情，我们需要将文本分割成多个片段
|
||
if re.search(emoji_pattern, content):
|
||
last_end = 0
|
||
for match in re.finditer(emoji_pattern, content):
|
||
# 添加表情前的文本
|
||
if match.start() > last_end:
|
||
text_part = content[last_end:match.start()]
|
||
if text_part:
|
||
message_list.append(OneBotMessageSegment.text(text_part))
|
||
|
||
# 添加表情作为图片
|
||
emoji_name = match.group(1)
|
||
emoji_id = match.group(2)
|
||
is_animated = match.group(0).startswith('<a:')
|
||
ext = 'gif' if is_animated else 'png'
|
||
emoji_url = f"https://cdn.discordapp.com/emojis/{emoji_id}.{ext}"
|
||
|
||
seg = OneBotMessageSegment.image(emoji_url)
|
||
seg.data["filename"] = f"{emoji_name}.{ext}"
|
||
message_list.append(seg)
|
||
|
||
last_end = match.end()
|
||
|
||
# 添加剩余的文本
|
||
if last_end < len(content):
|
||
text_part = content[last_end:]
|
||
if text_part:
|
||
message_list.append(OneBotMessageSegment.text(text_part))
|
||
else:
|
||
message_list.append(OneBotMessageSegment.text(content))
|
||
|
||
# 如果消息只包含表情（没有文本），更新 raw_message 以包含表情信息
|
||
if not raw_message.strip() or raw_message.strip().startswith('<'):
|
||
import re
|
||
raw_message = re.sub(r'<a?:([^:]+):(\d+)>', r'[\1]', raw_message)
|
||
|
||
# 添加附件信息
|
||
if discord_message.attachments:
|
||
mod_logger.debug(f"[DiscordToOneBotConverter] 检测到 {len(discord_message.attachments)} 个附件")
|
||
for attachment in discord_message.attachments:
|
||
filename = attachment.filename.lower()
|
||
mod_logger.debug(f"[DiscordToOneBotConverter] 处理附件: {attachment.filename}, MIME: {attachment.content_type}")
|
||
# 检查是否是语音文件
|
||
if filename.endswith(('.amr', '.silk', '.mp3', '.wav', '.ogg', '.m4a')):
|
||
seg = OneBotMessageSegment.record(attachment.url)
|
||
seg.data["filename"] = attachment.filename
|
||
message_list.append(seg)
|
||
raw_message += f"\n[语音: {attachment.filename}]"
|
||
mod_logger.debug(f"[DiscordToOneBotConverter] 识别为语音文件: {attachment.filename}")
|
||
elif filename.endswith(('.mp4', '.avi', '.mkv', '.mov', '.flv', '.wmv')):
|
||
seg = OneBotMessageSegment.video(attachment.url)
|
||
seg.data["filename"] = attachment.filename
|
||
message_list.append(seg)
|
||
raw_message += f"\n[视频: {attachment.filename}]"
|
||
mod_logger.debug(f"[DiscordToOneBotConverter] 识别为视频文件: {attachment.filename}")
|
||
elif filename.endswith(('.png', '.jpg', '.jpeg', '.gif', '.webp')):
|
||
image_type = "gif" if filename.endswith('.gif') else None
|
||
seg = OneBotMessageSegment.image(attachment.url, image_type=image_type)
|
||
seg.data["filename"] = attachment.filename
|
||
message_list.append(seg)
|
||
raw_message += f"\n[图片: {attachment.filename}]"
|
||
mod_logger.debug(f"[DiscordToOneBotConverter] 识别为图片文件: {attachment.filename}")
|
||
else:
|
||
seg = OneBotMessageSegment.file(attachment.url)
|
||
seg.data["filename"] = attachment.filename
|
||
message_list.append(seg)
|
||
raw_message += f"\n[文件: {attachment.filename}]"
|
||
mod_logger.success(f"[DiscordToOneBotConverter] 识别为普通文件: {attachment.filename}")
|
||
|
||
# 添加贴纸 (Stickers) 信息
|
||
if hasattr(discord_message, 'stickers') and discord_message.stickers:
|
||
for sticker in discord_message.stickers:
|
||
seg = OneBotMessageSegment.image(sticker.url)
|
||
seg.data["filename"] = f"{sticker.name}.png"
|
||
message_list.append(seg)
|
||
raw_message += f"\n[贴纸: {sticker.name}]"
|
||
bot_mention = f"<@{adapter.user.id}>"
|
||
if raw_message.startswith(bot_mention):
|
||
raw_message = raw_message[len(bot_mention):].strip()
|
||
# 如果 message_list 的第一个元素是文本，也需要去掉 @ 提及
|
||
if message_list and message_list[0].type == "text":
|
||
text_content = message_list[0].data.get("text", "")
|
||
if text_content.startswith(bot_mention):
|
||
message_list[0].data["text"] = text_content[len(bot_mention):].strip()
|
||
|
||
# 构造发送者信息
|
||
sender = Sender(
|
||
user_id=user_id,
|
||
nickname=discord_message.author.display_name,
|
||
card=getattr(discord_message.author, 'nick', ''), # 群名片
|
||
role="member" # 简化处理，默认都是普通成员
|
||
)
|
||
|
||
# 2. 判断是群聊还是私聊
|
||
is_private = isinstance(discord_message.channel, discord.DMChannel)
|
||
|
||
import time
|
||
current_time = int(time.time())
|
||
self_id = adapter.user.id if adapter.user else 0
|
||
|
||
# 注入 Discord 特定信息（用于跨平台插件识别）
|
||
discord_channel_id = discord_message.channel.id if not isinstance(discord_message.channel, discord.DMChannel) else None
|
||
# 使用 global_name (显示名称/昵称) 如果存在，否则使用 name (用户名)
|
||
discord_username = getattr(discord_message.author, 'global_name', None) or discord_message.author.name
|
||
discord_discriminator = f"#{discord_message.author.discriminator}" if discord_message.author.discriminator != "0" else ""
|
||
|
||
if is_private:
|
||
# 构造私聊事件
|
||
event = PrivateMessageEvent(
|
||
time=current_time,
|
||
self_id=self_id,
|
||
platform="discord",
|
||
message_type="private",
|
||
sub_type="friend",
|
||
message_id=message_id,
|
||
user_id=user_id,
|
||
raw_message=raw_message,
|
||
message=message_list,
|
||
sender=sender
|
||
)
|
||
else:
|
||
# 构造群聊事件
|
||
group_id = discord_message.channel.id
|
||
event = GroupMessageEvent(
|
||
time=current_time,
|
||
self_id=self_id,
|
||
platform="discord",
|
||
message_type="group",
|
||
sub_type="normal",
|
||
message_id=message_id,
|
||
user_id=user_id,
|
||
group_id=group_id,
|
||
raw_message=raw_message,
|
||
message=message_list,
|
||
sender=sender
|
||
)
|
||
|
||
# 注入 Discord 特定属性（用于跨平台插件识别）
|
||
event._is_discord_message = True
|
||
event.discord_channel_id = discord_channel_id
|
||
event.discord_username = discord_username
|
||
event.discord_discriminator = discord_discriminator
|
||
|
||
# 注入 DiscordBotWrapper
|
||
event.bot = DiscordBotWrapper(adapter)
|
||
|
||
return event
|
||
|
||
@staticmethod
|
||
async def send_discord_message(
|
||
channel: 'discord.abc.Messageable',
|
||
message: Union[str, OneBotMessageSegment, List[OneBotMessageSegment]],
|
||
adapter: Any
|
||
):
|
||
"""
|
||
将 OneBot 的消息段转换为 Discord 格式并发送。
|
||
|
||
Args:
|
||
channel: Discord 频道对象 (TextChannel, DMChannel 等)
|
||
message: 插件返回的 OneBot 消息内容 (字符串或 MessageSegment 列表)
|
||
adapter: DiscordAdapter 实例
|
||
"""
|
||
content = ""
|
||
files = []
|
||
|
||
try:
|
||
# 统一转换为列表处理
|
||
if not isinstance(message, list):
|
||
message = [message]
|
||
|
||
import re
|
||
|
||
for segment in message:
|
||
if isinstance(segment, str):
|
||
# 尝试解析 CQ 码
|
||
cq_pattern = r'\[CQ:([^,]+)(?:,([^\]]+))?\]'
|
||
matches = list(re.finditer(cq_pattern, segment))
|
||
|
||
if not matches:
|
||
content += segment
|
||
continue
|
||
|
||
last_end = 0
|
||
for match in matches:
|
||
# 添加 CQ 码之前的纯文本
|
||
if match.start() > last_end:
|
||
content += segment[last_end:match.start()]
|
||
|
||
cq_type = match.group(1)
|
||
cq_params_str = match.group(2) or ""
|
||
|
||
# 解析参数
|
||
params = {}
|
||
if cq_params_str:
|
||
for param in cq_params_str.split(','):
|
||
if '=' in param:
|
||
k, v = param.split('=', 1)
|
||
params[k] = v
|
||
|
||
if cq_type in ("image", "video", "record"):
|
||
file_url = params.get("url") or params.get("file")
|
||
if file_url:
|
||
if str(file_url).startswith("http"):
|
||
content += f"\n{file_url}"
|
||
elif str(file_url).startswith("base64://") or "data:image" in str(file_url) or "data:audio" in str(file_url) or "data:video" in str(file_url):
|
||
import base64
|
||
import io
|
||
b64_data = str(file_url)
|
||
if b64_data.startswith("base64://"):
|
||
b64_data = b64_data[9:]
|
||
if b64_data.startswith("data:image") or b64_data.startswith("data:audio") or b64_data.startswith("data:video"):
|
||
b64_data = b64_data.split(",", 1)[1]
|
||
try:
|
||
file_bytes = base64.b64decode(b64_data)
|
||
filename = "file.png" if cq_type == "image" else ("file.mp4" if cq_type == "video" else "file.ogg")
|
||
files.append(discord.File(fp=io.BytesIO(file_bytes), filename=filename))
|
||
except Exception as e:
|
||
logger.error(f"解析 Base64 文件失败: {e}")
|
||
else:
|
||
try:
|
||
files.append(discord.File(file_url))
|
||
except Exception as e:
|
||
logger.error(f"无法读取本地文件 {file_url}: {e}")
|
||
elif cq_type == "face":
|
||
face_id = params.get("id")
|
||
content += f"[表情:{face_id}]"
|
||
elif cq_type == "at":
|
||
qq_id = params.get("qq")
|
||
if qq_id == "all":
|
||
content += "@everyone "
|
||
else:
|
||
content += f"<@{qq_id}> "
|
||
|
||
last_end = match.end()
|
||
|
||
# 添加最后一个 CQ 码之后的纯文本
|
||
if last_end < len(segment):
|
||
content += segment[last_end:]
|
||
|
||
elif isinstance(segment, OneBotMessageSegment):
|
||
# 解析 OneBot 的 MessageSegment
|
||
seg_type = segment.type
|
||
seg_data = segment.data
|
||
|
||
if seg_type == "text":
|
||
content += seg_data.get("text", "")
|
||
elif seg_type in ("image", "video", "record"):
|
||
# OneBot 的图片/视频/语音通常有 file (URL或本地路径) 或 url 字段
|
||
file_url = seg_data.get("url") or seg_data.get("file")
|
||
|
||
if file_url:
|
||
# 处理 bytes 类型
|
||
if isinstance(file_url, bytes):
|
||
import io
|
||
try:
|
||
filename = "file.png" if seg_type == "image" else ("file.mp4" if seg_type == "video" else "file.ogg")
|
||
files.append(discord.File(fp=io.BytesIO(file_url), filename=filename))
|
||
except Exception as e:
|
||
logger.error(f"解析 bytes 文件失败: {e}")
|
||
elif str(file_url).startswith("http"):
|
||
# 如果是网络 URL，直接拼接到文本中，Discord 会自动解析预览
|
||
content += f"\n{file_url}"
|
||
elif str(file_url).startswith("base64://") or "data:image" in str(file_url) or "data:audio" in str(file_url) or "data:video" in str(file_url):
|
||
# 处理 Base64 文件 (需要解码并作为文件上传)
|
||
import base64
|
||
import io
|
||
b64_data = str(file_url)
|
||
if b64_data.startswith("base64://"):
|
||
b64_data = b64_data[9:]
|
||
if b64_data.startswith("data:image") or b64_data.startswith("data:audio") or b64_data.startswith("data:video"):
|
||
b64_data = b64_data.split(",", 1)[1]
|
||
try:
|
||
file_bytes = base64.b64decode(b64_data)
|
||
filename = "file.png" if seg_type == "image" else ("file.mp4" if seg_type == "video" else "file.ogg")
|
||
files.append(discord.File(fp=io.BytesIO(file_bytes), filename=filename))
|
||
except Exception as e:
|
||
logger.error(f"解析 Base64 文件失败: {e}")
|
||
else:
|
||
# 假设是本地文件路径
|
||
try:
|
||
files.append(discord.File(file_url))
|
||
except Exception as e:
|
||
logger.error(f"无法读取本地文件 {file_url}: {e}")
|
||
elif seg_type == "face":
|
||
face_id = seg_data.get("id")
|
||
content += f"[表情:{face_id}]"
|
||
elif seg_type == "at":
|
||
qq_id = seg_data.get("qq")
|
||
if qq_id == "all":
|
||
content += "@everyone "
|
||
else:
|
||
# 尝试将 QQ 号映射回 Discord ID (这里简单处理，直接拼接)
|
||
content += f"<@{qq_id}> "
|
||
elif seg_type == "reply":
|
||
# 忽略回复段，或者你可以尝试映射 message_id
|
||
pass
|
||
|
||
# 发送消息到 Discord
|
||
# 如果内容为空但有文件，Discord 允许发送
|
||
if content or files:
|
||
await channel.send(content=content, files=files if files else None)
|
||
else:
|
||
logger.warning("尝试发送空消息到 Discord，已拦截")
|
||
except Exception as e:
|
||
logger.error(f"发送 Discord 消息失败: {e}")
|
||
import traceback
|
||
logger.error(f"异常堆栈: {traceback.format_exc()}")
|