From 835c49ae79cacf00a09ce2d7a9575e00707c56a6 Mon Sep 17 00:00:00 2001
From: Koitharu <nvasya95@gmail.com>
Date: Fri, 15 Sep 2023 13:10:22 +0300
Subject: [PATCH] Download updates directly

---
 app/src/main/AndroidManifest.xml              |  8 ++++
 .../kotatsu/settings/about/AppUpdateDialog.kt | 24 ++++++++++--
 .../settings/about/UpdateDownloadReceiver.kt  | 38 +++++++++++++++++++
 3 files changed, 66 insertions(+), 4 deletions(-)
 create mode 100644 app/src/main/kotlin/org/koitharu/kotatsu/settings/about/UpdateDownloadReceiver.kt

diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index b665da264..ff15dabe3 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -19,6 +19,7 @@
 	<uses-permission android:name="android.permission.WRITE_SYNC_SETTINGS" />
 	<uses-permission android:name="android.permission.POST_NOTIFICATIONS" />
 	<uses-permission android:name="android.permission.FOREGROUND_SERVICE_DATA_SYNC" />
+	<uses-permission android:name="android.permission.REQUEST_INSTALL_PACKAGES" />
 	<uses-permission
 		android:name="android.permission.WRITE_EXTERNAL_STORAGE"
 		android:maxSdkVersion="29" />
@@ -319,6 +320,13 @@
 				android:name="android.appwidget.provider"
 				android:resource="@xml/widget_recent" />
 		</receiver>
+		<receiver
+			android:name="org.koitharu.kotatsu.settings.about.UpdateDownloadReceiver"
+			android:exported="true">
+			<intent-filter>
+				<action android:name="android.intent.action.DOWNLOAD_COMPLETE" />
+			</intent-filter>
+		</receiver>
 
 		<meta-data
 			android:name="android.webkit.WebView.EnableSafeBrowsing"
diff --git a/app/src/main/kotlin/org/koitharu/kotatsu/settings/about/AppUpdateDialog.kt b/app/src/main/kotlin/org/koitharu/kotatsu/settings/about/AppUpdateDialog.kt
index e57058e09..8b22218d6 100644
--- a/app/src/main/kotlin/org/koitharu/kotatsu/settings/about/AppUpdateDialog.kt
+++ b/app/src/main/kotlin/org/koitharu/kotatsu/settings/about/AppUpdateDialog.kt
@@ -1,7 +1,10 @@
 package org.koitharu.kotatsu.settings.about
 
+import android.app.DownloadManager
 import android.content.Context
 import android.content.Intent
+import android.os.Environment
+import android.widget.Toast
 import androidx.core.net.toUri
 import androidx.core.text.buildSpannedString
 import com.google.android.material.dialog.MaterialAlertDialogBuilder
@@ -29,13 +32,26 @@ class AppUpdateDialog(private val context: Context) {
 			.setTitle(R.string.app_update_available)
 			.setMessage(message)
 			.setIcon(R.drawable.ic_app_update)
-			.setPositiveButton(R.string.download) { _, _ ->
-				val intent = Intent(Intent.ACTION_VIEW, version.apkUrl.toUri())
+			.setNeutralButton(R.string.open_in_browser) { _, _ ->
+				val intent = Intent(Intent.ACTION_VIEW, version.url.toUri())
 				context.startActivity(Intent.createChooser(intent, context.getString(R.string.open_in_browser)))
-			}
-			.setNegativeButton(R.string.close, null)
+			}.setPositiveButton(R.string.update) { _, _ ->
+				downloadUpdate(version)
+			}.setNegativeButton(android.R.string.cancel, null)
 			.setCancelable(false)
 			.create()
 			.show()
 	}
+
+	private fun downloadUpdate(version: AppVersion) {
+		val url = version.apkUrl.toUri()
+		val dm = context.getSystemService(Context.DOWNLOAD_SERVICE) as DownloadManager
+		val request = DownloadManager.Request(url)
+			.setTitle("${context.getString(R.string.app_name)} v${version.name}")
+			.setDestinationInExternalPublicDir(Environment.DIRECTORY_DOWNLOADS, url.lastPathSegment)
+			.setNotificationVisibility(DownloadManager.Request.VISIBILITY_VISIBLE_NOTIFY_COMPLETED)
+			.setMimeType("application/vnd.android.package-archive")
+		dm.enqueue(request)
+		Toast.makeText(context, R.string.download_started, Toast.LENGTH_SHORT).show()
+	}
 }
diff --git a/app/src/main/kotlin/org/koitharu/kotatsu/settings/about/UpdateDownloadReceiver.kt b/app/src/main/kotlin/org/koitharu/kotatsu/settings/about/UpdateDownloadReceiver.kt
new file mode 100644
index 000000000..bb2b8e9c9
--- /dev/null
+++ b/app/src/main/kotlin/org/koitharu/kotatsu/settings/about/UpdateDownloadReceiver.kt
@@ -0,0 +1,38 @@
+package org.koitharu.kotatsu.settings.about
+
+import android.app.DownloadManager
+import android.content.ActivityNotFoundException
+import android.content.BroadcastReceiver
+import android.content.Context
+import android.content.Intent
+import org.koitharu.kotatsu.core.util.ext.printStackTraceDebug
+
+
+class UpdateDownloadReceiver : BroadcastReceiver() {
+
+	override fun onReceive(context: Context, intent: Intent) {
+		when (intent.action) {
+			DownloadManager.ACTION_DOWNLOAD_COMPLETE -> {
+				val downloadId = intent.getLongExtra(DownloadManager.EXTRA_DOWNLOAD_ID, 0L)
+				if (downloadId == 0L) {
+					return
+				}
+				val dm = context.getSystemService(Context.DOWNLOAD_SERVICE) as DownloadManager
+
+				@Suppress("DEPRECATION")
+				val installIntent = Intent(Intent.ACTION_INSTALL_PACKAGE)
+				installIntent.setFlags(Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_GRANT_READ_URI_PERMISSION)
+				installIntent.setDataAndType(
+					dm.getUriForDownloadedFile(downloadId),
+					dm.getMimeTypeForDownloadedFile(downloadId),
+				)
+				installIntent.putExtra(Intent.EXTRA_NOT_UNKNOWN_SOURCE, true)
+				try {
+					context.startActivity(installIntent)
+				} catch (e: ActivityNotFoundException) {
+					e.printStackTraceDebug()
+				}
+			}
+		}
+	}
+}