"""Collections/franchises as first-class entities (e.g. "Alien", "MCU")."""

from __future__ import annotations

import uuid

from sqlalchemy import ForeignKey, Integer, String, Text
from sqlalchemy.orm import Mapped, mapped_column

from app.models.base import Base, TimestampMixin, UUIDPrimaryKeyMixin


class Collection(Base, UUIDPrimaryKeyMixin, TimestampMixin):
    __tablename__ = "collections"

    name: Mapped[str] = mapped_column(String(300), nullable=False, index=True)
    overview: Mapped[str | None] = mapped_column(Text, nullable=True)
    slug: Mapped[str] = mapped_column(String(350), unique=True, nullable=False)


class CollectionMovie(Base):
    __tablename__ = "collection_movies"

    collection_id: Mapped[uuid.UUID] = mapped_column(
        ForeignKey("collections.id", ondelete="CASCADE"), primary_key=True
    )
    movie_id: Mapped[uuid.UUID] = mapped_column(
        ForeignKey("movies.id", ondelete="CASCADE"), primary_key=True
    )
    sort_order: Mapped[int | None] = mapped_column(Integer, nullable=True)


class CollectionSeries(Base):
    __tablename__ = "collection_series"

    collection_id: Mapped[uuid.UUID] = mapped_column(
        ForeignKey("collections.id", ondelete="CASCADE"), primary_key=True
    )
    series_id: Mapped[uuid.UUID] = mapped_column(
        ForeignKey("series.id", ondelete="CASCADE"), primary_key=True
    )
    sort_order: Mapped[int | None] = mapped_column(Integer, nullable=True)


class CollectionRelation(Base):
    """Relationships between collections (e.g. shared universe)."""

    __tablename__ = "collection_relations"

    collection_id: Mapped[uuid.UUID] = mapped_column(
        ForeignKey("collections.id", ondelete="CASCADE"), primary_key=True
    )
    related_collection_id: Mapped[uuid.UUID] = mapped_column(
        ForeignKey("collections.id", ondelete="CASCADE"), primary_key=True
    )
    relation_type: Mapped[str] = mapped_column(String(50), default="related", nullable=False)


__all__ = ["Collection", "CollectionMovie", "CollectionSeries", "CollectionRelation"]