import os.path
import shutil
import traceback

from core.builtins import Bot, Image
from core.component import module
from core.utils.cache import random_cache_path
from core.utils.http import get_url, download_to_cache
from .dbutils import PgrBindInfoManager
from .game_record import parse_game_record
from .genb19 import drawb19
from .update import update_assets, p_headers

phi = module('phigros', developers=['OasisAkari'], desc='{phigros.help.desc}',
             alias={'p': 'phigros', 'pgr': 'phigros', 'phi': 'phigros'})


@phi.command('bind <sessiontoken> {{phigros.help.bind}}')
async def _(msg: Bot.MessageSession, sessiontoken: str):
    need_revoke = False
    send_msg = []
    if msg.target.target_from in [
        'QQ|Group',
        'QQ|Guild',
        'Discord|Channel',
        'Telegram|group',
        'Telegram|supergroup',
        'Kook|GROUP']:
        send_msg.append(await msg.send_message(msg.locale.t("phigros.message.bind.warning"), quote=False))
        need_revoke = True
    headers = p_headers.copy()
    headers['X-LC-Session'] = sessiontoken
    get_user_info = await get_url('https://rak3ffdi.cloud.tds1.tapapis.cn/1.1/users/me', headers=headers, fmt='json')
    if 'nickname' in get_user_info:
        bind = PgrBindInfoManager(msg).set_bind_info(sessiontoken=sessiontoken, username=get_user_info['nickname'])
        if bind:
            send_msg.append(await msg.send_message(msg.locale.t("phigros.message.bind.success",
                                                                username=get_user_info['nickname']), quote=False))
        if need_revoke:
            await msg.sleep(15)
            for i in send_msg:
                await i.delete()


@phi.command('unbind {{phigros.help.unbind}}')
async def _(msg: Bot.MessageSession):
    if PgrBindInfoManager(msg).remove_bind_info():
        await msg.finish(msg.locale.t("phigros.message.unbind.success"))


@phi.command('b19 {{phigros.help.b19}}')
async def _(msg: Bot.MessageSession):
    if (bind := PgrBindInfoManager(msg).get_bind_info()) is None:
        await msg.finish(msg.locale.t("phigros.message.user_unbound", prefix=msg.prefixes[0]))
    else:
        try:
            headers = p_headers.copy()
            headers['X-LC-Session'] = bind[0]
            get_save_url = await get_url('https://rak3ffdi.cloud.tds1.tapapis.cn/1.1/classes/_GameSave',
                                         headers=headers,
                                         fmt='json')
            save_url = get_save_url['results'][0]['gameFile']['url']
            download = await download_to_cache(save_url)
            rd_path = random_cache_path()
            shutil.unpack_archive(download, rd_path)
            game_records = parse_game_record(os.path.join(rd_path, 'gameRecord'))
            sort_by_rks = sorted({f'{level}.{song}':
                                  game_records[song][level] for song in game_records for level in
                                  game_records[song]}.items(),
                                 key=lambda x: x[1]['rks'], reverse=True)
            phi_list = [s for s in sort_by_rks if s[1]['score'] == 1000000]
            b19_data = ([sorted(phi_list, key=lambda x: x[1]['rks'], reverse=True)[0]]
                        if phi_list else []) + sort_by_rks[0: 19]
            if len(rks_acc := [i[1]['rks'] for i in b19_data]) < 20:
                rks_acc += [0] * (20 - len(rks_acc))
            await msg.send_message(Image(drawb19(bind[1], round(sum(rks_acc) / len(rks_acc), 2), b19_data)))
        except Exception as e:
            traceback.print_exc()
            await msg.send_message(msg.locale.t("phigros.message.b19.get_failed", err=str(e)))


@phi.command('update', required_superuser=True)
async def _(msg: Bot.MessageSession):
    if await update_assets():
        await msg.finish(msg.locale.t("success"))
    else:
        await msg.finish(msg.locale.t("failed"))