QChatGPT/pkg/command/operator.py

from __future__ import annotations

import typing
import abc

from ..core import app, entities as core_entities
from . import entities


preregistered_operators: list[typing.Type[CommandOperator]] = []
"""预注册命令算子列表。在初始化时，所有算子类会被注册到此列表中。"""


def operator_class(
    name: str,
    help: str = "",
    usage: str = None,
    alias: list[str] = [],
    privilege: int=1,  # 1为普通用户，2为管理员
    parent_class: typing.Type[CommandOperator] = None
) -> typing.Callable[[typing.Type[CommandOperator]], typing.Type[CommandOperator]]:
    """命令类装饰器
    
    Args:
        name (str): 名称
        help (str, optional): 帮助信息. Defaults to "".
        usage (str, optional): 使用说明. Defaults to None.
        alias (list[str], optional): 别名. Defaults to [].
        privilege (int, optional): 权限，1为普通用户可用，2为仅管理员可用. Defaults to 1.
        parent_class (typing.Type[CommandOperator], optional): 父节点，若为None则为顶级命令. Defaults to None.

    Returns:
        typing.Callable[[typing.Type[CommandOperator]], typing.Type[CommandOperator]]: 注册后的命令类
    """

    def decorator(cls: typing.Type[CommandOperator]) -> typing.Type[CommandOperator]:
        assert issubclass(cls, CommandOperator)
        
        cls.name = name
        cls.alias = alias
        cls.help = help
        cls.usage = usage
        cls.parent_class = parent_class
        cls.lowest_privilege = privilege

        preregistered_operators.append(cls)

        return cls

    return decorator


class CommandOperator(metaclass=abc.ABCMeta):
    """命令算子抽象类
    """

    ap: app.Application

    name: str
    """名称，搜索到时若符合则使用"""

    path: str
    """路径，所有父节点的name的连接，用于定义命令权限"""

    alias: list[str]
    """同name"""

    help: str
    """此节点的帮助信息"""

    usage: str = None

    parent_class: typing.Union[typing.Type[CommandOperator], None] = None
    """父节点类。标记以供管理器在初始化时编织父子关系。"""

    lowest_privilege: int = 0
    """最低权限。若权限低于此值，则不予执行。"""

    children: list[CommandOperator]
    """子节点。解析命令时，若节点有子节点，则以下一个参数去匹配子节点，
    若有匹配中的，转移到子节点中执行，若没有匹配中的或没有子节点，执行此节点。"""

    def __init__(self, ap: app.Application):
        self.ap = ap
        self.children = []

    async def initialize(self):
        pass

    @abc.abstractmethod
    async def execute(
        self,
        context: entities.ExecuteContext
    ) -> typing.AsyncGenerator[entities.CommandReturn, None]:
        pass