AquaStream/app/src/main/java/com/lagradost/cloudstream3/ui/download/button/BaseFetchButton.kt

package com.lagradost.cloudstream3.ui.download.button

import android.content.Context
import android.text.format.Formatter
import android.util.AttributeSet
import android.widget.FrameLayout
import android.widget.TextView
import androidx.annotation.LayoutRes
import androidx.core.view.isVisible
import androidx.core.widget.ContentLoadingProgressBar
import com.lagradost.cloudstream3.R
import com.lagradost.cloudstream3.utils.VideoDownloadManager

typealias DownloadStatusTell = VideoDownloadManager.DownloadType

data class DownloadMetadata(
    var id: Int,
    var downloadedLength: Long,
    var totalLength: Long,
    var status: DownloadStatusTell? = null
) {
    val progressPercentage: Long
        get() = if (downloadedLength < 1024) 0 else maxOf(
            0,
            minOf(100, (downloadedLength * 100L) / totalLength)
        )
}

abstract class BaseFetchButton(context: Context, attributeSet: AttributeSet) :
    FrameLayout(context, attributeSet) {

    var persistentId: Int? = null // used to save sessions

    lateinit var progressBar: ContentLoadingProgressBar
    var progressText: TextView? = null

    /*val gid: String? get() = sessionIdToGid[persistentId]

    // used for resuming data
    var _lastRequestOverride: UriRequest? = null
    var lastRequest: UriRequest?
        get() = _lastRequestOverride ?: sessionIdToLastRequest[persistentId]
        set(value) {
            _lastRequestOverride = value
        }

    var files: List<AbstractClient.JsonFile> = emptyList()*/
    protected var isZeroBytes: Boolean = true

    fun inflate(@LayoutRes layout: Int) {
        inflate(context, layout, this)
    }

    init {
        resetViewData()
    }

    open fun resetViewData() {
        // lastRequest = null
        isZeroBytes = true
        persistentId = null
    }

    var currentMetaData: DownloadMetadata =
        DownloadMetadata(0, 0, 0, null)

    fun setPersistentId(id: Int) {
        persistentId = id
        currentMetaData.id = id

        VideoDownloadManager.getDownloadFileInfoAndUpdateSettings(context, id)?.let { savedData ->
            val downloadedBytes = savedData.fileLength
            val totalBytes = savedData.totalBytes

            /*lastRequest = savedData.uriRequest
            files = savedData.files

            var totalBytes: Long = 0
            var downloadedBytes: Long = 0
            for (file in savedData.files) {
                downloadedBytes += file.completedLength
                totalBytes += file.length
            }*/
            setProgress(downloadedBytes, totalBytes)
            // some extra padding for just in case
            val status = VideoDownloadManager.downloadStatus[id]
                ?: if (downloadedBytes > 1024L && downloadedBytes + 1024L >= totalBytes) DownloadStatusTell.IsDone else DownloadStatusTell.IsPaused
            currentMetaData.apply {
                this.id = id
                this.downloadedLength = downloadedBytes
                this.totalLength = totalBytes
                this.status = status
            }
            setStatus(status)
        } ?: run {
            resetView()
        }
    }

    abstract fun setStatus(status: VideoDownloadManager.DownloadType?)

    open fun setProgress(downloadedBytes: Long, totalBytes: Long) {
        isZeroBytes = downloadedBytes == 0L
        val steps = 10000L
        progressBar.max = steps.toInt()
        // div by zero error and 1 byte off is ok impo
        val progress = (downloadedBytes * steps / (totalBytes + 1L)).toInt()

        val animation = ProgressBarAnimation(
            progressBar,
            progressBar.progress.toFloat(),
            progress.toFloat()
        ).apply {
            fillAfter = true
            duration =
                if (progress > progressBar.progress) // we don't want to animate backward changes in progress
                    100
                else
                    0L
        }

        if (isZeroBytes) {
            progressText?.isVisible = false
        } else {
            progressText?.apply {
                val currentMbString = Formatter.formatShortFileSize(context, downloadedBytes)
                val totalMbString = Formatter.formatShortFileSize(context, totalBytes)
                text =
                        //if (isTextPercentage) "%d%%".format(setCurrentBytes * 100L / setTotalBytes) else
                    context?.getString(R.string.download_size_format)
                        ?.format(currentMbString, totalMbString)
            }
        }

        progressBar.startAnimation(animation)
    }

    fun downloadStatusEvent(data: Pair<Int, VideoDownloadManager.DownloadType>) {
        val (id, status) = data
        if (id == persistentId) {
            currentMetaData.status = status
            setStatus(status)
        }
    }

    /*fun downloadDeleteEvent(data: Int) {

    }*/

    /*fun downloadEvent(data: Pair<Int, VideoDownloadManager.DownloadActionType>) {
        val (id, action) = data

    }*/

    fun downloadProgressEvent(data: Triple<Int, Long, Long>) {
        val (id, bytesDownloaded, bytesTotal) = data
        if (id == persistentId) {
            currentMetaData.downloadedLength = bytesDownloaded
            currentMetaData.totalLength = bytesTotal

            setProgress(bytesDownloaded, bytesTotal)
        }
    }

    override fun onAttachedToWindow() {
        VideoDownloadManager.downloadStatusEvent += ::downloadStatusEvent
        //VideoDownloadManager.downloadDeleteEvent += ::downloadDeleteEvent
        //VideoDownloadManager.downloadEvent += ::downloadEvent
        VideoDownloadManager.downloadProgressEvent += ::downloadProgressEvent

        val pid = persistentId
        if (pid != null) {
            // refresh in case of onDetachedFromWindow -> onAttachedToWindow while still being ???????
            setPersistentId(pid)
        }

        super.onAttachedToWindow()
    }

    override fun onDetachedFromWindow() {
        VideoDownloadManager.downloadStatusEvent -= ::downloadStatusEvent
        //VideoDownloadManager.downloadDeleteEvent -= ::downloadDeleteEvent
        //VideoDownloadManager.downloadEvent -= ::downloadEvent
        VideoDownloadManager.downloadProgressEvent -= ::downloadProgressEvent

        super.onDetachedFromWindow()
    }

    /**
     * No checks required. Arg will always include a download with current id
     * */
    abstract fun updateViewOnDownload(metadata: DownloadMetadata)

    /**
     * Get a clean slate again, might be useful in recyclerview?
     * */
    abstract fun resetView()

}