QChatGPT/pkg/command/operators/cmd.py

from __future__ import annotations

import typing

from .. import operator, entities, cmdmgr, errors


@operator.operator_class(
    name="cmd",
    help='显示命令列表',
    usage='!cmd\n!cmd <命令名称>'
)
class CmdOperator(operator.CommandOperator):
    """命令列表
    """

    async def execute(
        self,
        context: entities.ExecuteContext
    ) -> typing.AsyncGenerator[entities.CommandReturn, None]:
        """执行
        """
        if len(context.crt_params) == 0:
            reply_str = "当前所有命令: \n\n"

            for cmd in self.ap.cmd_mgr.cmd_list:
                if cmd.parent_class is None:
                    reply_str += f"{cmd.name}: {cmd.help}\n"
        
            reply_str += "\n使用 !cmd <命令名称> 查看命令的详细帮助"

            yield entities.CommandReturn(text=reply_str.strip())
        
        else:
            cmd_name = context.crt_params[0]

            cmd = None

            for _cmd in self.ap.cmd_mgr.cmd_list:
                if (cmd_name == _cmd.name or cmd_name in _cmd.alias) and (_cmd.parent_class is None):
                    cmd = _cmd
                    break

            if cmd is None:
                yield entities.CommandReturn(error=errors.CommandNotFoundError(cmd_name))
            else:
                reply_str = f"{cmd.name}: {cmd.help}\n\n"
                reply_str += f"使用方法: \n{cmd.usage}"

                yield entities.CommandReturn(text=reply_str.strip())