"""NoneBot OneBot V11 适配器：负责拉取群历史消息、群组/成员信息与发送。""" from __future__ import annotations import asyncio import base64 import time from datetime import datetime, timedelta from pathlib import Path from typing import Any from .core.domain.value_objects.unified_group import UnifiedGroup, UnifiedMember from .core.domain.value_objects.unified_message import ( MessageContent, MessageContentType, UnifiedMessage, ) from .core.utils.logger import logger class OneBotAdapter: """面向 NoneBot OneBot V11 的最小适配器。""" platform_name = "onebot" USER_AVATAR_TEMPLATE = "https://q1.qlogo.cn/g?b=qq&nk={user_id}&s=640" def __init__(self, bot: Any, config: dict | None = None): self.bot = bot self.config = config or {} self.platform_id = str(self.config.get("platform_id") or "onebot") self.bot_self_ids = [str(x) for x in self.config.get("bot_self_ids", [])] self.filter_bot_messages = bool(self.config.get("filter_bot_messages", True)) # —— 消息拉取 —— async def fetch_messages( self, group_id: str, days: int = 1, max_count: int = 1000, before_id: str | None = None, since_ts: int | None = None, ) -> list[UnifiedMessage]: try: chunk_size = 100 all_raw = [] seen_raw_ids: set[str] = set() if since_ts and since_ts > 0: start_ts = since_ts else: start_ts = int( (datetime.now() - timedelta(days=days)).timestamp() ) current_anchor = before_id while len(all_raw) < max_count: fetch_count = min(chunk_size, max_count - len(all_raw)) params: dict[str, Any] = { "group_id": int(group_id), "count": fetch_count, } if current_anchor: params["message_seq"] = current_anchor params["reverseOrder"] = True result = None for attempt in range(1, 4): try: result = await self.bot.call_api( "get_group_msg_history", **params ) break except Exception as exc: if attempt < 3: logger.warning( f"OneBot 分页拉取失败(第{attempt}次): {exc}" ) await asyncio.sleep(attempt) else: logger.warning( f"OneBot 分页拉取重试耗尽: {exc}" ) if not result or "messages" not in result: break messages = result.get("messages", []) if not messages: break first = messages[0] last = messages[-1] earliest = first if first.get("time", 0) <= last.get("time", 0) else last chunk_earliest_ts = earliest.get("time", 0) for raw in messages: msg_time = raw.get("time", 0) msg_id = str(raw.get("message_id", "")) if not msg_id or msg_id in seen_raw_ids: continue if start_ts <= msg_time <= int(datetime.now().timestamp()): all_raw.append(raw) seen_raw_ids.add(msg_id) seq_val = ( earliest.get("message_seq") or earliest.get("real_id") or earliest.get("seq") ) mid_val = earliest.get("message_id") new_anchor = seq_val if seq_val is not None else mid_val if chunk_earliest_ts <= start_ts: break if current_anchor and str(new_anchor) == str(current_anchor): break current_anchor = new_anchor await asyncio.sleep(0.05) unified: list[UnifiedMessage] = [] seen: set[str] = set() for raw in all_raw: mid = str(raw.get("message_id", "")) if not mid or mid in seen: continue u = self._convert_message(raw, group_id) if u: unified.append(u) seen.add(mid) unified.sort(key=lambda m: m.timestamp) return unified except Exception as e: logger.warning(f"OneBot 分页获取消息失败: {e}") return [] def _convert_message(self, raw: dict, group_id: str) -> UnifiedMessage | None: try: sender = raw.get("sender", {}) chain = raw.get("message", []) if isinstance(chain, str): chain = [{"type": "text", "data": {"text": chain}}] contents: list[MessageContent] = [] text_parts: list[str] = [] for seg in chain: seg_t = seg.get("type", "") seg_d = seg.get("data", {}) if seg_t == "text": text = seg_d.get("text", "") text_parts.append(text) contents.append(MessageContent(type=MessageContentType.TEXT, text=text)) elif seg_t == "image": sub_type = seg_d.get("subType", seg_d.get("sub_type")) try: is_sticker = int(sub_type) == 1 except (TypeError, ValueError): is_sticker = False raw_data: dict[str, Any] = {"summary": seg_d.get("summary", "")} if sub_type is not None: try: raw_data["sub_type"] = int(sub_type) except (TypeError, ValueError): pass contents.append( MessageContent( type=MessageContentType.EMOJI if is_sticker else MessageContentType.IMAGE, url=seg_d.get("url", seg_d.get("file", "")), raw_data=raw_data, ) ) elif seg_t == "at": contents.append( MessageContent(type=MessageContentType.AT, at_user_id=str(seg_d.get("qq", ""))) ) elif seg_t in ("face", "mface", "bface", "sface"): contents.append( MessageContent( type=MessageContentType.EMOJI, emoji_id=str(seg_d.get("id", "")), raw_data={"face_type": seg_t}, ) ) elif seg_t == "reply": contents.append( MessageContent(type=MessageContentType.REPLY, raw_data={"reply_id": seg_d.get("id", "")}) ) else: contents.append(MessageContent(type=MessageContentType.UNKNOWN, raw_data=seg)) reply_to = None for c in contents: if c.type == MessageContentType.REPLY and c.raw_data: reply_to = str(c.raw_data.get("reply_id", "")) break return UnifiedMessage( message_id=str(raw.get("message_id", "")), sender_id=str(sender.get("user_id", "")), sender_name=sender.get("nickname", ""), sender_card=sender.get("card", "") or None, group_id=group_id, text_content="".join(text_parts), contents=tuple(contents), timestamp=raw.get("time", 0), platform="onebot", reply_to_id=reply_to, ) except Exception as e: logger.debug(f"OneBot _convert_message error: {e}") return None # —— 群/成员信息 —— async def get_group_info(self, group_id: str) -> UnifiedGroup | None: try: gid = int(group_id) data = await self.bot.call_api("get_group_info", group_id=gid) if isinstance(data, list): data = next((d for d in data if str(d.get("group_id")) == str(gid)), data[0] if data else {}) return UnifiedGroup( group_id=str(gid), group_name=data.get("group_name", ""), member_count=int(data.get("member_count", 0) or 0), platform="onebot", ) except Exception as e: logger.debug(f"get_group_info({group_id}) 失败: {e}") return None async def get_user_avatar_url(self, user_id: str) -> str | None: return self.USER_AVATAR_TEMPLATE.format(user_id=user_id) async def get_member_info(self, group_id: str, user_id: str) -> UnifiedMember | None: try: data = await self.bot.call_api( "get_group_member_info", group_id=int(group_id), user_id=int(user_id), ) return UnifiedMember( user_id=str(user_id), nickname=data.get("nickname", ""), card=data.get("card", "") or None, role=str(data.get("role", "member")), ) except Exception as e: logger.debug(f"get_member_info({group_id},{user_id}) 失败: {e}") return None async def is_group_muted(self, group_id: str) -> bool: return False def get_platform_name(self) -> str: return self.platform_name # —— 发送（在手动/自动报告中可用；手动指令会走 SAA 发送） —— async def send_text(self, group_id: str, text: str) -> bool: try: await self.bot.call_api("send_group_msg", group_id=int(group_id), message=str(text)) return True except Exception as e: logger.warning(f"send_text 失败: {e}") return False async def send_image(self, group_id: str, image_url: str, caption: str = "") -> bool: try: if image_url.startswith("base64://"): b64 = image_url.split("base64://", 1)[1] from nonebot.adapters.onebot.v11 import MessageSegment msg = MessageSegment.image(file=f"base64://{b64}") else: from nonebot.adapters.onebot.v11 import MessageSegment msg = MessageSegment.image(file=image_url) await self.bot.call_api("send_group_msg", group_id=int(group_id), message=msg) return True except Exception as e: logger.warning(f"send_image 失败: {e}") return False async def send_file(self, group_id: str, file_path: str, caption: str = "") -> bool: return False async def send_forward_msg(self, group_id: str, nodes: list[dict]) -> bool: """发送 OneBot v11 合并转发消息 (send_forward_msg)。""" try: from nonebot.adapters.onebot.v11 import MessageSegment normalized = [] for n in nodes: data = dict(n.get("data", {})) content = data.get("content") if isinstance(content, str): content = [MessageSegment.text(content)] elif content is None: content = [] data["content"] = content normalized.append({"type": "node", "data": data}) await self.bot.call_api( "send_forward_msg", group_id=int(group_id), messages=normalized ) return True except Exception as e: logger.warning(f"send_forward_msg 失败: {e}") return False async def set_reaction(self, group_id: str, message_id: str, emoji: str | int, is_add: bool = True) -> bool: return False async def prepare_group_member_cache(self, group_id: str): return True, None