import json
from pathlib import Path
from datetime import datetime

from httpx import get

from .get_gachalogs import save_gachalogs

PLAYER_PATH = Path(__file__).parents[1] / 'player'
INT_TO_TYPE = {
    '100': '新手祈愿',
    '200': '常驻祈愿',
    '301': '角色祈愿',
    '400': '角色祈愿',
    '302': '武器祈愿',
}


async def import_gachalogs(history_url: str, uid: str) -> str:
    history_data: dict = json.loads(get(history_url).text)
    data_uid = history_data['info']['uid']
    if data_uid != uid:
        return f'该抽卡记录UID{data_uid}与你绑定UID{uid}不符合！'
    raw_data = history_data['list']
    result = {'新手祈愿': [], '常驻祈愿': [], '角色祈愿': [], '武器祈愿': []}
    for item in raw_data:
        item['uid'] = uid
        item['item_id'] = ''
        item['count'] = '1'
        item['lang'] = 'zh-cn'
        del item['uigf_gacha_type']
        result[INT_TO_TYPE[item['gacha_type']]].append(item)
    im = await save_gachalogs(uid, result)
    return im


async def export_gachalogs(uid: str) -> dict:
    path = PLAYER_PATH / uid
    if not path.exists():
        path.mkdir(parents=True, exist_ok=True)

    # 获取当前时间
    now = datetime.now()
    current_time = now.strftime('%Y-%m-%d %H:%M:%S')

    # 抽卡记录json路径
    gachalogs_path = path / 'gacha_logs.json'
    if gachalogs_path.exists():
        with open(gachalogs_path, "r", encoding='UTF-8') as f:
            raw_data = json.load(f)
        result = {
            'info': {
                'uid': uid,
                'lang': 'zh-cn',
                'export_time': current_time,
                'export_app': 'GenshinUID',
                'export_app_version': '3.1',
                'uigf_version': '2.1',
            },
            'list': [],
        }
        for i in ['新手祈愿', '常驻祈愿', '角色祈愿', '武器祈愿']:
            for item in raw_data['data'][i]:
                if item['gacha_type'] == '400':
                    item['uigf_gacha_type'] = '301'
                else:
                    item['uigf_gacha_type'] = item['gacha_type']
                result['list'].append(item)
        # 保存文件
        with open(path / f'UIGF_{uid}.json', 'w', encoding='UTF-8') as file:
            json.dump(result, file, ensure_ascii=False)
        im = {
            'retcode': 'ok',
            'data': '导出成功!',
            'name': f'UIGF_{uid}.json',
            'url': str(path / f'UIGF_{uid}.json'),
        }
    else:
        im = {
            'retcode': 'error',
            'data': '你还没有抽卡记录可以导出!',
            'name': '',
            'url': '',
        }

    return im