import asyncio
from random import randint

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

from ATRI.service import Service as sv


__doc__ = """
群发内容
权限组：维护者
用法：
  广播 内容
"""

broadcast = sv.on_command(
    cmd="广播",
    docs=__doc__,
    permission=SUPERUSER
)

@broadcast.args_parser  # type: ignore
async def _broadcast_load(bot: Bot,
                          event: MessageEvent,
                          state: T_State) -> None:
    msg = str(event.message)
    quit_list = ['算了', '罢了', '取消']
    if msg in quit_list:
        await broadcast.finish('好吧...')
    if not msg:
        await broadcast.reject('想群发啥呢0w0')
    else:
        state['msg'] = msg

@broadcast.handle()
async def _broadcast(bot: Bot, event: MessageEvent, state: T_State) -> None:
    msg = str(event.message).strip()
    if msg:
        state['msg'] = msg
    
@broadcast.got('msg', prompt='请告诉咱需要群发的内容~！')
async def _deal_broadcast(bot: Bot,
                          event: MessageEvent,
                          state: T_State) -> None:
    msg = state['msg']
    group_list = await bot.get_group_list()
    succ_list = []
    err_list = []
    
    for group in group_list:
        await asyncio.sleep(randint(0, 2))
        try:
            await bot.send_group_msg(group_id=group["group_id"],
                                     message=msg)
        except BaseException:
            err_list.append(group["group_id"])
    
    msg0 = ""
    for i in err_list:
        msg0 += f"    {i}\n"
    
    repo_msg = (
        f"推送消息：\n{msg}\n"
        "————————\n"
        f"总共：{len(group_list)}\n"
        f"成功推送：{len(succ_list)}\n"
        f"失败[{len(err_list)}]个：\n"
    ) + msg0

    await broadcast.finish(repo_msg)