From cfca18262c1ff48dcb683ddab7d03cf8e55573ff Mon Sep 17 00:00:00 2001
From: Joris
Date: Fri, 24 Mar 2017 09:21:04 +0000
Subject: Features/categories

---
 src/server/Model/Init.hs | 25 ++++++++++++-------------
 1 file changed, 12 insertions(+), 13 deletions(-)

(limited to 'src/server/Model/Init.hs')

diff --git a/src/server/Model/Init.hs b/src/server/Model/Init.hs
index 09ac627..7610b25 100644
--- a/src/server/Model/Init.hs
+++ b/src/server/Model/Init.hs
@@ -10,22 +10,21 @@ import Database.Persist
 
 import Model.Database
 
-import Model.Json.Init (Init, Init(Init))
+import Model.Json.Init (Init)
 import qualified Model.Payment as Payment
-import Model.User (getUsers, getJsonUser)
-import Model.Income (getIncomes, getJsonIncome)
+import qualified Model.User as User
+import qualified Model.Income as Income
+import qualified Model.Category as Category
+import qualified Model.PaymentCategory as PaymentCategory
 
 import qualified Model.Json.Init as Init
 
 getInit :: Entity User -> Persist Init
 getInit user =
-  liftIO . runDb $ do
-    users <- getUsers
-    payments <- Payment.list
-    incomes <- getIncomes
-    return $ Init
-      { Init.users = map getJsonUser users
-      , Init.me = entityKey user
-      , Init.payments = payments
-      , Init.incomes = map getJsonIncome incomes
-      }
+  liftIO . runDb $ Init.Init <$>
+    (map User.getJson <$> User.list) <*>
+    (return . entityKey $ user) <*>
+    Payment.list <*>
+    Income.list <*>
+    Category.list <*>
+    PaymentCategory.list
-- 
cgit v1.2.3