import re
import asyncio
from random import choice

from nonebot.typing import T_State
from nonebot.adapters.cqhttp import (
    Bot,
    MessageEvent,
    GroupMessageEvent,
    PrivateMessageEvent,
)

from ATRI.service import Service as sv
from ATRI.rule import is_in_service
from ATRI.utils.limit import is_too_exciting

from .data_source import Tsuma


__doc__ = """
好欸！是老婆！
权限组：所有人
用法：
  抽老婆  # 获取一位老婆
  查老婆  # 查询老婆，如+at对象可查询对方
  我要离婚  # 离婚...
"""

roll_wife = sv.on_command(cmd="抽老婆", docs=__doc__, rule=is_in_service("抽老婆"))


@roll_wife.handle()
async def _roll_wife(bot: Bot, event: GroupMessageEvent) -> None:
    user = event.user_id
    gender = event.sender.sex
    group = event.group_id
    user_name = await bot.get_group_member_info(group_id=group, user_id=user)
    user_name = user_name["nickname"]
    run = is_too_exciting(user, 1, seconds=5)
    if not run:
        return

    check_repo, if_h = Tsuma.check_tsuma(str(user))
    if if_h:
        await roll_wife.finish(check_repo)

    msg = "5秒后咱将随机抽取一位群友成为\n" f"{user_name} 的老婆！究竟是谁呢~？"
    await bot.send(event, msg)
    await asyncio.sleep(5)

    async def get_luck_user():
        luck_list = await bot.get_group_member_list(group_id=group)
        return choice(luck_list)

    while True:
        luck_user = await get_luck_user()
        luck_qq = luck_user["user_id"]
        if user != luck_qq:
            break

    luck_gender = luck_user["sex"]
    luck_user = luck_user["nickname"]
    d = {
        "nickname": user_name,
        "gender": gender,
        "lassie": {"nickname": luck_user, "qq": luck_qq, "gender": luck_gender},
    }

    if str(luck_qq) == str(event.self_id):
        Tsuma.got_tsuma(str(user), d)
        msg = "老婆竟是我自己~❤"
    else:
        msg = Tsuma.got_tsuma(str(user), d)

    await roll_wife.finish(msg)


@roll_wife.handle()
async def _no_pr(bot: Bot, event: PrivateMessageEvent) -> None:
    await roll_wife.finish("对8起...该功能只对群聊开放（")


inquire_wife = sv.on_command(cmd="查老婆", rule=is_in_service("抽老婆"))


@inquire_wife.handle()
async def _inq_wife(bot: Bot, event: MessageEvent) -> None:
    msg = str(event.message).split(" ")
    if msg[0] == "":
        user = str(event.user_id)
        await inquire_wife.finish(Tsuma.inquire_tsuma(user))
    else:
        aim = re.findall(r"qq=(.*?)]", msg[0])[0]
        await inquire_wife.finish(Tsuma.inquire_tsuma(aim).replace("你", "ta"))


want_divorce = sv.on_command(cmd="我要离婚", rule=is_in_service("抽老婆"))


@want_divorce.handle()
async def _want_div(bot: Bot, event: MessageEvent, state: T_State) -> None:
    msg = str(event.message).strip()
    if msg:
        state["is_d"] = msg


@want_divorce.got("is_d", prompt="你确定吗？(是/否)")
async def _deal_div(bot: Bot, event: MessageEvent, state: T_State) -> None:
    msg = state["is_d"]
    user = str(event.user_id)
    name = event.sender.nickname

    if msg in ["是", "确定"]:
        await want_divorce.finish(Tsuma.divorce(user))
    else:
        await want_divorce.finish(f"({name})回心转意了！")