from __future__ import annotations

from datetime import datetime

from sqlalchemy import BigInteger, Boolean, DateTime, ForeignKey, Integer, String, Text, 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__ = {"schema": "operationbi"}

    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")


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"), 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")