commit 766cac16af225de5fa90598b28bbcab98fe49ed9
author George Lin <giolin@google.com> Fri Nov 01 21:12:02 2024 +0000
committer Android (Google) Code Review <android-gerrit@google.com> Fri Nov 01 21:12:02 2024 +0000
tree 298da923cc9818b45f135e771fe3dd53ecbdc479
parent 1a8461e5f8d9137fc442c33d5764f9ac81b874e5
parent 15aec091f66b09b487a232b70cdab74fe7a91a7e

Merge "Only go back when on complete apply" into main

src/com/android/wallpaper/customization/ui/binder/ThemePickerToolbarBinder.kt

diff --git a/src/com/android/wallpaper/customization/ui/binder/ThemePickerToolbarBinder.kt b/src/com/android/wallpaper/customization/ui/binder/ThemePickerToolbarBinder.kt
index 6282842..18b9584 100644
--- a/src/com/android/wallpaper/customization/ui/binder/ThemePickerToolbarBinder.kt
+++ b/src/com/android/wallpaper/customization/ui/binder/ThemePickerToolbarBinder.kt
@@ -64,12 +64,7 @@
             lifecycleOwner.repeatOnLifecycle(Lifecycle.State.STARTED) {
                 launch {
                     viewModel.onApplyButtonClicked.collect { onApplyButtonClicked ->
-                        applyButton.setOnClickListener {
-                            if (onApplyButtonClicked != null) {
-                                onApplyButtonClicked.invoke()
-                                onNavBack.invoke()
-                            }
-                        }
+                        applyButton.setOnClickListener { onApplyButtonClicked?.invoke(onNavBack) }
                     }
                 }
 

src/com/android/wallpaper/customization/ui/viewmodel/ThemePickerCustomizationOptionsViewModel.kt

diff --git a/src/com/android/wallpaper/customization/ui/viewmodel/ThemePickerCustomizationOptionsViewModel.kt b/src/com/android/wallpaper/customization/ui/viewmodel/ThemePickerCustomizationOptionsViewModel.kt
index 2e500e2..b2e2039 100644
--- a/src/com/android/wallpaper/customization/ui/viewmodel/ThemePickerCustomizationOptionsViewModel.kt
+++ b/src/com/android/wallpaper/customization/ui/viewmodel/ThemePickerCustomizationOptionsViewModel.kt
@@ -131,7 +131,7 @@
         }
 
     @OptIn(ExperimentalCoroutinesApi::class)
-    val onApplyButtonClicked =
+    val onApplyButtonClicked: Flow<((onComplete: () -> Unit) -> Unit)?> =
         selectedOption
             .flatMapLatest {
                 when (it) {
@@ -155,8 +155,11 @@
             }
             .map { onApply ->
                 if (onApply != null) {
-                    fun() {
-                        viewModelScope.launch { onApply() }
+                    fun(onComplete: () -> Unit) {
+                        viewModelScope.launch {
+                            onApply()
+                            onComplete()
+                        }
                     }
                 } else {
                     null