from pathlib import Path
import shutil
import sys

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))

from fastapi.testclient import TestClient

import server


client = TestClient(server.app)


class FakeTemporaryDirectory:
    def __init__(self, root: Path, prefix: str, created_paths: list[Path]) -> None:
        self.path = root / f"{prefix}{len(created_paths)}"
        self.created_paths = created_paths

    def __enter__(self) -> str:
        self.path.mkdir(parents=True, exist_ok=False)
        self.created_paths.append(self.path)
        return str(self.path)

    def __exit__(self, exc_type, exc, tb) -> None:
        shutil.rmtree(self.path, ignore_errors=True)


def test_validate_output_format_rejects_unknown() -> None:
    try:
        server.validate_output_format("docx")
    except Exception as exc:  # pragma: no cover - structure assertion below
        assert getattr(exc, "status_code", None) == 400
    else:  # pragma: no cover
        raise AssertionError("Expected HTTPException")


def test_transcriptions_returns_generated_artifact(monkeypatch, tmp_path: Path) -> None:
    created_paths: list[Path] = []
    monkeypatch.setattr(
        server,
        "TemporaryDirectory",
        lambda prefix="": FakeTemporaryDirectory(tmp_path, prefix, created_paths),
    )

    def fake_run(command: list[str], check: bool, capture_output: bool, text: bool, timeout: int):
        output_dir = Path(command[command.index("--output_dir") + 1])
        (output_dir / "clip.txt").write_text("hello world", encoding="utf-8")

        class Result:
            returncode = 0
            stdout = ""
            stderr = ""

        return Result()

    monkeypatch.setattr(server.subprocess, "run", fake_run)

    response = client.post(
        "/transcriptions",
        data={"model": "base", "language": "en", "output_format": "txt"},
        files={"file": ("clip.wav", b"audio", "audio/wav")},
    )

    assert response.status_code == 200
    assert response.text == "hello world"
    assert response.headers["x-whisper-output-format"] == "txt"
    assert created_paths
    assert all(not path.exists() for path in created_paths)


def test_transcriptions_maps_subprocess_failure(monkeypatch, tmp_path: Path) -> None:
    created_paths: list[Path] = []
    monkeypatch.setattr(
        server,
        "TemporaryDirectory",
        lambda prefix="": FakeTemporaryDirectory(tmp_path, prefix, created_paths),
    )

    def fake_run(command: list[str], check: bool, capture_output: bool, text: bool, timeout: int):
        class Result:
            returncode = 1
            stdout = ""
            stderr = "bad whisper day"

        return Result()

    monkeypatch.setattr(server.subprocess, "run", fake_run)

    response = client.post(
        "/transcriptions",
        data={"model": "base", "output_format": "txt"},
        files={"file": ("clip.wav", b"audio", "audio/wav")},
    )

    assert response.status_code == 502
    assert response.json()["detail"] == "bad whisper day"
    assert created_paths
    assert all(not path.exists() for path in created_paths)


def test_transcriptions_maps_subprocess_timeout(monkeypatch, tmp_path: Path) -> None:
    created_paths: list[Path] = []
    monkeypatch.setattr(
        server,
        "TemporaryDirectory",
        lambda prefix="": FakeTemporaryDirectory(tmp_path, prefix, created_paths),
    )

    def fake_run(command: list[str], check: bool, capture_output: bool, text: bool, timeout: int):
        raise server.subprocess.TimeoutExpired(cmd=command, timeout=timeout)

    monkeypatch.setattr(server.subprocess, "run", fake_run)

    response = client.post(
        "/transcriptions",
        data={"model": "base", "output_format": "txt"},
        files={"file": ("clip.wav", b"audio", "audio/wav")},
    )

    assert response.status_code == 504
    assert (
        response.json()["detail"]
        == f"Whisper CLI timed out after {server.WHISPER_PROCESS_TIMEOUT_SECONDS}s and was terminated."
    )
    assert created_paths
    assert all(not path.exists() for path in created_paths)