grayjay/app/src/main/java/com/futo/platformplayer/UpdateDownloadService.kt

package com.futo.platformplayer

import android.app.Service
import android.content.Intent
import android.os.IBinder
import android.os.SystemClock
import androidx.core.app.NotificationManagerCompat
import com.futo.platformplayer.logging.Logger
import com.futo.platformplayer.models.ImageVariable
import com.futo.platformplayer.states.AnnouncementType
import com.futo.platformplayer.states.SessionAnnouncement
import com.futo.platformplayer.states.StateAnnouncement
import com.futo.platformplayer.states.StateUpdate
import kotlinx.coroutines.*
import java.io.File
import java.io.FileOutputStream
import java.net.HttpURLConnection
import java.net.URL
import java.time.OffsetDateTime

class UpdateDownloadService : Service() {

    companion object {
        private const val TAG = "UpdateDownloadService"
        const val EXTRA_VERSION = "version"
        const val EXTRA_CANCEL = "cancel"
        private const val MAX_RETRIES = 5
        private const val INITIAL_BACKOFF_MS = 5_000L
        private const val BUFFER_SIZE = 8 * 1024
        private const val MIN_PROGRESS_UPDATE_INTERVAL_MS = 500L
    }

    private val job = SupervisorJob()
    private val scope = CoroutineScope(Dispatchers.IO + job)

    @Volatile
    private var isDownloading: Boolean = false

    @Volatile
    private var cancelRequested: Boolean = false

    private var lastProgressUpdateElapsedMs: Long = 0L

    override fun onBind(intent: Intent?): IBinder? = null

    override fun onStartCommand(intent: Intent?, flags: Int, startId: Int): Int {
        if (intent == null) {
            stopSelf()
            return START_NOT_STICKY
        }

        if (intent.getBooleanExtra(EXTRA_CANCEL, false)) {
            cancelRequested = true
            Logger.i(TAG, "Download cancel requested")
            StateUpdate.Companion.instance.clearUi()
            stopForeground(Service.STOP_FOREGROUND_REMOVE)
            stopSelf()
            return START_NOT_STICKY
        }

        val version = intent.getIntExtra(EXTRA_VERSION, 0)
        if (version == 0) {
            stopSelf()
            return START_NOT_STICKY
        }

        if (isDownloading) {
            Logger.i(TAG, "Download already in progress, ignoring new start")
            return START_STICKY
        }

        isDownloading = true
        cancelRequested = false

        StateUpdate.Companion.instance.setUiDownloading(version, 0, indeterminate = true)

        NotificationManagerCompat.from(this).cancel(UpdateNotificationManager.NOTIF_ID_READY)

        val notification = UpdateNotificationManager.buildDownloadProgressNotification(this, version, 0, true)
        startForeground(UpdateNotificationManager.NOTIF_ID_DOWNLOADING, notification)

        scope.launch {
            downloadApk(version)
        }

        return START_STICKY
    }

    override fun onDestroy() {
        super.onDestroy()
        job.cancel()
    }

    private fun throttledUpdateDownloadProgress(version: Int, progress: Int, indeterminate: Boolean, onProgress: ((Int) -> Unit)? = null) {
        val now = SystemClock.elapsedRealtime()
        val force = progress == 100 && !indeterminate

        if (force || now - lastProgressUpdateElapsedMs >= MIN_PROGRESS_UPDATE_INTERVAL_MS) {
            lastProgressUpdateElapsedMs = now
            UpdateNotificationManager.updateDownloadProgress(this, version, progress, indeterminate);
            StateUpdate.Companion.instance.setUiDownloading(version, progress, indeterminate)

            if(onProgress != null)
                onProgress.invoke(progress);
        }
    }

