fix: edit widget action

Author: jvsena42

app/src/main/java/to/bitkit/ui/screens/wallets/HomeScreen.kt

@@ -237,8 +237,7 @@ fun HomeScreen(
                 rootNavController.navigate(Routes.AddWidget)
             }
         },
-        onClickEnableEdit = homeViewModel::enableEditMode,
-        onClickConfirmEdit = homeViewModel::confirmWidgetOrder,
+        onClickConfirmEdit = homeViewModel::onClickEditWidget,
         onClickEditWidget = { widgetType ->
             when (widgetType) {
                 WidgetType.BLOCK -> rootNavController.navigate(Routes.BlocksPreview)
@@ -276,7 +275,6 @@ private fun Content(
     onRemoveSuggestion: (Suggestion) -> Unit = {},
     onClickSuggestion: (Suggestion) -> Unit = {},
     onClickAddWidget: () -> Unit = {},
-    onClickEnableEdit: () -> Unit = {},
     onClickConfirmEdit: () -> Unit = {},
     onClickEditWidget: (WidgetType) -> Unit = {},
     onClickDeleteWidget: (WidgetType) -> Unit = {},

app/src/main/java/to/bitkit/ui/screens/wallets/HomeViewModel.kt

@@ -205,14 +205,6 @@ class HomeViewModel @Inject constructor(
         }
     }
 
-    fun enableEditMode() {
-        _uiState.update { it.copy(isEditingWidgets = true) }
-    }
-
-    fun disableEditMode() {
-        _uiState.update { it.copy(isEditingWidgets = false) }
-    }
-
     fun moveWidget(fromIndex: Int, toIndex: Int) {
         val currentWidgets = _uiState.value.widgetsWithPosition.toMutableList()
         if (fromIndex in currentWidgets.indices && toIndex in currentWidgets.indices) {
@@ -228,11 +220,15 @@ class HomeViewModel @Inject constructor(
         }
     }
 
-    fun confirmWidgetOrder() {
-        viewModelScope.launch {
-            val widgets = _uiState.value.widgetsWithPosition
-            widgetsRepo.updateWidgets(widgets)
-            disableEditMode()
+    fun onClickEditWidget() {
+        if (_uiState.value.isEditingWidgets) {
+            viewModelScope.launch {
+                val widgets = _uiState.value.widgetsWithPosition
+                widgetsRepo.updateWidgets(widgets)
+                disableEditMode()
+            }
+        } else {
+            enableEditMode()
         }
     }
 
@@ -255,6 +251,14 @@ class HomeViewModel @Inject constructor(
         }
     }
 
+    private fun enableEditMode() {
+        _uiState.update { it.copy(isEditingWidgets = true) }
+    }
+
+    private fun disableEditMode() {
+        _uiState.update { it.copy(isEditingWidgets = false) }
+    }
+
     private fun createSuggestionsFlow() = combine(
         walletRepo.balanceState,
         settingsStore.data,