Show dialog messages when restoring (#1247)

RevenueCat · Sep 13, 2023 · d164409 · d164409
1 parent 6f3d8ae
commit d164409
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 12 deletions.
diff --git a/...ll-tester/src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreen.kt b/...ll-tester/src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreen.kt
@@ -37,10 +37,8 @@ fun PaywallScreen(
                         .setListener(viewModel)
                         .build(),
                 )
-                if (state.displayCompletedPurchaseMessage) {
-                    PurchaseAlertDialog(viewModel, "Purchase was successful")
-                } else if (state.displayErrorPurchasingMessage) {
-                    PurchaseAlertDialog(viewModel, "There was a purchasing error")
+                state?.dialogText?.let {
+                    PurchaseAlertDialog(viewModel, it)
                 }
             }
         }

diff --git a/...ster/src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreenState.kt b/...ster/src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreenState.kt
@@ -7,7 +7,6 @@ sealed class PaywallScreenState {
     data class Error(val errorMessage: String) : PaywallScreenState()
     data class Loaded(
         val offering: Offering,
-        val displayCompletedPurchaseMessage: Boolean = false,
-        val displayErrorPurchasingMessage: Boolean = false,
+        val dialogText: String? = null,
     ) : PaywallScreenState()
 }
diff --git a/.../src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreenViewModel.kt b/.../src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreenViewModel.kt
@@ -39,13 +39,45 @@ class PaywallScreenViewModelImpl(
         updateOffering()
     }
 
+    override fun onRestoreStarted() {
+        val value = _state.value
+        if (value is PaywallScreenState.Loaded) {
+            _state.update {
+                value.copy(
+                    dialogText = "Restoring purchases...",
+                )
+            }
+        }
+    }
+
+    override fun onRestoreCompleted(customerInfo: CustomerInfo) {
+        val value = _state.value
+        if (value is PaywallScreenState.Loaded) {
+            _state.update {
+                value.copy(
+                    dialogText = "Restore completed",
+                )
+            }
+        }
+    }
+
+    override fun onRestoreError(error: PurchasesError) {
+        val value = _state.value
+        if (value is PaywallScreenState.Loaded) {
+            _state.update {
+                value.copy(
+                    dialogText = "There was an error restoring purchases",
+                )
+            }
+        }
+    }
+
     override fun onPurchaseCompleted(customerInfo: CustomerInfo, storeTransaction: StoreTransaction) {
         val value = _state.value
         if (value is PaywallScreenState.Loaded) {
             _state.update {
                 value.copy(
-                    displayCompletedPurchaseMessage = true,
-                    displayErrorPurchasingMessage = false,
+                    dialogText = "Purchase was successful",
                 )
             }
         }
@@ -56,8 +88,7 @@ class PaywallScreenViewModelImpl(
         if (value is PaywallScreenState.Loaded) {
             _state.update {
                 value.copy(
-                    displayCompletedPurchaseMessage = false,
-                    displayErrorPurchasingMessage = true,
+                    dialogText = "There was an error purchasing",
                 )
             }
         }
@@ -68,8 +99,7 @@ class PaywallScreenViewModelImpl(
         if (value is PaywallScreenState.Loaded) {
             _state.update {
                 value.copy(
-                    displayCompletedPurchaseMessage = false,
-                    displayErrorPurchasingMessage = false,
+                    dialogText = null,
                 )
             }
         }