package com.lovegaoshi.kotlinaudio.models

import android.content.Context
import android.graphics.Bitmap
import android.os.Bundle
import androidx.core.net.toUri
import androidx.media3.common.MediaItem
import androidx.media3.common.MediaMetadata
import com.lovegaoshi.kotlinaudio.utils.getEmbeddedBitmapArray
import com.lovegaoshi.kotlinaudio.utils.saveMediaCoverToPng
import java.util.UUID


data class DefaultAudioItem(
    override var audioUrl: String,

    /**
     * Set to [MediaType.DEFAULT] by default.
     */
    override val type: MediaType = MediaType.DEFAULT,

    override var artist: String? = null,
    override var title: String? = null,
    override var albumTitle: String? = null,
    override var artwork: String? = null,
    override val duration: Long? = null,
    override val options: AudioItemOptions? = null,
    override val mediaId: String? = null,
) : AudioItem

class AudioItemHolder(
    var audioItem: AudioItem
) {
    var artworkBitmap: Bitmap? = null
}

interface AudioItem {
    var audioUrl: String
    val type: MediaType
    var artist: String?
    var title: String?
    var albumTitle: String?
    val artwork: String?
    val duration: Long?
    val options: AudioItemOptions?
    val mediaId: String?
}

data class AudioItemOptions(
    val headers: HashMap<String, String>? = null,
    val userAgent: String? = null,
    val resourceId: Int? = null
)

enum class MediaType(val value: String) {
    /**
     * The default media type. Should be used for streams over HTTP or files
     */
    DEFAULT("default"),

    /**
     * The DASH media type for adaptive streams. Should be used with DASH manifests.
     */
    DASH("dash"),

    /**
     * The HLS media type for adaptive streams. Should be used with HLS playlists.
     */
    HLS("hls"),

    /**
     * The SmoothStreaming media type for adaptive streams. Should be used with SmoothStreaming manifests.
     */
    SMOOTH_STREAMING("smoothstreaming");
}



fun audioItem2MediaItem(audioItem: AudioItem, context: Context? = null): MediaItem {
    // Check if this is a placeholder track (notPlayable flag from TrackAudioItem)
    val isNotPlayable = (audioItem as? com.doublesymmetry.trackplayer.model.TrackAudioItem)?.notPlayable ?: false
    val hasValidUrl = audioItem.audioUrl.isNotBlank()

    return MediaItem.Builder()
        .setMediaId(audioItem.mediaId ?: UUID.randomUUID().toString())
        // Always set URI (even empty string) so ExoPlayer adds item to timeline
        .setUri(audioItem.audioUrl)
        .setMediaMetadata(
            MediaMetadata.Builder()
            .setTitle(audioItem.title)
            .setArtist(audioItem.artist)
            // Mark placeholder tracks as not playable so Android Auto shows them correctly in queue
            .setIsPlayable(true)
            .setIsBrowsable(false)
            .setArtworkUri((
                if (context != null && hasValidUrl && audioItem.audioUrl.startsWith("file://")) {
                    saveMediaCoverToPng(
                        audioItem.audioUrl,
                        context.contentResolver,
                        audioItem.mediaId ?: audioItem.audioUrl
                    )
                        ?: audioItem.artwork
                }
                else audioItem.artwork)?.toUri())
            .setArtworkData(if (hasValidUrl && audioItem.audioUrl.startsWith("file://")) getEmbeddedBitmapArray(
                audioItem.audioUrl.substring(7)) else null, MediaMetadata.PICTURE_TYPE_MEDIA)
            .setExtras(Bundle().apply {
                audioItem.options?.headers?.let {
                    putSerializable("headers", audioItem.options!!.headers)
                }
                audioItem.options?.userAgent?.let {
                    putString("user-agent", it)
                }
                audioItem.options?.resourceId?.let {
                    putInt("resource-id", it)
                }
                putString("type", audioItem.type.toString())
                putString("uri", audioItem.audioUrl)
                putBoolean("notPlayable", isNotPlayable)
            }).build())
        .setTag(audioItem)
        .build()
}

fun mediaItem2AudioItem(item: MediaItem?): AudioItem? {
    return item?.localConfiguration?.tag as AudioItem?
}