From 7f33d6efc7f8d18c5d4107025eacab31eb9c948b Mon Sep 17 00:00:00 2001
From: machiav3lli <antonios.hazim@tutanota.com>
Date: Wed, 30 Mar 2022 01:55:17 +0200
Subject: [PATCH] Fix: Refresh updates list after update/install

---
 src/main/kotlin/com/looker/droidify/database/DAOs.kt     | 3 +++
 .../droidify/ui/viewmodels/MainNavFragmentViewModelX.kt  | 9 +++++++++
 2 files changed, 12 insertions(+)

diff --git a/src/main/kotlin/com/looker/droidify/database/DAOs.kt b/src/main/kotlin/com/looker/droidify/database/DAOs.kt
index b2d4f9b7..94fe4b94 100644
--- a/src/main/kotlin/com/looker/droidify/database/DAOs.kt
+++ b/src/main/kotlin/com/looker/droidify/database/DAOs.kt
@@ -272,6 +272,9 @@ interface InstalledDao : BaseDao<Installed> {
         installed.forEach { insertReplace(it) }
     }
 
+    @get:Query("SELECT * FROM memory_installed")
+    val allLive: LiveData<List<Installed>>
+
     @Query("SELECT * FROM memory_installed WHERE package_name = :packageName")
     fun get(packageName: String): Installed?
 
diff --git a/src/main/kotlin/com/looker/droidify/ui/viewmodels/MainNavFragmentViewModelX.kt b/src/main/kotlin/com/looker/droidify/ui/viewmodels/MainNavFragmentViewModelX.kt
index cced33b3..b41e89e9 100644
--- a/src/main/kotlin/com/looker/droidify/ui/viewmodels/MainNavFragmentViewModelX.kt
+++ b/src/main/kotlin/com/looker/droidify/ui/viewmodels/MainNavFragmentViewModelX.kt
@@ -99,6 +99,15 @@ class MainNavFragmentViewModelX(
                 }
             }
         }
+        if (secondarySource == Source.UPDATES) {
+            secondaryProducts.addSource(db.installedDao.allLive) {
+                viewModelScope.launch {
+                    withContext(Dispatchers.IO) {
+                        secondaryProducts.postValue(db.productDao.queryObject(secondaryRequest))
+                    }
+                }
+            }
+        }
     }
 
     fun setSection(newSection: Section) {