from typing import Any from nonebot.adapters import Bot, Event from zhenxun.services.ai.utils.scope import ScopeBuilder class ContextUtils: """ 从底层依赖容器 (deps) 中提取运行环境信息的纯静态工具类。 """ @staticmethod def extract_user_id(deps: Any) -> str | None: if not deps: return None if hasattr(deps, "user_id") and getattr(deps, "user_id") is not None: return str(getattr(deps, "user_id")) event = getattr(deps, "event", None) if event: try: return str(event.get_user_id()) except Exception: return ( str(getattr(event, "user_id", "")) or str(getattr(event, "sender_id", "")) or None ) return None @staticmethod def extract_group_id(deps: Any) -> str | None: if not deps: return None if hasattr(deps, "group_id") and getattr(deps, "group_id") is not None: return str(getattr(deps, "group_id")) event = getattr(deps, "event", None) if event: return str(getattr(event, "group_id", "")) or None return None @staticmethod def extract_platform(deps: Any) -> str: if not deps: return "unknown" if hasattr(deps, "platform") and getattr(deps, "platform") is not None: return str(getattr(deps, "platform")) bot = getattr(deps, "bot", None) if bot: from zhenxun.utils.platform import PlatformUtils return PlatformUtils.get_platform(bot) return "unknown" @staticmethod def extract_concurrency_lock_id( context: Any, scope: Any, default_session_id: str ) -> str: from zhenxun.services.ai.flow.base import ConcurrencyScope scope = scope or ConcurrencyScope.GROUP ns = getattr(getattr(context, "session", None), "namespace", "global") ns_suffix = f"_{ns}" if ns and ns != "global" else "" if scope == ConcurrencyScope.GLOBAL: return f"lock_global{ns_suffix}" elif scope == ConcurrencyScope.GROUP: gid = ContextUtils.extract_group_id(getattr(context, "deps", None)) uid = ContextUtils.extract_user_id(getattr(context, "deps", None)) return ( f"lock_group_{gid}{ns_suffix}" if gid else f"lock_user_{uid}{ns_suffix}" ) elif scope == ConcurrencyScope.USER: uid = ContextUtils.extract_user_id(getattr(context, "deps", None)) return ( f"lock_user_{uid}{ns_suffix}" if uid else f"lock_default_user{ns_suffix}" ) else: return f"lock_session_{default_session_id}" @staticmethod def generate_session_meta( bot: Bot, event: Event, scope_builder: ScopeBuilder | None = None, prefix: str = "", namespace: str | None = None, agent_name: str | None = None, ) -> Any: """根据事件和隔离级别,自动提取生成基于路径作用域的 SessionMetadata""" from zhenxun.services.ai.context.memory.types import ( Isolation, SessionMetadata, ) from zhenxun.services.ai.run.context import NoneBotDeps if scope_builder is None: scope_builder = Isolation.AGENT_USER() deps = NoneBotDeps(bot=bot, event=event) selector = scope_builder.resolve( deps=deps, prefix=prefix, default_namespace=namespace, default_agent=agent_name, ) parts = selector.get_scope_parts() session_id = selector.scope_prefix scope_prefix = selector.scope_prefix all_scopes = {"/"} current_path = "" for part in parts: current_path += f"/{part}" all_scopes.add(current_path) accessible_scopes = list(all_scopes) accessible_scopes.sort(key=lambda x: len(x.split("/"))) return SessionMetadata( session_id=session_id, scope_prefix=scope_prefix, accessible_scopes=accessible_scopes, selector=selector, isolation_level=scope_builder, ) class PermissionUtils: """运行时权限校验通用工具类""" @staticmethod async def check_superuser(context: Any) -> bool: bot = context.get_bot() event = context.get_event() if bot and event: from nonebot.permission import SUPERUSER return await SUPERUSER(bot, event) return False @staticmethod async def check_admin_level(context: Any, min_level: int) -> bool: if await PermissionUtils.check_superuser(context): return True user_id = context.get_user_id() group_id = context.get_group_id() if not user_id: return False from zhenxun.services.cache.runtime_cache import LevelUserMemoryCache global_user, group_users = await LevelUserMemoryCache.get_levels( user_id, group_id ) user_level = global_user.user_level if global_user else 0 if group_id and group_users: user_level = max(user_level, group_users.user_level) return user_level >= min_level