matebeamter/src/Util.hs

146 lines
3.3 KiB
Haskell
Raw Normal View History

2019-09-08 10:48:57 +00:00
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE TypeFamilies #-}
{-# LANGUAGE OverloadedStrings #-}
2019-09-18 13:24:23 +00:00
{-# LANGUAGE FlexibleContexts #-}
2019-10-14 15:40:20 +00:00
{-# LANGUAGE PackageImports #-}
2019-10-16 16:07:12 +00:00
{-# LANGUAGE LambdaCase #-}
2019-09-08 15:27:38 +00:00
module Util where
2019-09-08 10:48:57 +00:00
import Servant hiding (addHeader)
2019-10-14 15:40:20 +00:00
import Servant.Client
2019-09-11 04:27:54 +00:00
import Servant.Client.Core.Request
2019-09-08 10:48:57 +00:00
2019-10-21 18:41:58 +00:00
import qualified Text.Blaze.Html5.Attributes as HA
2019-09-11 04:27:54 +00:00
import qualified Data.Text as T
import Data.Text.Lazy.Encoding
import Data.ByteString.Builder
import Data.List (intercalate)
import Data.List.Split (chunksOf)
import Data.String (fromString)
import Web.Cookie
import Text.Printf (printf)
2019-10-14 15:40:20 +00:00
import Network.HTTP.Types.Status
-- imports from "mateamt"
import qualified "mateamt" Types as MT
-- internal imports
import Types
import API
2019-09-11 04:27:54 +00:00
setCookie
:: SetCookie
-> Request
-> Request
setCookie c = addHeader "Set-Cookie" (decodeUtf8 . toLazyByteString $ renderSetCookie c)
formatMoney :: Int -> T.Text
formatMoney amount = pre <> t <> "," <> c
where
pre = if amount < 0
then "-"
else ""
t = fromString $ reverse (intercalate "." $ chunksOf 3 $ reverse $ fst sp)
c = fromString $ snd sp
sp = tail <$>
break (== '.') (printf "%.2f" (fromIntegral $ abs amount :: Float))
2019-09-18 13:24:23 +00:00
redirect303
2019-10-04 07:04:23 +00:00
:: Link
-> ServerError
2019-09-18 13:24:23 +00:00
redirect303 link =
2019-10-04 07:04:23 +00:00
err303
2019-09-18 13:24:23 +00:00
{ errHeaders =
2019-10-16 16:07:12 +00:00
[ ("Location", "/" <> fromString (show $ linkURI link))
2019-09-18 13:24:23 +00:00
]
}
2019-10-04 07:04:23 +00:00
parseCookieText
:: T.Text
-> [(T.Text, T.Text)]
parseCookieText = map
(\x -> let (one, two) = T.breakOn "=" x in (one, T.tail two))
. T.splitOn "; "
2019-10-14 15:40:20 +00:00
handleErrors
:: (Traversable t, Show (t (Maybe ClientError)))
=> Maybe Int
2019-10-14 19:41:28 +00:00
-> Maybe Link
2019-10-14 15:40:20 +00:00
-> t (Maybe ClientError)
-> UserHandler UserOverviewPage
2019-10-14 19:41:28 +00:00
handleErrors mAuthUser mLink merrs = do
2019-10-14 15:40:20 +00:00
codes <- mapM
2019-10-16 16:07:12 +00:00
(\case
2019-10-14 15:40:20 +00:00
Just (FailureResponse _ resp) ->
return $ statusCode (responseStatusCode resp)
Just err ->
throwError $ err500
{ errBody = fromString (show err)
}
Nothing ->
return 200
)
merrs
2019-10-16 16:07:12 +00:00
if 401 `elem` codes
2019-10-14 15:40:20 +00:00
then
2019-10-14 19:41:28 +00:00
redirectOverAuth mAuthUser mLink Nothing
2019-10-14 15:40:20 +00:00
else
throwError $ err500
{ errBody = fromString (show merrs)
}
redirectOverAuth
:: Maybe Int
2019-10-14 19:41:28 +00:00
-> Maybe Link
2019-10-16 16:07:12 +00:00
-> Maybe MT.ProductRefine
2019-10-14 15:40:20 +00:00
-> UserHandler UserOverviewPage
2019-10-14 19:41:28 +00:00
redirectOverAuth muid mLink mRefine = do
2019-10-14 15:40:20 +00:00
let redirectHeaders = errHeaders $
redirect303 (authLink (Just $ "/" <>
2019-10-14 19:41:28 +00:00
fromString (show $ linkURI (case (mLink, muid) of
(Just link, _) -> link
2019-10-16 16:07:12 +00:00
(Nothing, Just uid) -> userOverviewLink uid mRefine
2019-10-14 19:41:28 +00:00
_ -> userSelectLink
2019-10-14 15:40:20 +00:00
))))
throwError
(err303
{ errHeaders = redirectHeaders ++ (case muid of
Just uid ->
[ ( "Set-Cookie"
2019-10-16 16:07:12 +00:00
, "x-auth-user=" <> fromString (show uid) <> "; Path=/"
2019-10-14 15:40:20 +00:00
)
]
Nothing ->
[
]
)
++
[ ( "Set-Cookie"
, "x-token=; Path=/; Expires=Thu, 01 Jan 1970 00:00:00 GMT"
)
, ( "Set-Cookie"
, "x-method=; Path=/; Expires=Thu, 01 Jan 1970 00:00:00 GMT"
)
]
}
)
2019-10-21 18:41:58 +00:00
productBgStyle aid = HA.style $ mconcat
[ "background-image: url(#" <> aid <> ");" -- FILLME
, "backgronud-color: blue;"
, "text-shadow:"
, " -1px 0 1px black,"
, " 0 1px 1px black,"
, " 1px 0 1px black,"
, " 0 -1px 1px black;"
]