import uuid
from datetime import datetime
from typing import Any

from sqlalchemy import Boolean, DateTime, String, text
from sqlalchemy.dialects.postgresql import JSONB, UUID
from sqlalchemy.orm import Mapped, mapped_column

from app.db import Base


class Company(Base):
    __tablename__ = "companies"

    id: Mapped[uuid.UUID] = mapped_column(
        UUID(as_uuid=True), primary_key=True, server_default=text("gen_random_uuid()")
    )
    nif: Mapped[str] = mapped_column(String(9), unique=True, nullable=False)
    legal_name: Mapped[str] = mapped_column(String, nullable=False)
    trade_name: Mapped[str | None] = mapped_column(String)
    cae: Mapped[str | None] = mapped_column(String)
    address: Mapped[str | None] = mapped_column(String)
    status: Mapped[str | None] = mapped_column(String)
    ptdata_payload: Mapped[dict[str, Any] | None] = mapped_column(JSONB)
    ptdata_fetched_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    monitored: Mapped[bool] = mapped_column(Boolean, nullable=False, server_default=text("TRUE"))
    monitoring_type: Mapped[str] = mapped_column(
        String, nullable=False, server_default=text("'internal'")
    )
    data_coverage_start: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    data_coverage_end: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
    risk_score: Mapped[int | None] = mapped_column()
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=text("now()")
    )
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=text("now()")
    )