"""Realtime domain — SQLAlchemy models."""
from __future__ import annotations

import enum
from datetime import datetime
from typing import Optional
from uuid import UUID, uuid4

from sqlalchemy import Boolean, ForeignKey, Integer, String, UniqueConstraint
from sqlalchemy.dialects.postgresql import UUID as PGUUID
from sqlalchemy.orm import Mapped, mapped_column
from sqlalchemy.sql import func
from sqlalchemy.types import DateTime

from src.core.database import Base


class RealtimeAccessPolicy(str, enum.Enum):
    public = "public"
    authenticated = "authenticated"
    private = "private"


class ProjectRealtimeConfig(Base):
    """One row per project — created on first enable (upsert pattern)."""

    __tablename__ = "project_realtime_config"

    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True, default=uuid4)
    project_id: Mapped[UUID] = mapped_column(
        PGUUID(as_uuid=True),
        ForeignKey("projects.id", ondelete="CASCADE"),
        nullable=False,
        unique=True,
    )
    enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
    max_concurrent_connections: Mapped[int] = mapped_column(Integer, nullable=False, default=10)
    history_enabled: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
    history_buffer_size: Mapped[int] = mapped_column(Integer, nullable=False, default=50)
    updated_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
    )


class ProjectRealtimeChannel(Base):
    """Custom channel registered for a project (codegen or dashboard-created).

    Built-in channels (records:*, jobs, auth, storage) need no row here —
    pc-realtime handles them directly.
    """

    __tablename__ = "project_realtime_channels"

    id: Mapped[UUID] = mapped_column(PGUUID(as_uuid=True), primary_key=True, default=uuid4)
    project_id: Mapped[UUID] = mapped_column(
        PGUUID(as_uuid=True),
        ForeignKey("projects.id", ondelete="CASCADE"),
        nullable=False,
    )
    name: Mapped[str] = mapped_column(String(128), nullable=False)
    access_policy: Mapped[str] = mapped_column(String(20), nullable=False, default="authenticated")
    rls_filter: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
    created_at: Mapped[datetime] = mapped_column(
        DateTime(timezone=True), server_default=func.now()
    )

    __table_args__ = (
        UniqueConstraint("project_id", "name", name="uq_realtime_channel_project_name"),
    )