From fc9a3ca6a5226f78a3a0ec51af3d8d287ccb2e1b Mon Sep 17 00:00:00 2001
From: Joris
Date: Sat, 2 Jan 2016 22:45:20 +0100
Subject: Merge deletePayment with deleteMonthlyPayment

---
 src/client/elm/Update/LoggedIn.elm | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

(limited to 'src/client/elm/Update/LoggedIn.elm')

diff --git a/src/client/elm/Update/LoggedIn.elm b/src/client/elm/Update/LoggedIn.elm
index ca4690b..704dd7f 100644
--- a/src/client/elm/Update/LoggedIn.elm
+++ b/src/client/elm/Update/LoggedIn.elm
@@ -44,12 +44,18 @@ updateLoggedIn model action loggedInView =
                }
     ToggleEdit id ->
       { loggedInView | paymentEdition = if loggedInView.paymentEdition == Just id then Nothing else Just id }
-    DeletePayment payment ->
-      { loggedInView
-      | account = updateAccount (UpdatePayer payment.userId (Date.toTime payment.creation) -payment.cost) loggedInView.account
-      , payments = deletePayment payment.id loggedInView.payments
-      , paymentsCount = loggedInView.paymentsCount - 1
-      }
+    DeletePayment payment frequency ->
+      case frequency of
+        Monthly ->
+          { loggedInView
+          | monthly = updateMonthly (Monthly.DeletePayment payment) loggedInView.monthly
+          }
+        Punctual ->
+          { loggedInView
+          | account = updateAccount (UpdatePayer payment.userId (Date.toTime payment.creation) -payment.cost) loggedInView.account
+          , payments = deletePayment payment.id loggedInView.payments
+          , paymentsCount = loggedInView.paymentsCount - 1
+          }
     UpdatePage page ->
       { loggedInView
       | currentPage = page
-- 
cgit v1.2.3