from __future__ import annotations

from pathlib import Path

from ozan_radio.dj import TrackPlan
from ozan_radio.lyria import GeneratedTrack
from ozan_radio.queue import RadioQueue


def _track(tmp_path: Path, track_id: str, title: str) -> GeneratedTrack:
    safe = title.replace(" ", "_")
    path = tmp_path / f"{track_id}_{safe}.mp3"
    path.write_bytes(b"ID3")
    plan = TrackPlan(
        id=track_id,
        title=title,
        mood="test",
        dj_line="on air",
        lyria_prompt="test prompt",
    )
    return GeneratedTrack(plan=plan, audio_path=path, lyrics="")


def test_queue_add_and_advance(tmp_path: Path):
    q = RadioQueue(tmp_path)
    q.add(_track(tmp_path, "aaaa1111", "One"))
    q.add(_track(tmp_path, "bbbb2222", "Two"))
    assert q.length == 2
    assert q.current().plan.title == "One"
    q.advance()
    assert q.current().plan.title == "Two"


def test_queue_prunes_stale_manifest_entries(tmp_path: Path):
    q = RadioQueue(tmp_path)
    q.add(_track(tmp_path, "aaaa1111", "Only"))
    manifest = tmp_path / "manifest.json"
    import json

    data = json.loads(manifest.read_text(encoding="utf-8"))
    data["tracks"].append(
        {
            "id": "ghost",
            "title": "Ghost",
            "mood": "",
            "dj_line": "",
            "lyria_prompt": "",
            "lyrics": "",
            "file": "ghost_Ghost.mp3",
        }
    )
    manifest.write_text(json.dumps(data), encoding="utf-8")

    q2 = RadioQueue(tmp_path)
    assert q2.length == 1
    cleaned = json.loads(manifest.read_text(encoding="utf-8"))
    assert cleaned["count"] == 1
    assert all(t["id"] != "ghost" for t in cleaned["tracks"])