from __future__ import annotations

from datetime import datetime
from enum import Enum

from sqlalchemy import (
    String,
    Integer,
    DateTime,
    ForeignKey,
    UniqueConstraint,
    Numeric,
    Enum as SQLEnum,
    Index,
)
from sqlalchemy.orm import Mapped, mapped_column, relationship

from adapters.db.session import Base


class CheckType(str, Enum):
    RECEIVED = "RECEIVED"
    TRANSFERRED = "TRANSFERRED"


class Check(Base):
    __tablename__ = "checks"
    __table_args__ = (
        # پیشنهاد: یکتا بودن شماره چک در سطح کسب‌وکار
        UniqueConstraint('business_id', 'check_number', name='uq_checks_business_check_number'),
        # پیشنهاد: یکتا بودن شناسه صیاد در سطح کسب‌وکار (چند NULL مجاز است)
        UniqueConstraint('business_id', 'sayad_code', name='uq_checks_business_sayad_code'),
        Index('ix_checks_business_type', 'business_id', 'type'),
        Index('ix_checks_business_person', 'business_id', 'person_id'),
        Index('ix_checks_business_issue_date', 'business_id', 'issue_date'),
        Index('ix_checks_business_due_date', 'business_id', 'due_date'),
    )

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    business_id: Mapped[int] = mapped_column(Integer, ForeignKey("businesses.id", ondelete="CASCADE"), nullable=False, index=True)

    type: Mapped[CheckType] = mapped_column(SQLEnum(CheckType, name="check_type"), nullable=False, index=True)
    person_id: Mapped[int | None] = mapped_column(Integer, ForeignKey("persons.id", ondelete="SET NULL"), nullable=True, index=True)

    issue_date: Mapped[datetime] = mapped_column(DateTime, nullable=False, index=True)
    due_date: Mapped[datetime] = mapped_column(DateTime, nullable=False, index=True)

    check_number: Mapped[str] = mapped_column(String(50), nullable=False, index=True)
    sayad_code: Mapped[str | None] = mapped_column(String(16), nullable=True, index=True)

    bank_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
    branch_name: Mapped[str | None] = mapped_column(String(255), nullable=True)

    amount: Mapped[float] = mapped_column(Numeric(18, 2), nullable=False)
    currency_id: Mapped[int] = mapped_column(Integer, ForeignKey("currencies.id", ondelete="RESTRICT"), nullable=False, index=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)

    # روابط
    business = relationship("Business", backref="checks")
    person = relationship("Person", lazy="joined")
    currency = relationship("Currency")