from sqlalchemy import Column, DateTime, ForeignKey, Index, Integer, String
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func

from app.database import Base


class NotificationEvent(Base):
    __tablename__ = "notification_events"

    id = Column(Integer, primary_key=True, autoincrement=True)
    user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
    bill_id = Column(String, ForeignKey("bills.bill_id", ondelete="CASCADE"), nullable=False)
    # new_document | new_amendment | bill_updated
    event_type = Column(String(50), nullable=False)
    # {bill_title, bill_label, brief_summary, bill_url}
    payload = Column(JSONB)
    created_at = Column(DateTime(timezone=True), server_default=func.now())
    dispatched_at = Column(DateTime(timezone=True), nullable=True)

    user = relationship("User", back_populates="notification_events")

    __table_args__ = (
        Index("ix_notification_events_user_id", "user_id"),
        Index("ix_notification_events_dispatched_at", "dispatched_at"),
    )