import json
from datetime import datetime, timedelta

from nonebot.params import ArgPlainText
from nonebot.adapters.onebot.v11 import PrivateMessageEvent, GroupMessageEvent

from ATRI import conf
from ATRI.log import log
from ATRI.service import Service
from ATRI.permission import MASTER
from ATRI.message import MessageBuilder
from ATRI.exceptions import WriteFileError
from ATRI.utils.apscheduler import scheduler

from .data_source import Console, CONSOLE_DIR
from .models import AuthData


__AUTH_FILE_PATH = CONSOLE_DIR / "data.json"


plugin = (
    Service("控制台")
    .document("前端管理页面")
    .only_admin(True)
    .permission(MASTER)
    .main_cmd("/con")
)


def __del_auth_key():
    with open(__AUTH_FILE_PATH, "w", encoding="utf-8") as w:
        w.write(json.dumps({}))
    log.warning("控制台验证密钥已过期")


gen_console_key = plugin.cmd_as_group("auth", "获取进入网页后台的凭证")


@gen_console_key.got("is_pub_n", "咱的运行环境是否有公网(y/n)")
async def _(event: PrivateMessageEvent, is_pub_n: str = ArgPlainText("is_pub_n")):
    is_access_key = conf.BotConfig.access_token
    if not is_access_key:
        await gen_console_key.finish(
            MessageBuilder("缺少设置: access_token")
            .text("请先填写该内容, 以保证ATRI与协议端链接的安全性")
            .text("填写并重启, 方可启用控制台")
            .text("Tip: 该内容请尽可能地复杂, 请勿使用中文")
        )

    if not __AUTH_FILE_PATH.is_file():
        with open(__AUTH_FILE_PATH, "w", encoding="utf-8") as w:
            w.write(json.dumps(dict()))
    else:
        raw_data = json.loads(__AUTH_FILE_PATH.read_bytes())

        data = raw_data.get("data")
        if data:
            now_time = datetime.now().timestamp()
            if now_time < data["dead_time"] and data.get("dead_time"):
                raw_last_time = data["dead_time"] - now_time
                last_time = datetime.fromtimestamp(raw_last_time).minute
                await gen_console_key.finish(
                    MessageBuilder("之前生成的密钥还在有效时间内奥")
                    .text(f"Token: {data['token']}")
                    .text(f"剩余有效时间: {last_time} min")
                )
            else:
                with open(__AUTH_FILE_PATH, "w", encoding="utf-8") as w:
                    w.write(json.dumps(dict()))

    if is_pub_n != "y":
        host = str(await Console().get_host_ip(False))
        await gen_console_key.send("没有公网吗...嗯知道了")
    else:
        host = str(await Console().get_host_ip(True))

    port = conf.BotConfig.port
    token = Console().get_random_str(20)
    dead_time = datetime.now() + timedelta(minutes=15)

    data = json.loads(__AUTH_FILE_PATH.read_bytes())
    data["data"] = AuthData(token=token, dead_time=dead_time.timestamp()).dict()
    with open(__AUTH_FILE_PATH, "w", encoding="utf-8") as w:
        w.write(json.dumps(data))

    msg = (
        MessageBuilder("控制台信息已生成!")
        .text(f"请访问: {host}:{port}")
        .text(f"Token: {token}")
        .text("该 token 有效时间为 15min")
    )

    scheduler.add_job(
        __del_auth_key,
        name="清除后台验证凭证",
        next_run_time=dead_time,  # type: ignore
        misfire_grace_time=15,  # type: ignore
    )

    await gen_console_key.finish(msg)


@gen_console_key.handle()
async def _(event: GroupMessageEvent):
    await gen_console_key.finish("请私戳咱获取（")


del_console_key = plugin.cmd_as_group("del", "销毁进入网页后台的凭证")


@del_console_key.got("is_sure_d", "...你确定吗(y/n)")
async def _(is_sure: str = ArgPlainText("is_sure_d")):
    if is_sure != "y":
        await del_console_key.finish("反悔了呢...")

    df = CONSOLE_DIR / "data.json"
    if not df.is_file():
        await del_console_key.finish("你还没向咱索取凭证呢...私戳咱键入 /con.auth 以获取")

    try:
        data = json.loads(df.read_bytes())

        del data["data"]

        with open(df, "w", encoding="utf-8") as w:
            w.write(json.dumps(data))
    except Exception:
        await del_console_key.send("销毁失败了...请至此处自行删除文件:\n" + str(df))
        raise WriteFileError("Writing / Reading file: " + str(df) + " failed!")

    await del_console_key.finish("销毁成功！如需再次获取: /con.auth")


from ATRI import driver as dr
from .data_source import init_resource
from .driver import init_driver


dr().on_startup(init_resource)
dr().on_startup(init_driver)