fix(mobile): unset default budget (#338)

Fixes #334
get6pm · Sep 19, 2024 · ad7a19f · ad7a19f
1 parent 55c43cf
commit ad7a19f
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/apps/mobile/app/(app)/budget/[budgetId]/edit.tsx b/apps/mobile/app/(app)/budget/[budgetId]/edit.tsx
@@ -22,7 +22,7 @@ export default function EditBudgetScreen() {
   const { i18n } = useLingui()
   const { budgetId } = useLocalSearchParams<{ budgetId: string }>()
   const { budget } = useBudget(budgetId!)
-  const { mutateAsync } = useUpdateBudget()
+  const { mutateAsync: updateUpdate } = useUpdateBudget()
   const { mutateAsync: mutateDelete } = useDeleteBudget()
   const { setDefaultBudgetId, defaultBudgetId } = useUserMetadata()
   const { sideOffset, ...rootProps } = useModalPortalRoot()
@@ -68,10 +68,14 @@ export default function EditBudgetScreen() {
   )
 
   const handleUpdate = async ({ isDefault, ...data }: BudgetFormValues) => {
-    if (isDefault) {
-      await setDefaultBudgetId(budget?.id)
+    if (isDefault && defaultBudgetId !== undefined) {
+      // only set if changed
+      setDefaultBudgetId(budget?.id)
+    } else if (defaultBudgetId === budget?.id) {
+      // unset default budget
+      setDefaultBudgetId(undefined)
     }
-    mutateAsync({
+    updateUpdate({
       data: data,
       id: budget?.id!,
     }).catch(() => {