"""Map Asset Types to Standards."""

from asset_transcoder.constants import asset_types


def map_image_asset_type(asset_type):
    """Convert asset image types to standard types.

    Args:
        asset_type (str): general image asset type

    Returns:
        string: asset type converted to standard type
    """
    if asset_type.upper() in asset_types.ASSET_JPG_FILE_TYPES:
        return asset_types.TYPE_FILE_JPG
    if asset_type.upper() in asset_types.ASSET_TIF_FILE_TYPES:
        return asset_types.TYPE_FILE_TIF
    return asset_type


def map_asset_to_file_types(asset_type):
    """Get subset of types that asset_type belongs to.

    Args:
        asset_type (str): asset type

    Return:
        sequence[str] : list of file types
    """
    if is_asset_audio(asset_type):
        return asset_types.AUDIO_FILE_TYPES
    if asset_type.upper() in asset_types.IMAGE_FILE_TYPES:
        return asset_types.IMAGE_FILE_TYPES
    return [asset_type]


def is_asset_audio(asset_type):
    """Determine if asset type is an audio type.

    Args:
        asset_type (str): asset type

    Return:
        boolean
    """
    return asset_type.upper() in asset_types.AUDIO_FILE_TYPES