module Network.Wai.Middleware.Gzip (gzip) where
import Network.Wai
import Network.Wai.Zlib
import Data.Maybe (fromMaybe)
import qualified Data.ByteString.Char8 as B
import Data.Enumerator (($$), joinI)
gzip :: Bool
-> Middleware
gzip files app env = do
res <- app env
return $
case res of
ResponseFile{} | not files -> res
_ -> if "gzip" `elem` enc
then ResponseEnumerator $ compressE $ responseEnumerator res
else res
where
enc = fromMaybe [] $ (splitCommas . B.unpack)
`fmap` lookup "Accept-Encoding" (requestHeaders env)
compressE :: (forall a. ResponseEnumerator a)
-> (forall a. ResponseEnumerator a)
compressE re f =
re f'
where
f' s hs =
joinI $ compress $$ f s hs'
where
hs' = ("Content-Encoding", "gzip") : hs
splitCommas :: String -> [String]
splitCommas [] = []
splitCommas x =
let (y, z) = break (== ',') x
in y : splitCommas (drop 1 z)