"""Agent trigger model — self-managed wake conditions for autonomous agents."""

import uuid
from datetime import datetime

from sqlalchemy import Boolean, DateTime, ForeignKey, Integer, String, Text, UniqueConstraint, func
from sqlalchemy.dialects.postgresql import UUID, JSONB
from sqlalchemy.orm import Mapped, mapped_column

from app.database import Base


class AgentTrigger(Base):
    """A trigger that an agent sets for itself to be woken up at a specific time or condition.

    Trigger types:
    - cron: croniter expression, e.g. {"expr": "0 9 * * 1-5"}
    - once: fire at a specific time, e.g. {"at": "2026-03-10T09:00:00+08:00"}
    - interval: fire every N minutes, e.g. {"minutes": 30}
    - poll: HTTP poll with change detection, e.g. {"url": "...", "json_path": "$.status", ...}
    - on_message: fire when receiving a message from a specific agent
    """

    __tablename__ = "agent_triggers"

    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
    agent_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), ForeignKey("agents.id", ondelete="CASCADE"), nullable=False, index=True
    )
    name: Mapped[str] = mapped_column(String(100), nullable=False)
    type: Mapped[str] = mapped_column(String(20), nullable=False)  # cron|once|interval|poll|on_message
    config: Mapped[dict] = mapped_column(JSONB, nullable=False, default=dict)
    reason: Mapped[str] = mapped_column(Text, nullable=False, default="")
    focus_ref: Mapped[str | None] = mapped_column(String(200))  # optional: related focus item identifier
    is_enabled: Mapped[bool] = mapped_column(Boolean, default=True)
    last_fired_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    fire_count: Mapped[int] = mapped_column(Integer, default=0)
    max_fires: Mapped[int | None] = mapped_column(Integer)  # None = unlimited
    cooldown_seconds: Mapped[int] = mapped_column(Integer, default=60)  # 1 min default
    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
    expires_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))

    __table_args__ = (
        UniqueConstraint("agent_id", "name", name="uq_agent_trigger_name"),
    )