from apollo_utils.core.utils.ext_enum import ExtEnum


class TrackPlaylistInclude(ExtEnum):
    """Additional fields to include."""

    PLAYLIST_STREAMS = "playlist_streams"
    FOLLOWERS = "followers"
    TRACK_COUNT = "track_count"
    USERNAME = "username"
    OWNER = "owner"
    CATEGORIES = "categories"


class TrackPlaylistHistoryInclude(ExtEnum):
    """Additional fields to include."""

    OWNER = "owner"
    CATEGORIES = "categories"
    FOLLOWERS = "followers"


class NMFTrackInclude(ExtEnum):
    """Additional fields to include."""

    TOP_PLAYLIST_IMAGE_URL = "top_playlist_image_url"
    REGIONS = "regions"
    INACTIVE_MARKETS = "inactive_markets"


class HHTrackInclude(ExtEnum):
    """Additional fields to include."""

    REGIONS = "regions"
    INACTIVE_MARKETS = "inactive_markets"


class HHLatestDateInclude(ExtEnum):
    """Additional fields to include."""

    TIME = "time"