PaiGram/metadata/genshin.py

"""此文件用于储存 honey impact 中的部分基础数据"""

from __future__ import annotations

import functools
from typing import Any, Generic, ItemsView, Iterator, KeysView, Optional, TypeVar, ValuesView

from utils.const import PROJECT_ROOT
from utils.log import logger
from utils.typedefs import StrOrInt

try:
    import ujson as jsonlib
except ImportError:
    import json as jsonlib

__all__ = (
    "HONEY_DATA",
    "AVATAR_DATA",
    "WEAPON_DATA",
    "MATERIAL_DATA",
    "ARTIFACT_DATA",
    "NAMECARD_DATA",
    "honey_id_to_game_id",
    "game_id_to_role_id",
    "Data",
    "weapon_to_game_id",
    "avatar_to_game_id",
)

K = TypeVar("K")
V = TypeVar("V")

data_dir = PROJECT_ROOT.joinpath("metadata/data/")
data_dir.mkdir(parents=True, exist_ok=True)

_cache = {}


class Data(dict, Generic[K, V]):
    _dict: dict[K, V]
    _file_name: str

    @property
    def data(self) -> dict[K, V]:
        if (result := _cache.get(self._file_name)) not in [None, {}]:
            self._dict = result
        else:
            path = data_dir.joinpath(self._file_name).with_suffix(".json")
            if not path.exists():
                logger.error(
                    '暂未找到名为 "%s" 的 metadata , ' "请先使用 [yellow bold]/refresh_metadata[/] 命令下载",
                    self._file_name,
                    extra={"markup": True},
                )
                self._dict = {}
            with open(path, encoding="utf-8") as file:
                self._dict = jsonlib.load(file)
            _cache.update({self._file_name: self._dict})
        return self._dict

    def __init__(self, file_name: str):
        self._file_name = file_name
        self._dict = {}
        super(Data, self).__init__()

    def __str__(self) -> str:
        return self.data.__str__()

    def __repr__(self) -> str:
        return self.data.__repr__()

    def get(self, key: K, value: Any = None) -> V | None:
        return self.data.get(key, value)

    def __getitem__(self, key: K) -> V:
        return self.data.__getitem__(key)

    def __setitem__(self, key: K, value: V) -> None:
        return self.data.__setitem__(key, value)

    def __delitem__(self, value: V) -> None:
        self.data.__delitem__(value)

    def __iter__(self) -> Iterator[K]:
        return self.data.__iter__()

    def keys(self) -> KeysView[K, V]:
        return self.data.keys()

    def values(self) -> ValuesView[K, V]:
        return self.data.values()

    def items(self) -> ItemsView[K, V]:
        return self.data.items()


HONEY_DATA: dict[str, dict[StrOrInt, list[str | int]]] = Data("honey")

AVATAR_DATA: dict[str, dict[str, int | str | list[int]]] = Data("avatar")
WEAPON_DATA: dict[str, dict[str, int | str]] = Data("weapon")
MATERIAL_DATA: dict[str, dict[str, int | str]] = Data("material")
ARTIFACT_DATA: dict[str, dict[str, int | str | list[int] | dict[str, str]]] = Data("reliquary")
NAMECARD_DATA: dict[str, dict[str, int | str]] = Data("namecard")


@functools.lru_cache()
def honey_id_to_game_id(honey_id: str, item_type: str) -> str | None:
    return next((key for key, value in HONEY_DATA[item_type].items() if value[0] == honey_id), None)


@functools.lru_cache
def game_id_to_role_id(gid: str) -> int | None:
    return next(
        (int(key.split("-")[0]) for key, value in AVATAR_DATA.items() if value["icon"].split("_")[-1] == gid), None
    )


@functools.lru_cache()
def weapon_to_game_id(name: str) -> Optional[int]:
    return next((int(key) for key, value in WEAPON_DATA.items() if value["name"] == name), None)


@functools.lru_cache()
def avatar_to_game_id(name: str) -> Optional[int]:
    return next((int(key) for key, value in AVATAR_DATA.items() if value["name"] == name), None)