import uuid
from datetime import date, datetime
from typing import Any

from sqlalchemy import CheckConstraint, Date, DateTime, ForeignKey, String, text
from sqlalchemy.dialects.postgresql import JSONB, UUID
from sqlalchemy.orm import Mapped, mapped_column

from app.db import Base


class Process(Base):
    __tablename__ = "processes"

    id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), primary_key=True, server_default=text("gen_random_uuid()")
    )
    company_id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), ForeignKey("companies.id", ondelete="CASCADE"), nullable=False
    )
    source: Mapped[str] = mapped_column(String, nullable=False)
    process_number: Mapped[str] = mapped_column(String, nullable=False)
    tribunal: Mapped[str] = mapped_column(String, nullable=False)
    juizo: Mapped[str | None] = mapped_column(String)
    species: Mapped[str | None] = mapped_column(String)
    role_in_process: Mapped[str | None] = mapped_column(String)
    date_filed: Mapped[date] = mapped_column(Date, nullable=False)
    dedup_hash: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
    raw: Mapped[dict[str, Any]] = mapped_column(JSONB, nullable=False)
    raw_html: Mapped[str | None] = mapped_column(String)
    first_seen_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=text("now()")
    )
    last_seen_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=text("now()")
    )

    __table_args__ = (
        CheckConstraint("source IN ('distribuicao','cire')", name="processes_source_check"),
    )