import asyncio
import datetime
import functools
import hashlib
import inspect
import time
import zipfile
from collections import defaultdict
from pathlib import Path

from LittlePaimon.config import config
from .logger import logger
from .requests import aiorequests

RESOURCE_BASE_PATH = Path() / 'resources'


class FreqLimiter:
    """
    频率限制器（冷却时间限制器）
    """

    def __init__(self):
        """
        初始化一个频率限制器
        """
        self.next_time = defaultdict(float)

    def check(self, key: str) -> bool:
        """
        检查是否冷却结束
            :param key: key
            :return: 布尔值
        """
        return time.time() >= self.next_time[key]

    def start(self, key: str, cooldown_time: int = 0):
        """
        开始冷却
            :param key: key
            :param cooldown_time: 冷却时间(秒)
        """
        self.next_time[key] = time.time() + (cooldown_time if cooldown_time > 0 else 60)

    def left(self, key: str) -> int:
        """
        剩余冷却时间
            :param key: key
            :return: 剩余冷却时间
        """
        return int(self.next_time[key] - time.time()) + 1


freq_limiter = FreqLimiter()


def cache(ttl=datetime.timedelta(hours=1)):
    """
    缓存装饰器
        :param ttl: 过期时间
    """

    def wrap(func):
        cache_data = {}

        @functools.wraps(func)
        async def wrapped(*args, **kw):
            nonlocal cache_data
            bound = inspect.signature(func).bind(*args, **kw)
            bound.apply_defaults()
            ins_key = '|'.join([f'{k}_{v}' for k, v in bound.arguments.items()])
            default_data = {"time": None, "value": None}
            data = cache_data.get(ins_key, default_data)
            now = datetime.datetime.now()
            if not data['time'] or now - data['time'] > ttl:
                try:
                    data['value'] = await func(*args, **kw)
                    data['time'] = now
                    cache_data[ins_key] = data
                except Exception as e:
                    raise e
            return data['value']

        return wrapped

    return wrap


async def check_resource():
    logger.info('资源检查', '开始检查资源')
    if not (
            (RESOURCE_BASE_PATH / 'LittlePaimon').is_dir() and
            len(list((RESOURCE_BASE_PATH / 'LittlePaimon').rglob('*'))) >= 50):
        try:
            await aiorequests.download(
                url=f'{config.github_proxy}https://raw.githubusercontent.com/CMHopeSunshine/LittlePaimonRes/main/resources.zip',
                save_path=RESOURCE_BASE_PATH / '小派蒙基础资源.zip')
            zipfile.ZipFile(RESOURCE_BASE_PATH / '小派蒙基础资源.zip').extractall(RESOURCE_BASE_PATH)
            (RESOURCE_BASE_PATH / '小派蒙基础资源.zip').unlink()

            await aiorequests.download(
                url=f'{config.github_proxy}https://raw.githubusercontent.com/CMHopeSunshine/GenshinWikiMap/master/resources/genshin_resources.zip',
                save_path=RESOURCE_BASE_PATH / '原神图标资源.zip')
            zipfile.ZipFile(RESOURCE_BASE_PATH / '原神图标资源.zip').extractall(RESOURCE_BASE_PATH / 'LittlePaimon')
            (RESOURCE_BASE_PATH / '原神图标资源.zip').unlink()
            logger.info('资源检查', '<g>资源下载完成</g>')
        except Exception:
            logger.warning('资源检查', '下载<m>资源包</m>时<r>出错</r>，请尝试更换<m>github资源地址</m>')
    else:
        if not (RESOURCE_BASE_PATH / 'LittlePaimon' / 'star_rail').is_dir():
            try:
                await aiorequests.download(
                    url=f'{config.github_proxy}https://raw.githubusercontent.com/CMHopeSunshine/LittlePaimonRes/main/star_rail.zip',
                    save_path=RESOURCE_BASE_PATH / 'star_rail.zip')
                zipfile.ZipFile(RESOURCE_BASE_PATH / 'star_rail.zip').extractall(RESOURCE_BASE_PATH / 'LittlePaimon' / 'star_rail')
                (RESOURCE_BASE_PATH / 'star_rail.zip').unlink()
                logger.info('资源检查', '<g>星穹铁道相关资源下载完成</g>')
            except Exception:
                logger.warning('资源检查', '下载<m>星穹铁道资源</m>时<r>出错</r>，请尝试更换<m>github资源地址</m>')
        try:
            resource_list = await aiorequests.get(
                f'{config.github_proxy}https://raw.githubusercontent.com/CMHopeSunshine/LittlePaimonRes/main/resources_list.json',
                follow_redirects=True)
            resource_list = resource_list.json()
        except Exception:
            logger.warning('资源检查', '读取资源列表<r>失败</r>，请尝试更换<m>github资源地址</m>')
            return
        flag = False
        for resource in resource_list:
            file_path = RESOURCE_BASE_PATH / resource['path']
            if file_path.exists():
                if not resource['lock'] or hashlib.md5(file_path.read_bytes()).hexdigest() == resource['hash']:
                    continue
                else:
                    file_path.unlink()
            try:
                await aiorequests.download(
                    url=f'{config.github_proxy}https://raw.githubusercontent.com/CMHopeSunshine/LittlePaimonRes/main/{resource["path"]}',
                    save_path=file_path, exclude_json=resource['path'].split('.')[-1] != 'json')
                await asyncio.sleep(0.2)
                flag = True
            except Exception:
                logger.warning('资源检查', f'下载<m>{resource["path"]}</m>时<r>出错</r>，请尝试更换<m>github资源地址</m>')
        logger.info('资源检查', '<g>资源下载完成</g>' if flag else '<g>资源完好，无需下载</g>')