from md2gost.parser import Parser
from md2gost.render import Renderer

FRONT_MATTER = "---\nназвание: Тестовая работа\nтип: Отчёт\n---\n\n"


def render(md: str) -> str:
    parser = Parser(FRONT_MATTER + md)
    return Renderer(parser.tokens, parser.front_matter).render()


def test_heading_numbering() -> None:
    tex = render("# Исследовательская часть\n\n## Обзор\n")
    assert "\\section{Исследовательская часть}" in tex
    assert "\\subsection{Обзор}" in tex


def test_unnumbered_heading() -> None:
    tex = render("# ВВЕДЕНИЕ\n\nТекст.\n")
    assert "\\section{" not in tex
    assert "\\clearpage" in tex
    assert "ВВЕДЕНИЕ" in tex


def test_list_punctuation() -> None:
    tex = render("- один\n- два\n- три\n")
    assert "  \\item один;" in tex
    assert "  \\item три." in tex


def test_figure_with_caption(tmp_path) -> None:
    tex = render(
        "![[images/antenna.png]]\n\nРисунок 1 — Антенна\n\n"
        "Данная антенна используется для ...\n"
    )
    assert "\\begin{figure}[ht]" in tex
    assert "\\centering" in tex
    assert (
        "\\includegraphics[width=0.8\\textwidth]{\\detokenize{images/antenna.png}}"
        in tex
    )
    assert "\\caption{Антенна}" in tex
    assert "Данная антенна используется для ..." in tex


def test_figure_without_caption() -> None:
    tex = render("![[images/photo.png]]\n\nОбычный абзац после картинки.\n")
    assert "\\begin{figure}[ht]" in tex
    assert "\\caption{" not in tex
    assert "Обычный абзац после картинки." in tex


def test_figure_custom_width() -> None:
    tex = render("![[images/scheme.png|50]]\n\nРисунок 2 —— Схема\n")
    assert (
        "\\includegraphics[width=0.5\\textwidth]{\\detokenize{images/scheme.png}}"
        in tex
    )


def test_caption_number_is_discarded() -> None:
    tex = render("![[images/a.png]]\n\nРисунок 7 —— Что-то\n")
    assert "7" not in _figure_block(tex)


def test_inline_image_skipped_with_warning(capsys) -> None:
    tex = render("Текст с ![alt](img.png) внутри.\n")
    assert "предупреждение: картинка внутри текста пропущена" in capsys.readouterr().err
    assert "\\begin{figure}" not in tex
    assert "img.png" not in tex


def test_toc_inserted_once() -> None:
    tex = render("# СОДЕРЖАНИЕ\n\n# ВВЕДЕНИЕ\n\nТекст.\n\n# ЗАКЛЮЧЕНИЕ\n\nВыводы.\n")
    assert tex.count("\\tableofcontents") == 1


def test_handler_must_advance_pos() -> None:
    from markdown_it import MarkdownIt

    tokens = MarkdownIt().parse(FRONT_MATTER + "# Заголовок\n")
    renderer = Renderer(tokens, {"название": "Т", "тип": "Отчёт"})
    renderer.render()
    broken = Renderer(tokens, {"название": "Т", "тип": "Отчёт"})
    from md2gost.registry import BLOCK

    original = BLOCK["heading_open"]
    BLOCK["heading_open"] = lambda r, tok: ""
    try:
        try:
            broken.render()
            raise AssertionError("ожидался RuntimeError")
        except RuntimeError as e:
            assert "heading_open" in str(e)
    finally:
        BLOCK["heading_open"] = original


def _figure_block(tex: str) -> str:
    start = tex.index("\\begin{figure}")
    end = tex.index("\\end{figure}") + len("\\end{figure}")
    return tex[start:end]