package com.suno.android.common_data.media.usecase

import android.media.MediaExtractor
import android.media.MediaFormat
import javax.inject.Inject

class GetAudioTrackIndexUseCase @Inject constructor() {

    operator fun invoke(
        mediaExtractor: MediaExtractor,
    ): Int? {
        for (index in 0 until mediaExtractor.trackCount) {
            val format = mediaExtractor.getTrackFormat(index)
            val mime = format.getString(MediaFormat.KEY_MIME) ?: continue

            if (mime.startsWith("audio/")) {
                return index
            }
        }
        return null
    }
}