271 lines
12 KiB
Python
271 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
跨平台消息互通插件事件处理器模块
|
|
"""
|
|
import os
|
|
import html
|
|
from typing import List, Any
|
|
from core.managers.command_manager import matcher
|
|
from models.events.message import GroupMessageEvent, MessageEvent
|
|
from models.message import MessageSegment
|
|
from core.permission import Permission
|
|
from core.utils.logger import ModuleLogger
|
|
from .config import config
|
|
from .parser import parse_forward_nodes
|
|
from .sender import forward_discord_to_qq, forward_qq_to_discord
|
|
|
|
# 创建模块专用日志记录器
|
|
logger = ModuleLogger("CrossPlatform")
|
|
|
|
async def handle_discord_message(
|
|
username: str,
|
|
discriminator: str,
|
|
content: str,
|
|
channel_id: int,
|
|
attachments: List[dict] = None,
|
|
embed: dict = None
|
|
):
|
|
"""处理 Discord 消息并转发"""
|
|
if not config.ENABLE_CROSS_PLATFORM:
|
|
return
|
|
|
|
logger.info(f"[CrossPlatform] 收到 Discord 消息: {username}#{discriminator} in {channel_id}")
|
|
logger.debug(f"[CrossPlatform] 消息内容: '{content}', 附件: {attachments}")
|
|
await forward_discord_to_qq(username, discriminator, content, channel_id, attachments)
|
|
|
|
async def handle_qq_message(
|
|
nickname: str,
|
|
user_id: int,
|
|
group_name: str,
|
|
group_id: int,
|
|
content: str,
|
|
attachments: List[dict] = None
|
|
):
|
|
"""处理 QQ 消息并转发"""
|
|
if not config.ENABLE_CROSS_PLATFORM:
|
|
return
|
|
|
|
logger.info(f"[CrossPlatform] 收到 QQ 消息: {nickname} ({user_id}) in {group_name}({group_id})")
|
|
await forward_qq_to_discord(nickname, user_id, group_name, group_id, content, attachments)
|
|
|
|
@matcher.on_message()
|
|
async def handle_qq_group_message(event: GroupMessageEvent):
|
|
"""处理 QQ 群消息，转发到 Discord"""
|
|
if not config.ENABLE_CROSS_PLATFORM:
|
|
return
|
|
|
|
group_id = event.group_id
|
|
mapped_channel = None
|
|
for discord_channel_id, info in config.CROSS_PLATFORM_MAP.items():
|
|
if info["qq_group_id"] == group_id:
|
|
mapped_channel = discord_channel_id
|
|
break
|
|
|
|
if mapped_channel is None:
|
|
return
|
|
|
|
content = ""
|
|
attachments = []
|
|
|
|
if isinstance(event.message, list):
|
|
has_forward_node = any(isinstance(seg, MessageSegment) and seg.type == "node" for seg in event.message)
|
|
|
|
if has_forward_node:
|
|
forward_nodes = [seg for seg in event.message if isinstance(seg, MessageSegment) and seg.type == "node"]
|
|
forward_nodes_dict = [{"type": seg.type, "data": seg.data} for seg in forward_nodes]
|
|
content, attachments = await parse_forward_nodes(forward_nodes_dict)
|
|
else:
|
|
for segment in event.message:
|
|
if isinstance(segment, MessageSegment):
|
|
if segment.type == "text":
|
|
content += segment.data.get("text", "")
|
|
elif segment.type == "image":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_url = html.unescape(str(file_url))
|
|
if not file_name:
|
|
file_name = os.path.basename(file_url.split('?')[0]) or f"image_{len(attachments)}.jpg"
|
|
attachments.append({"type": "image", "url": file_url, "filename": file_name})
|
|
elif segment.type == "video":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_url = html.unescape(str(file_url))
|
|
if not file_name:
|
|
file_name = os.path.basename(file_url.split('?')[0]) or f"video_{len(attachments)}.mp4"
|
|
attachments.append({"type": "video", "url": file_url, "filename": file_name})
|
|
elif segment.type == "record":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_url = html.unescape(str(file_url))
|
|
if not file_name:
|
|
file_name = os.path.basename(file_url.split('?')[0]) or f"record_{len(attachments)}.amr"
|
|
attachments.append({"type": "record", "url": file_url, "filename": file_name})
|
|
content += f"\n[语音: {file_name}]\n"
|
|
elif segment.type == "file":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_url = html.unescape(str(file_url))
|
|
if not file_name:
|
|
file_name = os.path.basename(file_url.split('?')[0]) or f"file_{len(attachments)}"
|
|
attachments.append({"type": "file", "url": file_url, "filename": file_name})
|
|
content += f"\n[文件: {file_name}]\n"
|
|
logger.debug(f"[CrossPlatform] QQ 消息识别到文件: {file_name}, URL: {file_url}")
|
|
elif segment.type == "at":
|
|
qq_id = segment.data.get("qq")
|
|
if qq_id and qq_id != "all":
|
|
content += f"@{qq_id} "
|
|
elif qq_id == "all":
|
|
content += "@所有人 "
|
|
elif isinstance(segment, str):
|
|
content += segment
|
|
elif isinstance(event.message, str):
|
|
content = event.message
|
|
|
|
import re
|
|
local_file_pattern = r'(http://[\w\.-]+:\d+/download\?id=file_[a-zA-Z0-9_]+)'
|
|
matches = re.finditer(local_file_pattern, content)
|
|
for match in matches:
|
|
file_url = match.group(1)
|
|
file_name = f"video_{len(attachments)}.mp4"
|
|
attachments.append({"type": "video", "url": file_url, "filename": file_name})
|
|
|
|
content = content.strip()
|
|
|
|
group_name = ""
|
|
try:
|
|
group_info = await event.bot.get_group_info(event.group_id)
|
|
group_name = group_info.get("group_name", "")
|
|
except Exception:
|
|
group_name = f"群{group_id}"
|
|
|
|
await handle_qq_message(
|
|
nickname=event.sender.nickname or event.sender.card or str(event.user_id),
|
|
user_id=event.user_id,
|
|
group_name=group_name,
|
|
group_id=group_id,
|
|
content=content,
|
|
attachments=attachments
|
|
)
|
|
|
|
@matcher.on_message()
|
|
async def handle_discord_message_event(event: Any):
|
|
"""处理 Discord 消息事件（通过适配器注入）"""
|
|
if not config.ENABLE_CROSS_PLATFORM:
|
|
return
|
|
|
|
logger.debug(f"[CrossPlatform] handle_discord_message_event 触发: {event}")
|
|
if not hasattr(event, '_is_discord_message'):
|
|
logger.debug(f"[CrossPlatform] 事件没有 _is_discord_message 属性，跳过")
|
|
return
|
|
|
|
logger.debug(f"[CrossPlatform] 检测到 Discord 事件")
|
|
discord_channel_id = getattr(event, 'discord_channel_id', None)
|
|
if discord_channel_id is None:
|
|
logger.debug(f"[CrossPlatform] discord_channel_id 为 None")
|
|
return
|
|
|
|
content = ""
|
|
attachments = []
|
|
|
|
logger.debug(f"[CrossPlatform] 开始处理 Discord 事件消息: channel_id={discord_channel_id}")
|
|
|
|
if hasattr(event, 'message') and isinstance(event.message, list):
|
|
has_text_content = False
|
|
for segment in event.message:
|
|
if isinstance(segment, MessageSegment):
|
|
if segment.type == "text":
|
|
content += segment.data.get("text", "")
|
|
has_text_content = True
|
|
elif segment.type == "image":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_name = file_name or os.path.basename(str(file_url).split('?')[0]) or "image"
|
|
attachment_item = {"type": "image", "url": str(file_url), "filename": file_name}
|
|
if attachment_item not in attachments:
|
|
attachments.append(attachment_item)
|
|
content += f"\n[图片: {file_name}]\n"
|
|
elif segment.type == "video":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_name = file_name or os.path.basename(str(file_url).split('?')[0]) or "video"
|
|
attachment_item = {"type": "video", "url": str(file_url), "filename": file_name}
|
|
if attachment_item not in attachments:
|
|
attachments.append(attachment_item)
|
|
content += f"\n[视频: {file_name}]\n"
|
|
elif segment.type == "record":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_name = file_name or os.path.basename(str(file_url).split('?')[0]) or "record"
|
|
attachment_item = {"type": "record", "url": str(file_url), "filename": file_name}
|
|
if attachment_item not in attachments:
|
|
attachments.append(attachment_item)
|
|
content += f"\n[语音: {file_name}]\n"
|
|
elif segment.type == "file":
|
|
file_url = segment.data.get("url") or segment.data.get("file")
|
|
file_name = segment.data.get("filename")
|
|
if file_url:
|
|
file_name = file_name or os.path.basename(str(file_url).split('?')[0]) or "file"
|
|
attachment_item = {"type": "file", "url": str(file_url), "filename": file_name}
|
|
if attachment_item not in attachments:
|
|
attachments.append(attachment_item)
|
|
content += f"\n[文件: {file_name}]\n"
|
|
logger.debug(f"[CrossPlatform] Discord 消息识别到文件: {file_name}, URL: {file_url}")
|
|
else:
|
|
content = event.raw_message or ""
|
|
|
|
content = content.strip()
|
|
|
|
# 如果 content 为空但有附件（如只有表情），使用 raw_message 作为 content
|
|
if not content and attachments:
|
|
content = event.raw_message or ""
|
|
|
|
logger.debug(f"[CrossPlatform] Discord 消息内容: '{content}', 附件数量: {len(attachments)}")
|
|
|
|
discord_username = getattr(event, 'discord_username', 'Unknown')
|
|
discord_discriminator = getattr(event, 'discord_discriminator', '')
|
|
|
|
logger.debug(f"[CrossPlatform] 调用 handle_discord_message: username={discord_username}, channel_id={discord_channel_id}")
|
|
await handle_discord_message(
|
|
username=discord_username,
|
|
discriminator=discord_discriminator,
|
|
content=content,
|
|
channel_id=discord_channel_id,
|
|
attachments=attachments,
|
|
embed=None
|
|
)
|
|
|
|
@matcher.command("cross_config", "跨平台配置", permission=Permission.ADMIN)
|
|
async def cross_config_command(event: MessageEvent):
|
|
"""查看跨平台配置"""
|
|
if not config.ENABLE_CROSS_PLATFORM:
|
|
await event.reply("跨平台功能已禁用")
|
|
return
|
|
|
|
config_lines = ["=== 跨平台映射配置 ==="]
|
|
|
|
if not config.CROSS_PLATFORM_MAP:
|
|
config_lines.append("当前没有配置任何映射")
|
|
else:
|
|
for discord_id, info in config.CROSS_PLATFORM_MAP.items():
|
|
discord_channel = f"Discord: {discord_id}"
|
|
qq_group = f"QQ: {info['qq_group_id']}"
|
|
name = info.get("name", "")
|
|
if name:
|
|
config_lines.append(f"• {discord_channel} ↔ {qq_group} ({name})")
|
|
else:
|
|
config_lines.append(f"• {discord_channel} ↔ {qq_group}")
|
|
|
|
await event.reply("\n".join(config_lines))
|
|
|
|
@matcher.command("cross_reload", "跨平台重载", permission=Permission.ADMIN)
|
|
async def cross_reload_command(event: MessageEvent):
|
|
"""重新加载跨平台配置"""
|
|
await config.reload()
|
|
await event.reply("跨平台配置已重载") |