module Network.Api.Support.Request (
RequestTransformer
, setApiKey
, setParams
, setHeaders
, setHeader
, addHeader
, stripHeader
, setCookieJar
, setMethod
, setBody
, setBodyLazy
, setJson
, (<>)
) where
import Control.Monad.Trans.Resource
import qualified Data.ByteString as B
import qualified Data.ByteString.Lazy as BL
import Data.Aeson
import Data.CaseInsensitive
import Data.Monoid
import Data.Time
import Network.HTTP.Conduit
import Network.HTTP.Conduit.Internal (insertCookiesIntoRequest)
type RequestTransformer m = Endo (Request (ResourceT m))
setApiKey :: B.ByteString -> RequestTransformer m
setApiKey key = Endo $ applyBasicAuth key ""
setParams :: Monad m => [(B.ByteString, B.ByteString)] -> RequestTransformer m
setParams params = Endo $ urlEncodedBody params
setHeaders :: [(CI B.ByteString, B.ByteString)] -> RequestTransformer m
setHeaders hs = Endo $ \r -> r { requestHeaders = hs }
setHeader :: (CI B.ByteString, B.ByteString) -> RequestTransformer m
setHeader h = stripHeader (fst h) <> addHeader h
addHeader :: (CI B.ByteString, B.ByteString) -> RequestTransformer m
addHeader h = Endo $ \r -> r { requestHeaders = requestHeaders r ++ [h] }
stripHeader :: CI B.ByteString -> RequestTransformer m
stripHeader n = (Endo $ \r -> r {
requestHeaders = filter (\x -> fst x == n) (requestHeaders r)
})
setCookieJar :: CookieJar -> UTCTime -> RequestTransformer m
setCookieJar cj now = Endo $ \r -> fst $ insertCookiesIntoRequest r cj now
setMethod :: B.ByteString -> RequestTransformer m
setMethod m = Endo $ \r -> r { method = m }
setBody :: B.ByteString -> RequestTransformer m
setBody b = Endo $ \r -> r { requestBody = RequestBodyBS b }
setBodyLazy :: BL.ByteString -> RequestTransformer m
setBodyLazy b = Endo $ \r -> r { requestBody = RequestBodyLBS b }
setJson :: ToJSON a => a -> RequestTransformer m
setJson v = setHeader ("Content-Type", "application/json") <>
(setBodyLazy . encode . toJSON $ v)
#if __GLASGOW_HASKELL__ < 704
infixr 5 <>
(<>) :: Monoid m => m -> m -> m
(<>) = mappend
#endif