Prevent crash if used in an app that has non-AppCompatActivity activi…

…ties
steamclock · Mar 24, 2022 · b026e1c · b026e1c
1 parent e39aa65
commit b026e1c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/debugmenu-ui/src/main/java/com/steamclock/debugmenu_ui/ComposeDebugMenuDisplay.kt b/debugmenu-ui/src/main/java/com/steamclock/debugmenu_ui/ComposeDebugMenuDisplay.kt
@@ -33,13 +33,16 @@ class ComposeDebugMenuDisplay(app: Application) : DebugMenuDisplay {
     private fun monitor(application: Application) {
         application.registerActivityLifecycleCallbacks(object : Application.ActivityLifecycleCallbacks {
             override fun onActivityCreated(p0: Activity, p1: Bundle?) {
-                activityReference = WeakReference(p0 as AppCompatActivity)
+                val activity = p0 as? AppCompatActivity
+                activityReference = WeakReference(activity)
             }
             override fun onActivityStarted(p0: Activity) {
-                activityReference = WeakReference(p0 as AppCompatActivity)
+                val activity = p0 as? AppCompatActivity
+                activityReference = WeakReference(activity)
             }
             override fun onActivityResumed(p0: Activity) {
-                activityReference = WeakReference(p0 as AppCompatActivity)
+                val activity = p0 as? AppCompatActivity
+                activityReference = WeakReference(activity)
             }
             override fun onActivityPaused(p0: Activity) {
                 activityReference = null