Features/categories

1 files changed, 12 insertions, 13 deletions

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