from __future__ import annotations

from datetime import datetime

from sqlalchemy import BigInteger, Boolean, DateTime, ForeignKey, Integer, String, Text, UniqueConstraint, func
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Mapped, mapped_column, relationship

from app.db.base import Base


class RawOpdCheckpoint(Base):
    __tablename__ = "raw_opd_checkpoint"
    __table_args__ = (
        UniqueConstraint("hn", "vn", "location", name="uq_raw_opd_checkpoint_hn_vn_location"),
        {"schema": "rawdata"},
    )

    id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
    hn: Mapped[int] = mapped_column(BigInteger, nullable=False)
    vn: Mapped[int] = mapped_column(BigInteger, nullable=False)
    location: Mapped[str] = mapped_column(Text, nullable=False)
    type: Mapped[str] = mapped_column(String(64), nullable=False)
    timestamp_in: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
    timestamp_out: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
    waiting_time: Mapped[int | None] = mapped_column(Integer, nullable=True)
    bu: Mapped[str | None] = mapped_column(String(128), nullable=True)


class ApiClient(Base):
    __tablename__ = "api_client"
    __table_args__ = {"schema": "fastapi"}

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    name: Mapped[str] = mapped_column(String(128), unique=True, nullable=False)
    is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)

    api_keys: Mapped[list[ApiKey]] = relationship(
        back_populates="client",
        cascade="all, delete-orphan",
        passive_deletes=True,
    )

    def __str__(self) -> str:
        client_id = getattr(self, "id", None)
        if client_id is None:
            return self.name
        return f"{self.name} ({client_id})"

    def __repr__(self) -> str:
        return str(self)


class ApiKey(Base):
    __tablename__ = "api_key"
    __table_args__ = {"schema": "fastapi"}

    id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
    client_id: Mapped[int] = mapped_column(
        ForeignKey("fastapi.api_client.id", ondelete="CASCADE"), nullable=False
    )
    name: Mapped[str | None] = mapped_column(String(128), nullable=True)

    key_prefix: Mapped[str] = mapped_column(String(12), nullable=False)
    key_hash: Mapped[str] = mapped_column(Text, nullable=False)

    permissions: Mapped[list[str]] = mapped_column(JSONB, nullable=False, default=list)
    is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)

    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), nullable=False, server_default=func.now()
    )

    client: Mapped[ApiClient] = relationship(back_populates="api_keys")