import json
from random import choice

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

from ATRI.service import Service as sv
from ATRI.utils.request import get_bytes
from ATRI.exceptions import RequestError

from .data_source import SetuData


API_URL: str = "https://api.kyomotoi.moe/api/pixiv/illust?id="


__doc__ = """
为本地添加涩图！
权限组：维护者
用法：
  添加涩图 (pid)
补充：
  pid: Pixiv 作品id
"""


add_setu = sv.on_command(cmd="添加涩图", docs=__doc__, permission=SUPERUSER)


@add_setu.args_parser  # type: ignore
async def _load_add_setu(bot: Bot, event: MessageEvent, state: T_State) -> None:
    msg = str(event.message).strip()
    cancel = ["算了", "罢了"]
    if msg in cancel:
        await add_setu.finish("好吧...")
    if not msg:
        await add_setu.reject("涩图(pid)速发！")
    else:
        state["setu_add"] = msg


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


@add_setu.got("setu_add", prompt="涩图(pid)速发！")
async def _deal_add_setu(bot: Bot, event: MessageEvent, state: T_State) -> None:
    pid = state["setu_add"]

    URL = API_URL + pid
    try:
        data = json.loads(await get_bytes(URL))["illust"]
    except RequestError:
        raise RequestError("Request failed!")

    try:
        pic = data["meta_single_page"]["original_image_url"].replace(
            "pximg.net", "pixiv.cat"
        )
    except Exception:
        pic = choice(data["meta_pages"])["image_urls"]["original"].replace(
            "pximg.net", "pixiv.cat"
        )

    d = {
        "pid": pid,
        "title": data["title"],
        "tags": str(data["tags"]),
        "user_id": data["user"]["id"],
        "user_name": data["user"]["name"],
        "user_account": data["user"]["account"],
        "url": pic,
    }
    await SetuData.add_data(d)

    show_img = data["image_urls"]["medium"].replace("pximg.net", "pixiv.cat")
    msg = (
        "好欸！是新涩图：\n"
        f"Pid: {pid}\n"
        f"Title: {data['title']}\n"
        f"{MessageSegment.image(show_img)}"
    )
    await add_setu.finish(Message(msg))


__doc__ = """
删除涩图！
权限组：维护者
用法：
  删除涩图 (pid)
补充：
  pid: Pixiv 作品id
"""


del_setu = sv.on_command(cmd="删除涩图", docs=__doc__, permission=SUPERUSER)


@del_setu.args_parser  # type: ignore
async def _load_del_setu(bot: Bot, event: MessageEvent, state: T_State) -> None:
    msg = str(event.message).strip()
    cancel = ["算了", "罢了"]
    if msg in cancel:
        await add_setu.finish("好吧...")
    if not msg:
        await add_setu.reject("涩图(pid)速发！")
    else:
        state["setu_del"] = msg


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


@del_setu.got("setu_del", prompt="涩图(pid)速发！")
async def _deal_del_setu(bot: Bot, event: MessageEvent, state: T_State) -> None:
    pid = int(state["setu_del"])
    await SetuData.del_data(pid)
    await del_setu.finish(f"涩图({pid})已删除...")


count_setu = sv.on_command(cmd="涩图总量", permission=SUPERUSER)


@count_setu.handle()
async def _count_setu(bot: Bot, event: MessageEvent) -> None:
    msg = f"咱本地搭载了 {await SetuData.count()} 张涩图！"
    await count_setu.finish(msg)