feat: media session as scope provider

Mufanc · May 23, 2024 · 08a8d69 · 08a8d69
1 parent 6af7956
commit 08a8d69
Show file tree

Hide file tree

Showing 9 changed files with 104 additions and 25 deletions.
diff --git a/api-stub/src/main/java/android/app/ActivityThread.java b/api-stub/src/main/java/android/app/ActivityThread.java
@@ -0,0 +1,11 @@
+package android.app;
+
+public class ActivityThread {
+    public static ActivityThread currentActivityThread() {
+        throw new RuntimeException("STUB");
+    }
+
+    public ContextImpl getSystemContext() {
+        throw new RuntimeException("STUB");
+    }
+}
diff --git a/api-stub/src/main/java/android/app/ContextImpl.java b/api-stub/src/main/java/android/app/ContextImpl.java
@@ -0,0 +1,7 @@
+package android.app;
+
+import android.content.Context;
+
+public abstract class ContextImpl extends Context {
+
+}
diff --git a/app/src/main/java/xyz/mufanc/applock/core/AppLockService.kt b/app/src/main/java/xyz/mufanc/applock/core/AppLockService.kt
@@ -41,7 +41,7 @@ object AppLockService : IAppLockService.Stub() {
         )
     }
 
-    fun client(): IAppLockService {
+    fun client(): IAppLockService? {
         return client
     }
 

diff --git a/app/src/main/java/xyz/mufanc/applock/core/ModuleMain.kt b/app/src/main/java/xyz/mufanc/applock/core/ModuleMain.kt
@@ -1,5 +1,7 @@
 package xyz.mufanc.applock.core
 
+import android.os.Handler
+import android.os.Looper
 import androidx.annotation.Keep
 import io.github.libxposed.api.XposedInterface
 import io.github.libxposed.api.XposedModule
@@ -13,6 +15,7 @@ import xyz.mufanc.applock.core.util.GraftClassLoader
 import xyz.mufanc.applock.core.util.Log
 import xyz.mufanc.applock.util.Configs
 import xyz.mufanc.autox.annotation.XposedEntry
+import kotlin.concurrent.thread
 
 @Keep
 @XposedEntry(["system"])
@@ -37,12 +40,33 @@ class ModuleMain(
         GraftClassLoader.init(param.classLoader)
 
         ProcessRecordUtil.init()
-        AppLockService.init(ixp)
-        ScopeManager.init(ixp)
-        ProcessGuard.install(ixp)
 
-        if (Configs.isDebug) {
-            KillProcessMonitor.init(ixp)
+        Log.d(TAG, "waiting for system...")
+
+        runOnSystemReady {
+            AppLockService.init(ixp)
+            ScopeManager.init(ixp)
+            ProcessGuard.install(ixp)
+
+            if (Configs.isDebug) {
+                KillProcessMonitor.init(ixp)
+            }
+        }
+    }
+
+    private fun runOnSystemReady(block: () -> Unit) {
+        thread {
+            try {
+                while (Looper.getMainLooper() == null) {
+                    Thread.sleep(1000)
+                }
+
+                Handler(Looper.getMainLooper()).post {
+                    block()
+                }
+            } catch (err: Throwable) {
+                Log.e(TAG, "", err)
+            }
         }
     }
 }
diff --git a/app/src/main/java/xyz/mufanc/applock/core/scope/ScopeManager.kt b/app/src/main/java/xyz/mufanc/applock/core/scope/ScopeManager.kt
@@ -44,7 +44,7 @@ object ScopeManager {
                 Log.i(TAG, "initializing scope provider: ${klass.simpleName}")
                 provider.init(ixp)
             } catch (err: Throwable) {
-                Log.e(TAG, "failed to initialize scope provider: ${klass.simpleName}")
+                Log.e(TAG, "failed to initialize scope provider: ${klass.simpleName}", err)
             }
         }
     }

