"""
DiskGuard - 磁盘空间守护（2026-08-28 新增）

背景：Oracle 机器磁盘曾经被写满（gdrive_videos 持续下载新素材、旧文件迟迟
没被清理），触发了 rclone 的一个安全机制——同步过程中一旦遇到 IO 错误（写
不进去）就整体拒绝执行删除，导致"越满越删不掉，越删不掉越满"的死循环，
最终连新视频都下载不了。这个模块是一道独立于 rclone 同步逻辑之外的兜底：
不管上游同步是否正常，只要本机磁盘剩余空间跌破阈值，就主动清理最旧的、已
经处理完成的整段素材文件，把空间抢回来。

清理对象：仅限"整段素材"（gdrive_videos 里落地的原始录像文件，filename
不是 motion_ 前缀）且 status='done'（已完成分割阶段，不会再被 Video-Queue
重新捡起）——不碰运动片段（motion_clips 里的文件，是独立的分析产物，事件
时间轴展示、人物头像裁剪都依赖它，删了会导致图片丢失）、不碰还在
pending/processing 中的素材（避免删掉还没来得及处理的数据）。
"""
import shutil
import threading
import time
import os

from .logger import setup_logger
from .config_loader import load_config

logger = setup_logger('fam-edge.disk_guard')


class DiskGuard:
    def __init__(self, db):
        self.db = db
        cfg = load_config().get('disk_guard', {})
        self.enabled = bool(cfg.get('enabled', True))
        self.check_interval_sec = int(cfg.get('check_interval_sec', 300))
        self.min_free_gb = float(cfg.get('min_free_gb', 10))
        # 清理到这个水位就停：留出缓冲，避免刚清理完又立刻因为新文件写入
        # 跌破阈值、频繁触发清理循环
        self.target_free_gb = float(cfg.get('target_free_gb', 15))
        self.watch_path = cfg.get('watch_path', '/opt/fam-edge')
        # 单轮最多清理几个文件：防止候选异常多时一次性删太多，先清一部分
        # 观察效果，下一轮检查再继续（check_interval_sec 后很快就会再跑一次）
        self.max_delete_per_round = int(cfg.get('max_delete_per_round', 50))
        self._running = False
        self._thread = None

    def start(self):
        if not self.enabled:
            logger.info("DiskGuard 未启用")
            return
        self._running = True
        self._thread = threading.Thread(target=self._run, daemon=True, name='disk-guard')
        self._thread.start()
        logger.info(
            f"DiskGuard 已启动（阈值 {self.min_free_gb}GB，"
            f"目标水位 {self.target_free_gb}GB，检查间隔 {self.check_interval_sec}s）")

    def stop(self):
        self._running = False
        if self._thread:
            self._thread.join(timeout=5)

    def is_alive(self) -> bool:
        return self._thread is not None and self._thread.is_alive()

    def _free_gb(self) -> float:
        return shutil.disk_usage(self.watch_path).free / (1024 ** 3)

    def _run(self):
        while self._running:
            try:
                self.check_once()
            except Exception as e:
                logger.error(f"DiskGuard 检查异常: {e}", exc_info=True)
            for _ in range(self.check_interval_sec):
                if not self._running:
                    return
                time.sleep(1)

    def check_once(self):
        """检查一次磁盘空间，不足则清理最旧的已完成素材直到恢复到目标水位。

        供后台循环调用，也可单独调用做一次性检查（比如手动触发/测试）。
        """
        free_gb = self._free_gb()
        if free_gb >= self.min_free_gb:
            return
        logger.warning(f"磁盘剩余 {free_gb:.1f}GB 低于阈值 {self.min_free_gb}GB，开始清理旧素材")
        self.db.record_activity(
            'disk_guard', 'low_space',
            f"剩余 {free_gb:.1f}GB 低于阈值 {self.min_free_gb}GB，开始清理")

        deleted = 0
        freed_bytes = 0
        while deleted < self.max_delete_per_round and self._free_gb() < self.target_free_gb:
            candidate = self.db.get_oldest_purgeable_material()
            if not candidate:
                logger.warning("磁盘空间仍然紧张，但已经没有可清理的素材了")
                self.db.record_activity(
                    'disk_guard', 'no_candidate',
                    f"剩余 {self._free_gb():.1f}GB 仍不足，且没有可清理的素材")
                break
            video_id, local_path = candidate['id'], candidate.get('local_path')
            size = os.path.getsize(local_path) if local_path and os.path.isfile(local_path) else 0
            self.db.delete_video(video_id)
            deleted += 1
            freed_bytes += size
            logger.info(f"DiskGuard 清理素材 video_id={video_id}（约 {size/1024/1024:.0f}MB）")

        if deleted:
            free_gb = self._free_gb()
            self.db.record_activity(
                'disk_guard', 'cleaned',
                f"清理 {deleted} 个素材，释放约 {freed_bytes/1024**3:.1f}GB，"
                f"当前剩余 {free_gb:.1f}GB")
            logger.info(f"DiskGuard 本轮清理完成：{deleted} 个文件，当前剩余 {free_gb:.1f}GB")