{-# LANGUAGE GADTs #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE PatternGuards #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE UnicodeSyntax #-}
module Network.Wai.Middleware.Cors
( Origin
, CorsResourcePolicy(..)
, simpleCorsResourcePolicy
, cors
, simpleCors
, isSimple
, simpleResponseHeaders
, simpleHeaders
, simpleContentTypes
, simpleMethods
) where
import Control.Monad.Error.Class
import Control.Monad.Trans.Except
import qualified Data.Attoparsec.ByteString.Char8 as P
import qualified Data.ByteString.Char8 as B8
import qualified Data.ByteString.Lazy.Char8 as LB8
import qualified Data.CaseInsensitive as CI
import Data.List (intersect, (\\), union)
import Data.Maybe (catMaybes)
import Data.Monoid.Unicode
import Data.String
import qualified Network.HTTP.Types as HTTP
import qualified Network.Wai as WAI
import Prelude.Unicode
type Origin = B8.ByteString
data CorsResourcePolicy = CorsResourcePolicy
{
CorsResourcePolicy -> Maybe ([ByteString], Bool)
corsOrigins ∷ !(Maybe ([Origin], Bool))
, CorsResourcePolicy -> [ByteString]
corsMethods ∷ ![HTTP.Method]
, ∷ ![HTTP.HeaderName]
, ∷ !(Maybe [HTTP.HeaderName])
, CorsResourcePolicy -> Maybe Int
corsMaxAge ∷ !(Maybe Int)
, CorsResourcePolicy -> Bool
corsVaryOrigin ∷ !Bool
, CorsResourcePolicy -> Bool
corsRequireOrigin ∷ !Bool
, CorsResourcePolicy -> Bool
corsIgnoreFailures ∷ !Bool
}
deriving (Int -> CorsResourcePolicy -> ShowS
[CorsResourcePolicy] -> ShowS
CorsResourcePolicy -> [Char]
(Int -> CorsResourcePolicy -> ShowS)
-> (CorsResourcePolicy -> [Char])
-> ([CorsResourcePolicy] -> ShowS)
-> Show CorsResourcePolicy
forall a.
(Int -> a -> ShowS) -> (a -> [Char]) -> ([a] -> ShowS) -> Show a
showList :: [CorsResourcePolicy] -> ShowS
$cshowList :: [CorsResourcePolicy] -> ShowS
show :: CorsResourcePolicy -> [Char]
$cshow :: CorsResourcePolicy -> [Char]
showsPrec :: Int -> CorsResourcePolicy -> ShowS
$cshowsPrec :: Int -> CorsResourcePolicy -> ShowS
Show,ReadPrec [CorsResourcePolicy]
ReadPrec CorsResourcePolicy
Int -> ReadS CorsResourcePolicy
ReadS [CorsResourcePolicy]
(Int -> ReadS CorsResourcePolicy)
-> ReadS [CorsResourcePolicy]
-> ReadPrec CorsResourcePolicy
-> ReadPrec [CorsResourcePolicy]
-> Read CorsResourcePolicy
forall a.
(Int -> ReadS a)
-> ReadS [a] -> ReadPrec a -> ReadPrec [a] -> Read a
readListPrec :: ReadPrec [CorsResourcePolicy]
$creadListPrec :: ReadPrec [CorsResourcePolicy]
readPrec :: ReadPrec CorsResourcePolicy
$creadPrec :: ReadPrec CorsResourcePolicy
readList :: ReadS [CorsResourcePolicy]
$creadList :: ReadS [CorsResourcePolicy]
readsPrec :: Int -> ReadS CorsResourcePolicy
$creadsPrec :: Int -> ReadS CorsResourcePolicy
Read,CorsResourcePolicy -> CorsResourcePolicy -> Bool
(CorsResourcePolicy -> CorsResourcePolicy -> Bool)
-> (CorsResourcePolicy -> CorsResourcePolicy -> Bool)
-> Eq CorsResourcePolicy
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
$c/= :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
== :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
$c== :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
Eq,Eq CorsResourcePolicy
Eq CorsResourcePolicy
-> (CorsResourcePolicy -> CorsResourcePolicy -> Ordering)
-> (CorsResourcePolicy -> CorsResourcePolicy -> Bool)
-> (CorsResourcePolicy -> CorsResourcePolicy -> Bool)
-> (CorsResourcePolicy -> CorsResourcePolicy -> Bool)
-> (CorsResourcePolicy -> CorsResourcePolicy -> Bool)
-> (CorsResourcePolicy -> CorsResourcePolicy -> CorsResourcePolicy)
-> (CorsResourcePolicy -> CorsResourcePolicy -> CorsResourcePolicy)
-> Ord CorsResourcePolicy
CorsResourcePolicy -> CorsResourcePolicy -> Bool
CorsResourcePolicy -> CorsResourcePolicy -> Ordering
CorsResourcePolicy -> CorsResourcePolicy -> CorsResourcePolicy
forall a.
Eq a
-> (a -> a -> Ordering)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> Bool)
-> (a -> a -> a)
-> (a -> a -> a)
-> Ord a
min :: CorsResourcePolicy -> CorsResourcePolicy -> CorsResourcePolicy
$cmin :: CorsResourcePolicy -> CorsResourcePolicy -> CorsResourcePolicy
max :: CorsResourcePolicy -> CorsResourcePolicy -> CorsResourcePolicy
$cmax :: CorsResourcePolicy -> CorsResourcePolicy -> CorsResourcePolicy
>= :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
$c>= :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
> :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
$c> :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
<= :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
$c<= :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
< :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
$c< :: CorsResourcePolicy -> CorsResourcePolicy -> Bool
compare :: CorsResourcePolicy -> CorsResourcePolicy -> Ordering
$ccompare :: CorsResourcePolicy -> CorsResourcePolicy -> Ordering
Ord)
simpleCorsResourcePolicy ∷ CorsResourcePolicy
simpleCorsResourcePolicy :: CorsResourcePolicy
simpleCorsResourcePolicy = CorsResourcePolicy :: Maybe ([ByteString], Bool)
-> [ByteString]
-> [CI ByteString]
-> Maybe [CI ByteString]
-> Maybe Int
-> Bool
-> Bool
-> Bool
-> CorsResourcePolicy
CorsResourcePolicy
{ corsOrigins :: Maybe ([ByteString], Bool)
corsOrigins = Maybe ([ByteString], Bool)
forall a. Maybe a
Nothing
, corsMethods :: [ByteString]
corsMethods = [ByteString]
simpleMethods
, corsRequestHeaders :: [CI ByteString]
corsRequestHeaders = []
, corsExposedHeaders :: Maybe [CI ByteString]
corsExposedHeaders = Maybe [CI ByteString]
forall a. Maybe a
Nothing
, corsMaxAge :: Maybe Int
corsMaxAge = Maybe Int
forall a. Maybe a
Nothing
, corsVaryOrigin :: Bool
corsVaryOrigin = Bool
False
, corsRequireOrigin :: Bool
corsRequireOrigin = Bool
False
, corsIgnoreFailures :: Bool
corsIgnoreFailures = Bool
False
}
cors
∷ (WAI.Request → Maybe CorsResourcePolicy)
→ WAI.Middleware
cors :: (Request -> Maybe CorsResourcePolicy) -> Middleware
cors Request -> Maybe CorsResourcePolicy
policyPattern Application
app Request
r Response -> IO ResponseReceived
respond
| Request -> Bool
isWebSocketsReq Request
r = IO ResponseReceived
runApp
| Just CorsResourcePolicy
policy ← Request -> Maybe CorsResourcePolicy
policyPattern Request
r = case Maybe ByteString
hdrOrigin of
Maybe ByteString
Nothing → if CorsResourcePolicy -> Bool
corsRequireOrigin CorsResourcePolicy
policy
then Response -> IO ResponseReceived
res (Response -> IO ResponseReceived)
-> Response -> IO ResponseReceived
forall a b. (a -> b) -> a -> b
$ ByteString -> Response
corsFailure ByteString
"Origin header is missing"
else IO ResponseReceived
runApp
Just ByteString
origin → CorsResourcePolicy -> ByteString -> IO ResponseReceived
applyCorsPolicy CorsResourcePolicy
policy ByteString
origin
| Bool
otherwise = IO ResponseReceived
runApp
where
res :: Response -> IO ResponseReceived
res = Response -> IO ResponseReceived
respond
runApp :: IO ResponseReceived
runApp = Application
app Request
r Response -> IO ResponseReceived
respond
hdrOrigin :: Maybe ByteString
hdrOrigin = CI ByteString -> ResponseHeaders -> Maybe ByteString
forall a b. Eq a => a -> [(a, b)] -> Maybe b
lookup CI ByteString
"origin" (Request -> ResponseHeaders
WAI.requestHeaders Request
r)
applyCorsPolicy :: CorsResourcePolicy -> ByteString -> IO ResponseReceived
applyCorsPolicy CorsResourcePolicy
policy ByteString
origin = do
let err :: [Char] -> IO ResponseReceived
err [Char]
e = if CorsResourcePolicy -> Bool
corsIgnoreFailures CorsResourcePolicy
policy
then IO ResponseReceived
runApp
else Response -> IO ResponseReceived
res (Response -> IO ResponseReceived)
-> Response -> IO ResponseReceived
forall a b. (a -> b) -> a -> b
$ ByteString -> Response
corsFailure ([Char] -> ByteString
B8.pack [Char]
e)
let respOriginOrErr :: Either [Char] (Maybe (ByteString, Bool))
respOriginOrErr = case CorsResourcePolicy -> Maybe ([ByteString], Bool)
corsOrigins CorsResourcePolicy
policy of
Maybe ([ByteString], Bool)
Nothing → Maybe (ByteString, Bool)
-> Either [Char] (Maybe (ByteString, Bool))
forall (m :: * -> *) a. Monad m => a -> m a
return Maybe (ByteString, Bool)
forall a. Maybe a
Nothing
Just ([ByteString]
originList, Bool
withCreds) → if ByteString
origin ByteString -> [ByteString] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [ByteString]
originList
then Maybe (ByteString, Bool)
-> Either [Char] (Maybe (ByteString, Bool))
forall a b. b -> Either a b
Right (Maybe (ByteString, Bool)
-> Either [Char] (Maybe (ByteString, Bool)))
-> Maybe (ByteString, Bool)
-> Either [Char] (Maybe (ByteString, Bool))
forall a b. (a -> b) -> a -> b
$ (ByteString, Bool) -> Maybe (ByteString, Bool)
forall a. a -> Maybe a
Just (ByteString
origin, Bool
withCreds)
else [Char] -> Either [Char] (Maybe (ByteString, Bool))
forall a b. a -> Either a b
Left ([Char] -> Either [Char] (Maybe (ByteString, Bool)))
-> [Char] -> Either [Char] (Maybe (ByteString, Bool))
forall a b. (a -> b) -> a -> b
$ [Char]
"Unsupported origin: " [Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ ByteString -> [Char]
B8.unpack ByteString
origin
case Either [Char] (Maybe (ByteString, Bool))
respOriginOrErr of
Left [Char]
e → [Char] -> IO ResponseReceived
err [Char]
e
Right Maybe (ByteString, Bool)
respOrigin → do
let ch :: ResponseHeaders
ch = Maybe (ByteString, Bool) -> Bool -> ResponseHeaders
commonCorsHeaders Maybe (ByteString, Bool)
respOrigin (CorsResourcePolicy -> Bool
corsVaryOrigin CorsResourcePolicy
policy)
case Request -> ByteString
WAI.requestMethod Request
r of
ByteString
"OPTIONS" → ExceptT [Char] IO ResponseHeaders
-> IO (Either [Char] ResponseHeaders)
forall e (m :: * -> *) a. ExceptT e m a -> m (Either e a)
runExceptT (CorsResourcePolicy -> ExceptT [Char] IO ResponseHeaders
forall (μ :: * -> *).
(Functor μ, Monad μ) =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
preflightHeaders CorsResourcePolicy
policy) IO (Either [Char] ResponseHeaders)
-> (Either [Char] ResponseHeaders -> IO ResponseReceived)
-> IO ResponseReceived
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
Left [Char]
e → [Char] -> IO ResponseReceived
err [Char]
e
Right ResponseHeaders
headers → Response -> IO ResponseReceived
res (Response -> IO ResponseReceived)
-> Response -> IO ResponseReceived
forall a b. (a -> b) -> a -> b
$ Status -> ResponseHeaders -> ByteString -> Response
WAI.responseLBS Status
HTTP.ok200 (ResponseHeaders
ch ResponseHeaders -> ResponseHeaders -> ResponseHeaders
forall α. Monoid α => α -> α -> α
⊕ ResponseHeaders
headers) ByteString
""
ByteString
_ → ResponseHeaders -> Middleware
addHeaders (ResponseHeaders
ch ResponseHeaders -> ResponseHeaders -> ResponseHeaders
forall α. Monoid α => α -> α -> α
⊕ CorsResourcePolicy -> ResponseHeaders
respCorsHeaders CorsResourcePolicy
policy) Application
app Request
r Response -> IO ResponseReceived
respond
preflightHeaders ∷ (Functor μ, Monad μ) ⇒ CorsResourcePolicy → ExceptT String μ HTTP.ResponseHeaders
preflightHeaders :: forall (μ :: * -> *).
(Functor μ, Monad μ) =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
preflightHeaders CorsResourcePolicy
policy = [ResponseHeaders] -> ResponseHeaders
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
concat ([ResponseHeaders] -> ResponseHeaders)
-> ExceptT [Char] μ [ResponseHeaders]
-> ExceptT [Char] μ ResponseHeaders
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> [ExceptT [Char] μ ResponseHeaders]
-> ExceptT [Char] μ [ResponseHeaders]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
sequence
[ CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
forall (μ :: * -> *).
Monad μ =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
hdrReqMethod CorsResourcePolicy
policy
, CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
forall (μ :: * -> *).
Monad μ =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
hdrRequestHeader CorsResourcePolicy
policy
, CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
forall (μ :: * -> *).
Monad μ =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
hdrMaxAge CorsResourcePolicy
policy
]
hdrMaxAge ∷ Monad μ ⇒ CorsResourcePolicy → ExceptT String μ HTTP.ResponseHeaders
hdrMaxAge :: forall (μ :: * -> *).
Monad μ =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
hdrMaxAge CorsResourcePolicy
policy = case CorsResourcePolicy -> Maybe Int
corsMaxAge CorsResourcePolicy
policy of
Maybe Int
Nothing → ResponseHeaders -> ExceptT [Char] μ ResponseHeaders
forall (m :: * -> *) a. Monad m => a -> m a
return []
Just Int
secs → ResponseHeaders -> ExceptT [Char] μ ResponseHeaders
forall (m :: * -> *) a. Monad m => a -> m a
return [(CI ByteString
"Access-Control-Max-Age", Int -> ByteString
forall α β. (IsString α, Show β) => β -> α
sshow Int
secs)]
hdrReqMethod ∷ Monad μ ⇒ CorsResourcePolicy → ExceptT String μ HTTP.ResponseHeaders
hdrReqMethod :: forall (μ :: * -> *).
Monad μ =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
hdrReqMethod CorsResourcePolicy
policy = case CI ByteString -> ResponseHeaders -> Maybe ByteString
forall a b. Eq a => a -> [(a, b)] -> Maybe b
lookup CI ByteString
"Access-Control-Request-Method" (Request -> ResponseHeaders
WAI.requestHeaders Request
r) of
Maybe ByteString
Nothing → [Char] -> ExceptT [Char] μ ResponseHeaders
forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError [Char]
"Access-Control-Request-Method header is missing in CORS preflight request"
Just ByteString
x → if ByteString
x ByteString -> [ByteString] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [ByteString]
supportedMethods
then ResponseHeaders -> ExceptT [Char] μ ResponseHeaders
forall (m :: * -> *) a. Monad m => a -> m a
return [(CI ByteString
"Access-Control-Allow-Methods", [ByteString] -> ByteString
hdrL [ByteString]
supportedMethods)]
else [Char] -> ExceptT [Char] μ ResponseHeaders
forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError
([Char] -> ExceptT [Char] μ ResponseHeaders)
-> [Char] -> ExceptT [Char] μ ResponseHeaders
forall a b. (a -> b) -> a -> b
$ [Char]
"Method requested in Access-Control-Request-Method of CORS request is not supported; requested: "
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ ByteString -> [Char]
B8.unpack ByteString
x
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ [Char]
"; supported are "
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ ByteString -> [Char]
B8.unpack ([ByteString] -> ByteString
hdrL [ByteString]
supportedMethods)
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ [Char]
"."
where
supportedMethods :: [ByteString]
supportedMethods = CorsResourcePolicy -> [ByteString]
corsMethods CorsResourcePolicy
policy [ByteString] -> [ByteString] -> [ByteString]
forall a. Eq a => [a] -> [a] -> [a]
`union` [ByteString]
simpleMethods
hdrRequestHeader ∷ Monad μ ⇒ CorsResourcePolicy → ExceptT String μ HTTP.ResponseHeaders
hdrRequestHeader :: forall (μ :: * -> *).
Monad μ =>
CorsResourcePolicy -> ExceptT [Char] μ ResponseHeaders
hdrRequestHeader CorsResourcePolicy
policy = case CI ByteString -> ResponseHeaders -> Maybe ByteString
forall a b. Eq a => a -> [(a, b)] -> Maybe b
lookup CI ByteString
"Access-Control-Request-Headers" (Request -> ResponseHeaders
WAI.requestHeaders Request
r) of
Maybe ByteString
Nothing → ResponseHeaders -> ExceptT [Char] μ ResponseHeaders
forall (m :: * -> *) a. Monad m => a -> m a
return []
Just ByteString
hdrsBytes → do
[CI ByteString]
hdrs ← ([Char] -> ExceptT [Char] μ [CI ByteString])
-> ([CI ByteString] -> ExceptT [Char] μ [CI ByteString])
-> Either [Char] [CI ByteString]
-> ExceptT [Char] μ [CI ByteString]
forall a c b. (a -> c) -> (b -> c) -> Either a b -> c
either [Char] -> ExceptT [Char] μ [CI ByteString]
forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError [CI ByteString] -> ExceptT [Char] μ [CI ByteString]
forall (m :: * -> *) a. Monad m => a -> m a
return (Either [Char] [CI ByteString] -> ExceptT [Char] μ [CI ByteString])
-> Either [Char] [CI ByteString]
-> ExceptT [Char] μ [CI ByteString]
forall a b. (a -> b) -> a -> b
$ Parser [CI ByteString]
-> ByteString -> Either [Char] [CI ByteString]
forall a. Parser a -> ByteString -> Either [Char] a
P.parseOnly Parser [CI ByteString]
httpHeaderNameListParser ByteString
hdrsBytes
if [CI ByteString]
hdrs [CI ByteString] -> [CI ByteString] -> Bool
forall α. Eq α => [α] -> [α] -> Bool
`isSubsetOf` [CI ByteString]
supportedHeaders
then ResponseHeaders -> ExceptT [Char] μ ResponseHeaders
forall (m :: * -> *) a. Monad m => a -> m a
return [(CI ByteString
"Access-Control-Allow-Headers", [CI ByteString] -> ByteString
hdrLI [CI ByteString]
supportedHeaders)]
else [Char] -> ExceptT [Char] μ ResponseHeaders
forall e (m :: * -> *) a. MonadError e m => e -> m a
throwError
([Char] -> ExceptT [Char] μ ResponseHeaders)
-> [Char] -> ExceptT [Char] μ ResponseHeaders
forall a b. (a -> b) -> a -> b
$ [Char]
"HTTP header requested in Access-Control-Request-Headers of CORS request is not supported; requested: "
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ ByteString -> [Char]
B8.unpack ([CI ByteString] -> ByteString
hdrLI [CI ByteString]
hdrs)
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ [Char]
"; supported are "
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ ByteString -> [Char]
B8.unpack ([CI ByteString] -> ByteString
hdrLI [CI ByteString]
supportedHeaders)
[Char] -> ShowS
forall α. Monoid α => α -> α -> α
⊕ [Char]
"."
where
supportedHeaders :: [CI ByteString]
supportedHeaders = CorsResourcePolicy -> [CI ByteString]
corsRequestHeaders CorsResourcePolicy
policy [CI ByteString] -> [CI ByteString] -> [CI ByteString]
forall a. Eq a => [a] -> [a] -> [a]
`union` [CI ByteString]
simpleHeadersWithoutContentType
simpleHeadersWithoutContentType :: [CI ByteString]
simpleHeadersWithoutContentType = [CI ByteString]
simpleHeaders [CI ByteString] -> [CI ByteString] -> [CI ByteString]
forall a. Eq a => [a] -> [a] -> [a]
\\ [CI ByteString
"content-type"]
commonCorsHeaders ∷ Maybe (Origin, Bool) → Bool → HTTP.ResponseHeaders
commonCorsHeaders :: Maybe (ByteString, Bool) -> Bool -> ResponseHeaders
commonCorsHeaders Maybe (ByteString, Bool)
Nothing Bool
_ = [(CI ByteString
"Access-Control-Allow-Origin", ByteString
"*")]
commonCorsHeaders (Just (ByteString
o, Bool
creds)) Bool
vary = []
ResponseHeaders -> ResponseHeaders -> ResponseHeaders
forall α. Monoid α => α -> α -> α
⊕ (Bool
True Bool -> (CI ByteString, ByteString) -> ResponseHeaders
forall {f :: * -> *} {a}.
(Applicative f, Monoid (f a)) =>
Bool -> a -> f a
?? (CI ByteString
"Access-Control-Allow-Origin", ByteString
o))
ResponseHeaders -> ResponseHeaders -> ResponseHeaders
forall α. Monoid α => α -> α -> α
⊕ (Bool
creds Bool -> (CI ByteString, ByteString) -> ResponseHeaders
forall {f :: * -> *} {a}.
(Applicative f, Monoid (f a)) =>
Bool -> a -> f a
?? (CI ByteString
"Access-Control-Allow-Credentials", ByteString
"true"))
ResponseHeaders -> ResponseHeaders -> ResponseHeaders
forall α. Monoid α => α -> α -> α
⊕ (Bool
vary Bool -> (CI ByteString, ByteString) -> ResponseHeaders
forall {f :: * -> *} {a}.
(Applicative f, Monoid (f a)) =>
Bool -> a -> f a
?? (CI ByteString
"Vary", ByteString
"Origin"))
where
?? :: Bool -> a -> f a
(??) Bool
a a
b = if Bool
a then a -> f a
forall (f :: * -> *) a. Applicative f => a -> f a
pure a
b else f a
forall a. Monoid a => a
mempty
respCorsHeaders ∷ CorsResourcePolicy → HTTP.ResponseHeaders
respCorsHeaders :: CorsResourcePolicy -> ResponseHeaders
respCorsHeaders CorsResourcePolicy
policy = [Maybe (CI ByteString, ByteString)] -> ResponseHeaders
forall a. [Maybe a] -> [a]
catMaybes
[ ([CI ByteString] -> (CI ByteString, ByteString))
-> Maybe [CI ByteString] -> Maybe (CI ByteString, ByteString)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (\[CI ByteString]
x → (CI ByteString
"Access-Control-Expose-Headers", [CI ByteString] -> ByteString
hdrLI [CI ByteString]
x)) (CorsResourcePolicy -> Maybe [CI ByteString]
corsExposedHeaders CorsResourcePolicy
policy)
]
simpleCors ∷ WAI.Middleware
simpleCors :: Middleware
simpleCors = (Request -> Maybe CorsResourcePolicy) -> Middleware
cors (Maybe CorsResourcePolicy -> Request -> Maybe CorsResourcePolicy
forall a b. a -> b -> a
const (Maybe CorsResourcePolicy -> Request -> Maybe CorsResourcePolicy)
-> Maybe CorsResourcePolicy -> Request -> Maybe CorsResourcePolicy
forall a b. (a -> b) -> a -> b
$ CorsResourcePolicy -> Maybe CorsResourcePolicy
forall a. a -> Maybe a
Just CorsResourcePolicy
simpleCorsResourcePolicy)
simpleResponseHeaders ∷ [HTTP.HeaderName]
=
[ CI ByteString
"Cache-Control"
, CI ByteString
"Content-Language"
, CI ByteString
"Content-Type"
, CI ByteString
"Expires"
, CI ByteString
"Last-Modified"
, CI ByteString
"Pragma"
]
simpleHeaders ∷ [HTTP.HeaderName]
=
[ CI ByteString
"Accept"
, CI ByteString
"Accept-Language"
, CI ByteString
"Content-Language"
, CI ByteString
"Content-Type"
]
simpleContentTypes ∷ [CI.CI B8.ByteString]
simpleContentTypes :: [CI ByteString]
simpleContentTypes =
[ CI ByteString
"application/x-www-form-urlencoded"
, CI ByteString
"multipart/form-data"
, CI ByteString
"text/plain"
]
simpleMethods ∷ [HTTP.Method]
simpleMethods :: [ByteString]
simpleMethods =
[ ByteString
"GET"
, ByteString
"HEAD"
, ByteString
"POST"
]
isSimple ∷ HTTP.Method → HTTP.RequestHeaders → Bool
isSimple :: ByteString -> ResponseHeaders -> Bool
isSimple ByteString
method ResponseHeaders
headers
= ByteString
method ByteString -> [ByteString] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [ByteString]
simpleMethods
Bool -> Bool -> Bool
∧ ((CI ByteString, ByteString) -> CI ByteString)
-> ResponseHeaders -> [CI ByteString]
forall a b. (a -> b) -> [a] -> [b]
map (CI ByteString, ByteString) -> CI ByteString
forall a b. (a, b) -> a
fst ResponseHeaders
headers [CI ByteString] -> [CI ByteString] -> Bool
forall α. Eq α => [α] -> [α] -> Bool
`isSubsetOf` [CI ByteString]
simpleHeaders
Bool -> Bool -> Bool
∧ case (ByteString
method, CI ByteString -> ResponseHeaders -> Maybe ByteString
forall a b. Eq a => a -> [(a, b)] -> Maybe b
lookup CI ByteString
"content-type" ResponseHeaders
headers) of
(ByteString
"POST", Just ByteString
x) → ByteString -> CI ByteString
forall s. FoldCase s => s -> CI s
CI.mk ByteString
x CI ByteString -> [CI ByteString] -> Bool
forall (t :: * -> *) a. (Foldable t, Eq a) => a -> t a -> Bool
`elem` [CI ByteString]
simpleContentTypes
(ByteString, Maybe ByteString)
_ → Bool
True
isHttpHeaderNameChar ∷ Char → Bool
Char
c = (Char
c Char -> Char -> Bool
forall α. Ord α => α -> α -> Bool
≥ Int -> Char
forall a. Enum a => Int -> a
toEnum Int
33) Bool -> Bool -> Bool
&& (Char
c Char -> Char -> Bool
forall α. Ord α => α -> α -> Bool
≤ Int -> Char
forall a. Enum a => Int -> a
toEnum Int
126) Bool -> Bool -> Bool
&& [Char] -> Char -> Bool
P.notInClass [Char]
"()<>@,;:\\\"/[]?={}" Char
c
httpHeaderNameParser ∷ P.Parser HTTP.HeaderName
= [Char] -> CI ByteString
forall a. IsString a => [Char] -> a
fromString ([Char] -> CI ByteString)
-> Parser ByteString [Char] -> Parser (CI ByteString)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Parser ByteString Char -> Parser ByteString [Char]
forall (f :: * -> *) a. Alternative f => f a -> f [a]
P.many1 ((Char -> Bool) -> Parser ByteString Char
P.satisfy Char -> Bool
isHttpHeaderNameChar) Parser (CI ByteString) -> [Char] -> Parser (CI ByteString)
forall i a. Parser i a -> [Char] -> Parser i a
P.<?> [Char]
"HTTP Header Name"
httpHeaderNameListParser ∷ P.Parser [HTTP.HeaderName]
= Parser ByteString [Char]
spaces Parser ByteString [Char]
-> Parser [CI ByteString] -> Parser [CI ByteString]
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f b
*> Parser (CI ByteString)
-> Parser ByteString Char -> Parser [CI ByteString]
forall (f :: * -> *) a s. Alternative f => f a -> f s -> f [a]
P.sepBy (Parser (CI ByteString)
httpHeaderNameParser Parser (CI ByteString)
-> Parser ByteString [Char] -> Parser (CI ByteString)
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* Parser ByteString [Char]
spaces) (Char -> Parser ByteString Char
P.char Char
',') Parser [CI ByteString]
-> Parser ByteString [Char] -> Parser [CI ByteString]
forall (f :: * -> *) a b. Applicative f => f a -> f b -> f a
<* Parser ByteString [Char]
spaces
where
spaces :: Parser ByteString [Char]
spaces = Parser ByteString Char -> Parser ByteString [Char]
forall (m :: * -> *) a. MonadPlus m => m a -> m [a]
P.many' Parser ByteString Char
P.space
sshow ∷ (IsString α, Show β) ⇒ β → α
sshow :: forall α β. (IsString α, Show β) => β -> α
sshow = [Char] -> α
forall a. IsString a => [Char] -> a
fromString ([Char] -> α) -> (β -> [Char]) -> β -> α
forall β γ α. (β -> γ) -> (α -> β) -> α -> γ
∘ β -> [Char]
forall a. Show a => a -> [Char]
show
isSubsetOf ∷ Eq α ⇒ [α] → [α] → Bool
isSubsetOf :: forall α. Eq α => [α] -> [α] -> Bool
isSubsetOf [α]
l1 [α]
l2 = [α] -> [α] -> [α]
forall a. Eq a => [a] -> [a] -> [a]
intersect [α]
l1 [α]
l2 [α] -> [α] -> Bool
forall α. Eq α => α -> α -> Bool
≡ [α]
l1
addHeaders ∷ HTTP.ResponseHeaders → WAI.Middleware
ResponseHeaders
hdrs Application
app Request
req Response -> IO ResponseReceived
respond = Application
app Request
req ((Response -> IO ResponseReceived) -> IO ResponseReceived)
-> (Response -> IO ResponseReceived) -> IO ResponseReceived
forall a b. (a -> b) -> a -> b
$ \Response
response → do
let (Status
st, ResponseHeaders
headers, (StreamingBody -> IO ResponseReceived) -> IO ResponseReceived
streamHandle) = Response
-> (Status, ResponseHeaders,
(StreamingBody -> IO ResponseReceived) -> IO ResponseReceived)
forall a.
Response
-> (Status, ResponseHeaders, (StreamingBody -> IO a) -> IO a)
WAI.responseToStream Response
response
(StreamingBody -> IO ResponseReceived) -> IO ResponseReceived
streamHandle ((StreamingBody -> IO ResponseReceived) -> IO ResponseReceived)
-> (StreamingBody -> IO ResponseReceived) -> IO ResponseReceived
forall a b. (a -> b) -> a -> b
$ \StreamingBody
streamBody →
Response -> IO ResponseReceived
respond (Response -> IO ResponseReceived)
-> Response -> IO ResponseReceived
forall a b. (a -> b) -> a -> b
$ Status -> ResponseHeaders -> StreamingBody -> Response
WAI.responseStream Status
st (ResponseHeaders
headers ResponseHeaders -> ResponseHeaders -> ResponseHeaders
forall α. Monoid α => α -> α -> α
⊕ ResponseHeaders
hdrs) StreamingBody
streamBody
hdrLI ∷ [HTTP.HeaderName] → B8.ByteString
hdrLI :: [CI ByteString] -> ByteString
hdrLI [CI ByteString]
l = ByteString -> [ByteString] -> ByteString
B8.intercalate ByteString
", " ((CI ByteString -> ByteString) -> [CI ByteString] -> [ByteString]
forall a b. (a -> b) -> [a] -> [b]
map CI ByteString -> ByteString
forall s. CI s -> s
CI.original [CI ByteString]
l)
hdrL ∷ [B8.ByteString] → B8.ByteString
hdrL :: [ByteString] -> ByteString
hdrL = ByteString -> [ByteString] -> ByteString
B8.intercalate ByteString
", "
corsFailure
∷ B8.ByteString
→ WAI.Response
corsFailure :: ByteString -> Response
corsFailure ByteString
msg = Status -> ResponseHeaders -> ByteString -> Response
WAI.responseLBS Status
HTTP.status400 [(CI ByteString
"Content-Type", ByteString
"text/html; charset-utf-8")] (ByteString -> ByteString
LB8.fromStrict ByteString
msg)
isWebSocketsReq
∷ WAI.Request
→ Bool
isWebSocketsReq :: Request -> Bool
isWebSocketsReq Request
req =
(ByteString -> CI ByteString)
-> Maybe ByteString -> Maybe (CI ByteString)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap ByteString -> CI ByteString
forall s. FoldCase s => s -> CI s
CI.mk (CI ByteString -> ResponseHeaders -> Maybe ByteString
forall a b. Eq a => a -> [(a, b)] -> Maybe b
lookup CI ByteString
"upgrade" (ResponseHeaders -> Maybe ByteString)
-> ResponseHeaders -> Maybe ByteString
forall a b. (a -> b) -> a -> b
$ Request -> ResponseHeaders
WAI.requestHeaders Request
req) Maybe (CI ByteString) -> Maybe (CI ByteString) -> Bool
forall α. Eq α => α -> α -> Bool
== CI ByteString -> Maybe (CI ByteString)
forall a. a -> Maybe a
Just CI ByteString
"websocket"