@file: OptIn(UnstableApi::class) package com.lovegaoshi.kotlinaudio.player

import android.content.Context
import androidx.annotation.OptIn
import androidx.media3.common.C
import androidx.media3.common.IllegalSeekPositionException
import androidx.media3.common.MediaItem
import androidx.media3.common.Player
import androidx.media3.common.util.UnstableApi
import com.lovegaoshi.kotlinaudio.models.*
import java.util.*
import kotlin.math.max
import kotlin.math.min
import timber.log.Timber

class QueuedAudioPlayer(
    private val context: Context,
    options: PlayerOptions = PlayerOptions()
) : AudioPlayer(context, options) {

    var parseEmbeddedArtwork: Boolean = false

    private val queue = LinkedList<MediaItem>()

    private fun parseAudioItem(audioItem: AudioItem, skipSavingEmbeddedArtwork: Boolean = false): MediaItem {
        return audioItem2MediaItem(audioItem, if (parseEmbeddedArtwork && !skipSavingEmbeddedArtwork) context else null, skipSavingEmbeddedArtwork)
    }

    var repeatMode: RepeatMode
        get() {
            return when (exoPlayer.repeatMode) {
                Player.REPEAT_MODE_ALL -> RepeatMode.ALL
                Player.REPEAT_MODE_ONE -> RepeatMode.ONE
                else -> RepeatMode.OFF
            }
        }
        set(value) {
            when (value) {
                RepeatMode.ALL -> players().forEach { p -> p.repeatMode = Player.REPEAT_MODE_ALL }
                RepeatMode.ONE -> players().forEach { p -> p.repeatMode = Player.REPEAT_MODE_ONE }
                RepeatMode.OFF -> players().forEach { p -> p.repeatMode = Player.REPEAT_MODE_OFF }
            }
        }

    val currentIndex
        get() = exoPlayer.currentMediaItemIndex

    var shuffleMode
        get() = exoPlayer.shuffleModeEnabled
        set(v) {
            players().forEach{ p -> p.shuffleModeEnabled = v }
        }

    override val currentItem: AudioItem?
        get() = mediaItem2AudioItem(queue.getOrNull(currentIndex))

    val nextIndex: Int?
        get() {
            return if (exoPlayer.nextMediaItemIndex == C.INDEX_UNSET) null
            else exoPlayer.nextMediaItemIndex
        }

    val previousIndex: Int?
        get() {
            return if (exoPlayer.previousMediaItemIndex == C.INDEX_UNSET) null
            else exoPlayer.previousMediaItemIndex
        }

    val items: List<AudioItem>
        get() = queue.map { mediaItem2AudioItem(it)!! }

    val previousItems: List<AudioItem>
        get() {
            return if (queue.isEmpty()) emptyList()
            else queue
                .subList(0, exoPlayer.currentMediaItemIndex)
                .map { mediaItem2AudioItem(it)!! }
        }

    val nextItems: List<AudioItem>
        get() {
            return if (queue.isEmpty()) emptyList()
            else queue
                .subList(exoPlayer.currentMediaItemIndex, queue.lastIndex)
                .map { mediaItem2AudioItem(it)!! }
        }

    val nextItem: AudioItem?
        get() = items.getOrNull(currentIndex + 1)

    val previousItem: AudioItem?
        get() = items.getOrNull(currentIndex - 1)

    override fun load(item: AudioItem, playWhenReady: Boolean) {
        load(item)
        exoPlayer.playWhenReady = playWhenReady
    }

    override fun load(item: AudioItem) {
        if (queue.isEmpty()) {
            add(item)
        } else {
            players().forEach { p ->
                p.replaceMediaItem(currentIndex, parseAudioItem(item))
                p.seekTo(currentIndex, C.TIME_UNSET)
            }
            exoPlayer.prepare()
        }
    }

    /**
     * Add a single item to the queue. If the AudioPlayer has no item loaded, it will load the `item`.
     * @param item The [AudioItem] to add.
     */
    fun add(item: AudioItem, playWhenReady: Boolean) {
        exoPlayer.playWhenReady = playWhenReady
        add(item)
    }

    /**
     * Add a single item to the queue. If the AudioPlayer has no item loaded, it will load the `item`.
     * @param item The [AudioItem] to add.
     */
    fun add(item: AudioItem) {
        val mediaSource = parseAudioItem(item)
        queue.add(mediaSource)
        players().forEach { p -> p.addMediaItem(mediaSource) }
        exoPlayer.prepare()
    }

    /**
     * Add multiple items to the queue. If the AudioPlayer has no item loaded, it will load the first item in the list.
     * @param items The [AudioItem]s to add.
     * @param playWhenReady Whether playback starts automatically.
     */
    fun add(items: List<AudioItem>, playWhenReady: Boolean) {
        exoPlayer.playWhenReady = playWhenReady
        add(items)
    }

    /**
     * Add multiple items to the queue. If the AudioPlayer has no item loaded, it will load the first item in the list.
     * @param items The [AudioItem]s to add.
     */
    fun add(items: List<AudioItem>) {
        try {
            // disable embedded artwork extraction for multiple items to prevent blocking I/O operations
            timber.log.Timber.tag("RNTP").d("QueuedAudioPlayer.add: Parsing ${items.size} items with skipSavingEmbeddedArtwork=true")
            val mediaSources = items.map { parseAudioItem(it, skipSavingEmbeddedArtwork = true) }
            
            timber.log.Timber.tag("RNTP").d("QueuedAudioPlayer.add: Adding ${mediaSources.size} media sources to queue")
            queue.addAll(mediaSources)
            
            timber.log.Timber.tag("RNTP").d("QueuedAudioPlayer.add: Adding media items to all players")
            players().forEach { p -> p.addMediaItems(mediaSources) }
            
            timber.log.Timber.tag("RNTP").d("QueuedAudioPlayer.add: Calling exoPlayer.prepare()")
            exoPlayer.prepare()
            
            timber.log.Timber.tag("RNTP").d("QueuedAudioPlayer.add: Successfully completed")
        } catch (exception: Exception) {
            timber.log.Timber.tag("RNTP").e(exception, "QueuedAudioPlayer.add: Error adding items")
            throw exception
        }
    }


    /**
     * Add multiple items to the queue.
     * @param items The [AudioItem]s to add.
     * @param atIndex  Index to insert items at, if no items loaded this will not automatically start playback.
     */
    fun add(items: List<AudioItem>, atIndex: Int) {
        // disable embedded artwork extraction for multiple items to prevent blocking I/O operations
        val mediaSources = items.map { parseAudioItem(it, skipSavingEmbeddedArtwork = true) }
        queue.addAll(atIndex, mediaSources)
        players().forEach { p -> p.addMediaItems(atIndex, mediaSources) }
        exoPlayer.prepare()
    }

    /**
     * Remove an item from the queue.
     * @param index The index of the item to remove.
     */
    fun remove(index: Int) {
        queue.removeAt(index)
        players().forEach { p -> p.removeMediaItem(index) }
    }

    /**
     * Remove items from the queue.
     * @param indexes The indexes of the items to remove.
     */
    fun remove(indexes: List<Int>) {
        val sorted = indexes.toMutableList()
        // Sort the indexes in descending order so we can safely remove them one by one
        // without having the next index possibly newly pointing to another item than intended:
        sorted.sortDescending()
        sorted.forEach {
            remove(it)
        }
    }

    /**
     * Skip to the next item in the queue, which may depend on the current repeat mode.
     * Does nothing if there is no next item to skip to.
     */
    fun next() {
        exoPlayer.seekToNextMediaItem()
        exoPlayer.prepare()
    }

    /**
     * Skip to the previous item in the queue, which may depend on the current repeat mode.
     * Does nothing if there is no previous item to skip to.
     */
    fun previous() {
        exoPlayer.seekToPreviousMediaItem()
        exoPlayer.prepare()
    }

    /**
     * Move an item in the queue from one position to another.
     * @param fromIndex The index of the item ot move.
     * @param toIndex The index to move the item to. If the index is larger than the size of the queue, the item is moved to the end of the queue instead.
     */
    fun move(fromIndex: Int, toIndex: Int) {
        players().forEach { p -> p.moveMediaItem(fromIndex, toIndex) }
        val item = queue[fromIndex]
        queue.removeAt(fromIndex)
        queue.add(max(0, min(items.size, if (toIndex > fromIndex) toIndex else toIndex - 1)), item)
    }

    /**
     * Jump to an item in the queue.
     * @param index the index to jump to
     * @param playWhenReady Whether playback starts automatically.
     */
    fun jumpToItem(index: Int, playWhenReady: Boolean) {
        exoPlayer.playWhenReady = playWhenReady
        jumpToItem(index)
    }

    /**
     * Jump to an item in the queue.
     * @param index the index to jump to
     */
    fun jumpToItem(index: Int) {
        try {
            exoPlayer.seekTo(index, C.TIME_UNSET)
            exoPlayer.prepare()
        } catch (e: IllegalSeekPositionException) {
            throw Error("This item index $index does not exist. The size of the queue is ${queue.size} items.")
        }
    }

    /**
     * Replaces item at index in queue.
     */
    fun replaceItem(index: Int, item: AudioItem) {
        val mediaSource = parseAudioItem(item)
        queue[index] = mediaSource
        players().forEach { p -> p.replaceMediaItem(index, mediaSource) }
    }

    /**
     * Removes all the upcoming items, if any (the ones returned by [next]).
     */
    fun removeUpcomingItems() {
        if (queue.lastIndex == -1 || currentIndex == -1) return
        val lastIndex = queue.lastIndex + 1
        val fromIndex = currentIndex + 1

        players().forEach { p -> p.removeMediaItems(fromIndex, lastIndex) }
        queue.subList(fromIndex, lastIndex).clear()
    }

    /**
     * Removes all the previous items, if any (the ones returned by [previous]).
     */
    fun removePreviousItems() {
        players().forEach { p -> p.removeMediaItems(0, currentIndex) }
        queue.subList(0, currentIndex).clear()
    }

    override fun destroy() {
        queue.clear()
        super.destroy()
    }

    override fun clear() {
        queue.clear()
        super.clear()
    }
}