from __future__ import annotations

import pytest
from fastapi.testclient import TestClient

from chat.app import app


@pytest.fixture
def client(tmp_path, monkeypatch):
    config_path = tmp_path / "config.toml"
    config_path.write_text('featherless_api_key = "test"\n')
    monkeypatch.setenv("CHAT_CONFIG_PATH", str(config_path))
    monkeypatch.setenv("CHAT_DB_PATH", str(tmp_path / "test.db"))
    with TestClient(app) as c:
        yield c


def test_get_new_bot_form_renders(client):
    response = client.get("/bots/new")
    assert response.status_code == 200
    body = response.text.lower()
    assert "<form" in body
    assert "name" in body
    assert "persona" in body
    assert "kickoff" in body


def test_post_new_bot_appends_event_and_redirects(client, tmp_path):
    response = client.post(
        "/bots/new",
        data={
            "id": "bot_a",
            "name": "BotA",
            "persona": "thoughtful, observant",
            "voice_samples": "first sample\n---\nsecond sample",
            "traits": "shy, quick to anger",
            "backstory": "grew up in a small town",
            "initial_relationship_to_you": "coworker",
            "kickoff_prose": "you stay late at the office",
        },
        follow_redirects=False,
    )
    assert response.status_code == 303
    assert response.headers["location"] == "/bots/bot_a/kickoff"

    from chat.db.connection import open_db
    from chat.state.entities import get_bot

    with open_db(tmp_path / "test.db") as conn:
        bot = get_bot(conn, "bot_a")
        assert bot is not None
        assert bot["name"] == "BotA"
        assert bot["voice_samples"] == ["first sample", "second sample"]
        assert bot["traits"] == ["shy", "quick to anger"]
        assert bot["backstory"] == "grew up in a small town"
        assert bot["initial_relationship_to_you"] == "coworker"
        assert bot["kickoff_prose"] == "you stay late at the office"

        # Confirm event was actually appended (state goes through event log).
        cur = conn.execute(
            "SELECT kind, payload_json FROM event_log WHERE kind = 'bot_authored'"
        )
        rows = cur.fetchall()
        assert len(rows) == 1


def test_post_new_bot_rejects_missing_required(client):
    response = client.post(
        "/bots/new",
        data={"id": "bot_b"},
        follow_redirects=False,
    )
    assert response.status_code == 400


def test_get_bots_list_renders(client):
    response = client.get("/bots")
    assert response.status_code == 200


def test_post_new_bot_empty_traits_parses_to_empty_list(client, tmp_path):
    response = client.post(
        "/bots/new",
        data={
            "id": "bot_c",
            "name": "BotC",
            "persona": "stoic",
            "voice_samples": "",
            "traits": "",
            "backstory": "",
            "initial_relationship_to_you": "stranger",
            "kickoff_prose": "the rain begins",
        },
        follow_redirects=False,
    )
    assert response.status_code == 303

    from chat.db.connection import open_db
    from chat.state.entities import get_bot

    with open_db(tmp_path / "test.db") as conn:
        bot = get_bot(conn, "bot_c")
        assert bot is not None
        assert bot["voice_samples"] == []
        assert bot["traits"] == []


def test_post_new_bot_traits_split_by_newlines(client, tmp_path):
    response = client.post(
        "/bots/new",
        data={
            "id": "bot_d",
            "name": "BotD",
            "persona": "curious",
            "voice_samples": "",
            "traits": "calm\nthoughtful\nguarded",
            "backstory": "",
            "initial_relationship_to_you": "neighbor",
            "kickoff_prose": "morning light",
        },
        follow_redirects=False,
    )
    assert response.status_code == 303

    from chat.db.connection import open_db
    from chat.state.entities import get_bot

    with open_db(tmp_path / "test.db") as conn:
        bot = get_bot(conn, "bot_d")
        assert bot is not None
        assert bot["traits"] == ["calm", "thoughtful", "guarded"]