from dataclasses import dataclass

FLAC = "FLAC"
MP3_128 = "MP3_128"
MP3_256 = "MP3_256"
MP3_320 = "MP3_320"
MP4_RA1 = "MP4_RA1"
MP4_RA2 = "MP4_RA2"
MP4_RA3 = "MP4_RA3"

FALLBACK_QUALITIES = [MP3_320, MP3_128, FLAC]
FORMAT_LIST = [MP3_128, MP3_256, MP3_320, FLAC]


@dataclass
class TrackFormat:
    code: int
    ext: str


TRACK_FORMAT_MAP = {
    FLAC: TrackFormat(code=9, ext=".flac"),
    MP3_128: TrackFormat(code=1, ext=".mp3"),
    MP3_256: TrackFormat(code=5, ext=".mp3"),
    MP3_320: TrackFormat(code=3, ext=".mp3"),
    MP4_RA1: TrackFormat(code=13, ext=".mp4"),
    MP4_RA2: TrackFormat(code=14, ext=".mp4"),
    MP4_RA3: TrackFormat(code=15, ext=".mp3"),
}