Update ActivityTaskTracker

Author: yongce

baseLib/src/main/java/me/ycdev/android/lib/common/activity/ActivityMeta.kt

@@ -0,0 +1,46 @@
+package me.ycdev.android.lib.common.activity
+
+import android.content.ComponentName
+import android.content.Context
+import android.content.pm.ActivityInfo
+import android.content.pm.PackageManager
+import androidx.annotation.VisibleForTesting
+import java.util.concurrent.ConcurrentHashMap
+
+data class ActivityMeta(
+    val componentName: ComponentName,
+    val taskAffinity: String,
+    val launchMode: Int,
+    val allowTaskReparenting: Boolean
+) {
+    companion object {
+        private val cache = ConcurrentHashMap<String, ActivityMeta>()
+
+        /**
+         * @throws PackageManager.NameNotFoundException if component not found in the system
+         */
+        fun get(context: Context, activity: ComponentName): ActivityMeta {
+            val key = activity.flattenToShortString()
+            var meta = cache[key]
+            if (meta != null) {
+                return meta
+            }
+
+            val info = context.packageManager.getActivityInfo(activity, 0)
+            val taskAffinity = info.taskAffinity ?: context.applicationInfo.taskAffinity
+            val allowTaskReparenting = (info.flags and ActivityInfo.FLAG_ALLOW_TASK_REPARENTING) > 0
+            meta = ActivityMeta(activity, taskAffinity, info.launchMode, allowTaskReparenting)
+            cache[key] = meta
+            return meta
+        }
+
+        @VisibleForTesting
+        internal fun initCache(vararg metas: ActivityMeta) {
+            cache.clear()
+            metas.forEach {
+                val key = it.componentName.flattenToShortString()
+                cache[key] = it
+            }
+        }
+    }
+}

baseLib/src/main/java/me/ycdev/android/lib/common/activity/ActivityInfo.kt renamed to baseLib/src/main/java/me/ycdev/android/lib/common/activity/ActivityRunningState.kt

@@ -2,13 +2,14 @@ package me.ycdev.android.lib.common.activity
 
 import android.content.ComponentName
 
-data class ActivityInfo(
+data class ActivityRunningState(
     val componentName: ComponentName,
-    val taskId: Int,
+    val hashCode: Int,
+    var taskId: Int,
     var state: State = State.None
 ) {
-    fun makeCopy(): ActivityInfo {
-        val cloned = ActivityInfo(componentName, taskId)
+    fun makeCopy(): ActivityRunningState {
+        val cloned = ActivityRunningState(componentName, hashCode, taskId)
         cloned.state = state
         return cloned
     }

baseLib/src/main/java/me/ycdev/android/lib/common/activity/ActivityTask.kt

@@ -3,38 +3,40 @@ package me.ycdev.android.lib.common.activity
 import android.content.ComponentName
 import java.util.Stack
 
-class ActivityTask(val taskId: Int) {
-    private val activities = arrayListOf<ActivityInfo>()
+class ActivityTask(val taskId: Int, val taskAffinity: String) {
+    private val activities = arrayListOf<ActivityRunningState>()
 
-    internal fun addActivity(activity: ActivityInfo) {
+    internal fun addActivity(activity: ActivityRunningState) {
         if (activity.taskId != taskId) {
             throw RuntimeException("Activity taskId[${activity.taskId}] != AppTask[$taskId]")
         }
         activities.add(activity)
     }
 
-    internal fun popActivity(componentName: ComponentName): ActivityInfo {
+    internal fun popActivity(componentName: ComponentName, hashCode: Int): ActivityRunningState {
         val it = activities.asReversed().iterator()
         while (it.hasNext()) {
             val activity = it.next()
-            if (activity.componentName == componentName) {
+            if (activity.componentName == componentName && activity.hashCode == hashCode) {
                 it.remove()
                 return activity
             }
         }
-        throw RuntimeException("Cannot find $componentName")
+        val hashHex = Integer.toHexString(hashCode)
+        throw RuntimeException("Cannot find $componentName@$hashHex")
     }
 
-    fun lastActivity(componentName: ComponentName): ActivityInfo {
+    fun lastActivity(componentName: ComponentName, hashCode: Int): ActivityRunningState {
         activities.asReversed().forEach {
-            if (it.componentName == componentName) {
+            if (it.componentName == componentName && it.hashCode == hashCode) {
                 return it
             }
         }
-        throw RuntimeException("Cannot find $componentName")
+        val hashHex = Integer.toHexString(hashCode)
+        throw RuntimeException("Cannot find $componentName@$hashHex")
     }
 
-    fun topActivity(): ActivityInfo {
+    fun topActivity(): ActivityRunningState {
         if (activities.isEmpty()) {
             throw RuntimeException("The task is empty. Cannot get the top Activity.")
         }
@@ -44,8 +46,8 @@ class ActivityTask(val taskId: Int) {
     /**
      * @return The last Activity in returned list is the top Activity
      */
-    fun getActivityStack(): Stack<ActivityInfo> {
-        val stack = Stack<ActivityInfo>()
+    fun getActivityStack(): Stack<ActivityRunningState> {
+        val stack = Stack<ActivityRunningState>()
         activities.forEach {
             stack.push(it)
         }
@@ -55,14 +57,14 @@ class ActivityTask(val taskId: Int) {
     fun isEmpty() = activities.isEmpty()
 
     fun makeCopy(): ActivityTask {
-        val task = ActivityTask(taskId)
+        val task = ActivityTask(taskId, taskAffinity)
         activities.forEach {
             task.activities.add(it.makeCopy())
         }
         return task
     }
 
     override fun toString(): String {
-        return "AppTask[taskId=$taskId, activities=$activities]"
+        return "AppTask[taskId=$taskId, taskAffinity=$taskAffinity, activities=$activities]"
     }
 }