from typing import Any

from assets.constants import asset_types, asset_upload_types
from assets.logic import asset_upload as asset_upload_logic
from assets.models import (
    asset_final,
    asset_upload as asset_upload_model,
    asset_upload_type,
)


def get_stereo_for_asset(asset_upload_filename: str) -> dict[str, Any]:
    asset_upload = asset_upload_logic.get_asset_upload_by_filename(
        asset_upload_filename
    )

    stereo_type_id = asset_upload_type.resolve_asset_upload_type_id(
        asset_upload_types.STEREO
    )
    latest_stereo_uploads = asset_upload_model.get_latest_asset_upload(
        product_id=asset_upload["product_id"],
        track_id=asset_upload["track_unique_id"],
        asset_upload_type_id=stereo_type_id,
        is_correction_mode=True,
        limit=1,
    )
    [stereo_upload] = latest_stereo_uploads

    stereo_flac = asset_final.get_asset_final_by_asset_upload_id_and_type(
        stereo_upload["id"],
        asset_types.TYPE_FILE_FLAC,
    )

    return {"bucket": stereo_flac["bucket"], "key": stereo_flac["filename"]}