import os
import json
import time
import logging

log = logging.getLogger(__name__)

# 定义缓存目录
CACHE_DIR = os.path.join(os.path.dirname(__file__), '..', 'cache')
os.makedirs(CACHE_DIR, exist_ok=True) # 自动创建目录

class CacheManager:
    """
    一个通用的JSON文件缓存管理器。
    """
    def __init__(self, cache_file_name: str, ttl_seconds: int = 3600):
        """
        :param cache_file_name: 缓存文件名, e.g., 'plex_library.json'
        :param ttl_seconds: 缓存的生命周期（秒），默认1小时
        """
        self.cache_path = os.path.join(CACHE_DIR, cache_file_name)
        self.ttl = ttl_seconds

    def _is_stale(self) -> bool:
        """检查缓存是否已过期"""
        if not os.path.exists(self.cache_path):
            return True # 文件不存在，视为过期
        
        file_mod_time = os.path.getmtime(self.cache_path)
        if (time.time() - file_mod_time) > self.ttl:
            return True # 文件存在但已超过TTL，视为过期
            
        return False

    def read_cache(self) -> list | dict | None:
        """从文件读取缓存"""
        try:
            with open(self.cache_path, 'r', encoding='utf-8') as f:
                return json.load(f)
        except (FileNotFoundError, json.JSONDecodeError):
            return None

    def write_cache(self, data: list | dict):
        """将数据写入缓存文件"""
        try:
            with open(self.cache_path, 'w', encoding='utf-8') as f:
                json.dump(data, f, ensure_ascii=False, indent=4)
        except Exception as e:
            log.error(f"写入缓存文件 {self.cache_path} 失败: {e}")

    def get_data(self, fetch_func, *args, **kwargs) -> list | dict:
        """
        获取数据的主要方法。如果缓存有效则从缓存读取，否则调用 fetch_func 获取新数据。
        
        :param fetch_func: 一个用于获取实时数据的函数 (e.g., plex_manager.get_items)
        :param args: 传递给 fetch_func 的位置参数
        :param kwargs: 传递给 fetch_func 的关键字参数
        :return: 数据
        """
        if not self._is_stale():
            log.debug(f"从缓存加载数据: {self.cache_path}")
            cached_data = self.read_cache()
            if cached_data is not None:
                return cached_data
        
        log.info(f"缓存失效或不存在，正在从源获取新数据: {fetch_func.__name__}")
        try:
            fresh_data = fetch_func(*args, **kwargs)
            self.write_cache(fresh_data)
            return fresh_data
        except Exception as e:
            log.error(f"调用 fetch_func ({fetch_func.__name__}) 失败: {e}", exc_info=True)
            # 如果获取新数据失败，尝试返回旧的缓存（即使已过期），这比返回错误要好
            log.warning("获取新数据失败，尝试使用旧缓存作为备用。")
            old_data = self.read_cache()
            return old_data if old_data is not None else []