fradrive/src/Handler/Utils/ContentDisposition.hs
2020-12-08 17:34:39 +01:00

24 lines
1005 B
Haskell

module Handler.Utils.ContentDisposition
( downloadFiles
, setContentDisposition'
) where
import Import.NoFoundation
import Foundation.Type
-- | Check whether the user's preference for files is inline-viewing or downloading
downloadFiles :: (MonadHandler m, HandlerSite m ~ UniWorX, YesodAuthPersist UniWorX, AuthEntity UniWorX ~ User, AuthId UniWorX ~ UserId) => m Bool
downloadFiles = do
mauth <- liftHandler maybeAuth
case mauth of
Just (Entity _ User{..}) -> return userDownloadFiles
Nothing -> do
UserDefaultConf{..} <- getsYesod $ view _appUserDefaults
return userDefaultDownloadFiles
setContentDisposition' :: (MonadHandler m, HandlerSite m ~ UniWorX, YesodAuthPersist UniWorX, AuthEntity UniWorX ~ User, AuthId UniWorX ~ UserId) => Maybe FilePath -> m ()
setContentDisposition' mFileName = do
wantsDownload <- maybeT downloadFiles . MaybeT $ lookupGlobalGetParam GetDownload
setContentDisposition (bool ContentInline ContentAttachment wantsDownload) mFileName