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/Category.hs | 56 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 src/server/Model/Category.hs

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

diff --git a/src/server/Model/Category.hs b/src/server/Model/Category.hs
new file mode 100644
index 0000000..50c3622
--- /dev/null
+++ b/src/server/Model/Category.hs
@@ -0,0 +1,56 @@
+module Model.Category
+  ( list
+  , create
+  , edit
+  , delete
+  ) where
+
+import Data.Text (Text)
+import Data.Maybe (isJust)
+import Data.Time.Clock (getCurrentTime)
+
+import Control.Monad.IO.Class (liftIO)
+
+import Database.Persist hiding (delete)
+
+import Model.Database
+import qualified Model.Json.Category as Json
+
+list :: Persist [Json.Category]
+list = map getJsonCategory <$> selectList [ CategoryDeletedAt ==. Nothing ] []
+
+getJsonCategory :: Entity Category -> Json.Category
+getJsonCategory categoryEntity =
+  Json.Category (entityKey categoryEntity) (categoryName category) (categoryColor category)
+  where category = entityVal categoryEntity
+
+create :: Text -> Text -> Persist CategoryId
+create name color = do
+  now <- liftIO getCurrentTime
+  insert (Category name color now Nothing Nothing)
+
+edit :: CategoryId -> Text -> Text -> Persist Bool
+edit categoryId name color = do
+  mbCategory <- get categoryId
+  if isJust mbCategory
+    then do
+      now <- liftIO getCurrentTime
+      update categoryId
+        [ CategoryEditedAt =. Just now
+        , CategoryName =. name
+        , CategoryColor =. color
+        ]
+      return True
+    else
+      return False
+
+delete :: CategoryId -> Persist Bool
+delete categoryId = do
+  mbCategory <- get categoryId
+  if isJust mbCategory
+    then do
+      now <- liftIO getCurrentTime
+      update categoryId [CategoryDeletedAt =. Just now]
+      return True
+    else
+      return False
-- 
cgit v1.2.3