from typing import TYPE_CHECKING

from sqlmodel import Field, Relationship, UniqueConstraint

from app.models.base import BaseModel

if TYPE_CHECKING:
    from app.models.playlist import Playlist
    from app.models.sound import Sound
    from app.models.user import User


class Favorite(BaseModel, table=True):
    """Database model for user favorites (sounds and playlists)."""

    user_id: int = Field(foreign_key="user.id", nullable=False)
    sound_id: int | None = Field(foreign_key="sound.id", default=None)
    playlist_id: int | None = Field(foreign_key="playlist.id", default=None)

    # constraints
    __table_args__ = (
        UniqueConstraint("user_id", "sound_id", name="uq_favorite_user_sound"),
        UniqueConstraint("user_id", "playlist_id", name="uq_favorite_user_playlist"),
    )

    # relationships
    user: "User" = Relationship(back_populates="favorites")
    sound: "Sound" = Relationship(back_populates="favorites")
    playlist: "Playlist" = Relationship(back_populates="favorites")