mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 08:39:59 +08:00
6da4f27b12
* ✨ feat(auth): 添加群组和机器人唤醒命令支持，优化权限检查逻辑 ✨ feat(llm): 增加额外请求头配置，改进API适配器请求头处理 * 🚨 auto fix by pre-commit hooks * ``` fix(auth): 优化bot权限验证逻辑并改进错误提示 - 将bot存在性检查与状态检查分离，提供更精确的错误信息 - 修复当bot为None时的状态访问问题 - 移除不必要的注释，保持代码简洁 - 优化日志记录的位置和条件判断 ``` --------- Co-authored-by: ATTomatoo <1126160939@qq.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
import re
|
|
import time
|
|
|
|
from zhenxun.models.group_console import GroupConsole
|
|
from zhenxun.models.plugin_info import PluginInfo
|
|
from zhenxun.services.cache.runtime_cache import GroupSnapshot
|
|
from zhenxun.services.log import logger
|
|
|
|
from .config import LOGGER_COMMAND, WARNING_THRESHOLD, SwitchEnum
|
|
from .exception import SkipPluginException
|
|
|
|
_GROUP_WAKE_PATTERN = re.compile(r"^醒来$", re.IGNORECASE)
|
|
_GROUP_WAKE_CANONICAL_PATTERN = re.compile(r"^group-status\s+wake$", re.IGNORECASE)
|
|
|
|
|
|
def _is_group_wake_command(plugin: PluginInfo, text: str) -> bool:
|
|
if "plugin_switch" not in (plugin.module or ""):
|
|
return False
|
|
normalized = re.sub(r"\s+", " ", (text or "").strip())
|
|
if not normalized:
|
|
return False
|
|
if (
|
|
_GROUP_WAKE_PATTERN.match(normalized) is not None
|
|
or _GROUP_WAKE_CANONICAL_PATTERN.match(normalized) is not None
|
|
):
|
|
return True
|
|
# 兼容 to_me 前缀场景：如“真寻 醒来”
|
|
tokens = normalized.split(" ")
|
|
return len(tokens) == 2 and tokens[-1] == SwitchEnum.ENABLE
|
|
|
|
|
|
async def auth_group(
|
|
plugin: PluginInfo,
|
|
group: GroupConsole | GroupSnapshot | None,
|
|
text: str | None,
|
|
group_id: str | None,
|
|
):
|
|
"""群黑名单检测 群总开关检测
|
|
|
|
参数:
|
|
plugin: PluginInfo
|
|
group: GroupConsole
|
|
message: UniMsg
|
|
"""
|
|
if not group_id:
|
|
return
|
|
|
|
start_time = time.time()
|
|
|
|
try:
|
|
text = text or ""
|
|
|
|
if not group:
|
|
raise SkipPluginException("群组信息不存在...")
|
|
if group.level < 0:
|
|
raise SkipPluginException("群组黑名单, 目标群组群权限权限-1...")
|
|
if not _is_group_wake_command(plugin, text) and not group.status:
|
|
raise SkipPluginException("群组休眠状态...")
|
|
if plugin.level > group.level:
|
|
raise SkipPluginException(
|
|
f"{plugin.name}({plugin.module}) 群等级限制，"
|
|
f"该功能需要的群等级: {plugin.level}..."
|
|
)
|
|
finally:
|
|
# 记录执行时间
|
|
elapsed = time.time() - start_time
|
|
if elapsed > WARNING_THRESHOLD: # 记录耗时超过500ms的检查
|
|
logger.warning(
|
|
f"auth_group 耗时: {elapsed:.3f}s, plugin={plugin.module}",
|
|
LOGGER_COMMAND,
|
|
group_id=group_id,
|
|
)
|