From b3d11085817255f8f6ab3b6e81be8aa02f08126a Mon Sep 17 00:00:00 2001
From: nek0 <neko.rloove@rocketmail.com>
Date: Mon, 8 Sep 2014 01:46:47 +0200
Subject: [PATCH] admin can now change album settings

---
 Handler/AdminAlbumSettings.hs  | 106 ++++++++++++++++++++++++++++++++-
 templates/adminAlbumSet.hamlet |   9 +++
 2 files changed, 112 insertions(+), 3 deletions(-)
 create mode 100644 templates/adminAlbumSet.hamlet

diff --git a/Handler/AdminAlbumSettings.hs b/Handler/AdminAlbumSettings.hs
index 7de9bde..f81bf9f 100644
--- a/Handler/AdminAlbumSettings.hs
+++ b/Handler/AdminAlbumSettings.hs
@@ -1,6 +1,9 @@
 module Handler.AdminAlbumSettings where
 
 import Import
+import qualified Data.Text as T
+import System.FilePath
+import System.Directory
 
 getAdminAlbumsR :: Handler Html
 getAdminAlbumsR = do
@@ -22,10 +25,107 @@ getAdminAlbumsR = do
       redirect $ LoginR
 
 getAdminAlbumSettingsR :: AlbumId -> Handler Html
-getAdminAlbumSettingsR = error "Not yet implemented: getAdminAlbumSettingsR"
+getAdminAlbumSettingsR albumId = do
+  tempAlbum <- runDB $ get albumId
+  case tempAlbum of
+    Just album -> do
+      msu <- lookupSession "userId"
+      case msu of
+        Just tempUserId -> do
+          userId <- return $ getUserIdFromText tempUserId
+          user <- runDB $ getJust userId
+          case userAdmin user of
+            True -> do
+              (adminAlbumSettingsWidget, enctype) <- generateFormPost $ adminAlbumSettingsForm album albumId
+              defaultLayout $ do
+                $(widgetFile "adminAlbumSet")
+            False -> do
+              setMessage "You must be admin"
+              redirect $ HomeR
+        Nothing -> do
+          setMessage "You must be logged in"
+          redirect $ LoginR
+    Nothing -> do
+      setMessage "This album does not exist"
+      redirect $ AdminR
 
 postAdminAlbumSettingsR :: AlbumId -> Handler Html
-postAdminAlbumSettingsR = error "Not yet implemented: postAdminAlbumSettingsR"
+postAdminAlbumSettingsR albumId = do
+  tempAlbum <- runDB $ get albumId
+  case tempAlbum of
+    Just album -> do
+      msu <- lookupSession "userId"
+      case msu of
+        Just tempUserId -> do
+          userId <- return $ getUserIdFromText tempUserId
+          user <- runDB $ getJust userId
+          case userAdmin user of
+            True -> do
+              ((res, adminAlbumSettingsWidget), enctype) <- runFormPost $ adminAlbumSettingsForm album albumId
+              case res of
+                FormSuccess temp -> do
+                  aId <- runDB $ update albumId
+                    [ AlbumTitle =. albumTitle temp
+                    , AlbumSamplePic =. albumSamplePic temp
+                    ]
+                  setMessage "Album settings changed successfully"
+                  redirect $ AdminR
+                _ -> do
+                  setMessage "There was an error while changing the settings"
+                  redirect $ AdminAlbumSettingsR albumId
+            False -> do
+              setMessage "You must be admin"
+              redirect $ HomeR
+        Nothing -> do
+          setMessage "You must be logged in"
+          redirect $ LoginR
+    Nothing -> do
+      setMessage "This album does not exist"
+      redirect $ AdminR
+
+adminAlbumSettingsForm :: Album -> AlbumId -> Form Album
+adminAlbumSettingsForm album albumId = renderDivs $ Album
+  <$> areq textField "Title" (Just $ albumTitle album)
+  <*> pure (albumOwner album)
+  <*> pure (albumContent album)
+  <*> aopt (selectField media) "Sample picture" (Just $ albumSamplePic album)
+  where
+    media = do
+      entities <- runDB $ selectList [MediumAlbum ==. albumId] [Desc MediumTitle]
+      optionsPairs $ map (\med -> (mediumTitle $ entityVal med, mediumPath (entityVal med))) entities
 
 getAdminAlbumDeleteR :: AlbumId -> Handler Html
-getAdminAlbumDeleteR albumId = error "Not yet implemented: getAdminAlbumDeleteR" 
+getAdminAlbumDeleteR albumId = do
+  tempAlbum <- runDB $ get albumId
+  case tempAlbum of
+    Just album -> do
+      msu <- lookupSession "userId"
+      case msu of
+        Just tempUserId -> do
+          userId <- return $ getUserIdFromText tempUserId
+          user <- runDB $ getJust userId
+          case userAdmin user of
+            True -> do
+              -- remove reference from owner
+              ownerId <- return $ albumOwner album
+              owner <- runDB $ getJust ownerId
+              albumList <- return $ userAlbums owner
+              newAlbumList <- return $ removeItem albumId albumList
+              -- delete album content
+              mapM (\a -> runDB $ delete a) (albumContent album)
+              -- delete album
+              runDB $ delete albumId
+              -- delete files
+              liftIO $ removeDirectoryRecursive $ "static" </> "data" </> (T.unpack $ extractKey ownerId) </> (T.unpack $ extractKey albumId)
+              -- outro
+              setMessage "Album deleted successfully"
+              redirect $ AdminR
+            _ -> do
+              setMessage "You are no admin"
+              redirect $ HomeR
+        Nothing -> do
+          setMessage "You must be logged in"
+          redirect $ LoginR
+    Nothing -> do
+      setMessage "This Album does not exist"
+      redirect $ AdminR
diff --git a/templates/adminAlbumSet.hamlet b/templates/adminAlbumSet.hamlet
new file mode 100644
index 0000000..4ad6ef0
--- /dev/null
+++ b/templates/adminAlbumSet.hamlet
@@ -0,0 +1,9 @@
+$newline always
+<h3>Album settings
+
+<form method="post" enctype=#{enctype}>
+  ^{adminAlbumSettingsWidget}
+  <div>
+    <input type=submit value="Change settings">
+
+<a href=@{AdminAlbumDeleteR albumId}>Delete this album