import asyncio
import re
import traceback

from config import Config
from core.builtins import Bot
from core.component import module
from core.utils.http import get_url

mod_dl = module(
    bind_prefix='mod_dl',
    desc='{mod_dl.help.desc}',
    developers=['HornCopper', 'OasisAkari', 'z0z0r4'],
    recommend_modules=['mcmod'],
    alias='moddl')

x_api_key = Config("curseforge_api_key")
if not x_api_key:
    # CurseForge API Key 未配置，使用镜像 https://mcim.z0z0r4.top ...(z0z0r4 不想解析网页)
    MCIM_MODE = True
else:
    MCIM_MODE = False


def cn_chk(string: str):
    for word in string:
        if u'\u4e00' <= word <= u'\u9fff':
            return True
    return False


@mod_dl.handle('<mod_name> [<version>] {{mod_dl.help}}')
async def main(msg: Bot.MessageSession, mod_name: str, version: str = None):
    ver = version
    if version is not None:
        match_ver = re.match(r'^\d+\.\d+\.\d+$|^\d+\.\d+$|\d+w\d+[abcd]', version)
        if match_ver is None:
            mod_name += ' ' + version
            ver = False
    if cn_chk(mod_name):
        await msg.finish(msg.locale.t("mod_dl.message.unsupport"))

    async def search_modrinth(name: str, ver: str):
        url = f'https://api.modrinth.com/v2/search?query={name}&limit=10'
        if ver:
            url += f'&facets=[["versions:{ver}"],["project_type:mod"]]'
        else:
            url += f'&facets=[["project_type:mod"]]'
        resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3)
        if resp is not None:
            results = []
            if len(resp["hits"]) == 0:
                return None
            for project in resp['hits']:
                results.append(("modrinth", project["title"], project["project_id"], project["versions"]))
            return results

    async def search_curseforge(name: str, ver: str):
        if MCIM_MODE:
            # https://mcim.z0z0r4.top/docs#/Curseforge/curseforge_search_curseforge_search_get
            url = f'https://mcim.z0z0r4.top/curseforge/search?gameId=432&searchFilter={name}&sortField=2&sortOrder=desc&pageSize=10&classId=6'
            headers = None
        else:
            headers = {
                'Accept': 'application/json',
                'x-api-key': x_api_key
            }
            url = f'https://api.curseforge.com/v1/mods/search?gameId=432&searchFilter={name}&sortField=2&sortOrder=desc&pageSize=10&classId=6'

        if ver:
            url += f'&gameVersion={ver}'
        results = []
        try:
            resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3, headers=headers)
            if resp is not None:
                if not MCIM_MODE:  # 没提供 pagination
                    if resp["pagination"]["resultCount"] == 0:
                        return None
                for mod in resp["data"]:
                    results.append(("curseforge", mod["name"], mod["id"], None))
        except Exception:
            traceback.print_exc()
        return results

    async def get_modrinth_project_version(project_id: str, ver: str):
        url = f'https://api.modrinth.com/v2/project/{project_id}/version?game_versions=["{ver}"]&featured=true'
        resp = (await get_url(url, 200, fmt="json", timeout=5, attempt=3))[0]
        if resp is not None:
            return resp

    async def get_curseforge_mod_version_index(modid: str):
        if MCIM_MODE:
            # https://mcim.z0z0r4.top/docs#/Curseforge/get_mod_curseforge_mod__modid_slug__get
            url = f'https://mcim.z0z0r4.top/curseforge/mod/{modid}'
            headers = None
        else:
            headers = {
                'Accept': 'application/json',
                'x-api-key': x_api_key
            }
            url = f'https://api.curseforge.com/v1/mods/{modid}'
        resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3, headers=headers)
        if resp is not None:
            return resp["data"]['latestFilesIndexes']

    async def get_curseforge_mod_file(modid: str, ver: str):
        if MCIM_MODE:
            url = f'https://mcim.z0z0r4.top/curseforge/mod/{modid}/files?gameVersion={ver}'
            headers = None
        else:
            headers = {
                'Accept': 'application/json',
                'x-api-key': x_api_key
            }
            url = f'https://api.curseforge.com/v1/mods/{modid}/files?gameVersion={ver}'

        try:
            resp = await get_url(url, 200, fmt="json", timeout=5, attempt=3, headers=headers)
            if resp is not None:
                return resp["data"][0]
        except Exception:
            traceback.print_exc()

    # 搜索 Mod
    result = await asyncio.gather(*(search_modrinth(mod_name, ver), search_curseforge(mod_name, ver)))
    cache_result = []
    if result[0] is None and result[1] is None:
        await msg.finish(msg.locale.t("mod_dl.message.not_found"))
    else:
        # 合并搜索结果
        reply_text, count = [], 0

        # 先显示 CurseForge 的结果
        if result[1] is None:
            reply_text.append(msg.locale.t("mod_dl.message.curseforge.not_found"))
        else:
            reply_text.append(msg.locale.t("mod_dl.message.curseforge.result"))
            for mod in result[1]:
                count += 1
                reply_text.append(f"{count}. {mod[1]}")
                cache_result.append(mod)

        if result[0] is None:
            reply_text.append(msg.locale.t("mod_dl.message.modrinth.not_found"))
        reply_text.append(msg.locale.t("mod_dl.message.modrinth.result"))
        for mod in result[0]:
            count += 1
            reply_text.append(f"{count}. {mod[1]}")
            cache_result.append(mod)

        reply = await msg.waitReply('\n'.join(reply_text) + '\n' + msg.locale.t("mod_dl.message.prompt"))
        replied = reply.asDisplay(text_only=True)

        # 查找 Mod
        if replied.isdigit():
            replied = int(replied)
            if replied > len(cache_result):
                return await msg.finish(msg.locale.t("mod_dl.message.invalid.out_of_range"))
            else:
                mod_info = cache_result[replied - 1]
        else:
            return await msg.finish(msg.locale.t("mod_dl.message.invalid.non_digital"))

        if mod_info[0] == "modrinth":  # modrinth mod
            if ver is None:
                reply2 = await msg.waitReply(f'{msg.locale.t("mod_dl.message.version")}\n'
                                             + "\n".join(mod_info[3])
                                             + f'\n{msg.locale.t("mod_dl.message.version.prompt")}')
                replied2 = reply2.asDisplay(text_only=True)
                if replied2 in mod_info[3]:
                    version_info = await get_modrinth_project_version(mod_info[2], replied2)
                    if version_info is not None:
                        await msg.finish(f'{" ".join(version_info["loaders"])}\n{msg.locale.t("mod_dl.message.download_url")}{version_info["files"][0]["url"]}\n{msg.locale.t("mod_dl.message.filename")}{version_info["files"][0]["filename"]}')
            elif ver not in mod_info[3]:
                await msg.finish(msg.locale.t("mod_dl.message.version.not_found"))
            elif ver in mod_info[3]:
                version_info = await get_modrinth_project_version(mod_info[2], ver)
                if version_info is not None:
                    await msg.finish(f'{" ".join(version_info["loaders"])}\n{msg.locale.t("mod_dl.message.download_url")}{version_info["files"][0]["url"]}\n{msg.locale.t("mod_dl.message.filename")}{version_info["files"][0]["filename"]}')
        else:  # curseforge mod
            version_index, ver_list = await get_curseforge_mod_version_index(mod_info[2]), []
            for version in version_index:
                if version["gameVersion"] not in ver_list:
                    ver_list.append(version["gameVersion"])
            if version_index is not None:
                if ver is None:
                    reply2 = await msg.waitReply(f'{msg.locale.t("mod_dl.message.version")}\n' +
                                                 '\n'.join(ver_list) +
                                                 f'\n{msg.locale.t("mod_dl.message.version.prompt")}')
                    ver = reply2.asDisplay(text_only=True)
                elif ver not in ver_list:
                    await msg.finish(msg.locale.t("mod_dl.message.version.not_found"))

                if ver in ver_list:
                    file_info = await get_curseforge_mod_file(mod_info[2], ver)
                    if file_info is not None:
                        await msg.finish(f'{" ".join(file_info["gameVersions"])} \
                                         \n{msg.locale.t("mod_dl.message.download_url")}{file_info["downloadUrl"]} \
                                         \n{msg.locale.t("mod_dl.message.filename")}{file_info["fileName"]}')
                else:
                    await msg.finish(msg.locale.t("mod_dl.message.version.not_found"))