import pytest


def _make_config():
    from pyra.config.schema import PyraConfig, ProviderConfig
    return PyraConfig(ai=ProviderConfig(provider_id="lmstudio", model="gemma"))


@pytest.fixture
def history(tmp_pyra_home, monkeypatch):
    monkeypatch.setattr("pyra.chat.history.load_context_for_session", lambda: "")
    from pyra.chat.history import ConversationHistory
    return ConversationHistory(_make_config())


def test_build_for_api_first_message_is_system(history):
    msgs = history.build_for_api()
    assert msgs[0]["role"] == "system"


def test_build_for_api_system_contains_pyra(history):
    msgs = history.build_for_api()
    assert "Pyra" in msgs[0]["content"]


def test_build_for_api_includes_memory_context(tmp_pyra_home, monkeypatch):
    monkeypatch.setattr(
        "pyra.chat.history.load_context_for_session",
        lambda: "## Long-term Memory\n\nSome remembered facts.",
    )
    from pyra.chat.history import ConversationHistory
    h = ConversationHistory(_make_config())
    msgs = h.build_for_api()
    assert "Long-term Memory" in msgs[0]["content"]


def test_add_user_appears_in_api_payload(history):
    history.add_user("hello from user")
    msgs = history.build_for_api()
    user_msgs = [m for m in msgs if m["role"] == "user"]
    assert any(m["content"] == "hello from user" for m in user_msgs)


def test_add_assistant_appears_in_api_payload(history):
    history.add_assistant("hello from assistant")
    msgs = history.build_for_api()
    asst_msgs = [m for m in msgs if m["role"] == "assistant"]
    assert any(m["content"] == "hello from assistant" for m in asst_msgs)


def test_add_tool_result(history):
    history.add_tool_result("call_abc", "tool output")
    msgs = history.build_for_api()
    tool_msgs = [m for m in msgs if m.get("role") == "tool"]
    assert len(tool_msgs) == 1
    assert tool_msgs[0]["tool_call_id"] == "call_abc"
    assert tool_msgs[0]["content"] == "tool output"


def test_clear_removes_messages(history):
    history.add_user("hello")
    history.add_assistant("hi")
    history.clear()
    msgs = history.build_for_api()
    assert all(m["role"] == "system" for m in msgs)


def test_trim_to_budget_drops_old_messages():
    from pyra.chat.history import _trim_to_budget
    msgs = [
        {"role": "user", "content": "a" * 4000},
        {"role": "assistant", "content": "b" * 4000},
        {"role": "user", "content": "new message"},
    ]
    trimmed = _trim_to_budget(list(msgs), 100)
    assert len(trimmed) < 3
    assert any(m["content"] == "new message" for m in trimmed)


def test_trim_to_budget_no_trim_when_under_budget():
    from pyra.chat.history import _trim_to_budget
    msgs = [{"role": "user", "content": "short"}]
    result = _trim_to_budget(list(msgs), 10000)
    assert len(result) == 1
    assert result[0]["content"] == "short"


def test_trim_to_budget_empty_list():
    from pyra.chat.history import _trim_to_budget
    assert _trim_to_budget([], 1000) == []