"""Poker-mode message classifier + fragment selection (pure, no DB)."""
from __future__ import annotations

from lyra import poker_prompts as pp


def c(msg, roster=()):
    return pp.classify(msg, roster)


# --- the spec's canonical cases ---

def test_read_villain_action_beats_hand():
    # A villain's action carries cards+position+verb but is NOT Brian's hand.
    assert c("TAG limped A4o in the SB (UTG straddled)") == "READ"
    assert c("Jonathan called the 3bet") == "READ"
    assert c("the neck-tattoo guy shoved the turn") == "READ"


def test_hand_is_first_person():
    assert c("Button straddle on. I limp UTG with 22. Flop 2d7cjh, I check-raise") == "HAND"
    assert c("I flopped a set with 99 on 9h4c2d and bet the turn") == "HAND"


def test_hand_narrated_without_I_still_hand_not_read():
    # No "I", but leads with a poker verb (not a name) + street/action → his hand.
    assert c("Flopped bottom set with 22, bet $40 on the river, he folded 88") == "HAND"


def test_table_ops():
    assert c("seat the table: TAG, Jonathan, Wheelz") == "TABLE"
    assert c("table broke, I'm at a new table") == "TABLE"
    assert c("I got moved to another table") == "TABLE"


def test_mental():
    assert c("I feel like I'm being mean when I raise") == "MENTAL"
    assert c("ugh I'm so tilted, card dead all night") == "MENTAL"


def test_status_is_not_a_mood():
    assert c("it's 11:50pm, waiting for a seat") == "STATUS"
    assert c("grabbing food, be right back") == "STATUS"


def test_log_bare_money():
    assert c("I'm at 317 now") == "LOG"
    assert c("stack is 540") == "LOG"


def test_chat_default():
    assert c("should I have folded the river?") == "CHAT"
    assert c("what do you think of this table so far") == "CHAT"


# --- the READ vs HAND boundary (the hard one) ---

def test_roster_handle_forces_read():
    # A seated handle as the actor → READ even if lowercase / plain.
    assert c("tag opened to 15 from the cutoff", roster=("TAG",)) == "READ"


def test_first_person_action_stays_hand_even_with_roster():
    # Brian is the actor → HAND, not a read on a seated player mentioned nearby.
    assert c("I 3bet TAG's open with AKs", roster=("TAG",)) == "HAND"


def test_all_caps_handle_reads_without_roster():
    assert c("JD min-raised the button") == "READ"


# --- fragment selection ---

def test_fragment_for_maps_each_type():
    for t in pp.MSG_TYPES:
        assert pp.fragment_for(t) is pp.FRAGMENTS[t]
    assert pp.fragment_for(None) is pp.FRAGMENTS["CHAT"]
    assert pp.fragment_for("bogus") is pp.FRAGMENTS["CHAT"]


def test_base_is_nonempty_and_names_the_hard_rules():
    assert "LOG FIRST" in pp.BASE
    assert "descriptor" in pp.BASE and "session_state" in pp.BASE