refactor: extract is_ban_cached method to BanConsole

2026-10-09 22:00:01 +08:00 · 2025-12-24 14:54:39 +08:00
parent ff0b37123e
commit be316a5caf
17 changed files with 478 additions and 1360 deletions
@@ -0,0 +1,78 @@
+from nonebot import on_message
+from nonebot.adapters import Bot
+from nonebot.plugin import PluginMetadata
+from nonebot_plugin_alconna import Image, UniMsg
+from nonebot_plugin_uninfo import Uninfo
+
+from zhenxun.configs.config import BotConfig
+from zhenxun.configs.utils import PluginExtraData
+from zhenxun.models.ban_console import BanConsole
+from zhenxun.services.data_access import DataAccess
+from zhenxun.services.log import logger
+from zhenxun.utils.enum import PluginType
+from zhenxun.utils.image_utils import get_download_image_hash
+from zhenxun.utils.message import MessageUtils
+from zhenxun.utils.platform import PlatformUtils
+from zhenxun.utils.utils import FreqLimiter, get_entity_ids
+
+from ._data_source import mute_manager
+
+__plugin_meta__ = PluginMetadata(
+    name="刷屏监听",
+    description="",
+    usage="",
+    extra=PluginExtraData(
+        author="HibiKier",
+        version="0.1-473ecd8",
+        menu_type="其他",
+        plugin_type=PluginType.DEPENDANT,
+    ).to_dict(),
+)
+
+
+async def rule(session: Uninfo) -> bool:
+    entity_ids = get_entity_ids(session)
+    if not session.group:
+        return False
+    ban_dao = DataAccess(BanConsole)
+    if not await ban_dao.safe_get_or_none(
+        user_id=entity_ids.user_id, group_id=entity_ids.group_id
+    ):
+        return False
+    if not await ban_dao.safe_get_or_none(user_id="", group_id=entity_ids.group_id):
+        return False
+    return True
+
+
+_matcher = on_message(rule=rule, priority=1, block=False)
+
+_flmt = FreqLimiter(30)
+
+
+@_matcher.handle()
+async def _(bot: Bot, session: Uninfo, message: UniMsg):
+    entity_ids = get_entity_ids(session)
+    plain_text = message.extract_plain_text()
+    image_list = [m.url for m in message if isinstance(m, Image) and m.url]
+    img_hash = ""
+    for url in image_list:
+        img_hash += await get_download_image_hash(url, "_mute_")
+    _message = plain_text + img_hash
+    if duration := mute_manager.add_message(
+        entity_ids.user_id, entity_ids.group_id or "0", _message
+    ):
+        try:
+            if _flmt.check(entity_ids.user_id):
+                _flmt.start_cd(entity_ids.user_id)
+                await PlatformUtils.ban_user(
+                    bot, entity_ids.user_id, entity_ids.group_id or "0", duration
+                )
+                await MessageUtils.build_message(
+                    f"检测到恶意刷屏，{BotConfig.self_nickname}要把你关进小黑屋！"
+                ).send(at_sender=True)
+                mute_manager.reset(entity_ids.user_id, entity_ids.group_id or "0")
+                logger.info(
+                    f"检测刷屏 被禁言 {duration} 分钟", "禁言检查", session=session
+                )
+        except Exception as e:
+            logger.error("禁言发送错误", "禁言检测", session=session, e=e)