Add onUpdateInstalled

Infomaniak · Dec 6, 2023 · af6dc75 · af6dc75
1 parent 0b0fade
commit af6dc75
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/Stores/src/fdroid/java/com.infomaniak.lib.stores/StoreUtils.kt b/Stores/src/fdroid/java/com.infomaniak.lib.stores/StoreUtils.kt
@@ -37,7 +37,7 @@ object StoreUtils {
     //endRegion
 
     //region In-App Update
-    fun initAppUpdateManager(context: Context, onInstall: () -> Unit) = Unit
+    fun initAppUpdateManager(context: Context, onUpdateDownloaded: () -> Unit, onUpdateInstalled: () -> Unit) = Unit
 
     fun FragmentActivity.checkUpdateIsAvailable(
         appId: String,

diff --git a/Stores/src/standard/java/com.infomaniak.lib.stores/StoreUtils.kt b/Stores/src/standard/java/com.infomaniak.lib.stores/StoreUtils.kt
@@ -36,14 +36,18 @@ object StoreUtils {
     private const val UPDATE_TYPE = AppUpdateType.FLEXIBLE
 
     private lateinit var appUpdateManager: AppUpdateManager
-    private lateinit var onInstallDownloaded: () -> Unit
+    private lateinit var onUpdateDownloaded: () -> Unit
+    private lateinit var onUpdateInstalled: () -> Unit
 
     // Create a listener to track request state updates.
     private val installStateUpdatedListener by lazy {
         InstallStateUpdatedListener { state ->
             when (state.installStatus()) {
-                InstallStatus.DOWNLOADED -> onInstallDownloaded()
-                InstallStatus.INSTALLED -> unregisterAppUpdateListener()
+                InstallStatus.DOWNLOADED -> onUpdateDownloaded()
+                InstallStatus.INSTALLED -> {
+                    onUpdateInstalled()
+                    unregisterAppUpdateListener()
+                }
                 else -> Unit
             }
         }
@@ -62,9 +66,10 @@ object StoreUtils {
     //endRegion
 
     //region In-App Update
-    fun initAppUpdateManager(context: Context, onInstall: () -> Unit) {
+    fun initAppUpdateManager(context: Context, onUpdateDownloaded: () -> Unit, onUpdateInstalled: () -> Unit) {
         appUpdateManager = AppUpdateManagerFactory.create(context)
-        onInstallDownloaded = onInstall
+        this.onUpdateDownloaded = onUpdateDownloaded
+        this.onUpdateInstalled = onUpdateInstalled
     }
 
     fun FragmentActivity.checkUpdateIsAvailable(
@@ -87,7 +92,7 @@ object StoreUtils {
         appUpdateInfo.addOnSuccessListener { appUpdateInfo ->
             if (appUpdateInfo.installStatus() == InstallStatus.DOWNLOADED) {
                 // If the update is downloaded but not installed, notify the user to complete the update.
-                onInstallDownloaded.invoke()
+                onUpdateDownloaded.invoke()
             }
         }
     }