from __future__ import annotations
from typing import AsyncGenerator

from .. import operator, entities, cmdmgr


@operator.operator_class(name="func", help="查看所有已注册的内容函数", usage='!func')
class FuncOperator(operator.CommandOperator):
    async def execute(
        self, context: entities.ExecuteContext
    ) -> AsyncGenerator[entities.CommandReturn, None]:
        reply_str = "当前已加载的内容函数: \n\n"

        index = 1

        all_functions = await self.ap.tool_mgr.get_all_functions()

        for func in all_functions:
            reply_str += "{}. {}{}:\n{}\n\n".format(
                index,
                ("(已禁用) " if not func.enable else ""),
                func.name,
                func.description,
            )
            index += 1

        yield entities.CommandReturn(text=reply_str)