Use a better project structure

1 files changed, 58 insertions, 0 deletions

diff --git a/server/src/Controller/Payment.hs b/server/src/Controller/Payment.hs
new file mode 100644
index 0000000..dc10311
--- /dev/null
+++ b/server/src/Controller/Payment.hs
@@ -0,0 +1,58 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module Controller.Payment
+  ( list
+  , create
+  , editOwn
+  , deleteOwn
+  ) where
+
+import Control.Monad.IO.Class (liftIO)
+import Network.HTTP.Types.Status (ok200, badRequest400)
+import Web.Scotty
+
+import Common.Model (PaymentId, User(..), CreatePayment(..), EditPayment(..))
+
+import Json (jsonId)
+import qualified Model.Payment as Payment
+import qualified Model.PaymentCategory as PaymentCategory
+import qualified Model.Query as Query
+import qualified Secure
+
+list :: ActionM ()
+list =
+  Secure.loggedAction (\_ ->
+    (liftIO . Query.run $ Payment.list) >>= json
+  )
+
+create :: CreatePayment -> ActionM ()
+create (CreatePayment name cost date category frequency) =
+  Secure.loggedAction (\user ->
+    (liftIO . Query.run $ do
+      PaymentCategory.save name category
+      Payment.create (_user_id user) name cost date frequency
+    ) >>= jsonId
+  )
+
+editOwn :: EditPayment -> ActionM ()
+editOwn (EditPayment paymentId name cost date category frequency) =
+  Secure.loggedAction (\user -> do
+    updated <- liftIO . Query.run $ do
+      edited <- Payment.editOwn (_user_id user) paymentId name cost date frequency
+      _ <- if edited
+        then PaymentCategory.save name category >> return ()
+        else return ()
+      return edited
+    if updated
+      then status ok200
+      else status badRequest400
+  )
+
+deleteOwn :: PaymentId -> ActionM ()
+deleteOwn paymentId =
+  Secure.loggedAction (\user -> do
+    deleted <- liftIO . Query.run $ Payment.deleteOwn (_user_id user) paymentId
+    if deleted
+      then status ok200
+      else status badRequest400
+  )