from nonebot.typing import T_State
from nonebot.adapters.cqhttp import Bot, MessageEvent

from ATRI.rule import to_bot
from .data_source import Helper


main_help = Helper().on_command(
    "菜单", "获取食用bot的方法", rule=to_bot(), aliases={"/help", "menu"}
)


@main_help.handle()
async def _main_help(bot: Bot, event: MessageEvent):
    repo = Helper().menu()
    await main_help.finish(repo)


about_me = Helper().on_command("关于", "获取关于bot的信息", rule=to_bot(), aliases={"about"})


@about_me.handle()
async def _about_me(bot: Bot, event: MessageEvent):
    repo = Helper().about()
    await about_me.finish(repo)


service_list = Helper().on_command("服务列表", "查看所有可用服务", rule=to_bot(), aliases={"功能列表"})


@service_list.handle()
async def _service_list(bot: Bot, event: MessageEvent):
    repo = Helper().service_list()
    await service_list.finish(repo)


service_info = Helper().on_command("帮助", "获取服务详细帮助", rule=to_bot())


@service_info.handle()
async def _ready_service_info(bot: Bot, event: MessageEvent, state: T_State):
    msg = str(event.message).split(" ")
    service = msg[0]
    try:
        cmd = msg[1]
    except BaseException:
        cmd = str()

    if not cmd:
        repo = Helper().service_info(service)
        await service_info.finish(repo)

    repo = Helper().cmd_info(service, cmd)
    await service_info.finish(repo)