from datetime import datetime, date
from typing import Optional
from sqlalchemy import String, Text, Boolean, DateTime, Date, Integer, ForeignKey, func
from sqlalchemy.orm import Mapped, mapped_column
from app.database import Base


class Surat(Base):
    __tablename__ = "surat"
    id: Mapped[int] = mapped_column(primary_key=True)
    no_surat: Mapped[str] = mapped_column(String(255), nullable=False, unique=True)
    subject: Mapped[str] = mapped_column(String(500), nullable=False)
    perihal: Mapped[str] = mapped_column(Text, nullable=False)
    company_id: Mapped[int] = mapped_column(ForeignKey("companies.id", ondelete="RESTRICT"), nullable=False)
    vendor_id: Mapped[Optional[int]] = mapped_column(ForeignKey("vendors.id", ondelete="SET NULL"))
    category_id: Mapped[int] = mapped_column(ForeignKey("categories.id", ondelete="RESTRICT"), nullable=False)
    tanggal: Mapped[Optional[date]] = mapped_column(Date)
    body: Mapped[Optional[str]] = mapped_column(Text)
    tembusan: Mapped[Optional[str]] = mapped_column(Text)
    needs_director_sig: Mapped[bool] = mapped_column(Boolean, default=False)
    needs_manager_sig: Mapped[bool] = mapped_column(Boolean, default=False)
    wet_signature: Mapped[bool] = mapped_column(Boolean, default=False)
    is_approved: Mapped[bool] = mapped_column(Boolean, default=False)
    approved_by: Mapped[Optional[int]] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"))
    approved_at: Mapped[Optional[datetime]] = mapped_column(DateTime)
    created_by: Mapped[Optional[int]] = mapped_column(ForeignKey("users.id", ondelete="SET NULL"))
    rejected_reason: Mapped[Optional[str]] = mapped_column(String(500), nullable=True)
    task_id: Mapped[Optional[int]] = mapped_column(ForeignKey("tasks.id", ondelete="SET NULL"))
    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
    updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now())