Use WakeLock for background operations

app/src/main/kotlin/org/koitharu/kotatsu/alternatives/ui/AutoFixService.kt

@@ -24,8 +24,10 @@ import org.koitharu.kotatsu.core.ui.CoroutineIntentService
 import org.koitharu.kotatsu.core.util.ext.checkNotificationPermission
 import org.koitharu.kotatsu.core.util.ext.getDisplayMessage
 import org.koitharu.kotatsu.core.util.ext.mangaSourceExtra
+import org.koitharu.kotatsu.core.util.ext.powerManager
 import org.koitharu.kotatsu.core.util.ext.printStackTraceDebug
 import org.koitharu.kotatsu.core.util.ext.toBitmapOrNull
+import org.koitharu.kotatsu.core.util.ext.withPartialWakeLock
 import org.koitharu.kotatsu.parsers.model.Manga
 import org.koitharu.kotatsu.parsers.util.runCatchingCancellable
 import javax.inject.Inject
@@ -51,6 +53,7 @@ class AutoFixService : CoroutineIntentService() {
 		val ids = requireNotNull(intent.getLongArrayExtra(DATA_IDS))
 		startForeground(this)
 		for (mangaId in ids) {
+			powerManager.withPartialWakeLock(TAG) {
 				val result = runCatchingCancellable {
 					autoFixUseCase.invoke(mangaId)
 				}
@@ -60,6 +63,7 @@ class AutoFixService : CoroutineIntentService() {
 				}
 			}
 		}
+	}
 
 	override fun IntentJobContext.onError(error: Throwable) {
 		if (applicationContext.checkNotificationPermission(CHANNEL_ID)) {

app/src/main/kotlin/org/koitharu/kotatsu/core/util/ext/Android.kt

@@ -56,6 +56,7 @@ import org.koitharu.kotatsu.parsers.util.runCatchingCancellable
 import org.xmlpull.v1.XmlPullParser
 import org.xmlpull.v1.XmlPullParserException
 import java.io.File
+import java.util.concurrent.TimeUnit
 import kotlin.math.roundToLong
 
 val Context.activityManager: ActivityManager?
@@ -228,3 +229,21 @@ fun Context.restartApplication() {
 	startActivity(intent)
 	activity?.finishAndRemoveTask()
 }
+
+internal inline fun <R> PowerManager?.withPartialWakeLock(tag: String, body: (PowerManager.WakeLock?) -> R): R {
+	val wakeLock = newPartialWakeLock(tag)
+	return try {
+		wakeLock?.acquire(TimeUnit.HOURS.toMillis(1))
+		body(wakeLock)
+	} finally {
+		wakeLock?.release()
+	}
+}
+
+private fun PowerManager?.newPartialWakeLock(tag: String): PowerManager.WakeLock? {
+	return if (this != null && isWakeLockLevelSupported(PowerManager.PARTIAL_WAKE_LOCK)) {
+		newWakeLock(PowerManager.PARTIAL_WAKE_LOCK, tag)
+	} else {
+		null
+	}
+}

app/src/main/kotlin/org/koitharu/kotatsu/local/ui/ImportService.kt

@@ -23,9 +23,11 @@ import org.koitharu.kotatsu.core.ui.CoroutineIntentService
 import org.koitharu.kotatsu.core.util.ext.checkNotificationPermission
 import org.koitharu.kotatsu.core.util.ext.getDisplayMessage
 import org.koitharu.kotatsu.core.util.ext.mangaSourceExtra
+import org.koitharu.kotatsu.core.util.ext.powerManager
 import org.koitharu.kotatsu.core.util.ext.printStackTraceDebug
 import org.koitharu.kotatsu.core.util.ext.toBitmapOrNull
 import org.koitharu.kotatsu.core.util.ext.toUriOrNull
+import org.koitharu.kotatsu.core.util.ext.withPartialWakeLock
 import org.koitharu.kotatsu.local.data.importer.SingleMangaImporter
 import org.koitharu.kotatsu.parsers.model.Manga
 import org.koitharu.kotatsu.parsers.util.runCatchingCancellable
@@ -50,6 +52,7 @@ class ImportService : CoroutineIntentService() {
 	override suspend fun IntentJobContext.processIntent(intent: Intent) {
 		val uri = requireNotNull(intent.getStringExtra(DATA_URI)?.toUriOrNull()) { "No input uri" }
 		startForeground(this)
+		powerManager.withPartialWakeLock(TAG) {
 			val result = runCatchingCancellable {
 				importer.import(uri).manga
 			}
@@ -58,6 +61,7 @@ class ImportService : CoroutineIntentService() {
 				notificationManager.notify(TAG, startId, notification)
 			}
 		}
+	}
 
 	override fun IntentJobContext.onError(error: Throwable) {
 		if (applicationContext.checkNotificationPermission(CHANNEL_ID)) {

app/src/main/kotlin/org/koitharu/kotatsu/local/ui/LocalChaptersRemoveService.kt

@@ -17,6 +17,8 @@ import org.koitharu.kotatsu.core.model.parcelable.ParcelableManga
 import org.koitharu.kotatsu.core.ui.CoroutineIntentService
 import org.koitharu.kotatsu.core.util.ext.getDisplayMessage
 import org.koitharu.kotatsu.core.util.ext.getParcelableExtraCompat
+import org.koitharu.kotatsu.core.util.ext.powerManager
+import org.koitharu.kotatsu.core.util.ext.withPartialWakeLock
 import org.koitharu.kotatsu.local.data.LocalMangaRepository
 import org.koitharu.kotatsu.local.data.LocalStorageChanges
 import org.koitharu.kotatsu.local.domain.model.LocalManga
@@ -44,13 +46,15 @@ class LocalChaptersRemoveService : CoroutineIntentService() {
 	}
 
 	override suspend fun IntentJobContext.processIntent(intent: Intent) {
+		startForeground(this)
 		val manga = intent.getParcelableExtraCompat<ParcelableManga>(EXTRA_MANGA)?.manga ?: return
 		val chaptersIds = intent.getLongArrayExtra(EXTRA_CHAPTERS_IDS)?.toSet() ?: return
-		startForeground(this)
+		powerManager.withPartialWakeLock(TAG) {
 			val mangaWithChapters = localMangaRepository.getDetails(manga)
 			localMangaRepository.deleteChapters(mangaWithChapters, chaptersIds)
 			localStorageChanges.emit(LocalManga(localMangaRepository.getDetails(manga)))
 		}
+	}
 
 	override fun IntentJobContext.onError(error: Throwable) {
 		val notification = NotificationCompat.Builder(applicationContext, CHANNEL_ID)
@@ -104,6 +108,8 @@ class LocalChaptersRemoveService : CoroutineIntentService() {
 		private const val EXTRA_MANGA = "manga"
 		private const val EXTRA_CHAPTERS_IDS = "chapters_ids"
 
+		private const val TAG = CHANNEL_ID
+
 		fun start(context: Context, manga: Manga, chaptersIds: Collection<Long>) {
 			if (chaptersIds.isEmpty()) {
 				return

app/src/main/kotlin/org/koitharu/kotatsu/settings/backup/BackupDialogFragment.kt

@@ -1,6 +1,5 @@
 package org.koitharu.kotatsu.settings.backup
 
-import android.net.Uri
 import android.os.Bundle
 import android.view.LayoutInflater
 import android.view.ViewGroup
@@ -19,20 +18,17 @@ import org.koitharu.kotatsu.core.util.ext.tryLaunch
 import org.koitharu.kotatsu.core.util.progress.Progress
 import org.koitharu.kotatsu.databinding.DialogProgressBinding
 import java.io.File
-import java.io.FileOutputStream
 
 @AndroidEntryPoint
 class BackupDialogFragment : AlertDialogFragment<DialogProgressBinding>() {
 
 	private val viewModel by viewModels<BackupViewModel>()
 
-	private var backup: File? = null
 	private val saveFileContract = registerForActivityResult(
 		ActivityResultContracts.CreateDocument("application/zip"),
 	) { uri ->
-		val file = backup
+		if (uri != null) {
-		if (uri != null && file != null) {
+			viewModel.saveBackup(uri)
-			saveBackup(file, uri)
 		} else {
 			dismiss()
 		}
@@ -51,6 +47,7 @@ class BackupDialogFragment : AlertDialogFragment<DialogProgressBinding>() {
 		viewModel.progress.observe(viewLifecycleOwner, this::onProgressChanged)
 		viewModel.onBackupDone.observeEvent(viewLifecycleOwner, this::onBackupDone)
 		viewModel.onError.observeEvent(viewLifecycleOwner, this::onError)
+		viewModel.onBackupSaved.observeEvent(viewLifecycleOwner) { onBackupSaved() }
 	}
 
 	override fun onBuildDialog(builder: MaterialAlertDialogBuilder): MaterialAlertDialogBuilder {
@@ -81,26 +78,14 @@ class BackupDialogFragment : AlertDialogFragment<DialogProgressBinding>() {
 	}
 
 	private fun onBackupDone(file: File) {
-		this.backup = file
 		if (!saveFileContract.tryLaunch(file.name)) {
 			Toast.makeText(requireContext(), R.string.operation_not_supported, Toast.LENGTH_SHORT).show()
 			dismiss()
 		}
 	}
 
-	private fun saveBackup(file: File, output: Uri) {
+	private fun onBackupSaved() {
-		try {
-			requireContext().contentResolver.openFileDescriptor(output, "w")?.use { fd ->
-				FileOutputStream(fd.fileDescriptor).use {
-					it.write(file.readBytes())
-				}
-			}
 		Toast.makeText(requireContext(), R.string.backup_saved, Toast.LENGTH_SHORT).show()
 		dismiss()
-		} catch (e: InterruptedException) {
-			throw e
-		} catch (e: Exception) {
-			onError(e)
-		}
 	}
 }

app/src/main/kotlin/org/koitharu/kotatsu/settings/backup/BackupViewModel.kt

@@ -1,8 +1,11 @@
 package org.koitharu.kotatsu.settings.backup
 
+import android.content.ContentResolver
 import android.content.Context
+import android.net.Uri
 import dagger.hilt.android.lifecycle.HiltViewModel
 import dagger.hilt.android.qualifiers.ApplicationContext
+import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.flow.MutableStateFlow
 import org.koitharu.kotatsu.core.backup.BackupRepository
 import org.koitharu.kotatsu.core.backup.BackupZipOutput
@@ -11,6 +14,7 @@ import org.koitharu.kotatsu.core.util.ext.MutableEventFlow
 import org.koitharu.kotatsu.core.util.ext.call
 import org.koitharu.kotatsu.core.util.progress.Progress
 import java.io.File
+import java.io.FileOutputStream
 import javax.inject.Inject
 
 @HiltViewModel
@@ -21,9 +25,13 @@ class BackupViewModel @Inject constructor(
 
 	val progress = MutableStateFlow(Progress.INDETERMINATE)
 	val onBackupDone = MutableEventFlow<File>()
+	val onBackupSaved = MutableEventFlow<Unit>()
+
+	private val contentResolver: ContentResolver = context.contentResolver
+	private var backupFile: File? = null
 
 	init {
-		launchLoadingJob {
+		launchLoadingJob(Dispatchers.Default) {
 			val file = BackupZipOutput.createTemp(context).use { backup ->
 				progress.value = Progress(0, 7)
 				backup.put(repository.createIndex())
@@ -52,7 +60,20 @@ class BackupViewModel @Inject constructor(
 				backup.finish()
 				backup.file
 			}
+			backupFile = file
 			onBackupDone.call(file)
 		}
 	}
+
+	fun saveBackup(output: Uri) {
+		launchLoadingJob(Dispatchers.Default) {
+			val file = checkNotNull(backupFile)
+			contentResolver.openFileDescriptor(output, "w")?.use { fd ->
+				FileOutputStream(fd.fileDescriptor).use {
+					it.write(file.readBytes())
+				}
+			}
+			onBackupSaved.call(Unit)
+		}
+	}
 }

app/src/main/kotlin/org/koitharu/kotatsu/settings/backup/RestoreService.kt

@@ -26,8 +26,10 @@ import org.koitharu.kotatsu.core.ui.CoroutineIntentService
 import org.koitharu.kotatsu.core.util.ext.checkNotificationPermission
 import org.koitharu.kotatsu.core.util.ext.getDisplayMessage
 import org.koitharu.kotatsu.core.util.ext.getFileDisplayName
+import org.koitharu.kotatsu.core.util.ext.powerManager
 import org.koitharu.kotatsu.core.util.ext.printStackTraceDebug
 import org.koitharu.kotatsu.core.util.ext.toUriOrNull
+import org.koitharu.kotatsu.core.util.ext.withPartialWakeLock
 import org.koitharu.kotatsu.core.util.progress.Progress
 import org.koitharu.kotatsu.parsers.util.mapToArray
 import org.koitharu.kotatsu.parsers.util.nullIfEmpty
@@ -59,6 +61,7 @@ class RestoreService : CoroutineIntentService() {
 		if (entries.isNullOrEmpty()) {
 			throw IllegalArgumentException("No entries specified")
 		}
+		powerManager.withPartialWakeLock(TAG) {
 			val result = runInterruptible(Dispatchers.IO) {
 				val tempFile = File.createTempFile("backup_", ".tmp")
 				(contentResolver.openInputStream(uri) ?: throw FileNotFoundException()).use { input ->
@@ -75,6 +78,7 @@ class RestoreService : CoroutineIntentService() {
 				notificationManager.notify(TAG, startId, notification)
 			}
 		}
+	}
 
 	override fun IntentJobContext.onError(error: Throwable) {
 		if (applicationContext.checkNotificationPermission(CHANNEL_ID)) {