import json

from gfa_core.mas.agents.conversation_tools import build_plantai_conversation_tools
from gfa_core.mas.llm import LLMManager
from gfa_core.mas.workflow_models import GoalLedger


def _registry(tmp_path, *, job_id="job-data-preparation-handoff"):
    ledger = GoalLedger(
        conversation_id="conversation-data-preparation-handoff",
        revision=0,
        user_terms={
            "active_job_id": job_id,
            "gene_family": "SPL",
            "target_species": ["Ft"],
            "reference_species": ["At"],
            "config_confirmed": True,
        },
    )
    return build_plantai_conversation_tools(
        llm_manager=LLMManager({}),
        project_root=tmp_path,
        bound_job_id=job_id,
        goal_ledger=ledger,
    )


def test_analysis_handoff_is_not_rewritten_to_genome_prepare(tmp_path):
    resolved = _registry(tmp_path).resolve_analysis_handoff_request(
        "执行亚家族分类",
        workflow_id="gene_family",
        guidance=True,
    )

    assert resolved.startswith("执行亚家族分类\n")
    assert "执行数据准备（genome_prepare）" not in resolved
    assert "目标物种 Ft" in resolved
    assert "参考物种 At" in resolved
    assert "正式 config.xlsx 已确认" in resolved


def test_genome_prepare_manifest_does_not_change_analysis_handoff(tmp_path):
    job_id = "job-manifest-independent-handoff"
    registry = _registry(tmp_path, job_id=job_id)
    before = registry.resolve_analysis_handoff_request(
        "执行亚家族分类",
        workflow_id="gene_family",
        guidance=True,
    )
    final_dir = tmp_path / "workspace" / job_id / "final"
    final_dir.mkdir(parents=True)
    (final_dir / "genome_prepare.genome_prepare.manifest.json").write_text(
        json.dumps({"stage": "genome_prepare", "status": "success"}),
        encoding="utf-8",
    )

    after = registry.resolve_analysis_handoff_request(
        "执行亚家族分类",
        workflow_id="gene_family",
        guidance=True,
    )

    assert after == before


def test_explicit_data_preparation_request_is_preserved(tmp_path):
    resolved = _registry(tmp_path).resolve_analysis_handoff_request(
        "执行数据准备",
        workflow_id="data_preparation",
        guidance=False,
    )

    assert resolved.startswith("执行数据准备\n")
    assert "当前业务动作已由数据准备契约唯一确定" not in resolved


def test_next_step_question_is_not_rewritten_to_genome_prepare(tmp_path):
    resolved = _registry(tmp_path).resolve_analysis_handoff_request(
        "接下来需要做什么？",
        workflow_id="gene_family",
        guidance=True,
    )

    assert resolved.startswith("接下来需要做什么？\n")
    assert "执行数据准备（genome_prepare）" not in resolved