chore: import deployable mai-bot source tree

plugin-templates/MaiBot-Napcat-Adapter/services/official_bot_guard.py

@@ -0,0 +1,59 @@
+"""NapCat 官方机器人消息拦截服务。"""
+
+from __future__ import annotations
+
+from typing import Any, Dict
+
+from .query_service import NapCatQueryService
+
+
+class NapCatOfficialBotGuard:
+    """根据群成员资料判断是否应拦截 QQ 官方机器人消息。"""
+
+    def __init__(self, logger: Any, query_service: NapCatQueryService) -> None:
+        """初始化官方机器人拦截服务。
+
+        Args:
+            logger: 插件日志对象。
+            query_service: NapCat 查询服务。
+        """
+        self._logger = logger
+        self._query_service = query_service
+        self._cache: Dict[str, bool] = {}
+
+    def clear_cache(self) -> None:
+        """清空机器人识别缓存。"""
+        self._cache.clear()
+
+    async def should_reject(self, sender_user_id: str, group_id: str, ban_qq_bot: bool) -> bool:
+        """判断是否应拦截当前消息。
+
+        Args:
+            sender_user_id: 发送者用户号。
+            group_id: 群号。
+            ban_qq_bot: 是否启用官方机器人拦截。
+
+        Returns:
+            bool: 若应拦截，则返回 ``True``。
+        """
+        if not ban_qq_bot or not group_id:
+            return False
+
+        cache_key = f"{group_id}:{sender_user_id}"
+        cached_result = self._cache.get(cache_key)
+        if cached_result is not None:
+            if cached_result:
+                self._logger.warning("QQ 官方机器人消息拦截已启用，消息被丢弃")
+            return cached_result
+
+        member_info = await self._query_service.get_group_member_info(group_id, sender_user_id, no_cache=True)
+        if member_info is None:
+            self._logger.warning("无法获取用户是否为机器人，默认放行当前消息")
+            self._cache[cache_key] = False
+            return False
+
+        should_reject = bool(member_info.get("is_robot"))
+        self._cache[cache_key] = should_reject
+        if should_reject:
+            self._logger.warning("QQ 官方机器人消息拦截已启用，消息被丢弃")
+        return should_reject