"""Состояние звонящего: базовый профиль, эмоциональная дуга, раздражение.

Дуга — фон, а не пьеса: она меняет подачу, а не факты. Директивы преподавателя
перекрывают дугу на время действия, повторные вопросы копят раздражение
(docs/product/CALL-SIM.md).
"""

from dataclasses import dataclass, field

from app.domain.events import Mood
from app.scenarios.schema import Persona

#: Настроение базового профиля, если дуга не задана.
BASE_MOOD: dict[str, Mood] = {
    "calm": Mood.CALM,
    "panic": Mood.PANIC,
    "aggressive": Mood.AGGRESSIVE,
    "elderly": Mood.CONFUSED,
    "drunk": Mood.CONFUSED,
    "evasive": Mood.WORRIED,
    "child": Mood.PANIC,
    "foreigner": Mood.WORRIED,
}

#: С какого числа повторов звонящий срывается в агрессию.
#: Первый повтор — «Я же сказал!», второй и дальше — уже крик.
AGGRESSION_AFTER_REPEATS = 2

#: Директивы, меняющие подачу. Остальные (обрыв связи, второй пострадавший)
#: правят факты и таймеры, а не настроение.
DIRECTIVE_MOOD: dict[str, Mood] = {
    "panic_rises": Mood.PANIC,
    "screaming": Mood.PANIC,
    "turns_aggressive": Mood.AGGRESSIVE,
    "distracted": Mood.CONFUSED,
}


@dataclass
class PersonaState:
    persona: Persona
    stage: str = "registration"
    repeats: int = 0
    directive: str | None = None
    history: list[Mood] = field(default_factory=list)

    @property
    def base(self) -> str:
        return self.persona.base

    def _arc_mood(self) -> Mood:
        for step in self.persona.arc:
            if step.stage == self.stage:
                return step.mood
        return BASE_MOOD.get(self.persona.base, Mood.CALM)

    @property
    def mood(self) -> Mood:
        """Приоритет: директива преподавателя → раздражение → дуга → база."""
        if self.directive in DIRECTIVE_MOOD:
            return DIRECTIVE_MOOD[self.directive]
        if self.repeats >= AGGRESSION_AFTER_REPEATS:
            return Mood.AGGRESSIVE
        return self._arc_mood()

    def on_repeat(self) -> None:
        """Повтор не штрафуется в оценке (иногда он оправдан), но звонящий
        его запоминает — и это попадает в разбор."""
        self.repeats += 1

    def advance(self, stage: str) -> None:
        self.stage = stage

    def remember(self) -> Mood:
        mood = self.mood
        self.history.append(mood)
        return mood