"""网页播放的按需转码缓存：源文件mp4容器HEVC轨道sample entry编码标签缺失，
浏览器<video>标签无法识别，第一次点击播放时用ffmpeg（VideoToolbox硬件加速）转码成
标准H264并缓存到 TRANSCODED_DIR，之后复用缓存不重复转码；不预先转码全部录像。
"""
import fcntl
import subprocess
from pathlib import Path

from analysis.config import TRANSCODED_DIR


def cached_path(video_id):
    return TRANSCODED_DIR / f"{video_id}.mp4"


def _lock_path(video_id):
    return TRANSCODED_DIR / f"{video_id}.lock"


def ensure_transcoded(video_id, source_path):
    """返回可直接播放的文件路径。已有缓存直接复用，否则先转码再缓存。

    用 fcntl 文件锁防止两个请求同时转码同一个视频：先到的转码，后到的等锁释放后
    直接读缓存。锁文件在转码完成后删除。
    """
    output_path = cached_path(video_id)
    if output_path.exists():
        return output_path

    TRANSCODED_DIR.mkdir(exist_ok=True)
    lock_file = _lock_path(video_id)
    lock_fd = open(lock_file, "w")

    try:
        # 阻塞等待锁，最多等 10 分钟（转码大文件可能慢）
        fcntl.flock(lock_fd, fcntl.LOCK_EX)

        # 拿到锁后再检查一次：可能前面的进程已经转码完成
        if output_path.exists():
            return output_path

        tmp_path = output_path.with_suffix(".mp4.tmp")
        result = subprocess.run(
            [
                "ffmpeg", "-i", str(source_path),
                "-c:v", "h264_videotoolbox", "-b:v", "1500k",
                "-c:a", "aac",
                "-movflags", "+faststart",
                "-f", "mp4",  # 输出文件名带.tmp后缀，ffmpeg无法从扩展名猜出格式，需显式指定
                "-y", str(tmp_path),
            ],
            capture_output=True, timeout=600,
        )
        if result.returncode != 0 or not tmp_path.exists():
            tmp_path.unlink(missing_ok=True)
            raise RuntimeError(result.stderr.decode("utf-8", errors="ignore"))

        tmp_path.rename(output_path)  # 转码完成后原子改名，避免并发请求读到半成品文件
        return output_path
    finally:
        fcntl.flock(lock_fd, fcntl.LOCK_UN)
        lock_fd.close()
        lock_file.unlink(missing_ok=True)


def cache_stats():
    if not TRANSCODED_DIR.exists():
        return {"file_count": 0, "total_size_bytes": 0, "files": []}

    files = []
    total = 0
    for f in TRANSCODED_DIR.glob("*.mp4"):
        size = f.stat().st_size
        total += size
        files.append({"video_id": int(f.stem), "size_bytes": size})

    return {"file_count": len(files), "total_size_bytes": total, "files": files}


def clear_cache():
    if not TRANSCODED_DIR.exists():
        return {"deleted": 0, "freed_bytes": 0}

    deleted = 0
    freed = 0
    for f in TRANSCODED_DIR.glob("*.mp4"):
        freed += f.stat().st_size
        f.unlink()
        deleted += 1
    # 清理残留的 .tmp 和 .lock 文件
    for f in TRANSCODED_DIR.glob("*.tmp"):
        f.unlink(missing_ok=True)
    for f in TRANSCODED_DIR.glob("*.lock"):
        f.unlink(missing_ok=True)

    return {"deleted": deleted, "freed_bytes": freed}