import ujson as json
from aiofiles import open as async_open
from httpx import AsyncClient, URL

from utils.const import AMBR_HOST, PROJECT_ROOT

__all__ = ['update_metadata_from_ambr', 'update_metadata_from_github']

client = AsyncClient()


async def update_metadata_from_ambr(overwrite: bool = True):
    result = []
    targets = ['material', 'weapon', 'avatar', 'reliquary']
    for target in targets:
        path = PROJECT_ROOT.joinpath(f'metadata/data/{target}.json')
        if not overwrite and path.exists():
            continue
        url = AMBR_HOST.join(f"v2/chs/{target}")
        path.parent.mkdir(parents=True, exist_ok=True)
        response = await client.get(url)
        json_data = json.loads(response.text)['data']['items']
        async with async_open(path, mode='w', encoding='utf-8') as file:
            data = json.dumps(json_data, ensure_ascii=False)
            await file.write(data)
        result.append(json_data)
    return result


async def update_metadata_from_github(overwrite: bool = True):
    path = PROJECT_ROOT.joinpath('metadata/data/namecard.json')
    if not overwrite and path.exists():
        return

    host = URL("https://raw.fastgit.org/Dimbreath/GenshinData/master/")

    text_map_url = host.join("TextMap/TextMapCHS.json")
    material_url = host.join("ExcelBinOutput/MaterialExcelConfigData.json")

    text_map_json_data = json.loads((await client.get(text_map_url)).text)
    material_json_data = json.loads((await client.get(material_url)).text)

    data = {}
    for namecard_data in filter(lambda x: x.get('materialType', None) == 'MATERIAL_NAMECARD', material_json_data):
        name = text_map_json_data[str(namecard_data['nameTextMapHash'])]
        icon = namecard_data['icon']
        navbar = namecard_data['picPath'][0]
        banner = namecard_data['picPath'][1]
        rank = namecard_data['rankLevel']
        description = text_map_json_data[str(namecard_data['descTextMapHash'])].replace('\\n', '\n')
        data.update({
            str(namecard_data['id']): {
                "id": namecard_data['id'],
                "name": name,
                "rank": rank,
                "icon": icon,
                "navbar": navbar,
                "profile": banner,
                "description": description,
            }
        })
    async with async_open(path, mode='w', encoding='utf-8') as file:
        data = json.dumps(data, ensure_ascii=False)
        await file.write(data)
    return data