"""Radarr tools — movie queries (read-only)."""

from ._common import arr_get


def _compact_movie(m):
    return {
        "id": m.get("id"),
        "title": m.get("title"),
        "year": m.get("year"),
        "monitored": m.get("monitored"),
        "has_file": m.get("hasFile"),
        "status": m.get("status"),
        "size_on_disk_gb": round((m.get("sizeOnDisk", 0) or 0) / 1e9, 1),
    }


def radarr_queue(limit=20):
    """Movies currently downloading or pending import."""
    data = arr_get("radarr", "/api/v3/queue", {"pageSize": limit, "includeMovie": "true"})
    items = []
    for r in data.get("records", []):
        items.append({
            "title": r.get("title"),
            "movie": r.get("movie", {}).get("title") if r.get("movie") else None,
            "status": r.get("status"),
            "timeleft": r.get("timeleft"),
            "size_gb": round((r.get("size", 0) or 0) / 1e9, 2),
            "size_left_gb": round((r.get("sizeleft", 0) or 0) / 1e9, 2),
            "download_client": r.get("downloadClient"),
        })
    return {"total_records": data.get("totalRecords", 0), "items": items}


def radarr_movie_search(query):
    """Search the user's Radarr library for a movie by title. Does NOT search indexers."""
    all_movies = arr_get("radarr", "/api/v3/movie")
    q = query.lower().strip()
    hits = [m for m in all_movies if q in (m.get("title") or "").lower()]
    return {"query": query, "count": len(hits), "movies": [_compact_movie(m) for m in hits[:20]]}


def radarr_library_stats():
    """Summary stats of the Radarr movie library."""
    all_movies = arr_get("radarr", "/api/v3/movie")
    on_disk = sum(1 for m in all_movies if m.get("hasFile"))
    total_size = sum((m.get("sizeOnDisk", 0) or 0) for m in all_movies)
    missing = [m.get("title") for m in all_movies if not m.get("hasFile") and m.get("monitored")][:10]
    return {
        "movie_count": len(all_movies),
        "movies_on_disk": on_disk,
        "movies_missing_monitored": len([m for m in all_movies if not m.get("hasFile") and m.get("monitored")]),
        "total_size_tb": round(total_size / 1e12, 2),
        "sample_missing": missing,
    }


TOOLS = [
    {
        "name": "radarr_queue",
        "description": "List movies Radarr is currently downloading or importing. Use when the user asks about movie downloads.",
        "input_schema": {
            "type": "object",
            "properties": {"limit": {"type": "integer", "default": 20}},
        },
        "read_only": True,
        "fn": radarr_queue,
    },
    {
        "name": "radarr_movie_search",
        "description": "Check whether a specific movie is in the user's Radarr library. Does NOT search for new movies to add.",
        "input_schema": {
            "type": "object",
            "properties": {"query": {"type": "string"}},
            "required": ["query"],
        },
        "read_only": True,
        "fn": radarr_movie_search,
    },
    {
        "name": "radarr_library_stats",
        "description": "High-level stats about the Radarr movie library: how many movies, how many on disk, total size.",
        "input_schema": {"type": "object", "properties": {}},
        "read_only": True,
        "fn": radarr_library_stats,
    },
]