@@ -31,8 +31,8 @@ module Servant.API.ResponseHeaders
3131 ) where
3232
3333import Data.ByteString.Char8 as BS (pack, unlines, init)
34−import Data.ByteString.Conversion (ToByteString, toByteString',
35− FromByteString, fromByteString)
34+import Web.HttpApiData (ToHttpApiData, toHeader,
35+ FromHttpApiData, parseHeader)
3636import qualified Data.CaseInsensitive as CI
3737import Data.Proxy
3838import GHC.TypeLits (KnownSymbol, symbolVal)
@@ -68,17 +68,17 @@ class BuildHeadersTo hs where
6868instance OVERLAPPING_ BuildHeadersTo '[] where
6969 buildHeadersTo _ = HNil
7070
71−instance OVERLAPPABLE_ ( FromByteString v, BuildHeadersTo xs, KnownSymbol h )
71+instance OVERLAPPABLE_ ( FromHttpApiData v, BuildHeadersTo xs, KnownSymbol h )
7272 => BuildHeadersTo ((Header h v) ': xs) where
7373 buildHeadersTo headers =
7474 let wantedHeader = CI.mk . pack $ symbolVal (Proxy :: Proxy h)
7575 matching = snd <$> filter (\(h, _) -> h == wantedHeader) headers
7676 in case matching of
7777 [] -> MissingHeader `HCons` buildHeadersTo headers
78− xs -> case fromByteString (BS.init $ BS.unlines xs) of
79− Nothing -> UndecodableHeader (BS.init $ BS.unlines xs)
78+ xs -> case parseHeader (BS.init $ BS.unlines xs) of
79+ Left _err -> UndecodableHeader (BS.init $ BS.unlines xs)
8080 `HCons` buildHeadersTo headers
81− Just h -> Header h `HCons` buildHeadersTo headers
81+ Right h -> Header h `HCons` buildHeadersTo headers
8282
8383-- * Getting
8484
@@ -88,18 +88,18 @@ class GetHeaders ls where
8888instance OVERLAPPING_ GetHeaders (HList '[]) where
8989 getHeaders _ = []
9090
91−instance OVERLAPPABLE_ ( KnownSymbol h, ToByteString x, GetHeaders (HList xs) )
91+instance OVERLAPPABLE_ ( KnownSymbol h, ToHttpApiData x, GetHeaders (HList xs) )
9292 => GetHeaders (HList (Header h x ': xs)) where
9393 getHeaders hdrs = case hdrs of
94− Header val `HCons` rest -> (headerName , toByteString' val):getHeaders rest
95− UndecodableHeader h `HCons` rest -> (headerName, h) : getHeaders rest
94+ Header val `HCons` rest -> (headerName , toHeader val):getHeaders rest
95+ UndecodableHeader h `HCons` rest -> (headerName, h) :getHeaders rest
9696 MissingHeader `HCons` rest -> getHeaders rest
9797 where headerName = CI.mk . pack $ symbolVal (Proxy :: Proxy h)
9898
9999instance OVERLAPPING_ GetHeaders (Headers '[] a) where
100100 getHeaders _ = []
101101
102−instance OVERLAPPABLE_ ( KnownSymbol h, GetHeaders (HList rest), ToByteString v )
102+instance OVERLAPPABLE_ ( KnownSymbol h, GetHeaders (HList rest), ToHttpApiData v )
103103 => GetHeaders (Headers (Header h v ': rest) a) where
104104 getHeaders hs = getHeaders $ getHeadersHList hs
105105
@@ -111,11 +111,11 @@ class AddHeader h v orig new
111111 addHeader :: v -> orig -> new -- ^ N.B.: The same header can't be added multiple times
112112
113113
114−instance OVERLAPPING_ ( KnownSymbol h, ToByteString v )
114+instance OVERLAPPING_ ( KnownSymbol h, ToHttpApiData v )
115115 => AddHeader h v (Headers (fst ': rest) a) (Headers (Header h v ': fst ': rest) a) where
116116 addHeader a (Headers resp heads) = Headers resp (HCons (Header a) heads)
117117
118−instance OVERLAPPABLE_ ( KnownSymbol h, ToByteString v
118+instance OVERLAPPABLE_ ( KnownSymbol h, ToHttpApiData v
119119 , new ~ (Headers '[Header h v] a) )
120120 => AddHeader h v a new where
121121 addHeader a resp = Headers resp (HCons (Header a) HNil)