From 6e74515e3760fab9057f4f03fe0861820fbabe08 Mon Sep 17 00:00:00 2001
From: Sarang S <85063520+eigengravy@users.noreply.github.com>
Date: Thu, 18 Aug 2022 21:28:10 +0530
Subject: [PATCH] Download update apk to cache folder (#20)

* custom downloader + typo

* mutex lock downloading, using app client

* fix mutex

* remove unused imports
---
 app/build.gradle                              |  8 +-
 .../cloudstream3/utils/InAppUpdater.kt        | 75 +++++--------------
 2 files changed, 24 insertions(+), 59 deletions(-)

diff --git a/app/build.gradle b/app/build.gradle
index b7226a49..1aa62378 100644
--- a/app/build.gradle
+++ b/app/build.gradle
@@ -8,9 +8,9 @@ plugins {
 
 def tmpFilePath = System.getProperty("user.home") + "/work/_temp/keystore/"
 def allFilesFromDir = new File(tmpFilePath).listFiles()
-def prerelaseStoreFile = null
+def prereleaseStoreFile = null
 if (allFilesFromDir != null) {
-    prerelaseStoreFile = allFilesFromDir.first()
+    prereleaseStoreFile = allFilesFromDir.first()
 }
 
 android {
@@ -19,8 +19,8 @@ android {
     }
     signingConfigs {
         prerelease {
-            if (prerelaseStoreFile != null) {
-                storeFile = file(prerelaseStoreFile)
+            if (prereleaseStoreFile != null) {
+                storeFile = file(prereleaseStoreFile)
                 storePassword System.getenv("SIGNING_STORE_PASSWORD")
                 keyAlias System.getenv("SIGNING_KEY_ALIAS")
                 keyPassword System.getenv("SIGNING_KEY_PASSWORD")
diff --git a/app/src/main/java/com/lagradost/cloudstream3/utils/InAppUpdater.kt b/app/src/main/java/com/lagradost/cloudstream3/utils/InAppUpdater.kt
index 94dac83b..32fa04af 100644
--- a/app/src/main/java/com/lagradost/cloudstream3/utils/InAppUpdater.kt
+++ b/app/src/main/java/com/lagradost/cloudstream3/utils/InAppUpdater.kt
@@ -1,18 +1,13 @@
 package com.lagradost.cloudstream3.utils
 
 import android.app.Activity
-import android.app.DownloadManager
-import android.content.BroadcastReceiver
 import android.content.Context
 import android.content.Intent
-import android.content.IntentFilter
 import android.net.Uri
-import android.os.Environment
 import android.util.Log
 import android.widget.Toast
 import androidx.appcompat.app.AlertDialog
 import androidx.core.content.FileProvider
-import androidx.core.content.getSystemService
 import androidx.preference.PreferenceManager
 import com.fasterxml.jackson.annotation.JsonProperty
 import com.lagradost.cloudstream3.BuildConfig
@@ -22,10 +17,15 @@ import com.lagradost.cloudstream3.app
 import com.lagradost.cloudstream3.mvvm.logError
 import com.lagradost.cloudstream3.mvvm.normalSafeApiCall
 import com.lagradost.cloudstream3.utils.AppUtils.parseJson
+import com.lagradost.cloudstream3.utils.Coroutines.ioSafe
 import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.sync.Mutex
+import kotlinx.coroutines.sync.withLock
+import okio.*
 import java.io.File
 import kotlin.concurrent.thread
 
+
 class InAppUpdater {
     companion object {
         const val GITHUB_USER_NAME = "recloudstream"
@@ -191,66 +191,31 @@ class InAppUpdater {
             }
         }
 
+
+        private val updateLock = Mutex()
+
         private fun Activity.downloadUpdate(url: String): Boolean {
-            Log.d(LOG_TAG, "Downloading update: ${url}")
 
-            val downloadManager = getSystemService<DownloadManager>()!!
-
-            val request = DownloadManager.Request(Uri.parse(url))
-                .setMimeType("application/vnd.android.package-archive")
-                .setTitle("CloudStream Update")
-                .setDestinationInExternalPublicDir(
-                    Environment.DIRECTORY_DOWNLOADS,
-                    "CloudStream.apk"
-                )
-                .setAllowedNetworkTypes(DownloadManager.Request.NETWORK_WIFI or DownloadManager.Request.NETWORK_MOBILE)
-                .setAllowedOverRoaming(true)
-                .setNotificationVisibility(DownloadManager.Request.VISIBILITY_VISIBLE_NOTIFY_COMPLETED)
+            Log.d(LOG_TAG, "Downloading update: $url")
 
             val localContext = this
 
-            val id = try {
-                downloadManager.enqueue(request)
-            } catch (e: Exception) {
-                logError(e)
-                showToast(this, R.string.storage_error, Toast.LENGTH_SHORT)
-                -1
+            val downloadedFile = File.createTempFile("CloudStream",".apk")
+            val sink: BufferedSink = downloadedFile.sink().buffer()
+
+
+            ioSafe {
+                updateLock.withLock {
+                    sink.writeAll(app.get(url).body.source() )
+                    sink.close()
+                    openApk(localContext, Uri.fromFile(downloadedFile))
+                }
             }
-            if (id == -1L) return true
-            registerReceiver(
-                object : BroadcastReceiver() {
-                    override fun onReceive(context: Context?, intent: Intent?) {
-                        try {
-                            val downloadId = intent?.getLongExtra(
-                                DownloadManager.EXTRA_DOWNLOAD_ID, id
-                            ) ?: id
 
-                            val query = DownloadManager.Query()
-                            query.setFilterById(downloadId)
-                            val c = downloadManager.query(query)
-
-                            if (c.moveToFirst()) {
-                                val columnIndex = c.getColumnIndex(DownloadManager.COLUMN_STATUS)
-                                if (DownloadManager.STATUS_SUCCESSFUL == c
-                                        .getInt(columnIndex)
-                                ) {
-                                    c.getColumnIndex(DownloadManager.COLUMN_MEDIAPROVIDER_URI)
-                                    val uri = Uri.parse(
-                                        c.getString(c.getColumnIndex(DownloadManager.COLUMN_LOCAL_URI))
-                                    )
-                                    openApk(localContext, uri)
-                                }
-                            }
-                        } catch (e: Exception) {
-                            logError(e)
-                        }
-                    }
-                }, IntentFilter(DownloadManager.ACTION_DOWNLOAD_COMPLETE)
-            )
             return true
         }
 
-        fun openApk(context: Context, uri: Uri) {
+        private fun openApk(context: Context, uri: Uri) {
             try {
                 uri.path?.let {
                     val contentUri = FileProvider.getUriForFile(