module Network.Wai.Middleware.Gzip (gzip) where
import Network.Wai
import Network.Wai.Enumerator (fromLBS', toLBS)
import Codec.Compression.GZip (compress)
import Data.Maybe (fromMaybe)
import Data.List.Split (splitOneOf)
import qualified Data.ByteString.Char8 as B
gzip :: Middleware
gzip app env = do
res <- app env
case responseBody res of
Left _ -> return res
Right _ -> do
let enc = fromMaybe []
$ (splitOneOf "," . B.unpack)
`fmap` lookup AcceptEncoding
(requestHeaders env)
if "gzip" `elem` enc
then return res
{ responseBody = compressE $ responseBody res
, responseHeaders = (ContentEncoding, B.pack "gzip")
: responseHeaders res
}
else return res
compressE :: Either FilePath Enumerator -> Either FilePath Enumerator
compressE (Left fp) = Left fp
compressE (Right e) = Right $ fromLBS' $ fmap compress $ toLBS e