From 34fe6274785e1c57497f6fa45aec0fc3c2391496 Mon Sep 17 00:00:00 2001
From: Canny <canny1913@outlook.com>
Date: Mon, 10 Oct 2022 20:49:02 +0300
Subject: [PATCH] feat: wakelock

---
 app/src/main/AndroidManifest.xml                   |  1 +
 .../manager/patcher/worker/PatcherWorker.kt        | 14 ++++++++++++--
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
index 1194e75..cc18227 100644
--- a/app/src/main/AndroidManifest.xml
+++ b/app/src/main/AndroidManifest.xml
@@ -12,6 +12,7 @@
     <uses-permission android:name="android.permission.READ_EXTERNAL_STORAGE" />
     <uses-permission android:name="android.permission.FOREGROUND_SERVICE"/>
     <uses-permission android:name="android.permission.WRITE_EXTERNAL_STORAGE" />
+    <uses-permission android:name="android.permission.WAKE_LOCK" />
     <uses-permission
         android:name="android.permission.MANAGE_EXTERNAL_STORAGE"
         tools:ignore="ScopedStorage" />
diff --git a/app/src/main/java/app/revanced/manager/patcher/worker/PatcherWorker.kt b/app/src/main/java/app/revanced/manager/patcher/worker/PatcherWorker.kt
index 0c9c44f..86a7a06 100644
--- a/app/src/main/java/app/revanced/manager/patcher/worker/PatcherWorker.kt
+++ b/app/src/main/java/app/revanced/manager/patcher/worker/PatcherWorker.kt
@@ -7,6 +7,7 @@ import android.app.PendingIntent
 import android.content.Context
 import android.content.Intent
 import android.graphics.drawable.Icon
+import android.os.PowerManager
 import android.util.Log
 import androidx.core.content.ContextCompat
 import androidx.work.CoroutineWorker
@@ -90,6 +91,13 @@ class PatcherWorker(context: Context, parameters: WorkerParameters, private val
     private suspend fun runPatcher(
         workdir: File
     ): Boolean {
+        val wakeLock: PowerManager.WakeLock =
+            (applicationContext.getSystemService(Context.POWER_SERVICE) as PowerManager).run {
+                newWakeLock(PowerManager.FULL_WAKE_LOCK, "$tag::Patcher").apply {
+                    acquire(10*60*1000L)
+                }
+            }
+        Log.d(tag, "Acquired wakelock.")
         val aaptPath = Aapt.binary(applicationContext)?.absolutePath
         if (aaptPath == null) {
             Logging.log += "AAPT2 not found.\n"
@@ -188,8 +196,10 @@ class PatcherWorker(context: Context, parameters: WorkerParameters, private val
             Signer("ReVanced", "s3cur3p@ssw0rd").signApk(patchedFile, outputFile)
             Logging.log += "Successfully patched!\n"
         } finally {
-            // Log.d(tag, "Deleting workdir")
-            // workdir.deleteRecursively()
+            Log.d(tag, "Deleting workdir")
+            workdir.deleteRecursively()
+            wakeLock.release()
+            Log.d(tag, "Released wakelock.")
         }
         return false
     }