mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
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.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
|
|
if mute_manager.get_group_data(entity_ids.group_id or "0").duration == 0:
|
|
return False
|
|
if await BanConsole.is_ban_cached(entity_ids.user_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)
|