mirror of
https://github.com/Nezumi-2711/astrbot_plugin_qq_group_daily_analysis.git
synced 2026-09-23 04:09:59 +00:00
refactor(Telegram): 重构 Telegram 逻辑并提取消息处理与注册表服务
- 提取消息处理服务及 Telegram 注册表，显著精简主插件逻辑。 - Telegram 群组获取回退逻辑下沉至适配器，移除调度中心硬编码。 - 优化依赖注入机制，使适配器层级可访问插件 KV 存储。 - 修正消息内容定义兼容性并清理冗余代码。
This commit is contained in:
SXP-Simon
@@ -0,0 +1,269 @@
|
||||
import re
|
||||
from collections import Counter
|
||||
|
||||
from astrbot.api.event import AstrMessageEvent
|
||||
from astrbot.api.star import Context
|
||||
|
||||
from ...utils.logger import logger
|
||||
from ...infrastructure.persistence.telegram_group_registry import TelegramGroupRegistry
|
||||
|
||||
|
||||
class MessageProcessingService:
|
||||
"""
|
||||
消息处理服务
|
||||
|
||||
负责处理接收到的消息事件：
|
||||
1. 解析消息内容（文本、图片、@提及等）
|
||||
2. 解析发送者信息（跨平台兼容）
|
||||
3. 存储消息历史
|
||||
4. 维护 Telegram 群组注册表（回退机制）
|
||||
"""
|
||||
|
||||
def __init__(self, context: Context, telegram_registry: TelegramGroupRegistry):
|
||||
self.context = context
|
||||
self.telegram_registry = telegram_registry
|
||||
|
||||
async def process_message(self, event: AstrMessageEvent) -> None:
|
||||
"""
|
||||
处理并在历史记录中存储消息。
|
||||
|
||||
Args:
|
||||
event: AstrBot 消息事件
|
||||
|
||||
Raises:
|
||||
ValueError: 当必要数据无法获取时
|
||||
RuntimeError: 当消息内容为空时
|
||||
"""
|
||||
# 1. 获取群组 ID（必需）
|
||||
group_id = self._get_group_id_from_event(event)
|
||||
if not group_id:
|
||||
raise ValueError("无法获取群组 ID，拒绝存储消息")
|
||||
|
||||
# 2. 获取发送者 ID（必需）
|
||||
sender_id = event.get_sender_id()
|
||||
if not sender_id:
|
||||
raise ValueError(f"群 {group_id}: 无法获取发送者 ID，拒绝存储消息")
|
||||
sender_id = str(sender_id)
|
||||
|
||||
# 3. 获取发送者名称（昵称优先，必要时回退）
|
||||
sender_name = self._resolve_sender_name(event, sender_id)
|
||||
|
||||
# 4. 获取平台 ID（必需）
|
||||
platform_id = event.get_platform_id()
|
||||
if not platform_id:
|
||||
raise ValueError(f"群 {group_id}: 无法获取平台 ID，拒绝存储消息")
|
||||
|
||||
# 5. 提取消息内容
|
||||
message_parts = self._extract_message_parts(event)
|
||||
if not message_parts:
|
||||
# 尝试记录一条警告但不中断流程（或者视为错误）
|
||||
# 原逻辑是抛出 RuntimeError
|
||||
raise RuntimeError(
|
||||
f"群 {group_id}: 消息内容为空 (sender={sender_name})，拒绝存储"
|
||||
)
|
||||
|
||||
# 6. 提取事件消息 ID（用于 Telegram 已见群/话题记录）
|
||||
msg_obj = getattr(event, "message_obj", None)
|
||||
event_message_id = str(getattr(msg_obj, "message_id", "") or "")
|
||||
|
||||
# 7. 存储到数据库
|
||||
await self.context.message_history_manager.insert(
|
||||
platform_id=platform_id,
|
||||
user_id=group_id,
|
||||
content={"type": "user", "message": message_parts},
|
||||
sender_id=sender_id,
|
||||
sender_name=sender_name,
|
||||
)
|
||||
|
||||
# Telegram: 记录已见群/话题
|
||||
if self._is_telegram_event(event, platform_id):
|
||||
try:
|
||||
await self.telegram_registry.upsert(
|
||||
platform_id=platform_id,
|
||||
group_id=group_id,
|
||||
sender_id=sender_id,
|
||||
sender_name=sender_name,
|
||||
event_message_id=event_message_id,
|
||||
)
|
||||
except Exception as e:
|
||||
logger.warning(
|
||||
"[TGRegistry] Upsert failed: "
|
||||
f"platform_id={platform_id} group_id={group_id} error={e}"
|
||||
)
|
||||
|
||||
logger.debug(
|
||||
f"[{platform_id}] 已缓存群 {group_id} 的消息 (发送者: {sender_name})"
|
||||
)
|
||||
|
||||
def _get_group_id_from_event(self, event: AstrMessageEvent) -> str | None:
|
||||
"""从消息事件中安全获取群组 ID"""
|
||||
try:
|
||||
group_id = event.get_group_id()
|
||||
return group_id if group_id else None
|
||||
except Exception:
|
||||
return None
|
||||
|
||||
def _resolve_sender_name(self, event: AstrMessageEvent, sender_id: str) -> str:
|
||||
"""解析发送者展示名"""
|
||||
platform_name = str(event.get_platform_name() or "").lower()
|
||||
candidates: list[str | None] = []
|
||||
|
||||
msg_obj = getattr(event, "message_obj", None)
|
||||
sender_obj = getattr(msg_obj, "sender", None)
|
||||
raw_message = getattr(msg_obj, "raw_message", None)
|
||||
raw_msg_obj = getattr(raw_message, "message", raw_message)
|
||||
from_user = getattr(raw_msg_obj, "from_user", None)
|
||||
|
||||
if platform_name == "telegram":
|
||||
if from_user is not None:
|
||||
candidates.extend(
|
||||
[
|
||||
getattr(from_user, "full_name", None),
|
||||
getattr(from_user, "first_name", None),
|
||||
]
|
||||
)
|
||||
candidates.append(event.get_sender_name())
|
||||
if sender_obj is not None:
|
||||
candidates.append(getattr(sender_obj, "nickname", None))
|
||||
if from_user is not None:
|
||||
candidates.append(getattr(from_user, "username", None))
|
||||
else:
|
||||
candidates.append(event.get_sender_name())
|
||||
if sender_obj is not None:
|
||||
candidates.append(getattr(sender_obj, "nickname", None))
|
||||
|
||||
if from_user is not None:
|
||||
candidates.extend(
|
||||
[
|
||||
getattr(from_user, "full_name", None),
|
||||
getattr(from_user, "first_name", None),
|
||||
getattr(from_user, "username", None),
|
||||
]
|
||||
)
|
||||
|
||||
for candidate in candidates:
|
||||
name = str(candidate or "").strip()
|
||||
if not self._is_placeholder_sender_name(name, sender_id):
|
||||
return name
|
||||
|
||||
return sender_id
|
||||
|
||||
def _extract_message_parts(self, event: AstrMessageEvent) -> list[dict]:
|
||||
"""从事件中提取消息内容"""
|
||||
message_parts = []
|
||||
message = event.message_obj
|
||||
|
||||
# 收集 @ 标记
|
||||
pending_mentions: Counter[str] = Counter()
|
||||
if message and hasattr(message, "message"):
|
||||
for seg in message.message:
|
||||
if not hasattr(seg, "type"):
|
||||
continue
|
||||
if seg.type not in ("At", "at"):
|
||||
continue
|
||||
|
||||
target = getattr(seg, "target", None) or getattr(seg, "qq", None)
|
||||
if target is None and hasattr(seg, "data"):
|
||||
target = seg.data.get("qq") or seg.data.get("target")
|
||||
|
||||
target_str = str(target or "").strip()
|
||||
if target_str:
|
||||
pending_mentions[target_str] += 1
|
||||
|
||||
display_name = str(getattr(seg, "name", "") or "").strip()
|
||||
if display_name and display_name != target_str:
|
||||
pending_mentions[display_name] += 1
|
||||
|
||||
if message and hasattr(message, "message"):
|
||||
for seg in message.message:
|
||||
if not hasattr(seg, "type"):
|
||||
continue
|
||||
|
||||
seg_type = seg.type
|
||||
if seg_type in ("Plain", "text"):
|
||||
text = getattr(seg, "text", None)
|
||||
if text is None and hasattr(seg, "data"):
|
||||
text = seg.data.get("text")
|
||||
if text:
|
||||
text = self._strip_known_mentions(text, pending_mentions)
|
||||
message_parts.append({"type": "plain", "text": text})
|
||||
|
||||
elif seg_type in ("Image", "image"):
|
||||
url = getattr(seg, "url", None) or (
|
||||
seg.data.get("url") if hasattr(seg, "data") else None
|
||||
)
|
||||
if url:
|
||||
message_parts.append({"type": "image", "url": url})
|
||||
|
||||
elif seg_type in ("At", "at"):
|
||||
target = getattr(seg, "target", None) or getattr(seg, "qq", None)
|
||||
if target is None and hasattr(seg, "data"):
|
||||
target = seg.data.get("qq") or seg.data.get("target")
|
||||
if target:
|
||||
message_parts.append(
|
||||
{
|
||||
"type": "at",
|
||||
"target_id": str(target),
|
||||
"name": str(getattr(seg, "name", "") or ""),
|
||||
}
|
||||
)
|
||||
|
||||
if not message_parts and event.message_str:
|
||||
message_parts.append({"type": "plain", "text": event.message_str})
|
||||
|
||||
# 清理空文本段
|
||||
message_parts = [
|
||||
part
|
||||
for part in message_parts
|
||||
if not (
|
||||
part.get("type") == "plain" and not str(part.get("text", "")).strip()
|
||||
)
|
||||
]
|
||||
|
||||
return message_parts
|
||||
|
||||
@staticmethod
|
||||
def _strip_known_mentions(text: str, pending_mentions: Counter[str]) -> str:
|
||||
"""从文本中移除已识别的 @ 提及"""
|
||||
cleaned = str(text)
|
||||
if not cleaned or not pending_mentions:
|
||||
return cleaned.strip()
|
||||
|
||||
for mention, remaining in list(pending_mentions.items()):
|
||||
if not mention or remaining <= 0:
|
||||
continue
|
||||
|
||||
pattern = re.compile(rf"(?<!\w)@{re.escape(mention)}(?!\w)")
|
||||
removed = 0
|
||||
while removed < remaining:
|
||||
cleaned, subn = pattern.subn("", cleaned, count=1)
|
||||
if subn == 0:
|
||||
break
|
||||
removed += 1
|
||||
|
||||
if removed > 0:
|
||||
pending_mentions[mention] -= removed
|
||||
if pending_mentions[mention] <= 0:
|
||||
pending_mentions.pop(mention, None)
|
||||
|
||||
return re.sub(r"\s{2,}", " ", cleaned).strip()
|
||||
|
||||
@staticmethod
|
||||
def _is_placeholder_sender_name(name: str | None, sender_id: str) -> bool:
|
||||
"""判断 sender_name 是否为占位值"""
|
||||
if not name:
|
||||
return True
|
||||
normalized = str(name).strip()
|
||||
if not normalized:
|
||||
return True
|
||||
if normalized.lower() in {"unknown", "none", "null", "nil", "undefined"}:
|
||||
return True
|
||||
return normalized == str(sender_id).strip()
|
||||
|
||||
@staticmethod
|
||||
def _is_telegram_event(event: AstrMessageEvent, platform_id: str) -> bool:
|
||||
"""判断当前事件是否为 Telegram 平台"""
|
||||
platform_name = str(event.get_platform_name() or "").strip().lower()
|
||||
if platform_name == "telegram":
|
||||
return True
|
||||
return str(platform_id or "").strip().lower().startswith("telegram")
|
||||
Reference in New Issue
Block a user