Update StatsViewModel.kt

handle deeplink granularity
wordpress-mobile · Feb 28, 2024 · 7c02ba5 · 7c02ba5
1 parent 3a12b87
commit 7c02ba5
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/WordPress/src/main/java/org/wordpress/android/ui/stats/refresh/StatsViewModel.kt b/WordPress/src/main/java/org/wordpress/android/ui/stats/refresh/StatsViewModel.kt
@@ -140,7 +140,20 @@ class StatsViewModel
     }
 
     private fun getInitialTimeFrame(intent: Intent): StatsSection? {
-        return when (intent.getSerializableExtraCompat<Serializable>(StatsActivity.ARG_DESIRED_TIMEFRAME)) {
+        val timeframe = intent.getSerializableExtraCompat<StatsTimeframe>(StatsActivity.ARG_DESIRED_TIMEFRAME)
+        if (statsTrafficTabFeatureConfig.isEnabled()) {
+            if (timeframe == DAY || timeframe == WEEK || timeframe == MONTH || timeframe == YEAR) {
+                when (timeframe) {
+                    DAY -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.DAYS)
+                    WEEK -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.WEEKS)
+                    MONTH -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.MONTHS)
+                    YEAR -> selectedTrafficGranularityManager.setSelectedTrafficGranularity(StatsGranularity.YEARS)
+                    else -> { /* Do nothing */}
+                }
+            }
+        }
+
+        return when (timeframe) {
             StatsTimeframe.INSIGHTS -> StatsSection.INSIGHTS
             DAY -> StatsSection.DAYS
             WEEK -> StatsSection.WEEKS