vervis/src/Vervis/Handler/Key.hs

{- This file is part of Vervis.
 -
 - Written in 2016 by fr33domlover <fr33domlover@riseup.net>.
 -
 - ♡ Copying is an act of love. Please copy, reuse and share.
 -
 - The author(s) have dedicated all copyright and related and neighboring
 - rights to this software to the public domain worldwide. This software is
 - distributed without any warranty.
 -
 - You should have received a copy of the CC0 Public Domain Dedication along
 - with this software. If not, see
 - <http://creativecommons.org/publicdomain/zero/1.0/>.
 -}

module Vervis.Handler.Key
    ( getKeysR
    , postKeysR
    , getKeyNewR
    , getKeyR
    , deleteKeyR
    , postKeyR
    )
where

import Prelude

import Data.ByteString.Base64 (encode)
import Data.Monoid ((<>))
import Data.Text (Text, intercalate)
import Data.Text.Encoding (decodeUtf8With)
import Data.Text.Encoding.Error (lenientDecode)
import Database.Persist
import Text.Blaze.Html (Html, toHtml)
import Yesod.Core (defaultLayout)
import Yesod.Core.Handler
import Yesod.Core.Widget (setTitle)
import Yesod.Form.Functions (runFormPost)
import Yesod.Form.Types (FormResult (..))
import Yesod.Persist.Core (runDB, getBy404)

import Vervis.Form.Key
import Vervis.Foundation
import Vervis.Model
import Vervis.Settings

getKeysR :: Text -> Handler Html
getKeysR user = do
    keys <- runDB $ do
        Entity sid _sharer <- getBy404 $ UniqueSharerIdent user
        Entity pid _person <- getBy404 $ UniquePersonIdent sid
        ks <- selectList [SshKeyPerson ==. pid] [Asc SshKeyName]
        return $ map (\ (Entity _ k) -> sshKeyName k) ks
    defaultLayout $ do
        setTitle $ toHtml $
            intercalate " > " ["Vervis", "People", user, "Keys"]
        $(widgetFile "key/keys")

postKeysR :: Text -> Handler Html
postKeysR user = do
    pid <- runDB $ do
        Entity s _sharer <- getBy404 $ UniqueSharerIdent user
        Entity p _person <- getBy404 $ UniquePersonIdent s
        return p
    ((result, widget), enctype) <- runFormPost $ newKeyForm pid
    case result of
        FormSuccess key -> do
            runDB $ insert_ key
            setMessage "Key added."
            redirect $ KeysR user
        FormMissing -> do
            setMessage "Field(s) missing"
            defaultLayout $(widgetFile "key/key-new")
        FormFailure _l -> do
            setMessage "Invalid input, see below"
            defaultLayout $(widgetFile "key/key-new")

getKeyNewR :: Text -> Handler Html
getKeyNewR user = do
    pid <- runDB $ do
        Entity s _sharer <- getBy404 $ UniqueSharerIdent user
        Entity p _person <- getBy404 $ UniquePersonIdent s
        return p
    ((_result, widget), enctype) <- runFormPost $ newKeyForm pid
    defaultLayout $ do
        setTitle $ toHtml $ "Vervis > People > " <> user <> " > New Key"
        $(widgetFile "key/key-new")

getKeyR :: Text -> Text -> Handler Html
getKeyR user tag = do
    Entity _kid key <- runDB $ do
        Entity sid _sharer <- getBy404 $ UniqueSharerIdent user
        Entity pid _person <- getBy404 $ UniquePersonIdent sid
        getBy404 $ UniqueSshKey pid tag
    let toText = decodeUtf8With lenientDecode
        content = toText $ encode $ sshKeyContent key
    defaultLayout $ do
        setTitle $ toHtml $
            intercalate " > " ["Vervis", "People", user, "Keys", tag]
        $(widgetFile "key/key")

deleteKeyR :: Text -> Text -> Handler Html
deleteKeyR user tag = do
    runDB $ do
        Entity sid _s <- getBy404 $ UniqueSharerIdent user
        Entity pid _p <- getBy404 $ UniquePersonIdent sid
        Entity kid _k <- getBy404 $ UniqueSshKey pid tag
        delete kid
    setMessage "Key deleted."
    redirect $ KeysR user

postKeyR :: Text -> Text -> Handler Html
postKeyR user tag = do
    mmethod <- lookupPostParam "_method"
    case mmethod of
        Just "DELETE" -> deleteKeyR user tag
        _             -> notFound