zhenxun_bot/utils/manager/models.py

from enum import Enum
from pathlib import Path
from typing import Any, Dict, List, Literal, Optional, Tuple, Union

from pydantic import BaseModel

from configs.config import Config
from configs.utils import Config as zConfig


class AdminSetting(BaseModel):
    """
    管理员设置
    """

    level: int = 5
    cmd: Optional[List[str]]


class BaseGroup(BaseModel):
    """
    基础群聊信息
    """

    level: int = Config.get_config("group_manager", "DEFAULT_GROUP_LEVEL")  # 群等级
    status: bool = Config.get_config(
        "group_manager", "DEFAULT_GROUP_BOT_STATUS"
    )  # 总开关状态
    close_plugins: List[str] = []  # 已关闭插件
    group_task_status: Dict[str, bool] = {}  # 被动状态


class BaseData(BaseModel):
    """
    群基本信息
    """

    white_group: List[int] = []
    """白名单"""
    close_task: List[str] = []
    """全局关闭的被动任务"""
    group_manager: Dict[str, BaseGroup] = {}
    """群组管理"""
    task: Dict[str, str] = {}
    """被动任务 英文:中文名"""

    def __len__(self) -> int:
        return len(self.group_manager)


class PluginBlock(BaseModel):
    """
    插件阻断
    """

    status: bool = True  # 限制状态
    check_type: Literal["private", "group", "all"] = "all"  # 检查类型
    limit_type: Literal["user", "group"] = "user"  # 监听对象
    rst: Optional[str]  # 阻断时回复


class PluginCd(BaseModel):
    """
    插件阻断
    """

    cd: int = 5  # cd
    status: bool = True  # 限制状态
    check_type: Literal["private", "group", "all"] = "all"  # 检查类型
    limit_type: Literal["user", "group"] = "user"  # 监听对象
    rst: Optional[str]  # 阻断时回复


class PluginCount(BaseModel):
    """
    插件阻断
    """

    max_count: int  # 次数
    status: bool = True  # 限制状态
    limit_type: Literal["user", "group"] = "user"  # 监听对象
    rst: Optional[str]  # 阻断时回复


class PluginSetting(BaseModel):
    """
    插件设置
    """

    cmd: List[str] = []  # 命令 或 命令别名
    default_status: bool = True  # 默认开关状态
    level: int = 5  # 功能权限等级
    limit_superuser: bool = False  # 功能状态是否限制超级用户
    plugin_type: Tuple[Union[str, int], ...] = ("normal",)  # 插件类型
    cost_gold: int = 0  # 需要消费的金币


class Plugin(BaseModel):
    """
    插件数据
    """

    plugin_name: str  # 模块名
    status: Optional[bool] = True  # 开关状态
    error: Optional[bool] = False  # 是否加载报错
    block_type: Optional[str] = None  # 关闭类型
    author: Optional[str] = None  # 作者
    version: Optional[Union[int, str]] = None  # 版本


class PluginType(Enum):
    """
    插件类型
    """

    NORMAL = "normal"
    ADMIN = "admin"
    HIDDEN = "hidden"
    SUPERUSER = "superuser"


class PluginData(BaseModel):
    model: str
    name: str
    plugin_type: PluginType  # 插件内部类型，根据name [Hidden] [Admin] [SUPERUSER]
    usage: Optional[str]
    superuser_usage: Optional[str]
    des: Optional[str]
    task: Optional[Dict[str, str]]
    menu_type: Tuple[Union[str, int], ...] = ("normal",)  # 菜单类型
    plugin_setting: Optional[PluginSetting]
    plugin_cd: Optional[PluginCd]
    plugin_block: Optional[PluginBlock]
    plugin_count: Optional[PluginCount]
    plugin_resources: Optional[Dict[str, Union[str, Path]]]
    plugin_configs: Optional[Dict[str, zConfig]]
    plugin_status: Plugin

    class Config:
        arbitrary_types_allowed = True

    def __eq__(self, other: "PluginData"):
        return (
            isinstance(other, PluginData)
            and self.name == other.name
            and self.menu_type == other.menu_type
        )

    def __hash__(self):
        return hash(self.name + str(self.menu_type[0]))