from datetime import datetime, timezone
from typing import TYPE_CHECKING

from sqlalchemy import DateTime, ForeignKey, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column, relationship

from db.base import Base

if TYPE_CHECKING:
    from db.models.thread import Thread


def _utcnow() -> datetime:
    return datetime.now(timezone.utc)


class Message(Base):
    __tablename__ = "messages"

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    thread_id: Mapped[int] = mapped_column(
        ForeignKey("threads.id", ondelete="CASCADE"), nullable=False, index=True
    )
    role: Mapped[str] = mapped_column(String(20), nullable=False)  # "user" | "assistant"
    text: Mapped[str] = mapped_column(Text, nullable=False)
    sources_json: Mapped[str | None] = mapped_column(Text, nullable=True)
    assembled_prompt: Mapped[str | None] = mapped_column(Text, nullable=True)
    # Ветка, которую выбрал роутер для этой реплики (проставляется со Спринта 4).
    intent_id: Mapped[int | None] = mapped_column(
        ForeignKey("intents.id", ondelete="SET NULL"), nullable=True, index=True
    )
    # JSON со снимком обработки реплики: решение роутера, шаг, список событий.
    # Используется в Песочнице для отображения подробных пилюль (со Спринта 6b).
    meta_json: Mapped[str | None] = mapped_column(Text, nullable=True)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=_utcnow, nullable=False)

    thread: Mapped["Thread"] = relationship(back_populates="messages")