#!/usr/bin/env python3
"""Самопроверка обвязки озвучки: python3 test_audiobook.py (нужен ffmpeg)"""
import json
import subprocess
import sys
import tempfile
from pathlib import Path

import audiobook as a


def mp3(path, seconds):
    """Тишина заданной длительности — реальный mp3 для ffprobe."""
    subprocess.run(
        ["ffmpeg", "-v", "error", "-y", "-f", "lavfi",
         "-i", "anullsrc=r=22050:cl=mono", "-t", str(seconds),
         "-b:a", "32k", str(path)],
        check=True)


def write_chapter(d, num, paragraphs):
    (d / ("chapter_%03d_translated.json" % num)).write_text(
        json.dumps({"number": num, "title": "Глава ⟦b⟧%d⟦/b⟧" % num,
                    "paragraphs": paragraphs}, ensure_ascii=False),
        encoding="utf-8")


def test_prep(tmp: Path):
    src, out = tmp / "translations", tmp / "tts"
    src.mkdir()
    write_chapter(src, 0, ["Он сказал ⟦i⟧привет⟧⟦/i⟧ громко.",
                           "Обычный  абзац.", "   "])
    ns = type("ns", (), {"translations": src, "output": out})
    a.cmd_prep(ns)
    data = json.loads((out / "chapter_000_translated.json").read_text(encoding="utf-8"))
    assert "⟦" not in json.dumps(data, ensure_ascii=False), "маркеры должны исчезнуть"
    assert data["title"] == "Глава 0"
    assert len(data["paragraphs"]) == 2, "пустой абзац должен отброситься"
    assert "  " not in data["paragraphs"][1], "двойные пробелы должны схлопнуться"


def test_verify_detects_gap(tmp: Path):
    tts, ab = tmp / "tts", tmp / "audiobook"
    tts.mkdir()
    temp = ab / "temp_audio"
    temp.mkdir(parents=True)

    # две главы по 4 абзаца одинаковой длины
    para = "х" * 100
    for n in (0, 1):
        write_chapter(tts, n, [para] * 4)
        mp3(temp / ("chapter_%03d_intro.mp3" % n), 1)
        # у главы 1 намеренно пропущен последний абзац
        count = 4 if n == 0 else 3
        for i in range(count):
            mp3(temp / ("chapter_%03d_para_%04d.mp3" % (n, i)), 6)

    ns = type("ns", (), {"translations": tts, "audiobook": ab})
    assert a.cmd_verify(ns) is False, "пропущенный фрагмент должен завалить проверку"


def test_verify_clean(tmp: Path):
    tts, ab = tmp / "tts", tmp / "audiobook"
    tts.mkdir()
    temp = ab / "temp_audio"
    temp.mkdir(parents=True)
    para = "х" * 100
    for n in (0, 1):
        write_chapter(tts, n, [para] * 3)
        mp3(temp / ("chapter_%03d_intro.mp3" % n), 1)
        for i in range(3):
            mp3(temp / ("chapter_%03d_para_%04d.mp3" % (n, i)), 6)
    ns = type("ns", (), {"translations": tts, "audiobook": ab})
    assert a.cmd_verify(ns) is True, "целая книга должна проходить проверку"


def test_verify_detects_empty_file(tmp: Path):
    tts, ab = tmp / "tts", tmp / "audiobook"
    tts.mkdir()
    temp = ab / "temp_audio"
    temp.mkdir(parents=True)
    para = "х" * 100
    write_chapter(tts, 0, [para] * 3)
    mp3(temp / "chapter_000_intro.mp3", 1)
    mp3(temp / "chapter_000_para_0000.mp3", 6)
    mp3(temp / "chapter_000_para_0001.mp3", 6)
    (temp / "chapter_000_para_0002.mp3").touch()  # нулевой размер
    ns = type("ns", (), {"translations": tts, "audiobook": ab})
    assert a.cmd_verify(ns) is False, "пустой файл должен завалить проверку"


if __name__ == "__main__":
    if subprocess.run(["which", "ffmpeg"], capture_output=True).returncode:
        sys.exit("нужен ffmpeg")
    for fn in (test_prep, test_verify_detects_gap, test_verify_clean,
               test_verify_detects_empty_file):
        with tempfile.TemporaryDirectory() as d:
            print("---", fn.__name__)
            fn(Path(d))
    print("OK")