#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
'''
@File    :   __init__.py
@Time    :   2020/11/07 14:20:08
@Author  :   Kyomotoi
@Contact :   kyomotoiowo@gmail.com
@Github  :   https://github.com/Kyomotoi
@License :   Copyright © 2018-2020 Kyomotoi, All Rights Reserved.
'''
__author__ = 'kyomotoi'

import os
import re
import json
import random
from pathlib import Path
from time import strftime
from zipfile import PyZipFile
from typing import Tuple, Dict, List
from datetime import datetime, timedelta

from nonebot.plugin import on_command
from nonebot.adapters.cqhttp import Bot, Event

from utils.utils_error import errorRepo
from utils.utils_rule import check_banlist, check_switch

from .roll import roll_dice
from .genshin import GetInfo, JsonAnalysis

file = Path('.') / 'ATRI' / 'data' / 'data_IDcard' / 'main.bin'


def infoID() -> Tuple[Dict[str, List[str]], Dict[str, str]]:
    with PyZipFile(os.path.abspath(file), "r") as zipFile:
        with zipFile.open("name.json", "r") as f:
            name = json.loads(f.read().decode())
        with zipFile.open("area.json", "r") as f:
            area = json.loads(f.read().decode())
    return name, area


NAME, AREA = infoID()

BIRTH_BEGIN = datetime(*[1980, 10, 10])  # type: ignore
BIRTH_END = datetime(*[2002, 10, 10])  # type: ignore


def numberID(area: int, sex: int, birth: int) -> str:
    def checkSum(fullCode: str) -> int or str:
        assert len(fullCode) == 17
        checkSum = sum([((1 << (17 - i)) % 11) * int(fullCode[i])
                        for i in range(0, 17)])
        checkDigit = (12 - (checkSum % 11)) % 11
        if checkDigit < 10:
            return checkDigit
        else:
            return "X"

    orderCode = str(random.randint(10, 99))
    sexCode = str(random.randrange(sex, 10, step=2))
    fullCode = str(area) + str(birth) + str(orderCode) + str(sexCode)
    fullCode += str(checkSum(fullCode))
    return fullCode


plugin_name_0 = "one-key-adult"
generateID = on_command("我要转大人，一天打25小时游戏",
                        rule=check_banlist() & check_switch(plugin_name_0))


@generateID.handle()  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    id_card_area = int(random.choice(list(AREA.keys())))
    id_card_area_name = AREA[str(id_card_area)]
    id_card_year_old = timedelta(
        days=random.randint(0, (BIRTH_END - BIRTH_BEGIN).days) + 1)
    id_card_birth_day = strftime("%Y%m%d",
                                 (BIRTH_BEGIN + id_card_year_old).timetuple())
    id_card_sex = random.choice([0, 1])
    id_card_name = random.choice(NAME[{0: "female", 1: "male"}[id_card_sex]])
    id_card_id = numberID(id_card_area, id_card_sex, id_card_birth_day)

    msg0 = "恭喜，你已经成大人了！\n"
    msg0 += "这是你一天25h游戏的通行证：\n"
    msg0 += f"NumberID: {id_card_id}\n"
    msg0 += f"Sex: {'男' if id_card_sex == 1 else '女'}\n"
    msg0 += f"Name: {id_card_name} || Address: {id_card_area_name}\n"
    msg0 += "注: 1、以上信息根据国家公开标准生成，非真实信息。\n"
    msg0 += "      2、不适用于网易和腾讯。"

    await generateID.finish(msg0)


rollD = on_command("roll", rule=check_banlist())


@rollD.handle()  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    args = str(event.message).strip()

    if args:
        state['resu'] = args


@rollD.got("resu",
           prompt="roll 参数不能为空~！\ndemo：1d10 或 2d10+2d10")  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    resu = state['resu']
    match = re.match(r'^([\dd+\s]+?)$', resu)

    if not match:
        await rollD.finish("请输入正确的参数！！\ndemo：1d10 或 2d10+2d10")

    await rollD.finish(roll_dice(resu))


plugin_name_1 = 'genshin-search'
genshinInfo = on_command('genshin',
                         rule=check_banlist() & check_switch(plugin_name_1))


@genshinInfo.handle()  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    args = str(event.message).strip()

    if args:
        state['uid'] = args


@genshinInfo.got('uid', prompt='请告诉咱需要查询的UID，暂时只支持国服嗷~（')  # type: ignore
async def _(bot: Bot, event: Event, state: dict) -> None:
    uid = str(state['uid'])

    if (len(uid) == 9 and uid[0] == '1'):
        await bot.send(event, '别急，在搜索了！')
        uid_info = ''

        try:
            uid_info = JsonAnalysis(GetInfo(uid))
        except:
            await genshinInfo.finish(errorRepo("数据请求错误，原因可能为ID输入错误或不存在"))

        msg0 = f'{uid} Genshin Info:\n'
        msg0 += uid_info
        await genshinInfo.finish(msg0)

    else:
        await genshinInfo.finish('UID检查未通过，请确保此ID为9位数或者是否为国服ID~！')