"""Tests for `.astro` extraction (#850).

Astro files have a TypeScript frontmatter block (`---...---`) at the top where
nearly all imports live, followed by an HTML-with-expressions template and
optionally `<script>` blocks. Tree-sitter-javascript fed the whole file produces
a top-level ERROR node because the template is not valid JS, so the JS AST pass
recovers nothing. The :func:`extract_astro` regex pass salvages imports from the
frontmatter and any `<script>` blocks — same strategy as :func:`extract_svelte`.
"""
from __future__ import annotations

from pathlib import Path

from graphify.detect import CODE_EXTENSIONS
from graphify.extract import (
    _make_id,
    extract_astro,
)


def _write(path: Path, body: str) -> Path:
    path.parent.mkdir(parents=True, exist_ok=True)
    path.write_text(body, encoding="utf-8")
    return path


def _import_targets(result: dict, *, relation: str | None = None) -> set[str]:
    return {
        str(e.get("target") or "")
        for e in result.get("edges", [])
        if relation is None or e.get("relation") == relation
    }


def test_astro_is_in_code_extensions():
    """Without this, detect.py silently drops `.astro` from the AST pass (#850)."""
    assert ".astro" in CODE_EXTENSIONS


def test_extract_astro_picks_up_frontmatter_static_imports(tmp_path):
    page = _write(
        tmp_path / "src/pages/index.astro",
        """---
import Layout from '../layouts/Layout.astro';
import Hero from '../components/Hero.astro';
const { title } = Astro.props;
---

<Layout title={title}>
  <Hero />
</Layout>
""",
    )
    # Sibling files so the resolver lands on real node ids, not phantoms.
    layout = _write(tmp_path / "src/layouts/Layout.astro", "---\n---\n<slot />\n")
    hero = _write(tmp_path / "src/components/Hero.astro", "---\n---\n<h1>hi</h1>\n")

    result = extract_astro(page)
    targets = _import_targets(result, relation="imports_from")
    assert _make_id(str(layout)) in targets
    assert _make_id(str(hero)) in targets


def test_extract_astro_handles_dynamic_import_in_frontmatter(tmp_path):
    page = _write(
        tmp_path / "src/pages/lazy.astro",
        """---
const Mod = await import('./Other.astro');
---

<div>{Mod.default}</div>
""",
    )
    other = _write(tmp_path / "src/pages/Other.astro", "---\n---\n<p>o</p>\n")

    result = extract_astro(page)
    targets = _import_targets(result, relation="dynamic_import")
    assert _make_id(str(other)) in targets


def test_extract_astro_picks_up_client_side_script_imports(tmp_path):
    page = _write(
        tmp_path / "src/pages/with-script.astro",
        """---
import Layout from '../layouts/Layout.astro';
---

<Layout>
  <button id="b">click</button>
</Layout>

<script>
  import { hydrate } from '../client/hydrate.ts';
  hydrate(document.getElementById('b'));
</script>
""",
    )
    layout = _write(tmp_path / "src/layouts/Layout.astro", "---\n---\n<slot />\n")
    hydrate = _write(tmp_path / "src/client/hydrate.ts", "export function hydrate(){}\n")

    result = extract_astro(page)
    targets = _import_targets(result, relation="imports_from")
    assert _make_id(str(layout)) in targets
    assert _make_id(str(hydrate)) in targets


def test_extract_astro_no_frontmatter_does_not_crash(tmp_path):
    """Astro permits frontmatter-less files (pure-HTML pages). Must not raise."""
    page = _write(
        tmp_path / "src/pages/plain.astro",
        "<h1>no frontmatter here</h1>\n",
    )
    result = extract_astro(page)
    # Empty/no-imports result is acceptable; the extractor must just not crash.
    assert isinstance(result, dict)
    assert _import_targets(result, relation="imports_from") == set()


def test_extract_astro_handles_tsconfig_path_alias(tmp_path):
    _write(
        tmp_path / "tsconfig.json",
        """{
  "compilerOptions": {
    "baseUrl": ".",
    "paths": { "@components/*": ["src/components/*"] }
  }
}
""",
    )
    page = _write(
        tmp_path / "src/pages/alias.astro",
        """---
import Hero from '@components/Hero.astro';
---

<Hero />
""",
    )
    hero = _write(tmp_path / "src/components/Hero.astro", "---\n---\n<h1>h</h1>\n")

    result = extract_astro(page)
    targets = _import_targets(result, relation="imports_from")
    assert _make_id(str(hero)) in targets