from enum import StrEnum


class AudioReferenceType(StrEnum):
    """参考素材类型。"""

    audio = "audio"


class AudioReferenceFormat(StrEnum):
    """允许上传的参考音频格式。"""

    mp3 = "mp3"
    wav = "wav"


class AudioReferenceMimeType(StrEnum):
    """允许上传的参考音频 MIME。"""

    mp3 = "audio/mpeg"
    wav = "audio/wav"


AUDIO_ALLOWED_EXTENSIONS = {item.value for item in AudioReferenceFormat}
AUDIO_ALLOWED_MIME_TYPES = {
    AudioReferenceFormat.mp3.value: AudioReferenceMimeType.mp3.value,
    AudioReferenceFormat.wav.value: AudioReferenceMimeType.wav.value,
}
AUDIO_MAX_FILE_SIZE_MB = 15
AUDIO_MIN_DURATION_SECONDS = 2
AUDIO_MAX_DURATION_SECONDS = 15
AUDIO_MAX_TOTAL_DURATION_SECONDS = 15
AUDIO_MAX_COUNT_LIMIT = 3