App migration minor refactoring

app/core/src/main/java/com/topjohnwu/magisk/core/AppContext.kt

@@ -22,7 +22,7 @@ import com.topjohnwu.magisk.core.base.UntrackedActivity
 import com.topjohnwu.magisk.core.base.launchPackage
 import com.topjohnwu.magisk.core.di.ServiceLocator
 import com.topjohnwu.magisk.core.ktx.writeTo
-import com.topjohnwu.magisk.core.tasks.HideAPK
+import com.topjohnwu.magisk.core.tasks.AppMigration
 import com.topjohnwu.magisk.core.utils.LocaleSetting
 import com.topjohnwu.magisk.core.utils.NetworkObserver
 import com.topjohnwu.magisk.core.utils.ProcessLifecycle
@@ -178,15 +178,15 @@ where T : ComponentActivity, T : IActivityExtension {
             // Version mismatch
             Info.stub!!.version != BuildConfig.STUB_VERSION ||
             // Not properly patched
-            intent.component!!.className.contains(HideAPK.PLACEHOLDER))
+            intent.component!!.className.contains(AppMigration.PLACEHOLDER))
     ) {
         withPermission(REQUEST_INSTALL_PACKAGES) { granted ->
             if (granted) {
-                lifecycleScope.launch(Dispatchers.IO) {
+                lifecycleScope.launch {
                     val apk = File(cacheDir, "stub.apk")
                     try {
                         assets.open("stub.apk").writeTo(apk)
-                        HideAPK.upgrade(this@initializeOnSplashScreen, apk)?.let {
+                        AppMigration.upgradeStub(this@initializeOnSplashScreen, apk)?.let {
                             startActivity(it)
                         }
                     } catch (e: IOException) {

app/core/src/main/java/com/topjohnwu/magisk/core/download/DownloadEngine.kt

@@ -33,7 +33,7 @@ import com.topjohnwu.magisk.core.ktx.forEach
 import com.topjohnwu.magisk.core.ktx.set
 import com.topjohnwu.magisk.core.ktx.withStreams
 import com.topjohnwu.magisk.core.ktx.writeTo
-import com.topjohnwu.magisk.core.tasks.HideAPK
+import com.topjohnwu.magisk.core.tasks.AppMigration
 import com.topjohnwu.magisk.core.utils.MediaStoreUtils.outputStream
 import com.topjohnwu.magisk.core.utils.ProgressInputStream
 import com.topjohnwu.magisk.utils.APKInstall
@@ -292,7 +292,7 @@ class DownloadEngine(
                 zf.close()
 
                 // Patch and install
-                subject.intent = HideAPK.upgrade(context, apk)
+                subject.intent = AppMigration.upgradeStub(context, apk)
                     ?: throw IOException("HideAPK patch error")
                 apk.delete()
             } catch (e: Exception) {

app/core/src/main/java/com/topjohnwu/magisk/core/tasks/AppMigration.kt

@@ -6,7 +6,6 @@ import android.content.Context
 import android.content.Intent
 import android.os.Build
 import android.widget.Toast
-import androidx.annotation.WorkerThread
 import com.topjohnwu.magisk.StubApk
 import com.topjohnwu.magisk.core.BuildConfig.APP_PACKAGE_NAME
 import com.topjohnwu.magisk.core.Config
@@ -33,7 +32,7 @@ import java.io.OutputStream
 import java.security.SecureRandom
 import kotlin.random.asKotlinRandom
 
-object HideAPK {
+object AppMigration {
 
     private const val ALPHA = "abcdefghijklmnopqrstuvwxyz"
     private const val ALPHADOTS = "$ALPHA....."
@@ -258,16 +257,17 @@ object HideAPK {
         if (!success) onFailure.run()
     }
 
-    @WorkerThread
-    fun upgrade(context: Context, apk: File): Intent? {
+    suspend fun upgradeStub(context: Context, apk: File): Intent? {
         val label = context.applicationInfo.nonLocalizedLabel
         val pkg = context.packageName
         val session = APKInstall.startSession(context)
-        session.openStream(context).use {
-            if (!patch(context, apk, it, pkg, label)) {
-                return null
+        return withContext(Dispatchers.IO) {
+            session.openStream(context).use {
+                if (!patch(context, apk, it, pkg, label)) {
+                    return@withContext null
+                }
             }
+            session.waitIntent()
         }
-        return session.waitIntent()
     }
 }