eidolon/Handler/NewAlbum.hs

86 lines
3.1 KiB
Haskell
Executable file

-- eidolon -- A simple gallery in Haskell and Yesod
-- Copyright (C) 2015 Amedeo Molnár
--
-- This program is free software: you can redistribute it and/or modify
-- it under the terms of the GNU Affero General Public License as published
-- by the Free Software Foundation, either version 3 of the License, or
-- (at your option) any later version.
--
-- This program is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU Affero General Public License for more details.
--
-- You should have received a copy of the GNU Affero General Public License
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
module Handler.NewAlbum where
import Import as I
import Data.List as L (head)
import Data.Text as T
import Text.Blaze (toMarkup)
import System.Directory
import System.FilePath
getNewAlbumR :: Handler Html
getNewAlbumR = do
musername <- maybeAuthId
case musername of
Just username -> do
(Just (Entity userId _)) <- runDB $ getBy $ UniqueUser username
(albumWidget, enctype) <- generateFormPost $
renderBootstrap3 BootstrapBasicForm $
albumForm userId
defaultLayout $ do
setTitle "Eidolon :: Create new Album"
$(widgetFile "newAlbum")
Nothing -> do
setMessage "You need to be logged in"
redirect $ AuthR LoginR
postNewAlbumR :: Handler Html
postNewAlbumR = do
musername <- maybeAuthId
case musername of
Just username -> do
(Just (Entity userId _)) <- runDB $ getBy $ UniqueUser username
((result, _), _) <- runFormPost $
renderBootstrap3 BootstrapBasicForm $
albumForm userId
case result of
FormSuccess album -> do
namesakes <- runDB $ selectList [AlbumTitle ==. albumTitle album, AlbumOwner ==. userId] []
if I.null namesakes
then do
-- Put album in Database
albumId <- runDB $ insert album
-- add album reference in user
user <- runDB $ getJust userId
let albumList = userAlbums user
let newAlbumList = albumId : albumList
runDB $ update userId [UserAlbums =. newAlbumList]
-- create folder
liftIO $ createDirectoryIfMissing True $ "static" </> "data" </> unpack (extractKey userId) </> unpack (extractKey albumId)
-- outro
setMessage "Album successfully created"
redirect $ AlbumR albumId
else do
setMessage $ toMarkup $ "You already have an album named " `T.append`
albumTitle album
redirect $ AlbumR $ entityKey $ L.head namesakes
_ -> do
setMessage "There was an error creating the album"
redirect NewAlbumR
Nothing -> do
setMessage "You must be logged in to create albums"
redirect $ AuthR LoginR
albumForm :: UserId -> AForm Handler Album
albumForm userId = Album
<$> areq textField (bfs ("Title" :: T.Text)) Nothing
<*> pure userId
<*> pure []
<*> pure []
<*> pure Nothing
<* bootstrapSubmit ("Create album" :: BootstrapSubmit Text)