"""Perceive: cheap heuristic read of the moment, and route turning it into a nudge."""
from __future__ import annotations

import importlib

import pytest

from lyra import perceive


def test_reads_tilt():
    m = perceive.read("I'm so fucking tilted, card dead all night, this is brutal!!")
    assert m["tilt"] >= 0.5 and m["sentiment"] < 0 and m["kind"] == "emotional"


def test_reads_strategy_calm():
    m = perceive.read("Should I fold the river here given his range and the board?")
    assert m["kind"] == "strategic" and m["tilt"] < 0.4


def test_reads_up_energy():
    m = perceive.read("Let's go!! crushing it tonight, feeling so good!")
    assert m["sentiment"] > 0 and m["kind"] == "emotional"


def test_reads_build_and_casual():
    assert perceive.read("let's refactor the cognition pipeline module").get("kind") == "build"
    assert perceive.read("ok sounds good to me").get("kind") == "casual"
    assert perceive.read("ok sounds good to me")["tilt"] == 0.0


@pytest.fixture
def mind(tmp_path, monkeypatch):
    monkeypatch.setenv("LYRA_DB_PATH", str(tmp_path / "test.db"))
    monkeypatch.setenv("CHAT_DELIBERATE", "false")
    from lyra import llm
    monkeypatch.setattr(llm, "embed", lambda texts: [[0.1, 0.2, 0.3] for _ in texts])
    import lyra.memory as memory
    importlib.reload(memory)
    import lyra.mind as mind
    importlib.reload(mind)
    memory.ensure_session("s1")
    return mind


def test_route_injects_tilt_nudge(mind):
    turn = mind.assemble("s1", "ugh I'm steaming, fucking coolered again!!", "cloud", None)
    assert turn.register == "steady"
    sys_blob = " ".join(m["content"] for m in turn.messages if m["role"] == "system")
    assert "on tilt" in sys_blob.lower() or "frustrated" in sys_blob.lower()


def test_route_quiet_on_neutral_turn(mind):
    turn = mind.assemble("s1", "what did we decide about the schema yesterday?", "cloud", None)
    assert turn.register is None                    # neutral -> no nudge
    assert not (turn.moment or {}).get("note")


# --- Phase A pipeline fixes: poker mode suppresses the two mush sources ---

def test_poker_mode_suppresses_tilt_nudge(mind):
    from lyra import memory
    memory.set_session_mode("s1", "poker_cash")
    turn = mind.assemble("s1", "ugh I'm steaming, fucking coolered again!!", "cloud", None)
    assert turn.register is None                     # at the table, register comes from fragments
    sys_blob = " ".join(m["content"] for m in turn.messages if m["role"] == "system")
    assert "on tilt" not in sys_blob.lower()         # false-positive lexicon nudge suppressed


def test_mode_menu_note_suppressed_in_poker(mind):
    from lyra import modes
    poker = " ".join(m["content"] for m in mind.build_messages("s1", "stack 350", mode=modes.CASH)
                     if m["role"] == "system")
    build = " ".join(m["content"] for m in mind.build_messages("s1", "let's refactor", mode=modes.get("build"))
                     if m["role"] == "system")
    assert "Your modes:" not in poker                # no "offer to switch" note at the table
    assert "Your modes:" in build                    # still present in a non-poker mode


# --- Phase B: sharded poker prompt (BASE + one fragment, no monolith) ---

def _poker_blob(mind, msg):
    from lyra import modes
    return " ".join(m["content"] for m in mind.build_messages("s1", msg, mode=modes.CASH)
                    if m["role"] == "system")


def test_poker_injects_base_plus_the_matching_fragment(mind):
    blob = _poker_blob(mind, "I flopped a set with 99 on 9h4c2d and bet the turn")
    assert "LOG FIRST" in blob                       # BASE is always on in poker
    assert "MESSAGE TYPE: HAND" in blob              # the fragment for THIS message
    assert "MESSAGE TYPE: STATUS" not in blob        # and not the others
    assert "MESSAGE TYPE: READ" not in blob


def test_poker_fragment_changes_with_message_type(mind):
    status = _poker_blob(mind, "it's 11:50pm, waiting for a seat")
    assert "MESSAGE TYPE: STATUS" in status and "MESSAGE TYPE: HAND" not in status


def test_poker_monolith_no_longer_injected(mind):
    from lyra import modes
    blob = _poker_blob(mind, "stack 350")
    assert "You move between two registers" not in blob   # the old _CASH_CARD opener is gone
    assert modes.CASH.card == ""                           # card sharded out