#343 - Handle income created / delete events on Summary.

maraf · May 16, 2021 · dd5c505 · dd5c505
1 parent 25b998b
commit dd5c505
Showing 1 changed file with 31 additions and 8 deletions.
diff --git a/src/Money.Blazor.Host/Pages/Summary.razor.cs b/src/Money.Blazor.Host/Pages/Summary.razor.cs
@@ -30,7 +30,9 @@ public partial class Summary<T> :
         IEventHandler<OutcomeDeleted>,
         IEventHandler<OutcomeAmountChanged>,
         IEventHandler<OutcomeWhenChanged>,
-        IEventHandler<PulledToRefresh>
+        IEventHandler<PulledToRefresh>,
+        IEventHandler<IncomeCreated>,
+        IEventHandler<IncomeDeleted>
     {
         private CurrencyFormatter formatter;
 
@@ -118,11 +120,7 @@ protected async Task LoadSelectedPeriodAsync()
                     {
                         Categories = await Queries.QueryAsync(CreateCategoriesQuery(SelectedPeriod));
 
-                        var incomeQuery = CreateIncomeTotalQuery(SelectedPeriod);
-                        if (incomeQuery != null)
-                            IncomeTotal = await Queries.QueryAsync(CreateIncomeTotalQuery(SelectedPeriod));
-                        else
-                            IncomeTotal = null;
+                        await LoadIncomeTotalAsync();
 
                         ExpenseTotal = await Queries.QueryAsync(CreateExpenseTotalQuery(SelectedPeriod));
                     }
@@ -136,6 +134,15 @@ protected async Task LoadSelectedPeriodAsync()
             }
         }
 
+        private async Task LoadIncomeTotalAsync()
+        {
+            var incomeQuery = CreateIncomeTotalQuery(SelectedPeriod);
+            if (incomeQuery != null)
+                IncomeTotal = await Queries.QueryAsync(CreateIncomeTotalQuery(SelectedPeriod));
+            else
+                IncomeTotal = null;
+        }
+
         protected virtual IQuery<Price> CreateIncomeTotalQuery(T item)
             => null;
 
@@ -203,7 +210,9 @@ private void BindEvents()
                 .Add<OutcomeDeleted>(this)
                 .Add<OutcomeAmountChanged>(this)
                 .Add<OutcomeWhenChanged>(this)
-                .Add<PulledToRefresh>(this);
+                .Add<PulledToRefresh>(this)
+                .Add<IncomeCreated>(this)
+                .Add<IncomeDeleted>(this);
         }
 
         private void UnBindEvents()
@@ -213,7 +222,9 @@ private void UnBindEvents()
                 .Remove<OutcomeDeleted>(this)
                 .Remove<OutcomeAmountChanged>(this)
                 .Remove<OutcomeWhenChanged>(this)
-                .Remove<PulledToRefresh>(this);
+                .Remove<PulledToRefresh>(this)
+                .Remove<IncomeCreated>(this)
+                .Remove<IncomeDeleted>(this);
         }
 
         Task IEventHandler<OutcomeCreated>.HandleAsync(OutcomeCreated payload)
@@ -247,6 +258,18 @@ Task IEventHandler<PulledToRefresh>.HandleAsync(PulledToRefresh payload)
             return Task.CompletedTask;
         }
 
+        async Task IEventHandler<IncomeCreated>.HandleAsync(IncomeCreated payload)
+        {
+            await LoadIncomeTotalAsync();
+            StateHasChanged();
+        }
+
+        async Task IEventHandler<IncomeDeleted>.HandleAsync(IncomeDeleted payload)
+        {
+            await LoadIncomeTotalAsync();
+            StateHasChanged();
+        }
+
         private async void OnMonthUpdatedEvent(DateTime changed)
         {
             if (Periods != null && !IsContained(changed))