fix: make generating budget from year work for both income and expenses

simonwep · Nov 14, 2024 · 9aa8ed0 · 9aa8ed0
1 parent 54b5841
commit 9aa8ed0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/store/state/utils.ts b/src/store/state/utils.ts
@@ -20,12 +20,15 @@ export const generateBudgetYear = (year: number): BudgetYear => ({
 });
 
 export const generateBudgetYearFromCurrent = (year: number, current: BudgetYear): BudgetYear => {
-  current.income = current.income.map((value) => {
+  function clearValues(value: BudgetGroup) {
     for (const budget of value.budgets) {
       budget.values = new Array(12).fill(0);
     }
     return value;
-  });
+  }
+
+  current.income = current.income.map(clearValues);
+  current.expenses = current.expenses.map(clearValues);
 
   return {
     year,