    private suspend fun downloadApk(version: Int) {
        val apkFile = StateUpdate.getApkFile(this, version)
        val partialFile = StateUpdate.getPartialApkFile(this, version)

        var announcement: SessionAnnouncement? = null;
        try {
            if (apkFile.exists() && apkFile.length() > 0L) {
                Logger.i(TAG, "APK already downloaded at ${apkFile.absolutePath}")
                onDownloadComplete(version, apkFile)
                return
            }

            try {
                announcement = StateAnnouncement.instance.registerLoading("Downloading new version [${version}]", "New version is being downloaded..",
                    ImageVariable.fromResource(R.drawable.foreground));
            }
            catch(ex: Exception){
                Logger.e(TAG, "Failed to set progress announcement", ex);
            }

            var backoffMs = INITIAL_BACKOFF_MS

            for (attempt in 0 until MAX_RETRIES) {
                if (cancelRequested) {
                    Logger.i(TAG, "Download cancelled before attempt ${attempt + 1}")
                    break
                }

                try {
                    performDownload(StateUpdate.getApkUrl(version), partialFile, version, {
                        try {
                            if (announcement != null)
                                announcement?.setProgress(it);
                        }
                        catch(ex: Throwable) {}
                    })

                    if (!cancelRequested) {
                        if (apkFile.exists()) {
                            apkFile.delete()
                        }
                        if (!partialFile.renameTo(apkFile)) {
                            throw IllegalStateException("Failed to rename partial APK file")
                        }
                        onDownloadComplete(version, apkFile)
                    }
                    break
                } catch (t: Throwable) {
                    if (cancelRequested) {
                        Logger.i(TAG, "Download cancelled by user", t)
                        break
                    }

                    if (attempt == MAX_RETRIES - 1) {
                        Logger.e(TAG, "Download failed after ${attempt + 1} attempts", t)
                        UpdateNotificationManager.showDownloadFailedNotification(this, version, t)
                        StateUpdate.Companion.instance.setUiFailed(version, t.message)
                        break
                    } else {
                        Logger.w(TAG, "Download attempt ${attempt + 1} failed, retrying in ${backoffMs / 1000}s", t)
                        delay(backoffMs)
                        backoffMs *= 2
                    }
                }
            }
        } finally {
            try {
                if (announcement != null) {
                    StateAnnouncement.instance.closeAnnouncement(announcement.id);
                }
            }
            catch(ex: Throwable){}
            isDownloading = false
            cancelRequested = false
            stopForeground(Service.STOP_FOREGROUND_REMOVE)
            stopSelf()
        }
    }

    private fun performDownload(url: String, partialFile: File, version: Int, onProgress: ((Int)->Unit)? = null) {
        var startOffset = if (partialFile.exists()) partialFile.length() else 0L
        Logger.i(TAG, "Starting download. url=$url, existingBytes=$startOffset")

        var connection: HttpURLConnection? = null
        try {
            connection = (URL(url).openConnection() as HttpURLConnection).apply {
                connectTimeout = 15_000
                readTimeout = 30_000
                if (startOffset > 0L) {
                    setRequestProperty("Range", "bytes=$startOffset-")
                }
            }

            connection.connect()
            val responseCode = connection.responseCode

            if (responseCode == HttpURLConnection.HTTP_OK && startOffset > 0L) {
                Logger.w(TAG, "Server ignored Range header, restarting download from scratch")
                partialFile.delete()
                startOffset = 0L
            } else if (responseCode != HttpURLConnection.HTTP_OK &&
                responseCode != HttpURLConnection.HTTP_PARTIAL) {
                throw IllegalStateException("Unexpected HTTP response code $responseCode")
            }

            val contentLength = connection.contentLengthLong
            val totalBytes = if (contentLength > 0L) startOffset + contentLength else -1L

            val buffer = ByteArray(BUFFER_SIZE)
            var downloaded = 0L
            var lastProgress = -1

            connection.inputStream.use { input ->
                FileOutputStream(partialFile, startOffset > 0L).use { output ->
                    while (!cancelRequested) {
                        val read = input.read(buffer)
                        if (read == -1) {
                            break
                        }
                        output.write(buffer, 0, read)
                        downloaded += read

                        if (totalBytes > 0L) {
                            val progress = (((startOffset + downloaded) * 100L) / totalBytes).toInt()
                            if (progress != lastProgress) {
                                lastProgress = progress
                                val safeProgress = when {
                                    progress < 0 -> 0
                                    progress > 100 -> 100
                                    else -> progress
                                }
                                throttledUpdateDownloadProgress(version, safeProgress, indeterminate = false, onProgress)
                            }
                        } else {
                            throttledUpdateDownloadProgress(version, progress = 0, indeterminate = true)
                        }
                    }

                    if (!cancelRequested && totalBytes > 0L) {
                        val finalProgress = 100
                        throttledUpdateDownloadProgress(version, finalProgress, indeterminate = false)
                    }

                    output.flush()
                }
            }

            if (cancelRequested) {
                throw CancellationException("Download cancelled")
            }

            if (totalBytes > 0L && startOffset + downloaded < totalBytes) {
                throw IllegalStateException("Download incomplete: expected=$totalBytes, got=${startOffset + downloaded}")
            }
        } finally {
            connection?.disconnect()
        }
    }

    private fun onDownloadComplete(version: Int, apkFile: File) {
        Logger.i(TAG, "Download complete for version=$version, file=${apkFile.absolutePath}")
        UpdateNotificationManager.showDownloadCompleteNotification(this, version, apkFile)
        StateUpdate.Companion.instance.setUiReady(version, apkFile)

        try {
            val ctx = applicationContext
            StateAnnouncement.instance.registerAnnouncement("install-update-apk", "Grayjay v${version} is ready!", "You can now install the new Grayjay version.", AnnouncementType.SESSION, OffsetDateTime.now(), "update", "Install") {
                UpdateNotificationManager.cancelAll(ctx)
                UpdateInstaller.startInstall(ctx, version, apkFile)
            }
        } catch (ex: Throwable) {
            Logger.w(TAG, "Failed to register install announcement", ex)
        }
    }
}