"""Regression guards for AI document updates while Markdown Preview is visible (#2182)."""

import re
from pathlib import Path
from tests.helpers.document_source import document_source




def _function_body(name: str) -> str:
    text = document_source()
    match = re.search(rf"\n\s*(?:export\s+)?(?:async\s+)?function\s+{name}\([^)]*\)\s*\{{", text)
    assert match, f"{name} not found"

    start = match.end()
    depth = 1
    i = start
    while i < len(text) and depth:
        if text[i] == "{":
            depth += 1
        elif text[i] == "}":
            depth -= 1
        i += 1
    assert depth == 0, f"{name} body did not close"
    return text[start : i - 1]


def test_markdown_preview_refresh_rerenders_visible_preview():
    body = _function_body("_refreshMarkdownPreviewIfVisible")

    assert "_isMarkdownPreviewVisible()" in body
    assert "lang !== 'markdown'" in body
    assert "textarea.value = content;" in body
    assert "syncHighlighting();" in body
    assert "_setMarkdownPreviewActive(true, { remember: false });" in body


def test_doc_update_refreshes_preview_instead_of_hidden_editor_animation():
    body = _function_body("handleDocUpdate")

    visible = "const markdownPreviewWasVisible = _isMarkdownPreviewVisible();"
    exit_preview = "if (markdownPreviewWasVisible) _setMarkdownPreviewActive(false, { remember: false });"
    diff = "enterDiffMode(oldContent, newContent);"
    refresh = "markdownPreviewWasVisible && _refreshMarkdownPreviewIfVisible(docId, newContent)"
    animate = "_animateDocEdit(textarea, newContent);"

    assert visible in body
    assert exit_preview in body
    assert diff in body
    assert body.index(exit_preview) < body.index(diff)
    assert refresh in body
    assert body.index(refresh) < body.index(animate)
    assert "_refreshMarkdownPreviewIfVisible(docId, newContent);" in body


def test_doc_update_shows_a_plain_text_diff_before_refreshing_rich_text():
    body = _function_body("handleDocUpdate")

    assert "const isRichTextUpdate = _isRichTextLang(docLang);" in body
    assert "if (isRichTextUpdate && updatedDocForRichText)" in body
    assert "_animateRichTextEdit(oldContent, newContent, updatedDocForRichText);" in body

    rich_diff = _function_body("_animateRichTextEdit")
    assert "_richTextContentToPlain(oldContent)" in rich_diff
    assert "_richTextContentToPlain(newContent)" in rich_diff
    assert "lineDiff(oldText, newText)" in rich_diff
    assert "_showRichTextEditor(updatedDoc);" in rich_diff