# SPDX-License-Identifier: AGPL-3.0-only
# Copyright (c) 2026 sol pbc

import json
from datetime import datetime
from pathlib import Path
from types import SimpleNamespace

import pytest

from convey.chat_stream import (
    append_chat_event,
    find_unresponded_trigger,
    read_chat_events,
    read_chat_tail,
    reduce_chat_state,
)


def _setup_journal(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> Path:
    journal = tmp_path / "journal"
    journal.mkdir()
    monkeypatch.setenv("_SOLSTONE_JOURNAL_OVERRIDE", str(journal))
    return journal


def _ms(year: int, month: int, day: int, hour: int, minute: int, second: int) -> int:
    return int(datetime(year, month, day, hour, minute, second).timestamp() * 1000)


def test_append_owner_message_creates_segment_and_jsonl(tmp_path, monkeypatch):
    journal = _setup_journal(tmp_path, monkeypatch)
    ts = _ms(2026, 4, 20, 12, 0, 0)

    event = append_chat_event(
        "owner_message",
        ts=ts,
        text="hello",
        app="sol",
        path="/chat",
        facet="work",
    )

    segment_dir = journal / "chronicle" / "20260420" / "chat" / "120000_300"
    chat_path = segment_dir / "chat.jsonl"

    assert event["kind"] == "owner_message"
    assert event["ts"] == ts
    assert segment_dir.is_dir()
    assert (segment_dir / "stream.json").is_file()
    assert chat_path.is_file()
    assert (journal / "streams" / "chat.json").is_file()

    entries = [
        json.loads(line) for line in chat_path.read_text(encoding="utf-8").splitlines()
    ]
    assert entries == [event]


def test_append_is_atomic(tmp_path, monkeypatch):
    journal = _setup_journal(tmp_path, monkeypatch)

    append_chat_event(
        "owner_message",
        ts=_ms(2026, 4, 20, 12, 0, 0),
        text="hello",
        app="sol",
        path="/chat",
        facet="work",
    )

    chat_path = (
        journal / "chronicle" / "20260420" / "chat" / "120000_300" / "chat.jsonl"
    )
    lines = chat_path.read_text(encoding="utf-8").splitlines()

    assert len(lines) == 1
    assert json.loads(lines[0])["text"] == "hello"


def test_append_broadcasts_on_chat_tract_with_stored_event_payload(
    tmp_path, monkeypatch
):
    _setup_journal(tmp_path, monkeypatch)
    import convey.chat as chat

    calls: list[tuple[str, str, dict]] = []

    class FakeCallosum:
        def emit(self, tract, event, **fields):
            calls.append((tract, event, fields))
            return True

    monkeypatch.setattr(chat, "_runtime", SimpleNamespace(callosum=FakeCallosum()))

    event = append_chat_event(
        "sol_message",
        ts=_ms(2026, 4, 20, 12, 0, 0),
        use_id="1713626000000",
        text="hello",
        notes="ready",
        requested_target=None,
        requested_task=None,
    )

    assert calls == [("chat", "sol_message", event)]


def test_append_broadcast_failure_is_swallowed(tmp_path, monkeypatch):
    journal = _setup_journal(tmp_path, monkeypatch)
    import convey.chat as chat

    class FakeCallosum:
        def emit(self, tract, event, **fields):
            raise RuntimeError("boom")

    monkeypatch.setattr(chat, "_runtime", SimpleNamespace(callosum=FakeCallosum()))

    event = append_chat_event(
        "owner_message",
        ts=_ms(2026, 4, 20, 12, 0, 0),
        text="hello",
        app="sol",
        path="/chat",
        facet="work",
    )

    chat_path = (
        journal / "chronicle" / "20260420" / "chat" / "120000_300" / "chat.jsonl"
    )

    assert event["kind"] == "owner_message"
    lines = chat_path.read_text(encoding="utf-8").splitlines()
    assert len(lines) == 1
    assert json.loads(lines[0]) == event


def test_append_rejects_unknown_kind(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)

    with pytest.raises(ValueError, match="Unknown chat event kind"):
        append_chat_event("unknown", ts=1)


def test_append_rejects_missing_required_fields(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)

    with pytest.raises(ValueError, match="owner_message requires fields: path, facet"):
        append_chat_event(
            "owner_message",
            ts=1,
            text="hello",
            app="sol",
        )


def test_append_rolls_at_300_seconds(tmp_path, monkeypatch):
    journal = _setup_journal(tmp_path, monkeypatch)
    start = _ms(2026, 4, 20, 12, 0, 0)

    append_chat_event(
        "owner_message",
        ts=start,
        text="first",
        app="sol",
        path="/chat",
        facet="work",
    )
    append_chat_event(
        "owner_message",
        ts=start + 299_999,
        text="second",
        app="sol",
        path="/chat",
        facet="work",
    )
    append_chat_event(
        "owner_message",
        ts=start + 300_000,
        text="third",
        app="sol",
        path="/chat",
        facet="work",
    )

    chat_root = journal / "chronicle" / "20260420" / "chat"
    assert sorted(path.name for path in chat_root.iterdir()) == [
        "120000_300",
        "120500_300",
    ]


def test_append_rolls_at_day_cross(tmp_path, monkeypatch):
    journal = _setup_journal(tmp_path, monkeypatch)
    first = _ms(2026, 4, 20, 23, 59, 59)
    second = _ms(2026, 4, 21, 0, 0, 0)

    append_chat_event(
        "owner_message",
        ts=first,
        text="late",
        app="sol",
        path="/chat",
        facet="work",
    )
    append_chat_event(
        "owner_message",
        ts=second,
        text="next day",
        app="sol",
        path="/chat",
        facet="work",
    )

    assert (
        journal / "chronicle" / "20260420" / "chat" / "235959_300" / "chat.jsonl"
    ).is_file()
    assert (
        journal / "chronicle" / "20260421" / "chat" / "000000_300" / "chat.jsonl"
    ).is_file()


def test_read_chat_events_returns_ordered(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)
    start = _ms(2026, 4, 20, 12, 0, 0)

    append_chat_event(
        "owner_message",
        ts=start,
        text="first",
        app="sol",
        path="/chat",
        facet="work",
    )
    append_chat_event(
        "owner_message",
        ts=start + 300_000,
        text="second",
        app="sol",
        path="/chat",
        facet="work",
    )
    append_chat_event(
        "owner_message",
        ts=start + 600_000,
        text="third",
        app="sol",
        path="/chat",
        facet="work",
    )

    events = read_chat_events("20260420")
    assert [event["text"] for event in events] == ["first", "second", "third"]
    assert [event["ts"] for event in events] == [
        start,
        start + 300_000,
        start + 600_000,
    ]


def test_read_chat_tail_last_n(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)
    start = _ms(2026, 4, 20, 12, 0, 0)

    for index in range(4):
        append_chat_event(
            "owner_message",
            ts=start + (index * 60_000),
            text=f"msg-{index}",
            app="sol",
            path="/chat",
            facet="work",
        )

    tail = read_chat_tail("20260420", limit=2)
    assert [event["text"] for event in tail] == ["msg-2", "msg-3"]


def test_reduce_chat_state_extracts_latest_sol_and_active_talents(
    tmp_path, monkeypatch
):
    _setup_journal(tmp_path, monkeypatch)
    start = _ms(2026, 4, 20, 12, 0, 0)

    append_chat_event(
        "owner_message",
        ts=start,
        text="hello",
        app="sol",
        path="/chat",
        facet="work",
    )
    append_chat_event(
        "sol_message",
        ts=start + 1_000,
        use_id="chat-1",
        text="dispatching",
        notes="planning",
        requested_target="exec",
        requested_task="compare drafts",
    )
    append_chat_event(
        "talent_spawned",
        ts=start + 2_000,
        use_id="exec-1",
        name="exec",
        task="compare drafts",
        started_at=start + 2_000,
    )
    append_chat_event(
        "talent_finished",
        ts=start + 3_000,
        use_id="exec-1",
        name="exec",
        summary="done",
    )
    append_chat_event(
        "talent_spawned",
        ts=start + 4_000,
        use_id="exec-2",
        name="exec",
        task="write summary",
        started_at=start + 4_000,
    )
    append_chat_event(
        "talent_errored",
        ts=start + 5_000,
        use_id="exec-3",
        name="exec",
        reason="bad input",
    )
    append_chat_event(
        "chat_error",
        ts=start + 6_000,
        reason="chat had trouble — try again",
        use_id=None,
    )
    append_chat_event(
        "reflection_ready",
        ts=start + 7_000,
        day="20260308",
        url="/app/reflections/20260308",
    )

    reduced = reduce_chat_state("20260420")

    assert reduced["latest_sol_message"] == {
        "ts": start + 1_000,
        "use_id": "chat-1",
        "text": "dispatching",
        "notes": "planning",
        "requested_target": "exec",
        "requested_task": "compare drafts",
    }
    assert reduced["active_talents"] == [
        {
            "use_id": "exec-2",
            "name": "exec",
            "task": "write summary",
            "started_at": start + 4_000,
        }
    ]
    assert reduced["completed_talents"] == [
        {
            "use_id": "exec-1",
            "name": "exec",
            "task": "compare drafts",
            "summary": "done",
            "finished_at": start + 3_000,
        }
    ]


def test_append_reflection_ready_event(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)
    ts = _ms(2026, 4, 20, 12, 0, 0)

    event = append_chat_event(
        "reflection_ready",
        ts=ts,
        day="20260308",
        url="/app/reflections/20260308",
    )

    assert event["kind"] == "reflection_ready"
    assert event["day"] == "20260308"
    assert event["url"] == "/app/reflections/20260308"


def test_find_unresponded_trigger_owner_message(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)
    ts = _ms(2026, 4, 20, 12, 0, 0)

    append_chat_event(
        "owner_message",
        ts=ts,
        text="hello",
        app="sol",
        path="/chat",
        facet="work",
    )

    trigger = find_unresponded_trigger("20260420")
    assert trigger is not None
    assert trigger["kind"] == "owner_message"
    assert trigger["text"] == "hello"


def test_find_unresponded_trigger_talent_finished(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)
    start = _ms(2026, 4, 20, 12, 0, 0)

    append_chat_event(
        "owner_message",
        ts=start,
        text="hello",
        app="sol",
        path="/chat",
        facet="work",
    )
    append_chat_event(
        "sol_message",
        ts=start + 1_000,
        use_id="chat-1",
        text="working",
        notes="",
        requested_target=None,
        requested_task=None,
    )
    append_chat_event(
        "talent_finished",
        ts=start + 2_000,
        use_id="exec-1",
        name="exec",
        summary="done",
    )

    trigger = find_unresponded_trigger("20260420")
    assert trigger is not None
    assert trigger["kind"] == "talent_finished"
    assert trigger["summary"] == "done"


def test_find_unresponded_trigger_resolved(tmp_path, monkeypatch):
    _setup_journal(tmp_path, monkeypatch)
    start = _ms(2026, 4, 20, 12, 0, 0)

    append_chat_event(
        "talent_finished",
        ts=start,
        use_id="exec-1",
        name="exec",
        summary="done",
    )
    append_chat_event(
        "sol_message",
        ts=start + 1_000,
        use_id="chat-1",
        text="thanks",
        notes="",
        requested_target=None,
        requested_task=None,
    )

    assert find_unresponded_trigger("20260420") is None