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

from app.database import Base


class BillBrief(Base):
    __tablename__ = "bill_briefs"

    id = Column(Integer, primary_key=True, autoincrement=True)
    bill_id = Column(String, ForeignKey("bills.bill_id", ondelete="CASCADE"), nullable=False)
    document_id = Column(Integer, ForeignKey("bill_documents.id", ondelete="SET NULL"), nullable=True)
    brief_type = Column(String(20), nullable=False, server_default="full")  # full | amendment
    summary = Column(Text)
    key_points = Column(JSONB)      # list[{text, citation, quote}]
    risks = Column(JSONB)           # list[{text, citation, quote}]
    deadlines = Column(JSONB)       # list[{date: str, description: str}]
    topic_tags = Column(JSONB)      # list[str]
    llm_provider = Column(String(50))
    llm_model = Column(String(100))
    govinfo_url = Column(String, nullable=True)
    share_token = Column(postgresql.UUID(as_uuid=False), nullable=True, server_default=func.gen_random_uuid())
    created_at = Column(DateTime(timezone=True), server_default=func.now())

    bill = relationship("Bill", back_populates="briefs")
    document = relationship("BillDocument", back_populates="briefs")

    __table_args__ = (
        Index("ix_bill_briefs_bill_id", "bill_id"),
        Index("ix_bill_briefs_topic_tags", "topic_tags", postgresql_using="gin"),
    )