from __future__ import annotations

import copy
from pathlib import Path

import pytest

from tools.graphrag.evaluation.freeze_graphrag import make_freeze
from tools.graphrag.evaluation.reproducibility import source_corpus_fingerprint


def freeze_inputs():
    questions = [
        {
            "id": "dev-a",
            "split": "dev",
            "graph": {"schema_version": 3, "source_profiles": ["pages/students/a/README.md"]},
        },
        {
            "id": "test-b",
            "split": "test",
            "graph": {"schema_version": 3, "source_profiles": ["pages/students/b/README.md"]},
        },
    ]
    fingerprints = {
        "implementation": {},
        "dataset_sha256": "dataset",
        "inputs": {},
        "source_corpus": {"sha256": "raw-corpus"},
    }
    audit = {
        "dataset_sha256": "dataset",
        "inputs": {},
        "source_corpus": {"sha256": "raw-corpus"},
        "questions": [{"id": q["id"], "source_consistent": True} for q in questions],
        "by_split": {},
    }
    dev = {
        "configuration": {"split": "dev", "limit": 40, "max_entities": 8},
        "questions": [{"id": "dev-a", "error": None}],
        "summary": {"error_count": 0},
        "run_metadata": {"fingerprints": fingerprints, "graph": {"sha256": "graph"}},
    }
    return dev, questions, fingerprints, audit


def test_primary_freeze_verifies_disjoint_profiles():
    args = freeze_inputs()
    manifest = make_freeze(*args)
    assert manifest["partition"]["overlap"] == 0
    assert manifest["partition"]["test_profile_count"] == 1


def test_primary_freeze_rejects_same_profile_across_splits():
    dev, questions, fingerprints, audit = freeze_inputs()
    questions[1]["graph"]["source_profiles"] = questions[0]["graph"]["source_profiles"]
    with pytest.raises(RuntimeError, match="profiles overlap"):
        make_freeze(dev, questions, fingerprints, audit)


def test_primary_freeze_rejects_unvalidated_source_expectation():
    dev, questions, fingerprints, audit = freeze_inputs()
    audit["questions"][1]["source_consistent"] = False
    with pytest.raises(RuntimeError, match="source-inconsistent"):
        make_freeze(dev, questions, fingerprints, audit)


def test_primary_freeze_rejects_stale_raw_markdown_audit():
    dev, questions, fingerprints, audit = freeze_inputs()
    audit["source_corpus"] = {"sha256": "stale"}
    with pytest.raises(RuntimeError, match="original Markdown"):
        make_freeze(dev, questions, fingerprints, audit)


def test_raw_source_fingerprint_detects_unindexed_change(tmp_path: Path, monkeypatch):
    wiki = tmp_path / "wiki"
    doc = wiki / "pages/students/example/README.md"
    doc.parent.mkdir(parents=True)
    doc.write_text("---\ntitle: Example\n---\nOriginal source\n")
    monkeypatch.setenv("ZPWIKI_ROOT", str(wiki))
    original = source_corpus_fingerprint()
    doc.write_text("---\ntitle: Example\n---\nChanged source\n")
    assert original["sha256"] != source_corpus_fingerprint()["sha256"]


def test_freeze_cli_does_not_overwrite_previous_manifest(tmp_path, monkeypatch):
    from tools.graphrag.evaluation.freeze_graphrag import main

    previous = tmp_path / "frozen_v1.json"
    previous.write_text("historical manifest")
    monkeypatch.setattr(
        "sys.argv",
        ["freeze", "--dev-results", "unused", "--audit", "unused", "--output", str(previous)],
    )
    with pytest.raises(SystemExit):
        main()
    assert previous.read_text() == "historical manifest"


def test_dev_overwrite_cannot_replace_archived_results(tmp_path):
    from argparse import Namespace
    from tools.graphrag.evaluation.evaluate_graphrag import validate_output_destination

    directory = tmp_path / "final_v1"
    directory.mkdir()
    previous = directory / "graphrag_dev.json"
    previous.write_text("archived DEV")
    args = Namespace(split="dev", ids="", max_questions=None, overwrite=True, output_dir=directory)
    with pytest.raises(RuntimeError, match="immutable"):
        validate_output_destination(args)
    assert previous.read_text() == "archived DEV"


def test_reproduction_requires_fresh_results_directory(tmp_path):
    from argparse import Namespace
    from tools.graphrag.evaluation.evaluate_graphrag import validate_output_destination

    previous = tmp_path / "graphrag_test.json"
    previous.write_text("archived TEST")
    args = Namespace(split="test", ids="", max_questions=None, overwrite=False, output_dir=tmp_path)
    with pytest.raises(RuntimeError, match="already exist"):
        validate_output_destination(args)
    assert previous.read_text() == "archived TEST"