fix(mobile): unset default bugdet

get6pm · Sep 19, 2024 · fecf889 · fecf889
1 parent 59c3ae0
commit fecf889
Showing 1 changed file with 6 additions and 3 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()
@@ -69,9 +69,12 @@ export default function EditBudgetScreen() {
 
   const handleUpdate = async ({ isDefault, ...data }: BudgetFormValues) => {
     if (isDefault) {
-      await setDefaultBudgetId(budget?.id)
+      setDefaultBudgetId(budget?.id)
+    } else if (defaultBudgetId === budget?.id) {
+      // unset default budget
+      setDefaultBudgetId(undefined)
     }
-    mutateAsync({
+    updateUpdate({
       data: data,
       id: budget?.id!,
     }).catch(() => {