36fce6ca98
- 为NEW_napcat_adapter插件实现了核心模块，包括消息处理、事件处理和路由。 - 创建了MessageHandler、MetaEventHandler和NoticeHandler来处理收到的消息和事件。 - 开发了SendHandler，用于向Napcat发送回消息。 引入了StreamRouter来管理多个聊天流，确保消息的顺序和高效处理。 - 增加了对各种消息类型和格式的支持，包括文本、图像和通知。 - 建立了一个用于监控和调试的日志系统。
342 lines
13 KiB
Python
342 lines
13 KiB
Python
"""
|
|
MessageEnvelope converter between mofox_bus schema and internal message structures.
|
|
|
|
- 优先处理 maim_message 风格的 message_info + message_segment。
|
|
- 兼容旧版 content/sender/channel 结构，方便逐步迁移。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from mofox_bus import (
|
|
BaseMessageInfo,
|
|
MessageBase,
|
|
MessageEnvelope,
|
|
Seg,
|
|
UserInfo,
|
|
GroupInfo,
|
|
)
|
|
|
|
from src.common.logger import get_logger
|
|
|
|
logger = get_logger("envelope_converter")
|
|
|
|
|
|
class EnvelopeConverter:
|
|
"""MessageEnvelope <-> MessageBase converter."""
|
|
|
|
@staticmethod
|
|
def to_message_base(envelope: MessageEnvelope) -> MessageBase:
|
|
"""
|
|
Convert MessageEnvelope to MessageBase.
|
|
"""
|
|
try:
|
|
# 优先使用 maim_message 样式字段
|
|
info_payload = envelope.get("message_info") or {}
|
|
seg_payload = envelope.get("message_segment") or envelope.get("message_chain")
|
|
|
|
if info_payload:
|
|
message_info = BaseMessageInfo.from_dict(info_payload)
|
|
else:
|
|
message_info = EnvelopeConverter._build_info_from_legacy(envelope)
|
|
|
|
if seg_payload is None:
|
|
seg_list = EnvelopeConverter._content_to_segments(envelope.get("content"))
|
|
seg_payload = seg_list
|
|
|
|
message_segment = EnvelopeConverter._ensure_seg(seg_payload)
|
|
raw_message = envelope.get("raw_message") or envelope.get("raw_platform_message")
|
|
|
|
return MessageBase(
|
|
message_info=message_info,
|
|
message_segment=message_segment,
|
|
raw_message=raw_message,
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"转换 MessageEnvelope 失败: {e}", exc_info=True)
|
|
raise
|
|
|
|
@staticmethod
|
|
def _build_info_from_legacy(envelope: MessageEnvelope) -> BaseMessageInfo:
|
|
"""将 legacy 字段映射为 BaseMessageInfo。"""
|
|
platform = envelope.get("platform")
|
|
channel = envelope.get("channel") or {}
|
|
sender = envelope.get("sender") or {}
|
|
|
|
message_id = envelope.get("id") or envelope.get("message_id")
|
|
timestamp_ms = envelope.get("timestamp_ms")
|
|
time_value = (timestamp_ms / 1000.0) if timestamp_ms is not None else None
|
|
|
|
group_info: Optional[GroupInfo] = None
|
|
channel_type = channel.get("channel_type")
|
|
if channel_type in ("group", "supergroup", "room"):
|
|
group_info = GroupInfo(
|
|
platform=platform,
|
|
group_id=channel.get("channel_id"),
|
|
group_name=channel.get("title"),
|
|
)
|
|
|
|
user_info: Optional[UserInfo] = None
|
|
if sender:
|
|
user_info = UserInfo(
|
|
platform=platform,
|
|
user_id=str(sender.get("user_id")) if sender.get("user_id") is not None else None,
|
|
user_nickname=sender.get("display_name") or sender.get("user_nickname"),
|
|
user_avatar=sender.get("avatar_url"),
|
|
)
|
|
|
|
return BaseMessageInfo(
|
|
platform=platform,
|
|
message_id=message_id,
|
|
time=time_value,
|
|
group_info=group_info,
|
|
user_info=user_info,
|
|
additional_config=envelope.get("metadata"),
|
|
)
|
|
|
|
@staticmethod
|
|
def _ensure_seg(payload: Any) -> Seg:
|
|
"""将任意 payload 转为 Seg dataclass。"""
|
|
if isinstance(payload, Seg):
|
|
return payload
|
|
if isinstance(payload, list):
|
|
# 直接传入 Seg 列表或 seglist data
|
|
return Seg(type="seglist", data=[EnvelopeConverter._ensure_seg(item) for item in payload])
|
|
if isinstance(payload, dict):
|
|
seg_type = payload.get("type") or "text"
|
|
data = payload.get("data")
|
|
if seg_type == "seglist" and isinstance(data, list):
|
|
data = [EnvelopeConverter._ensure_seg(item) for item in data]
|
|
return Seg(type=seg_type, data=data)
|
|
# 兜底：转成文本片段
|
|
return Seg(type="text", data="" if payload is None else str(payload))
|
|
|
|
@staticmethod
|
|
def _flatten_segments(seg: Seg) -> List[Seg]:
|
|
"""将 Seg/seglist 打平成列表，便于旧 content 转换。"""
|
|
if seg.type == "seglist" and isinstance(seg.data, list):
|
|
return [item if isinstance(item, Seg) else EnvelopeConverter._ensure_seg(item) for item in seg.data]
|
|
return [seg]
|
|
|
|
@staticmethod
|
|
def _content_to_segments(content: Any) -> List[Seg]:
|
|
"""
|
|
Convert legacy Content (type/data/metadata) to a flat list of Seg.
|
|
"""
|
|
segments: List[Seg] = []
|
|
|
|
def _walk(node: Any) -> None:
|
|
if node is None:
|
|
return
|
|
if isinstance(node, list):
|
|
for item in node:
|
|
_walk(item)
|
|
return
|
|
if not isinstance(node, dict):
|
|
logger.warning("未知的 content 节点类型: %s", type(node))
|
|
return
|
|
|
|
content_type = node.get("type")
|
|
data = node.get("data")
|
|
metadata = node.get("metadata") or {}
|
|
|
|
if content_type == "collection":
|
|
items = data if isinstance(data, list) else node.get("items", [])
|
|
for item in items:
|
|
_walk(item)
|
|
return
|
|
|
|
if content_type in ("text", "at"):
|
|
subtype = metadata.get("subtype") or ("at" if content_type == "at" else None)
|
|
text = "" if data is None else str(data)
|
|
if subtype in ("at", "mention"):
|
|
user_info = metadata.get("user") or {}
|
|
seg_data: Dict[str, Any] = {
|
|
"user_id": user_info.get("id") or user_info.get("user_id"),
|
|
"user_name": user_info.get("name") or user_info.get("display_name"),
|
|
"text": text,
|
|
"raw": user_info.get("raw") or user_info if user_info else None,
|
|
}
|
|
if any(v is not None for v in seg_data.values()):
|
|
segments.append(Seg(type="at", data=seg_data))
|
|
else:
|
|
segments.append(Seg(type="at", data=text))
|
|
else:
|
|
segments.append(Seg(type="text", data=text))
|
|
return
|
|
|
|
if content_type == "image":
|
|
url = ""
|
|
if isinstance(data, dict):
|
|
url = data.get("url") or data.get("file") or data.get("file_id") or ""
|
|
elif data is not None:
|
|
url = str(data)
|
|
segments.append(Seg(type="image", data=url))
|
|
return
|
|
|
|
if content_type == "audio":
|
|
url = ""
|
|
if isinstance(data, dict):
|
|
url = data.get("url") or data.get("file") or data.get("file_id") or ""
|
|
elif data is not None:
|
|
url = str(data)
|
|
segments.append(Seg(type="record", data=url))
|
|
return
|
|
|
|
if content_type == "video":
|
|
url = ""
|
|
if isinstance(data, dict):
|
|
url = data.get("url") or data.get("file") or data.get("file_id") or ""
|
|
elif data is not None:
|
|
url = str(data)
|
|
segments.append(Seg(type="video", data=url))
|
|
return
|
|
|
|
if content_type == "file":
|
|
file_name = ""
|
|
if isinstance(data, dict):
|
|
file_name = data.get("file_name") or data.get("name") or ""
|
|
text = file_name or "[file]"
|
|
segments.append(Seg(type="text", data=text))
|
|
return
|
|
|
|
if content_type == "command":
|
|
name = ""
|
|
args: Dict[str, Any] = {}
|
|
if isinstance(data, dict):
|
|
name = data.get("name", "")
|
|
args = data.get("args", {}) or {}
|
|
else:
|
|
name = str(data or "")
|
|
cmd_text = f"/{name}" if name else "/command"
|
|
if args:
|
|
cmd_text += " " + " ".join(f"{k}={v}" for k, v in args.items())
|
|
segments.append(Seg(type="text", data=cmd_text))
|
|
return
|
|
|
|
if content_type == "event":
|
|
event_type = ""
|
|
if isinstance(data, dict):
|
|
event_type = data.get("event_type", "")
|
|
else:
|
|
event_type = str(data or "")
|
|
segments.append(Seg(type="text", data=f"[事件: {event_type or 'unknown'}]"))
|
|
return
|
|
|
|
if content_type == "system":
|
|
text = "" if data is None else str(data)
|
|
segments.append(Seg(type="text", data=f"[系统] {text}"))
|
|
return
|
|
|
|
logger.warning(f"未知的消息类型: {content_type}")
|
|
segments.append(Seg(type="text", data=f"[未知消息类型: {content_type}]"))
|
|
|
|
_walk(content)
|
|
return segments
|
|
|
|
@staticmethod
|
|
def to_legacy_dict(envelope: MessageEnvelope) -> Dict[str, Any]:
|
|
"""
|
|
Convert MessageEnvelope to legacy dict for backward compatibility.
|
|
"""
|
|
message_base = EnvelopeConverter.to_message_base(envelope)
|
|
return message_base.to_dict()
|
|
|
|
@staticmethod
|
|
def from_message_base(message: MessageBase, direction: str = "outgoing") -> MessageEnvelope:
|
|
"""
|
|
Convert MessageBase to MessageEnvelope (maim_message style preferred).
|
|
"""
|
|
try:
|
|
info_dict = message.message_info.to_dict()
|
|
seg_dict = message.message_segment.to_dict()
|
|
|
|
envelope: MessageEnvelope = {
|
|
"direction": direction,
|
|
"message_info": info_dict,
|
|
"message_segment": seg_dict,
|
|
"platform": info_dict.get("platform"),
|
|
"message_id": info_dict.get("message_id"),
|
|
"schema_version": 1,
|
|
}
|
|
|
|
if message.message_info.time is not None:
|
|
envelope["timestamp_ms"] = int(message.message_info.time * 1000)
|
|
if message.raw_message is not None:
|
|
envelope["raw_message"] = message.raw_message
|
|
|
|
# legacy 补充，方便老代码继续工作
|
|
segments = EnvelopeConverter._flatten_segments(message.message_segment)
|
|
envelope["content"] = EnvelopeConverter._segments_to_content(segments)
|
|
if message.message_info.user_info:
|
|
envelope["sender"] = {
|
|
"user_id": message.message_info.user_info.user_id,
|
|
"role": "assistant" if direction == "outgoing" else "user",
|
|
"display_name": message.message_info.user_info.user_nickname,
|
|
"avatar_url": getattr(message.message_info.user_info, "user_avatar", None),
|
|
}
|
|
if message.message_info.group_info:
|
|
envelope["channel"] = {
|
|
"channel_id": message.message_info.group_info.group_id,
|
|
"channel_type": "group",
|
|
"title": message.message_info.group_info.group_name,
|
|
}
|
|
|
|
return envelope
|
|
|
|
except Exception as e:
|
|
logger.error(f"转换 MessageBase 失败: {e}", exc_info=True)
|
|
raise
|
|
|
|
@staticmethod
|
|
def _segments_to_content(segments: List[Seg]) -> Dict[str, Any]:
|
|
"""
|
|
Convert Seg list to legacy Content (type/data/metadata).
|
|
"""
|
|
if not segments:
|
|
return {"type": "text", "data": ""}
|
|
|
|
def _seg_to_content(seg: Seg) -> Dict[str, Any]:
|
|
data = seg.data
|
|
|
|
if seg.type == "text":
|
|
return {"type": "text", "data": data}
|
|
|
|
if seg.type == "at":
|
|
content: Dict[str, Any] = {"type": "text", "data": ""}
|
|
metadata: Dict[str, Any] = {"subtype": "at"}
|
|
if isinstance(data, dict):
|
|
content["data"] = data.get("text", "")
|
|
user = {
|
|
"id": data.get("user_id"),
|
|
"name": data.get("user_name"),
|
|
"raw": data.get("raw"),
|
|
}
|
|
if any(v is not None for v in user.values()):
|
|
metadata["user"] = user
|
|
else:
|
|
content["data"] = data
|
|
if metadata:
|
|
content["metadata"] = metadata
|
|
return content
|
|
|
|
if seg.type == "image":
|
|
return {"type": "image", "data": data}
|
|
|
|
if seg.type in ("record", "voice", "audio"):
|
|
return {"type": "audio", "data": data}
|
|
|
|
if seg.type == "video":
|
|
return {"type": "video", "data": data}
|
|
|
|
return {"type": seg.type, "data": data}
|
|
|
|
if len(segments) == 1:
|
|
return _seg_to_content(segments[0])
|
|
|
|
return {"type": "collection", "data": [_seg_to_content(seg) for seg in segments]}
|
|
|
|
|
|
__all__ = ["EnvelopeConverter"]
|