mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-04 11:20:01 +08:00
284 lines
9.4 KiB
Python
284 lines
9.4 KiB
Python
import base64
|
||
from collections import OrderedDict, deque
|
||
from pathlib import Path
|
||
import random
|
||
from typing import Any
|
||
|
||
from nonebot import on_message
|
||
from nonebot.adapters import Bot, Event
|
||
from nonebot.adapters.onebot.v11 import Message as V11Message
|
||
from nonebot.adapters.onebot.v11 import MessageSegment
|
||
from nonebot.plugin import PluginMetadata
|
||
from nonebot.utils import run_sync
|
||
from nonebot_plugin_alconna import Image as alcImg
|
||
from nonebot_plugin_alconna import UniMsg
|
||
from nonebot_plugin_uninfo import Uninfo
|
||
|
||
from zhenxun.configs.config import BotConfig, Config
|
||
from zhenxun.configs.path_config import DATA_PATH, TEMP_PATH
|
||
from zhenxun.configs.utils import PluginExtraData, RegisterConfig, Task
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils.common_utils import CommonUtils
|
||
from zhenxun.utils.enum import PluginType
|
||
from zhenxun.utils.http_utils import AsyncHttpx
|
||
from zhenxun.utils.image_utils import get_img_hash
|
||
from zhenxun.utils.message import MessageUtils
|
||
|
||
FUDU_IMAGE_PATH = DATA_PATH / "fudu"
|
||
FUDU_IMAGE_PATH.mkdir(parents=True, exist_ok=True)
|
||
FUDU_CACHE_PATH = TEMP_PATH / "fudu"
|
||
FUDU_CACHE_PATH.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 支持的图片扩展名
|
||
IMAGE_EXTENSIONS = {".png", ".jpg", ".jpeg", ".gif", ".webp"}
|
||
|
||
__plugin_meta__ = PluginMetadata(
|
||
name="复读",
|
||
description="群友的本质是什么？是复读机哒！",
|
||
usage="""
|
||
usage：
|
||
重复3次相同的消息时会复读
|
||
""".strip(),
|
||
extra=PluginExtraData(
|
||
author="HibiKier",
|
||
version="0.3",
|
||
menu_type="其他",
|
||
plugin_type=PluginType.DEPENDANT,
|
||
tasks=[Task(module="fudu", name="复读")],
|
||
ignore_prompt=True,
|
||
configs=[
|
||
RegisterConfig(
|
||
key="FUDU_PROBABILITY",
|
||
value=0.8,
|
||
help="复读概率",
|
||
default_value=0.8,
|
||
type=float,
|
||
),
|
||
RegisterConfig(
|
||
module="_task",
|
||
key="DEFAULT_FUDU",
|
||
value=True,
|
||
help="被动 复读 进群默认开关状态",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="FUDU_TRIGGER_COUNT",
|
||
value=3,
|
||
help="触发复读所需的消息重复次数",
|
||
default_value=3,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="FUDU_BREAK_PROBABILITY",
|
||
value=0.2,
|
||
help="打断复读的概率（基于复读概率）",
|
||
default_value=0.2,
|
||
type=float,
|
||
),
|
||
RegisterConfig(
|
||
key="FUDU_BREAK_TEXTS",
|
||
value=["打断施法！"],
|
||
help="用于打断复读时随机发送的文本列表",
|
||
type=list[str],
|
||
),
|
||
RegisterConfig(
|
||
key="FUDU_BREAK_USE_IMAGE",
|
||
value=True,
|
||
help="是否启用图片作为打断复读的内容 (图片存放于 data/fudu/ 目录下)",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="FUDU_BREAK_USE_TEXT",
|
||
value=True,
|
||
help="是否启用文本作为打断复读的内容",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
],
|
||
).to_dict(),
|
||
)
|
||
|
||
|
||
class Fudu:
|
||
"""复读数据管理器，使用 LRU 策略限制内存占用"""
|
||
|
||
MAX_GROUPS = 500 # 最大缓存群组数量
|
||
|
||
def __init__(self):
|
||
self._data: OrderedDict[str, dict[str, Any]] = OrderedDict()
|
||
|
||
def _get_or_create(self, key: str) -> dict[str, Any]:
|
||
"""获取或创建群组数据，同时维护 LRU 顺序"""
|
||
if key in self._data:
|
||
self._data.move_to_end(key)
|
||
return self._data[key]
|
||
|
||
# 超出限制时删除最旧的条目
|
||
while len(self._data) >= self.MAX_GROUPS:
|
||
self._data.popitem(last=False)
|
||
|
||
trigger_count = Config.get_config("fudu", "FUDU_TRIGGER_COUNT", 3)
|
||
self._data[key] = {
|
||
"is_repeater": False,
|
||
"data": deque(maxlen=trigger_count),
|
||
"message_obj": None,
|
||
"reply_info": None,
|
||
}
|
||
return self._data[key]
|
||
|
||
def append(self, key: str, content: str, msg_obj: Any, reply_info: Any) -> None:
|
||
"""添加消息内容及原始消息对象"""
|
||
group_data = self._get_or_create(key)
|
||
group_data["data"].append(content)
|
||
group_data["message_obj"] = msg_obj
|
||
group_data["reply_info"] = reply_info
|
||
|
||
def clear(self, key: str) -> None:
|
||
"""清空群组的复读数据"""
|
||
group_data = self._get_or_create(key)
|
||
group_data["data"].clear()
|
||
group_data["is_repeater"] = False
|
||
group_data["message_obj"] = None
|
||
group_data["reply_info"] = None
|
||
|
||
def size(self, key: str) -> int:
|
||
"""获取当前消息数量"""
|
||
return len(self._get_or_create(key)["data"])
|
||
|
||
def check(self, key: str, content: str) -> bool:
|
||
"""检查内容是否与第一条消息相同"""
|
||
data_list = self._get_or_create(key)["data"]
|
||
return bool(data_list) and data_list[0] == content
|
||
|
||
def get_repeat_target(self, key: str) -> tuple[Any, Any]:
|
||
"""获取要复读的消息对象和回复信息"""
|
||
group_data = self._get_or_create(key)
|
||
return group_data["message_obj"], group_data["reply_info"]
|
||
|
||
def is_repeater(self, key: str) -> bool:
|
||
"""检查是否已经复读过"""
|
||
return self._get_or_create(key)["is_repeater"]
|
||
|
||
def set_repeater(self, key: str) -> None:
|
||
"""标记已复读"""
|
||
self._get_or_create(key)["is_repeater"] = True
|
||
|
||
|
||
_manager = Fudu()
|
||
base_config = Config.get("fudu")
|
||
|
||
|
||
def get_break_images() -> list[Path]:
|
||
"""获取打断复读用的图片列表"""
|
||
if not FUDU_IMAGE_PATH.exists():
|
||
return []
|
||
return [
|
||
f
|
||
for f in FUDU_IMAGE_PATH.iterdir()
|
||
if f.is_file() and f.suffix.lower() in IMAGE_EXTENSIONS
|
||
]
|
||
|
||
|
||
async def send_break_response() -> None:
|
||
"""发送打断复读的响应"""
|
||
response_pool: list[str | Path] = []
|
||
|
||
if base_config.get("FUDU_BREAK_USE_TEXT"):
|
||
if break_texts := base_config.get("FUDU_BREAK_TEXTS"):
|
||
response_pool.extend(break_texts)
|
||
|
||
if base_config.get("FUDU_BREAK_USE_IMAGE"):
|
||
response_pool.extend(get_break_images())
|
||
|
||
if not response_pool:
|
||
response_pool.append("打断施法！")
|
||
|
||
response = random.choice(response_pool)
|
||
|
||
if isinstance(response, Path):
|
||
file_data = f"base64://{base64.b64encode(response.read_bytes()).decode()}"
|
||
await _matcher.finish(
|
||
MessageSegment("image", {"file": file_data, "sub_type": "1"})
|
||
)
|
||
else:
|
||
await MessageUtils.build_message(response).finish()
|
||
|
||
|
||
async def rule(message: UniMsg, session: Uninfo, event: Event) -> bool:
|
||
"""消息匹配规则：仅匹配群聊中的有效消息"""
|
||
if not session.group:
|
||
return False
|
||
if event.is_tome():
|
||
return False
|
||
plain_text = message.extract_plain_text()
|
||
image_list = [m.url for m in message if isinstance(m, alcImg) and m.url]
|
||
if not plain_text and not image_list:
|
||
return False
|
||
return not await CommonUtils.task_is_block(session, "fudu")
|
||
|
||
|
||
_matcher = on_message(rule=rule, priority=999)
|
||
|
||
|
||
@_matcher.handle()
|
||
async def _(bot: Bot, message: UniMsg, event: Event, session: Uninfo):
|
||
# rule 已确保 session.group 存在
|
||
group_id = session.group.id # type: ignore
|
||
|
||
raw_message = event.get_message()
|
||
reply_info = getattr(event, "reply", None)
|
||
plain_text = message.extract_plain_text()
|
||
image_list = [m.url for m in message if isinstance(m, alcImg) and m.url]
|
||
|
||
# 检测虚空艾特
|
||
if plain_text and plain_text.startswith(f"@可爱的{BotConfig.self_nickname}"):
|
||
await MessageUtils.build_message("复制粘贴的虚空艾特？").send(reply_to=True)
|
||
|
||
# 计算图片哈希
|
||
img_hash = ""
|
||
if image_list:
|
||
temp_image_path = FUDU_CACHE_PATH / f"fudu_cache_{group_id}.jpg"
|
||
try:
|
||
if await AsyncHttpx.download_file(image_list[0], temp_image_path):
|
||
img_hash = await run_sync(get_img_hash)(temp_image_path)
|
||
except Exception as e:
|
||
logger.warning("下载复读图片以获取Hash时出错", "复读", e=e)
|
||
|
||
add_msg = f"{plain_text}|-|{img_hash}"
|
||
|
||
# 更新复读状态
|
||
if _manager.size(group_id) == 0 or _manager.check(group_id, add_msg):
|
||
_manager.append(group_id, add_msg, raw_message, reply_info)
|
||
else:
|
||
_manager.clear(group_id)
|
||
_manager.append(group_id, add_msg, raw_message, reply_info)
|
||
|
||
# 检查是否触发复读
|
||
trigger_count = base_config.get("FUDU_TRIGGER_COUNT")
|
||
if _manager.size(group_id) < trigger_count:
|
||
return
|
||
if _manager.is_repeater(group_id):
|
||
return
|
||
if random.random() >= base_config.get("FUDU_PROBABILITY"):
|
||
return
|
||
|
||
# 判断是否打断复读
|
||
if random.random() < base_config.get("FUDU_BREAK_PROBABILITY"):
|
||
_manager.clear(group_id)
|
||
await send_break_response()
|
||
|
||
# 执行复读
|
||
message_to_send, reply = _manager.get_repeat_target(group_id)
|
||
if not message_to_send:
|
||
return
|
||
|
||
_manager.set_repeater(group_id)
|
||
|
||
# 处理回复消息
|
||
if reply and isinstance(message_to_send, V11Message):
|
||
message_to_send.insert(0, MessageSegment.reply(reply.message_id))
|
||
|
||
await _matcher.finish(message_to_send)
|