import os

from fastapi import APIRouter, Query
from fastapi.responses import FileResponse

from ..config import MEDIA_DIRS, VIDEO_EXTENSIONS

router = APIRouter()


def _scan_videos(root: str) -> list[dict]:
    results = []
    for dirpath, _, filenames in os.walk(root):
        for f in sorted(filenames):
            if os.path.splitext(f)[1].lower() in VIDEO_EXTENSIONS:
                full = os.path.join(dirpath, f)
                rel = os.path.relpath(full, root)
                results.append({
                    "name": f,
                    "path": rel,
                    "root": root,
                    "size": os.path.getsize(full),
                })
    return results


@router.get("/files")
def list_files(root: str | None = Query(None)):
    dirs = [root] if root and root in MEDIA_DIRS else MEDIA_DIRS
    files = []
    for d in dirs:
        files.extend(_scan_videos(d))
    return files


@router.get("/roots")
def list_roots():
    return MEDIA_DIRS


@router.get("/video/{path:path}")
def serve_video(path: str, root: str = Query(...)):
    if root not in MEDIA_DIRS:
        return {"error": "invalid root"}
    full = os.path.join(root, path)
    if not os.path.isfile(full):
        return {"error": "not found"}
    return FileResponse(full, media_type="video/mp4")