from pathlib import Path

from click.testing import CliRunner

from markitect_tool.backend import LocalSnapshotStore, local_index_path_for
from markitect_tool.cli import main


def test_local_snapshot_store_persists_state_and_document(tmp_path: Path):
    source = tmp_path / "doc.md"
    source.write_text("---\ntitle: Example\n---\n# Doc\n\nBody.\n", encoding="utf-8")
    store = LocalSnapshotStore(tmp_path / ".markitect" / "cache" / "index.sqlite3")

    state = store.put_file(source, root=tmp_path)
    loaded = store.load_state()
    document = store.get_document("doc.md")

    assert state.path == "doc.md"
    assert state.snapshot_id.startswith("snapshot:")
    assert loaded[0].path == "doc.md"
    assert loaded[0].content_hash == state.content_hash
    assert document["frontmatter"]["title"] == "Example"
    assert document["headings"][0]["text"] == "Doc"


def test_local_snapshot_store_build_is_incremental(tmp_path: Path):
    source = tmp_path / "doc.md"
    source.write_text("# Doc\n", encoding="utf-8")
    store = LocalSnapshotStore(local_index_path_for(tmp_path))

    first = store.build([tmp_path], root=tmp_path)
    second = store.build([tmp_path], root=tmp_path)

    assert first.parsed == ["doc.md"]
    assert first.indexed == ["doc.md"]
    assert second.parsed == []
    assert second.indexed == []
    assert not second.dirty

    source.write_text("# Doc\n\nChanged.\n", encoding="utf-8")
    changed = store.build([tmp_path], root=tmp_path)

    assert changed.parsed == ["doc.md"]
    assert changed.indexed == ["doc.md"]


def test_local_snapshot_store_deletes_removed_files(tmp_path: Path):
    source = tmp_path / "doc.md"
    source.write_text("# Doc\n", encoding="utf-8")
    store = LocalSnapshotStore(local_index_path_for(tmp_path))
    store.build([tmp_path], root=tmp_path)

    source.unlink()
    result = store.build([tmp_path], root=tmp_path)

    assert result.deleted == ["doc.md"]
    assert store.load_state() == []


def test_mkt_ast_show_and_stats(tmp_path: Path):
    source = tmp_path / "doc.md"
    source.write_text("# Doc\n\nBody.\n", encoding="utf-8")
    runner = CliRunner()

    shown = runner.invoke(main, ["ast", "show", str(source), "--format", "tree"])
    stats = runner.invoke(main, ["ast", "stats", str(source)])

    assert shown.exit_code == 0
    assert "# Doc" in shown.output
    assert stats.exit_code == 0
    assert "headings: 1" in stats.output
    assert "paragraph_open" in stats.output


def test_mkt_cache_init_and_index(tmp_path: Path):
    source = tmp_path / "doc.md"
    source.write_text("# Doc\n", encoding="utf-8")
    runner = CliRunner()

    initialized = runner.invoke(main, ["cache", "init", "--root", str(tmp_path)])
    indexed = runner.invoke(main, ["cache", "index", str(tmp_path), "--root", str(tmp_path)])
    clean = runner.invoke(main, ["cache", "index", str(tmp_path), "--root", str(tmp_path)])

    assert initialized.exit_code == 0
    assert "schema_version: 1" in initialized.output
    assert indexed.exit_code == 0
    assert "parsed: 1" in indexed.output
    assert clean.exit_code == 0
    assert "clean" in clean.output