"""Answer + resume routes. Resume now enqueues a command for the control worker (the tmux
session lives in the control container), so we assert on the queued command, not an
in-process seam call."""

from __future__ import annotations

from handler.db import repository as repo
from handler.db.engine import get_engine


def _seed_agent_with_question(env):
    """Seed a project + agent + an open question directly in the DB."""
    with get_engine().begin() as conn:
        repo.create_project(conn, "proj", "/tmp/proj")
        a = repo.create_agent(conn, "proj", "api", "/tmp/proj/api", status="paused_for_input")
        log_id = repo.insert_log_entry(
            conn, a["id"], status="paused_for_input", question="Which DB?"
        )
    return a, log_id


def test_answer_backfills_latest_open_question(client, auth, env):
    _seed_agent_with_question(env)
    r = client.post(
        "/projects/proj/agents/api/answer",
        json={"answer": "Postgres"},
        headers=auth,
    )
    assert r.status_code == 200
    assert r.json()["answered"] is True

    with get_engine().begin() as conn:
        a = repo.get_agent_by_name(conn, "proj", "api")
        assert repo.get_log(conn, a["id"])[0]["answer"] == "Postgres"


def test_answer_with_no_open_question_is_404(client, auth, env):
    with get_engine().begin() as conn:
        repo.create_project(conn, "proj", "/tmp/proj")
        repo.create_agent(conn, "proj", "api", "/tmp/proj/api")
    r = client.post(
        "/projects/proj/agents/api/answer", json={"answer": "x"}, headers=auth
    )
    assert r.status_code == 404


def test_resume_enqueues_command_with_the_answer(client, auth, env):
    _seed_agent_with_question(env)
    client.post(
        "/projects/proj/agents/api/answer", json={"answer": "Postgres"}, headers=auth
    )

    r = client.post("/projects/proj/agents/api/resume", json={}, headers=auth)
    assert r.status_code == 202
    body = r.json()
    assert body["type"] == "resume"
    assert body["agent_name"] == "api"
    assert body["status"] == "queued"
    # The API resolves the stored answer and hands it to the worker via the payload.
    assert body["payload"]["answer"] == "Postgres"

    with get_engine().begin() as conn:
        commands = repo.list_commands(conn, project_id="proj")
        assert [c["type"] for c in commands] == ["resume"]


def test_resume_without_answer_is_400(client, auth, env):
    with get_engine().begin() as conn:
        repo.create_project(conn, "proj", "/tmp/proj")
        repo.create_agent(conn, "proj", "api", "/tmp/proj/api")
    r = client.post("/projects/proj/agents/api/resume", json={}, headers=auth)
    assert r.status_code == 400