from datetime import datetime, timezone

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

from db.base import Base


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


class AgentConfig(Base):
    """Версионируемая конфигурация агента: системный промпт + правила.

    Принцип: сохранённые версии не редактируются, только создаются новые.
    Активна одна версия одновременно на каждую ветку (intent).

    Со Спринта 4 конфиг привязан к ветке: activate = «сделать активной в рамках intent_id».
    """
    __tablename__ = "agent_configs"
    __table_args__ = (UniqueConstraint("intent_id", "version", name="uq_config_intent_version"),)

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    intent_id: Mapped[int | None] = mapped_column(
        ForeignKey("intents.id", ondelete="SET NULL"), nullable=True, index=True
    )
    version: Mapped[int] = mapped_column(Integer, nullable=False, index=True)
    name: Mapped[str | None] = mapped_column(String(200), nullable=True)
    system_prompt: Mapped[str] = mapped_column(Text, nullable=False)
    rules_text: Mapped[str] = mapped_column(Text, nullable=False, default="")
    is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False, index=True)
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=_utcnow, nullable=False)