LibreTube/app/src/main/java/com/github/libretube/util/AutoPlayHelper.kt

package com.github.libretube.util

import com.github.libretube.Globals
import com.github.libretube.api.RetrofitInstance
import com.github.libretube.obj.StreamItem
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.withContext

class AutoPlayHelper(
    private val playlistId: String?
) {

    private val playlistStreamIds = mutableListOf<String>()
    private var playlistNextPage: String? = null

    suspend fun getNextVideoId(
        currentVideoId: String,
        relatedStreams: List<StreamItem>
    ): String? {
        return if (Globals.playingQueue.last() != currentVideoId) {
            val currentVideoIndex = Globals.playingQueue.indexOf(currentVideoId)
            Globals.playingQueue[currentVideoIndex + 1]
        } else if (playlistId == null) getNextTrendingVideoId(
            currentVideoId,
            relatedStreams
        ) else getNextPlaylistVideoId(
            currentVideoId
        )
    }

    private fun getNextTrendingVideoId(videoId: String, relatedStreams: List<StreamItem>): String? {
        // don't play a video if it got played before already
        var index = 0
        var nextStreamId: String? = null
        while (nextStreamId == null ||
            (
                Globals.playingQueue.contains(nextStreamId) &&
                    Globals.playingQueue.indexOf(videoId) > Globals.playingQueue.indexOf(
                        nextStreamId
                    )
                )
        ) {
            nextStreamId = relatedStreams[index].url.toID()
            if (index + 1 < relatedStreams.size) index += 1
            else break
        }
        return nextStreamId
    }

    private suspend fun getNextPlaylistVideoId(currentVideoId: String): String? {
        // if the playlists contain the video, then save the next video as next stream
        if (playlistStreamIds.contains(currentVideoId)) {
            val index = playlistStreamIds.indexOf(currentVideoId)
            // check whether there's a next video
            return if (index + 1 < playlistStreamIds.size) playlistStreamIds[index + 1]
            else if (playlistNextPage == null) null
            else getNextPlaylistVideoId(currentVideoId)
        } else if (playlistStreamIds.isEmpty() || playlistNextPage != null) {
            // fetch the next page of the playlist
            return withContext(Dispatchers.IO) {
                // fetch the playlists or its nextPage's videos
                val playlist =
                    if (playlistNextPage == null) RetrofitInstance.authApi.getPlaylist(playlistId!!)
                    else RetrofitInstance.authApi.getPlaylistNextPage(
                        playlistId!!,
                        playlistNextPage!!
                    )
                // save the playlist urls to the list
                playlistStreamIds += playlist.relatedStreams!!.map { it.url.toID() }
                // save playlistNextPage for usage if video is not contained
                playlistNextPage = playlist.nextpage
                return@withContext getNextPlaylistVideoId(currentVideoId)
            }
        }
        // return null when no nextPage is found
        return null
    }

    fun getNextPlayingQueueVideoId(
        currentVideoId: String
    ): String? {
        return if (Globals.playingQueue.last() != currentVideoId) {
            val currentVideoIndex = Globals.playingQueue.indexOf(currentVideoId)
            Globals.playingQueue[currentVideoIndex + 1]
        } else null
    }
}