From dd5c505f790c18364e79851e9c170fa3cdffab09 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marek=20Fi=C5=A1era?= <mara@neptuo.com>
Date: Sun, 16 May 2021 09:12:41 +0200
Subject: [PATCH] #343 - Handle income created / delete events on Summary.

---
 src/Money.Blazor.Host/Pages/Summary.razor.cs | 39 ++++++++++++++++----
 1 file changed, 31 insertions(+), 8 deletions(-)

diff --git a/src/Money.Blazor.Host/Pages/Summary.razor.cs b/src/Money.Blazor.Host/Pages/Summary.razor.cs
index e2719f04..76882a94 100644
--- 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))