from datetime import date, datetime
from decimal import Decimal

from sqlalchemy import Boolean, Date, DateTime, ForeignKey, Numeric, Text, UniqueConstraint
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.db.base import Base


class DocumentAdditionalField(Base):
    __tablename__ = "document_additional_fields"
    __table_args__ = (
        UniqueConstraint("document_id", name="uq_document_additional_fields_document_id"),
    )

    id: Mapped[int] = mapped_column(primary_key=True, index=True)
    document_id: Mapped[int] = mapped_column(ForeignKey("documents.id"), nullable=False, index=True)

    owner_primary: Mapped[str | None] = mapped_column(Text, nullable=True)
    owner_secondary: Mapped[str | None] = mapped_column(Text, nullable=True)
    paid_by_person: Mapped[str | None] = mapped_column(Text, nullable=True)
    occasion_note: Mapped[str | None] = mapped_column(Text, nullable=True)
    is_shared_expense: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)

    covered_people: Mapped[list | None] = mapped_column(JSONB, nullable=True)
    attendees: Mapped[list | None] = mapped_column(JSONB, nullable=True)
    reimbursement_expected_from: Mapped[list | None] = mapped_column(JSONB, nullable=True)

    reimbursement_paid_by: Mapped[str | None] = mapped_column(Text, nullable=True)
    reimbursement_paid_to: Mapped[str | None] = mapped_column(Text, nullable=True)
    reimbursement_paid_amount: Mapped[Decimal | None] = mapped_column(Numeric(12, 2), nullable=True)
    reimbursement_paid_date: Mapped[date | None] = mapped_column(Date, nullable=True)
    reimbursement_note: Mapped[str | None] = mapped_column(Text, nullable=True)

    created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
    updated_at: Mapped[datetime] = mapped_column(
        DateTime,
        default=datetime.utcnow,
        onupdate=datetime.utcnow,
        nullable=False,
    )

    document: Mapped["Document"] = relationship("Document", back_populates="additional_fields")