from __future__ import annotations

import re
from pathlib import Path

import yaml

from tools.catalog import load_catalog

ROOT = Path(__file__).resolve().parents[1]
# flypython.com links are first-party contextual continuations governed by
# docs/REPO_TO_WEBSITE.md, not catalog references.
FIRST_PARTY_PREFIX = "https://flypython.com"
GUIDES = {
    "en-US": ROOT / "guides" / "ai-coding" / "workflow.md",
    "zh-CN": ROOT / "guides" / "ai-coding" / "workflow_cn.md",
}


def load_guide(path: Path) -> tuple[dict, str]:
    content = path.read_text(encoding="utf-8")
    assert content.startswith("---\n")
    _, front_matter, body = content.split("---\n", maxsplit=2)
    return yaml.safe_load(front_matter), body


def test_ai_coding_guides_have_matching_identity_and_version() -> None:
    loaded = {lang: load_guide(path) for lang, path in GUIDES.items()}
    english, _ = loaded["en-US"]
    chinese, _ = loaded["zh-CN"]

    assert english["id"] == chinese["id"] == "python-ai-coding-workflow"
    assert english["content_version"] == chinese["content_version"] == 1
    assert english["reviewed_on"] == chinese["reviewed_on"]
    assert english["lang"] == "en-US"
    assert chinese["lang"] == "zh-CN"


def test_ai_coding_guides_cover_the_same_eight_step_workflow() -> None:
    for path in GUIDES.values():
        _, body = load_guide(path)
        headings = re.findall(r"^## ([1-8])\.", body, flags=re.MULTILINE)
        assert headings == [str(number) for number in range(1, 9)]


def test_guide_reference_urls_come_from_reviewed_catalog() -> None:
    catalog = load_catalog(ROOT / "catalog")
    catalog_urls = {resource["url"] for resource in catalog["resources"]}

    for path in GUIDES.values():
        _, body = load_guide(path)
        urls = set(re.findall(r"\]\((https://[^)]+)\)", body))
        assert urls
        assert {url for url in urls if not url.startswith(FIRST_PARTY_PREFIX)} <= catalog_urls


def test_readmes_link_to_the_matching_ai_coding_guide() -> None:
    english = (ROOT / "README.md").read_text(encoding="utf-8")
    chinese = (ROOT / "README_cn.md").read_text(encoding="utf-8")

    assert "guides/ai-coding/workflow.md" in english
    assert "guides/ai-coding/workflow_cn.md" in chinese