LiteyukiBot/src/utils/resource.py

import os

import nonebot
import yaml
from typing_extensions import Any

from src.utils.data import LiteModel

_resource_data = {}
_loaded_resource_packs = []  # 按照加载顺序排序


class ResourceMetadata(LiteModel):
    name: str = "Unknown"
    version: str = "0.0.1"
    description: str = "Unknown"
    path: str


def load_resource_from_dir(path: str):
    """
    把资源包按照文件相对路径加载到资源包中，后加载的优先级更高，顺便加载语言
    Args:
        path:  资源文件夹

    Returns:

    """
    for root, dirs, files in os.walk(path):
        for file in files:
            relative_path = os.path.relpath(os.path.join(root, file), path).replace("\\", "/")
            abs_path = os.path.join(root, file).replace("\\", "/")
            _resource_data[relative_path] = abs_path
            nonebot.logger.debug(f"Loaded {relative_path} -> {abs_path}")
    if os.path.exists(os.path.join(path, "metadata.yml")):
        with open(os.path.join(path, "metadata.yml"), "r", encoding="utf-8") as f:
            metadata = yaml.safe_load(f)
    else:
        metadata = ResourceMetadata()
    metadata["path"] = path
    if os.path.exists(os.path.join(path, "lang")):
        from src.utils.language import load_from_dir
        load_from_dir(os.path.join(path, "lang"))
    _loaded_resource_packs.append(ResourceMetadata(**metadata))


def get_res(path: str, default: Any = None) -> str | Any:
    """
    获取资源包中的文件
    Args:
        default: 默认
        path: 文件相对路径
    Returns: 文件绝对路径
    """
    return _resource_data.get(path, default)