diff --git a/app/src/main/java/xyz/mufanc/applock/core/scope/provider/MediaSessionProvider.kt b/app/src/main/java/xyz/mufanc/applock/core/scope/provider/MediaSessionProvider.kt
@@ -0,0 +1,32 @@
+package xyz.mufanc.applock.core.scope.provider
+
+import android.media.session.MediaSessionManager
+import android.media.session.MediaSessionManager.OnActiveSessionsChangedListener
+import android.app.ActivityThread
+import io.github.libxposed.api.XposedInterface
+import xyz.mufanc.applock.core.util.Log
+
+@Suppress("Unused")
+data object MediaSessionProvider : ScopeProvider() {
+
+    private const val TAG = "MediaSessionProvider"
+
+    private val manager: MediaSessionManager by lazy {
+        ActivityThread.currentActivityThread()
+            .systemContext
+            .getSystemService(MediaSessionManager::class.java)
+    }
+
+    private val listener = OnActiveSessionsChangedListener { controllers ->
+        if (controllers != null) {
+            emit(controllers.map { it.packageName }.toSet())
+        }
+    }
+
+    override fun isAvailable(): Boolean = true
+
+    override fun init(ixp: XposedInterface) {
+        manager.addOnActiveSessionsChangedListener(listener, null)
+        Log.d(TAG, "initialized")
+    }
+}
diff --git a/app/src/main/java/xyz/mufanc/applock/ui/fragment/settings/SettingsFragment.kt b/app/src/main/java/xyz/mufanc/applock/ui/fragment/settings/SettingsFragment.kt
@@ -9,8 +9,6 @@ import androidx.preference.Preference
 import androidx.preference.PreferenceFragmentCompat
 import androidx.preference.PreferenceGroup
 import androidx.preference.forEach
-import kotlinx.coroutines.flow.collect
-import kotlinx.coroutines.flow.collectLatest
 import xyz.mufanc.applock.R
 import xyz.mufanc.applock.ui.util.ThemeManager
 import xyz.mufanc.applock.ui.widgets.LicenseListDialog
@@ -60,7 +58,7 @@ class SettingsFragment : PreferenceFragmentCompat() {
                 true
             }
             "scope_providers" -> {
-                ScopeProviderSelectorDialog(requireActivity()).show()
+                ScopeProviderSelectorDialog(requireActivity())?.show()
                 true
             }
             else -> super.onPreferenceTreeClick(preference)

diff --git a/app/src/main/java/xyz/mufanc/applock/ui/widgets/ScopeProviderSelectorDialog.kt b/app/src/main/java/xyz/mufanc/applock/ui/widgets/ScopeProviderSelectorDialog.kt
@@ -4,6 +4,7 @@ import android.app.Activity
 import android.app.Application.ActivityLifecycleCallbacks
 import android.content.Context
 import android.content.DialogInterface
+import android.os.Build
 import android.os.Bundle
 import android.view.View
 import android.view.ViewGroup
@@ -26,8 +27,10 @@ class ScopeProviderSelectorDialog private constructor(
 
         private val disabledProviders = RemotePrefs.disabledProviders.asLiveData()
 
-        operator fun invoke(activity: Activity): ScopeProviderSelectorDialog {
-            val availableProviders = AppLockService.client().availableProviders.toSet()
+        operator fun invoke(activity: Activity): ScopeProviderSelectorDialog? {
+            val service = AppLockService.client() ?: return null
+            val availableProviders = service.availableProviders.toSet()
+
             Log.d(TAG, "available providers: $availableProviders")
             Log.d(TAG, "disabled providers: ${disabledProviders.value?.all?.keys}")
 
@@ -52,18 +55,22 @@ class ScopeProviderSelectorDialog private constructor(
                 .create()
 
 
-            activity.registerActivityLifecycleCallbacks(object : ActivityLifecycleCallbacks {
-                override fun onActivityDestroyed(activity: Activity) {
-                    dialog.dismiss()
-                }
-
-                override fun onActivityCreated(activity: Activity, bundle: Bundle?) = Unit
-                override fun onActivityStarted(activity: Activity) = Unit
-                override fun onActivityResumed(activity: Activity) = Unit
-                override fun onActivityPaused(activity: Activity) = Unit
-                override fun onActivityStopped(activity: Activity) = Unit
-                override fun onActivitySaveInstanceState(activity: Activity, bundle: Bundle) = Unit
-            })
+            if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) {
+                activity.registerActivityLifecycleCallbacks(object : ActivityLifecycleCallbacks {
+                    override fun onActivityDestroyed(activity: Activity) {
+                        dialog.dismiss()
+                    }
+
+                    override fun onActivityCreated(activity: Activity, bundle: Bundle?) = Unit
+                    override fun onActivityStarted(activity: Activity) = Unit
+                    override fun onActivityResumed(activity: Activity) = Unit
+                    override fun onActivityPaused(activity: Activity) = Unit
+                    override fun onActivityStopped(activity: Activity) = Unit
+                    override fun onActivitySaveInstanceState(activity: Activity, bundle: Bundle) = Unit
+                })
+            } else {
+                // Todo:
+            }
 
             return ScopeProviderSelectorDialog(dialog)
         }

diff --git a/app/src/main/res/values/arrays.xml b/app/src/main/res/values/arrays.xml
@@ -13,7 +13,7 @@
 
     <string-array name="scope_providers">
         <item>PreferenceScopeProvider</item>
-        <item>MediaScopeProvider</item>
+        <item>MediaSessionProvider</item>
     </string-array>
 
     <string-array name="scope_provider_names">