Show dialog after completed or error purchasing

examples/paywall-tester/src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreen.kt

@@ -3,7 +3,9 @@ package com.revenuecat.paywallstester.ui.screens.paywall
 import androidx.compose.foundation.layout.Arrangement
 import androidx.compose.foundation.layout.Column
 import androidx.compose.foundation.layout.fillMaxSize
+import androidx.compose.material.AlertDialog
 import androidx.compose.material.Text
+import androidx.compose.material.TextButton
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.collectAsState
 import androidx.compose.ui.Alignment
@@ -29,8 +31,40 @@ fun PaywallScreen(
                 Text(text = "Error: ${state.errorMessage}")
             }
             is PaywallScreenState.Loaded -> {
-                PaywallView(PaywallViewOptions.Builder().setOffering(state.offering).build())
+                PaywallView(
+                    PaywallViewOptions.Builder()
+                        .setOffering(state.offering)
+                        .setListener(viewModel)
+                        .build(),
+                )
+                state?.dialogText?.let {
+                    PurchaseAlertDialog(viewModel, it)
+                }
             }
         }
     }
 }
+
+@Composable
+private fun PurchaseAlertDialog(
+    viewModel: PaywallScreenViewModel,
+    text: String,
+) {
+    AlertDialog(
+        onDismissRequest = {
+            viewModel.onDialogDismissed()
+        },
+        buttons = {
+            TextButton(
+                onClick = {
+                    viewModel.onDialogDismissed()
+                },
+            ) {
+                Text("Ok")
+            }
+        },
+        text = {
+            Text(text)
+        },
+    )
+}

examples/paywall-tester/src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreenState.kt

@@ -5,5 +5,8 @@ import com.revenuecat.purchases.Offering
 sealed class PaywallScreenState {
     object Loading : PaywallScreenState()
     data class Error(val errorMessage: String) : PaywallScreenState()
-    data class Loaded(val offering: Offering) : PaywallScreenState()
+    data class Loaded(
+        val offering: Offering,
+        val dialogText: String? = null,
+    ) : PaywallScreenState()
 }

examples/paywall-tester/src/main/java/com/revenuecat/paywallstester/ui/screens/paywall/PaywallScreenViewModel.kt

@@ -4,20 +4,26 @@ import android.app.Application
 import androidx.lifecycle.AndroidViewModel
 import androidx.lifecycle.SavedStateHandle
 import androidx.lifecycle.viewModelScope
+import com.revenuecat.purchases.CustomerInfo
 import com.revenuecat.purchases.Purchases
+import com.revenuecat.purchases.PurchasesError
 import com.revenuecat.purchases.PurchasesException
 import com.revenuecat.purchases.awaitOfferings
+import com.revenuecat.purchases.models.StoreTransaction
+import com.revenuecat.purchases.ui.revenuecatui.PaywallViewListener
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
 
-interface PaywallScreenViewModel {
+interface PaywallScreenViewModel : PaywallViewListener {
     companion object {
         const val OFFERING_ID_KEY = "offering_id"
     }
     val state: StateFlow<PaywallScreenState>
+
+    fun onDialogDismissed()
 }
 
 class PaywallScreenViewModelImpl(
@@ -35,6 +41,72 @@ 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:\n${error.message}",
+                )
+            }
+        }
+    }
+
+    override fun onPurchaseCompleted(customerInfo: CustomerInfo, storeTransaction: StoreTransaction) {
+        val value = _state.value
+        if (value is PaywallScreenState.Loaded) {
+            _state.update {
+                value.copy(
+                    dialogText = "Purchase was successful",
+                )
+            }
+        }
+    }
+
+    override fun onPurchaseError(error: PurchasesError) {
+        val value = _state.value
+        if (value is PaywallScreenState.Loaded) {
+            _state.update {
+                value.copy(
+                    dialogText = "There was an error purchasing:\n${error.message}",
+                )
+            }
+        }
+    }
+
+    override fun onDialogDismissed() {
+        val value = _state.value
+        if (value is PaywallScreenState.Loaded) {
+            _state.update {
+                value.copy(
+                    dialogText = null,
+                )
+            }
+        }
+    }
+
     private fun updateOffering() {
         viewModelScope.launch {
             try {