Restore wakelock and change to PARTIAL_WAKE_LOCK

ReVanced · Dec 17, 2024 · e989c55 · e989c55
1 parent d546045
commit e989c55
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/app/src/main/AndroidManifest.xml b/app/src/main/AndroidManifest.xml
@@ -15,6 +15,7 @@
     <uses-permission android:name="android.permission.READ_EXTERNAL_STORAGE" android:maxSdkVersion="29" />
     <uses-permission android:name="android.permission.MANAGE_EXTERNAL_STORAGE"
         tools:ignore="ScopedStorage" />
+    <uses-permission android:name="android.permission.WAKE_LOCK" />
 
     <queries>
         <intent>

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
@@ -9,6 +9,7 @@ import android.content.Intent
 import android.content.pm.ServiceInfo
 import android.graphics.drawable.Icon
 import android.os.Build
+import android.os.PowerManager
 import android.util.Log
 import androidx.core.content.ContextCompat
 import androidx.work.ForegroundInfo
@@ -106,9 +107,21 @@ class PatcherWorker(
             Log.d(tag, "Failed to set foreground info:", e)
         }
 
+        val wakeLock: PowerManager.WakeLock =
+            (applicationContext.getSystemService(Context.POWER_SERVICE) as PowerManager)
+                .newWakeLock(PowerManager.PARTIAL_WAKE_LOCK, "$tag::Patcher")
+                .apply {
+                    acquire(10 * 60 * 1000L)
+                    Log.d(tag, "Acquired wakelock.")
+                }
+
         val args = workerRepository.claimInput(this)
 
-        return runPatcher(args)
+        return try {
+            runPatcher(args)
+        } finally {
+            wakeLock.release()
+        }
     }
 
     private suspend fun runPatcher(args: Args): Result {