@@ -1,34 +1,33 @@
1−{-# LANGUAGE AllowAmbiguousTypes #-}
1+{-# LANGUAGE CPP #-}
2{-# LANGUAGE ConstraintKinds #-}
2{-# LANGUAGE ConstraintKinds #-}
3{-# LANGUAGE DataKinds #-}
3{-# LANGUAGE DataKinds #-}
4{-# LANGUAGE DeriveDataTypeable #-}
4{-# LANGUAGE DeriveDataTypeable #-}
5{-# LANGUAGE FlexibleContexts #-}
5{-# LANGUAGE FlexibleContexts #-}
6{-# LANGUAGE FlexibleInstances #-}
6{-# LANGUAGE FlexibleInstances #-}
7−{-# LANGUAGE InstanceSigs #-}
8{-# LANGUAGE MultiParamTypeClasses #-}
7{-# LANGUAGE MultiParamTypeClasses #-}
9{-# LANGUAGE OverloadedStrings #-}
8{-# LANGUAGE OverloadedStrings #-}
10{-# LANGUAGE PolyKinds #-}
9{-# LANGUAGE PolyKinds #-}
11−{-# LANGUAGE QuantifiedConstraints #-}
12{-# LANGUAGE RankNTypes #-}
10{-# LANGUAGE RankNTypes #-}
13{-# LANGUAGE ScopedTypeVariables #-}
11{-# LANGUAGE ScopedTypeVariables #-}
14{-# LANGUAGE TupleSections #-}
12{-# LANGUAGE TupleSections #-}
15−{-# LANGUAGE TypeApplications #-}
16{-# LANGUAGE TypeFamilies #-}
13{-# LANGUAGE TypeFamilies #-}
17{-# LANGUAGE TypeOperators #-}
14{-# LANGUAGE TypeOperators #-}
18{-# LANGUAGE UndecidableInstances #-}
15{-# LANGUAGE UndecidableInstances #-}
17+#if MIN_VERSION_base(4,9,0) && __GLASGOW_HASKELL__ >= 802
18+#define HAS_TYPE_ERROR
21+#include "overlapping-compat.h"
20module Servant.Server.Internal
23module Servant.Server.Internal
21 ( module Servant.Server.Internal
24 ( module Servant.Server.Internal
22 , module Servant.Server.Internal.BasicAuth
25 , module Servant.Server.Internal.BasicAuth
23 , module Servant.Server.Internal.Context
26 , module Servant.Server.Internal.Context
24− , module Servant.Server.Internal.Delayed
25− , module Servant.Server.Internal.DelayedIO
26− , module Servant.Server.Internal.ErrorFormatter
27 , module Servant.Server.Internal.Handler
27 , module Servant.Server.Internal.Handler
28 , module Servant.Server.Internal.Router
28 , module Servant.Server.Internal.Router
29− , module Servant.Server.Internal.RouteResult
30 , module Servant.Server.Internal.RoutingApplication
29 , module Servant.Server.Internal.RoutingApplication
31− , module Servant.Server.Internal.ServerError
30+ , module Servant.Server.Internal.ServantErr
34import Control.Monad
33import Control.Monad
@@ -41,11 +40,12 @@ import qualified Data.ByteString as B
41import qualified Data.ByteString.Builder as BB
40import qualified Data.ByteString.Builder as BB
42import qualified Data.ByteString.Char8 as BC8
41import qualified Data.ByteString.Char8 as BC8
43import qualified Data.ByteString.Lazy as BL
42import qualified Data.ByteString.Lazy as BL
44−import Data.Constraint (Constraint, Dict(..))
45import Data.Either
43import Data.Either
46 (partitionEithers)
44 (partitionEithers)
47import Data.Maybe
45import Data.Maybe
48 (fromMaybe, isNothing, mapMaybe, maybeToList)
46 (fromMaybe, isNothing, mapMaybe, maybeToList)
49import Data.String
49import Data.String
50 (IsString (..))
50 (IsString (..))
51import Data.String.Conversions
51import Data.String.Conversions
@@ -54,9 +54,8 @@ import Data.Tagged
54 (Tagged (..), retag, untag)
54 (Tagged (..), retag, untag)
55import qualified Data.Text as T
55import qualified Data.Text as T
56import Data.Typeable
56import Data.Typeable
58import GHC.TypeLits
57import GHC.TypeLits
59− (KnownNat, KnownSymbol, TypeError, symbolVal)
58+ (KnownNat, KnownSymbol, natVal, symbolVal)
60import qualified Network.HTTP.Media as NHM
59import qualified Network.HTTP.Media as NHM
61import Network.HTTP.Types hiding
60import Network.HTTP.Types hiding
62 (Header, ResponseHeaders)
61 (Header, ResponseHeaders)
@@ -64,52 +63,42 @@ import Network.Socket
64 (SockAddr)
63 (SockAddr)
65import Network.Wai
64import Network.Wai
66 (Application, Request, httpVersion, isSecure, lazyRequestBody,
65 (Application, Request, httpVersion, isSecure, lazyRequestBody,
67− queryString, remoteHost, getRequestBodyChunk, requestHeaders,
66+ rawQueryString, remoteHost, requestHeaders, requestMethod,
68− requestMethod, responseLBS, responseStream, vault)
67+ responseLBS, responseStream, vault)
69import Prelude ()
68import Prelude ()
70import Prelude.Compat
69import Prelude.Compat
71import Servant.API
70import Servant.API
72− ((:<|>) (..), (:>), Accept (..), BasicAuth, Capture',
71+ ((:<|>) (..), (:>), Accept (..), BasicAuth,
73− CaptureAll, Description, EmptyAPI, Fragment,
72+ BoundaryStrategy (..), Capture', CaptureAll, Description,
74− FramingRender (..), FramingUnrender (..), FromSourceIO (..),
75− Header', If, IsSecure (..), NoContentVerb, QueryFlag,
73+ EmptyAPI, FramingRender (..), Header', If, IsSecure (..),
76− QueryParam', QueryParams, Raw, ReflectMethod (reflectMethod),
74+ QueryFlag, QueryParam', QueryParams, Raw,
77− RemoteHost, ReqBody', SBool (..), SBoolI (..), SourceIO,
75+ ReflectMethod (reflectMethod), RemoteHost, ReqBody',
76+ SBool (..), SBoolI (..), Stream, StreamGenerator (..),
78− Stream, StreamBody', Summary, ToSourceIO (..), Vault, Verb,
77+ Summary, ToStreamGenerator (..), Vault, Verb,
79− WithNamedContext, NamedRoutes)
78+ WithNamedContext)
80−import Servant.API.Generic (GenericMode(..), ToServant, ToServantApi, GServantProduct, toServant, fromServant)
81import Servant.API.ContentTypes
79import Servant.API.ContentTypes
82 (AcceptHeader (..), AllCTRender (..), AllCTUnrender (..),
80 (AcceptHeader (..), AllCTRender (..), AllCTUnrender (..),
83− AllMime, MimeRender (..), MimeUnrender (..), NoContent,
84− canHandleAcceptH)
81+ AllMime, MimeRender (..), canHandleAcceptH)
85import Servant.API.Modifiers
82import Servant.API.Modifiers
86 (FoldLenient, FoldRequired, RequestArgument,
83 (FoldLenient, FoldRequired, RequestArgument,
87 unfoldRequestArgument)
84 unfoldRequestArgument)
88import Servant.API.ResponseHeaders
85import Servant.API.ResponseHeaders
89 (GetHeaders, Headers, getHeaders, getResponse)
86 (GetHeaders, Headers, getHeaders, getResponse)
90−import Servant.API.Status
92−import qualified Servant.Types.SourceT as S
93−import Servant.API.TypeErrors
94import Web.HttpApiData
87import Web.HttpApiData
95− (FromHttpApiData, parseHeader, parseQueryParam, parseUrlPiece,
88+ (FromHttpApiData, parseHeader, parseQueryParam,
96− parseUrlPieces)
89+ parseUrlPieceMaybe, parseUrlPieces)
98import Servant.Server.Internal.BasicAuth
91import Servant.Server.Internal.BasicAuth
99import Servant.Server.Internal.Context
92import Servant.Server.Internal.Context
100−import Servant.Server.Internal.Delayed
101−import Servant.Server.Internal.DelayedIO
102−import Servant.Server.Internal.ErrorFormatter
103import Servant.Server.Internal.Handler
93import Servant.Server.Internal.Handler
104import Servant.Server.Internal.Router
94import Servant.Server.Internal.Router
105−import Servant.Server.Internal.RouteResult
106import Servant.Server.Internal.RoutingApplication
95import Servant.Server.Internal.RoutingApplication
107−import Servant.Server.Internal.ServerError
96+import Servant.Server.Internal.ServantErr
109import GHC.TypeLits
99import GHC.TypeLits
110 (ErrorMessage (..), TypeError)
100 (ErrorMessage (..), TypeError)
111−import Servant.API.TypeLevel
101+#endif
112− (AtLeastOneFragment, FragmentUnique)
114class HasServer api context where
103class HasServer api context where
115 type ServerT api (m :: * -> *) :: *
104 type ServerT api (m :: * -> *) :: *
@@ -173,31 +162,22 @@ instance (HasServer a context, HasServer b context) => HasServer (a :<|> b) cont
173-- > server = getBook
162-- > server = getBook
174-- > where getBook :: Text -> Handler Book
163-- > where getBook :: Text -> Handler Book
175-- > getBook isbn = ...
164-- > getBook isbn = ...
176−instance (KnownSymbol capture, FromHttpApiData a, Typeable a
165+instance (KnownSymbol capture, FromHttpApiData a, HasServer api context)
177− , HasServer api context, SBoolI (FoldLenient mods)
178− , HasContextEntry (MkContextWithErrorFormatter context) ErrorFormatters
180 => HasServer (Capture' mods capture a :> api) context where
166 => HasServer (Capture' mods capture a :> api) context where
182 type ServerT (Capture' mods capture a :> api) m =
168 type ServerT (Capture' mods capture a :> api) m =
183− If (FoldLenient mods) (Either String a) a -> ServerT api m
169+ a -> ServerT api m
185 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
171 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
187 route Proxy context d =
173 route Proxy context d =
188− CaptureRouter [hint] $
174+ CaptureRouter $
189 route (Proxy :: Proxy api)
175 route (Proxy :: Proxy api)
191− (addCapture d $ \ txt -> withRequest $ \ request ->
177+ (addCapture d $ \ txt -> case parseUrlPieceMaybe txt of
192− case ( sbool :: SBool (FoldLenient mods)
178+ Nothing -> delayedFail err400
193− , parseUrlPiece txt :: Either T.Text a) of
194− (SFalse, Left e) -> delayedFail $ formatError rep request $ cs e
195− (SFalse, Right v) -> return v
179+ Just v -> return v
196− (STrue, piece) -> return $ (either (Left . cs) Right) piece)
180+ )
198− rep = typeRep (Proxy :: Proxy Capture')
199− formatError = urlParseErrorFormatter $ getContextEntry (mkContextWithErrorFormatter context)
200− hint = CaptureHint (T.pack $ symbolVal $ Proxy @capture) (typeRep (Proxy :: Proxy a))
202-- | If you use 'CaptureAll' in one of the endpoints for your API,
182-- | If you use 'CaptureAll' in one of the endpoints for your API,
203-- this automatically requires your server-side handler to be a
183-- this automatically requires your server-side handler to be a
@@ -216,10 +196,7 @@ instance (KnownSymbol capture, FromHttpApiData a, Typeable a
216-- > server = getSourceFile
196-- > server = getSourceFile
217-- > where getSourceFile :: [Text] -> Handler Book
197-- > where getSourceFile :: [Text] -> Handler Book
218-- > getSourceFile pathSegments = ...
198-- > getSourceFile pathSegments = ...
219−instance (KnownSymbol capture, FromHttpApiData a, Typeable a
199+instance (KnownSymbol capture, FromHttpApiData a, HasServer api context)
220− , HasServer api context
221− , HasContextEntry (MkContextWithErrorFormatter context) ErrorFormatters
223 => HasServer (CaptureAll capture a :> api) context where
200 => HasServer (CaptureAll capture a :> api) context where
225 type ServerT (CaptureAll capture a :> api) m =
202 type ServerT (CaptureAll capture a :> api) m =
@@ -228,18 +205,14 @@ instance (KnownSymbol capture, FromHttpApiData a, Typeable a
228 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
205 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
230 route Proxy context d =
207 route Proxy context d =
231− CaptureAllRouter [hint] $
208+ CaptureAllRouter $
232 route (Proxy :: Proxy api)
209 route (Proxy :: Proxy api)
234− (addCapture d $ \ txts -> withRequest $ \ request ->
235− case parseUrlPieces txts of
211+ (addCapture d $ \ txts -> case parseUrlPieces txts of
236− Left e -> delayedFail $ formatError rep request $ cs e
212+ Left _ -> delayedFail err400
237− Right v -> return v
213+ Right v -> return v
240− rep = typeRep (Proxy :: Proxy CaptureAll)
241− formatError = urlParseErrorFormatter $ getContextEntry (mkContextWithErrorFormatter context)
242− hint = CaptureHint (T.pack $ symbolVal $ Proxy @capture) (typeRep (Proxy :: Proxy [a]))
244allowedMethodHead :: Method -> Request -> Bool
217allowedMethodHead :: Method -> Request -> Bool
245allowedMethodHead method request = method == methodGet && requestMethod request == methodHead
218allowedMethodHead method request = method == methodGet && requestMethod request == methodHead
@@ -259,10 +232,10 @@ methodCheck method request
259-- body check is no longer an option. However, we now run the accept
232-- body check is no longer an option. However, we now run the accept
260-- check before the body check and can therefore afford to make it
233-- check before the body check and can therefore afford to make it
261-- recoverable.
234-- recoverable.
262−acceptCheck :: (AllMime list) => Proxy list -> AcceptHeader -> DelayedIO ()
235+acceptCheck :: (AllMime list) => Proxy list -> B.ByteString -> DelayedIO ()
263acceptCheck proxy accH
236acceptCheck proxy accH
264− | canHandleAcceptH proxy accH = return ()
237+ | canHandleAcceptH proxy (AcceptHeader accH) = return ()
265− | otherwise = delayedFail err406
238+ | otherwise = delayedFail err406
267methodRouter :: (AllCTRender ctypes a)
240methodRouter :: (AllCTRender ctypes a)
268 => (b -> ([(HeaderName, B.ByteString)], a))
241 => (b -> ([(HeaderName, B.ByteString)], a))
@@ -272,29 +245,18 @@ methodRouter :: (AllCTRender ctypes a)
272methodRouter splitHeaders method proxy status action = leafRouter route'
245methodRouter splitHeaders method proxy status action = leafRouter route'
274 route' env request respond =
247 route' env request respond =
275− let accH = getAcceptHeader request
248+ let accH = fromMaybe ct_wildcard $ lookup hAccept $ requestHeaders request
276 in runAction (action `addMethodCheck` methodCheck method request
249 in runAction (action `addMethodCheck` methodCheck method request
277 `addAcceptCheck` acceptCheck proxy accH
250 `addAcceptCheck` acceptCheck proxy accH
278 ) env request respond $ \ output -> do
251 ) env request respond $ \ output -> do
279 let (headers, b) = splitHeaders output
252 let (headers, b) = splitHeaders output
280− case handleAcceptH proxy accH b of
253+ case handleAcceptH proxy (AcceptHeader accH) b of
281 Nothing -> FailFatal err406 -- this should not happen (checked before), so we make it fatal if it does
254 Nothing -> FailFatal err406 -- this should not happen (checked before), so we make it fatal if it does
282 Just (contentT, body) ->
255 Just (contentT, body) ->
283 let bdy = if allowedMethodHead method request then "" else body
256 let bdy = if allowedMethodHead method request then "" else body
284 in Route $ responseLBS status ((hContentType, cs contentT) : headers) bdy
257 in Route $ responseLBS status ((hContentType, cs contentT) : headers) bdy
286−noContentRouter :: Method
259+instance OVERLAPPABLE_
288− -> Delayed env (Handler b)
290−noContentRouter method status action = leafRouter route'
292− route' env request respond =
293− runAction (action `addMethodCheck` methodCheck method request)
294− env request respond $ \ _output ->
295− Route $ responseLBS status [] ""
297−instance {-# OVERLAPPABLE #-}
298 ( AllCTRender ctypes a, ReflectMethod method, KnownNat status
260 ( AllCTRender ctypes a, ReflectMethod method, KnownNat status
299 ) => HasServer (Verb method status ctypes a) context where
261 ) => HasServer (Verb method status ctypes a) context where
@@ -303,9 +265,9 @@ instance {-# OVERLAPPABLE #-}
304 route Proxy _ = methodRouter ([],) method (Proxy :: Proxy ctypes) status
266 route Proxy _ = methodRouter ([],) method (Proxy :: Proxy ctypes) status
305 where method = reflectMethod (Proxy :: Proxy method)
267 where method = reflectMethod (Proxy :: Proxy method)
306− status = statusFromNat (Proxy :: Proxy status)
268+ status = toEnum . fromInteger $ natVal (Proxy :: Proxy status)
308−instance {-# OVERLAPPING #-}
270+instance OVERLAPPING_
309 ( AllCTRender ctypes a, ReflectMethod method, KnownNat status
271 ( AllCTRender ctypes a, ReflectMethod method, KnownNat status
310 , GetHeaders (Headers h a)
272 , GetHeaders (Headers h a)
311 ) => HasServer (Verb method status ctypes (Headers h a)) context where
273 ) => HasServer (Verb method status ctypes (Headers h a)) context where
@@ -315,46 +277,37 @@ instance {-# OVERLAPPING #-}
316 route Proxy _ = methodRouter (\x -> (getHeaders x, getResponse x)) method (Proxy :: Proxy ctypes) status
278 route Proxy _ = methodRouter (\x -> (getHeaders x, getResponse x)) method (Proxy :: Proxy ctypes) status
317 where method = reflectMethod (Proxy :: Proxy method)
279 where method = reflectMethod (Proxy :: Proxy method)
318− status = statusFromNat (Proxy :: Proxy status)
280+ status = toEnum . fromInteger $ natVal (Proxy :: Proxy status)
320−instance (ReflectMethod method) =>
321− HasServer (NoContentVerb method) context where
323− type ServerT (NoContentVerb method) m = m NoContent
324− hoistServerWithContext _ _ nt s = nt s
326− route Proxy _ = noContentRouter method status204
327− where method = reflectMethod (Proxy :: Proxy method)
329−instance {-# OVERLAPPABLE #-}
283+instance OVERLAPPABLE_
330− ( MimeRender ctype chunk, ReflectMethod method, KnownNat status,
284+ ( MimeRender ctype a, ReflectMethod method, KnownNat status,
331− FramingRender framing, ToSourceIO chunk a
285+ FramingRender framing ctype, ToStreamGenerator b a
332− ) => HasServer (Stream method status framing ctype a) context where
286+ ) => HasServer (Stream method status framing ctype b) context where
334− type ServerT (Stream method status framing ctype a) m = m a
288+ type ServerT (Stream method status framing ctype b) m = m b
335 hoistServerWithContext _ _ nt s = nt s
289 hoistServerWithContext _ _ nt s = nt s
337 route Proxy _ = streamRouter ([],) method status (Proxy :: Proxy framing) (Proxy :: Proxy ctype)
291 route Proxy _ = streamRouter ([],) method status (Proxy :: Proxy framing) (Proxy :: Proxy ctype)
338 where method = reflectMethod (Proxy :: Proxy method)
292 where method = reflectMethod (Proxy :: Proxy method)
339− status = statusFromNat (Proxy :: Proxy status)
293+ status = toEnum . fromInteger $ natVal (Proxy :: Proxy status)
342−instance {-# OVERLAPPING #-}
295+instance OVERLAPPING_
343− ( MimeRender ctype chunk, ReflectMethod method, KnownNat status,
296+ ( MimeRender ctype a, ReflectMethod method, KnownNat status,
344− FramingRender framing, ToSourceIO chunk a,
297+ FramingRender framing ctype, ToStreamGenerator b a,
345− GetHeaders (Headers h a)
298+ GetHeaders (Headers h b)
346− ) => HasServer (Stream method status framing ctype (Headers h a)) context where
299+ ) => HasServer (Stream method status framing ctype (Headers h b)) context where
348− type ServerT (Stream method status framing ctype (Headers h a)) m = m (Headers h a)
301+ type ServerT (Stream method status framing ctype (Headers h b)) m = m (Headers h b)
349 hoistServerWithContext _ _ nt s = nt s
302 hoistServerWithContext _ _ nt s = nt s
351 route Proxy _ = streamRouter (\x -> (getHeaders x, getResponse x)) method status (Proxy :: Proxy framing) (Proxy :: Proxy ctype)
304 route Proxy _ = streamRouter (\x -> (getHeaders x, getResponse x)) method status (Proxy :: Proxy framing) (Proxy :: Proxy ctype)
352 where method = reflectMethod (Proxy :: Proxy method)
305 where method = reflectMethod (Proxy :: Proxy method)
353− status = statusFromNat (Proxy :: Proxy status)
306+ status = toEnum . fromInteger $ natVal (Proxy :: Proxy status)
356−streamRouter :: forall ctype a c chunk env framing. (MimeRender ctype chunk, FramingRender framing, ToSourceIO chunk a) =>
309+streamRouter :: (MimeRender ctype a, FramingRender framing ctype, ToStreamGenerator b a) =>
357− (c -> ([(HeaderName, B.ByteString)], a))
310+ (c -> ([(HeaderName, B.ByteString)], b))
358 -> Method
311 -> Method
359 -> Status
312 -> Status
360 -> Proxy framing
313 -> Proxy framing
@@ -362,7 +315,7 @@ streamRouter :: forall ctype a c chunk env framing. (MimeRender ctype chunk, Fra
362 -> Delayed env (Handler c)
315 -> Delayed env (Handler c)
363 -> Router env
316 -> Router env
364streamRouter splitHeaders method status framingproxy ctypeproxy action = leafRouter $ \env request respond ->
317streamRouter splitHeaders method status framingproxy ctypeproxy action = leafRouter $ \env request respond ->
365− let AcceptHeader accH = getAcceptHeader request
318+ let accH = fromMaybe ct_wildcard $ lookup hAccept $ requestHeaders request
366 cmediatype = NHM.matchAccept [contentType ctypeproxy] accH
319 cmediatype = NHM.matchAccept [contentType ctypeproxy] accH
367 accCheck = when (isNothing cmediatype) $ delayedFail err406
320 accCheck = when (isNothing cmediatype) $ delayedFail err406
368 contentHeader = (hContentType, NHM.renderHeader . maybeToList $ cmediatype)
321 contentHeader = (hContentType, NHM.renderHeader . maybeToList $ cmediatype)
@@ -370,19 +323,28 @@ streamRouter splitHeaders method status framingproxy ctypeproxy action = leafRou
370 `addAcceptCheck` accCheck
323 `addAcceptCheck` accCheck
371 ) env request respond $ \ output ->
324 ) env request respond $ \ output ->
372 let (headers, fa) = splitHeaders output
325 let (headers, fa) = splitHeaders output
373− sourceT = toSourceIO fa
326+ k = getStreamGenerator . toStreamGenerator $ fa in
374− S.SourceT kStepLBS = framingRender framingproxy (mimeRender ctypeproxy :: chunk -> BL.ByteString) sourceT
375− in Route $ responseStream status (contentHeader : headers) $ \write flush -> do
327+ Route $ responseStream status (contentHeader : headers) $ \write flush -> do
376− let loop S.Stop = flush
328+ write . BB.lazyByteString $ header framingproxy ctypeproxy
377− loop (S.Error err) = fail err -- TODO: throw better error
329+ case boundary framingproxy ctypeproxy of
378− loop (S.Skip s) = loop s
330+ BoundaryStrategyBracket f ->
379− loop (S.Effect ms) = ms >>= loop
331+ let go x = let bs = mimeRender ctypeproxy x
380− loop (S.Yield lbs s) = do
332+ (before, after) = f bs
333+ in write ( BB.lazyByteString before
334+ <> BB.lazyByteString bs
335+ <> BB.lazyByteString after) >> flush
337+ BoundaryStrategyIntersperse sep -> k
339+ write . BB.lazyByteString . mimeRender ctypeproxy $ x
381− write (BB.lazyByteString lbs)
342+ write . (BB.lazyByteString sep <>) . BB.lazyByteString . mimeRender ctypeproxy $ x
383− loop s
344+ BoundaryStrategyGeneral f ->
384−
345+ let go = (>> flush) . write . BB.lazyByteString . f . mimeRender ctypeproxy
385− kStepLBS loop
346+ in k go go
347+ write . BB.lazyByteString $ trailer framingproxy ctypeproxy
387-- | If you use 'Header' in one of the endpoints for your API,
349-- | If you use 'Header' in one of the endpoints for your API,
388-- this automatically requires your server-side handler to be a function
350-- this automatically requires your server-side handler to be a function
@@ -407,7 +369,6 @@ streamRouter splitHeaders method status framingproxy ctypeproxy action = leafRou
408 (KnownSymbol sym, FromHttpApiData a, HasServer api context
370 (KnownSymbol sym, FromHttpApiData a, HasServer api context
409 , SBoolI (FoldRequired mods), SBoolI (FoldLenient mods)
371 , SBoolI (FoldRequired mods), SBoolI (FoldLenient mods)
410− , HasContextEntry (MkContextWithErrorFormatter context) ErrorFormatters
412 => HasServer (Header' mods sym a :> api) context where
373 => HasServer (Header' mods sym a :> api) context where
@@ -419,9 +380,6 @@ instance
419 route Proxy context subserver = route (Proxy :: Proxy api) context $
380 route Proxy context subserver = route (Proxy :: Proxy api) context $
420 subserver `addHeaderCheck` withRequest headerCheck
381 subserver `addHeaderCheck` withRequest headerCheck
422− rep = typeRep (Proxy :: Proxy Header')
423− formatError = headerParseErrorFormatter $ getContextEntry (mkContextWithErrorFormatter context)
425 headerName :: IsString n => n
383 headerName :: IsString n => n
426 headerName = fromString $ symbolVal (Proxy :: Proxy sym)
384 headerName = fromString $ symbolVal (Proxy :: Proxy sym)
@@ -432,13 +390,15 @@ instance
432 mev :: Maybe (Either T.Text a)
390 mev :: Maybe (Either T.Text a)
433 mev = fmap parseHeader $ lookup headerName (requestHeaders req)
391 mev = fmap parseHeader $ lookup headerName (requestHeaders req)
435− errReq = delayedFailFatal $ formatError rep req
393+ errReq = delayedFailFatal err400
436− $ "Header " <> headerName <> " is required"
394+ { errBody = "Header " <> headerName <> " is required"
438− errSt e = delayedFailFatal $ formatError rep req
397+ errSt e = delayedFailFatal err400
439− $ cs $ "Error parsing header "
398+ { errBody = cs $ "Error parsing header "
440− <> headerName
399+ <> headerName
441− <> " failed: " <> e
400+ <> " failed: " <> e
443-- | If you use @'QueryParam' "author" Text@ in one of the endpoints for your API,
403-- | If you use @'QueryParam' "author" Text@ in one of the endpoints for your API,
444-- this automatically requires your server-side handler to be a function
404-- this automatically requires your server-side handler to be a function
@@ -464,7 +424,6 @@ instance
465 ( KnownSymbol sym, FromHttpApiData a, HasServer api context
425 ( KnownSymbol sym, FromHttpApiData a, HasServer api context
466 , SBoolI (FoldRequired mods), SBoolI (FoldLenient mods)
426 , SBoolI (FoldRequired mods), SBoolI (FoldLenient mods)
467− , HasContextEntry (MkContextWithErrorFormatter context) ErrorFormatters
469 => HasServer (QueryParam' mods sym a :> api) context where
428 => HasServer (QueryParam' mods sym a :> api) context where
@@ -474,12 +433,9 @@ instance
474 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
433 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
476 route Proxy context subserver =
435 route Proxy context subserver =
477− let querytext = queryToQueryText . queryString
436+ let querytext req = parseQueryText $ rawQueryString req
478 paramname = cs $ symbolVal (Proxy :: Proxy sym)
437 paramname = cs $ symbolVal (Proxy :: Proxy sym)
480− rep = typeRep (Proxy :: Proxy QueryParam')
481− formatError = urlParseErrorFormatter $ getContextEntry (mkContextWithErrorFormatter context)
483 parseParam :: Request -> DelayedIO (RequestArgument mods a)
439 parseParam :: Request -> DelayedIO (RequestArgument mods a)
484 parseParam req =
440 parseParam req =
485 unfoldRequestArgument (Proxy :: Proxy mods) errReq errSt mev
441 unfoldRequestArgument (Proxy :: Proxy mods) errReq errSt mev
@@ -487,12 +443,14 @@ instance
487 mev :: Maybe (Either T.Text a)
443 mev :: Maybe (Either T.Text a)
488 mev = fmap parseQueryParam $ join $ lookup paramname $ querytext req
444 mev = fmap parseQueryParam $ join $ lookup paramname $ querytext req
490− errReq = delayedFailFatal $ formatError rep req
446+ errReq = delayedFailFatal err400
491− $ cs $ "Query parameter " <> paramname <> " is required"
447+ { errBody = cs $ "Query parameter " <> paramname <> " is required"
493− errSt e = delayedFailFatal $ formatError rep req
450+ errSt e = delayedFailFatal err400
494− $ cs $ "Error parsing query parameter "
451+ { errBody = cs $ "Error parsing query parameter "
495− <> paramname <> " failed: " <> e
452+ <> paramname <> " failed: " <> e
497 delayed = addParameterCheck subserver . withRequest $ \req ->
455 delayed = addParameterCheck subserver . withRequest $ \req ->
498 parseParam req
456 parseParam req
@@ -518,8 +476,7 @@ instance
518-- > server = getBooksBy
476-- > server = getBooksBy
519-- > where getBooksBy :: [Text] -> Handler [Book]
477-- > where getBooksBy :: [Text] -> Handler [Book]
520-- > getBooksBy authors = ...return all books by these authors...
478-- > getBooksBy authors = ...return all books by these authors...
521−instance (KnownSymbol sym, FromHttpApiData a, HasServer api context
479+instance (KnownSymbol sym, FromHttpApiData a, HasServer api context)
522− , HasContextEntry (MkContextWithErrorFormatter context) ErrorFormatters)
523 => HasServer (QueryParams sym a :> api) context where
480 => HasServer (QueryParams sym a :> api) context where
525 type ServerT (QueryParams sym a :> api) m =
482 type ServerT (QueryParams sym a :> api) m =
@@ -530,23 +487,21 @@ instance (KnownSymbol sym, FromHttpApiData a, HasServer api context
530 route Proxy context subserver = route (Proxy :: Proxy api) context $
487 route Proxy context subserver = route (Proxy :: Proxy api) context $
531 subserver `addParameterCheck` withRequest paramsCheck
488 subserver `addParameterCheck` withRequest paramsCheck
533− rep = typeRep (Proxy :: Proxy QueryParams)
534− formatError = urlParseErrorFormatter $ getContextEntry (mkContextWithErrorFormatter context)
536 paramname = cs $ symbolVal (Proxy :: Proxy sym)
490 paramname = cs $ symbolVal (Proxy :: Proxy sym)
537 paramsCheck req =
491 paramsCheck req =
538 case partitionEithers $ fmap parseQueryParam params of
492 case partitionEithers $ fmap parseQueryParam params of
539 ([], parsed) -> return parsed
493 ([], parsed) -> return parsed
540− (errs, _) -> delayedFailFatal $ formatError rep req
494+ (errs, _) -> delayedFailFatal err400
541− $ cs $ "Error parsing query parameter(s) "
495+ { errBody = cs $ "Error parsing query parameter(s) "
542− <> paramname <> " failed: "
496+ <> paramname <> " failed: "
543− <> T.intercalate ", " errs
497+ <> T.intercalate ", " errs
545 params :: [T.Text]
500 params :: [T.Text]
546 params = mapMaybe snd
501 params = mapMaybe snd
547 . filter (looksLikeParam . fst)
502 . filter (looksLikeParam . fst)
548− . queryToQueryText
503+ . parseQueryText
549− . queryString
504+ . rawQueryString
552 looksLikeParam name = name == paramname || name == (paramname <> "[]")
507 looksLikeParam name = name == paramname || name == (paramname <> "[]")
@@ -572,7 +527,7 @@ instance (KnownSymbol sym, HasServer api context)
572 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
527 hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
574 route Proxy context subserver =
529 route Proxy context subserver =
575− let querytext = queryToQueryText . queryString
530+ let querytext r = parseQueryText $ rawQueryString r
576 param r = case lookup paramname (querytext r) of
531 param r = case lookup paramname (querytext r) of
577 Just Nothing -> True -- param is there, with no value
532 Just Nothing -> True -- param is there, with no value
578 Just (Just v) -> examine v -- param with a value
533 Just (Just v) -> examine v -- param with a value
@@ -614,7 +569,7 @@ instance HasServer Raw context where
614-- The @Content-Type@ header is inspected, and the list provided is used to
569-- The @Content-Type@ header is inspected, and the list provided is used to
615-- attempt deserialization. If the request does not have a @Content-Type@
570-- attempt deserialization. If the request does not have a @Content-Type@
616-- header, it is treated as @application/octet-stream@ (as specified in
571-- header, it is treated as @application/octet-stream@ (as specified in
617−-- [RFC 7231 section 3.1.1.5](http://tools.ietf.org/html/rfc7231#section-3.1.1.5)).
572+-- <http://tools.ietf.org/html/rfc7231#section-3.1.1.5 RFC7231>.
618-- This lets servant worry about extracting it from the request and turning
573-- This lets servant worry about extracting it from the request and turning
619-- it into a value of the type you specify.
574-- it into a value of the type you specify.
@@ -630,7 +585,6 @@ instance HasServer Raw context where
630-- > where postBook :: Book -> Handler Book
585-- > where postBook :: Book -> Handler Book
631-- > postBook book = ...insert into your db...
586-- > postBook book = ...insert into your db...
632instance ( AllCTUnrender list a, HasServer api context, SBoolI (FoldLenient mods)
587instance ( AllCTUnrender list a, HasServer api context, SBoolI (FoldLenient mods)
633− , HasContextEntry (MkContextWithErrorFormatter context) ErrorFormatters
634 ) => HasServer (ReqBody' mods list a :> api) context where
588 ) => HasServer (ReqBody' mods list a :> api) context where
636 type ServerT (ReqBody' mods list a :> api) m =
590 type ServerT (ReqBody' mods list a :> api) m =
@@ -642,9 +596,6 @@ instance ( AllCTUnrender list a, HasServer api context, SBoolI (FoldLenient mods
642 = route (Proxy :: Proxy api) context $
596 = route (Proxy :: Proxy api) context $
643 addBodyCheck subserver ctCheck bodyCheck
597 addBodyCheck subserver ctCheck bodyCheck
645− rep = typeRep (Proxy :: Proxy ReqBody')
646− formatError = bodyParserErrorFormatter $ getContextEntry (mkContextWithErrorFormatter context)
648 -- Content-Type check, we only lookup we can try to parse the request body
599 -- Content-Type check, we only lookup we can try to parse the request body
649 ctCheck = withRequest $ \ request -> do
600 ctCheck = withRequest $ \ request -> do
650 -- See HTTP RFC 2616, section 7.2.1
601 -- See HTTP RFC 2616, section 7.2.1
@@ -663,34 +614,9 @@ instance ( AllCTUnrender list a, HasServer api context, SBoolI (FoldLenient mods
663 case sbool :: SBool (FoldLenient mods) of
614 case sbool :: SBool (FoldLenient mods) of
664 STrue -> return mrqbody
615 STrue -> return mrqbody
665 SFalse -> case mrqbody of
616 SFalse -> case mrqbody of
666− Left e -> delayedFailFatal $ formatError rep request e
617+ Left e -> delayedFailFatal err400 { errBody = cs e }
667 Right v -> return v
618 Right v -> return v
670− ( FramingUnrender framing, FromSourceIO chunk a, MimeUnrender ctype chunk
671− , HasServer api context
672− ) => HasServer (StreamBody' mods framing ctype a :> api) context
674− type ServerT (StreamBody' mods framing ctype a :> api) m = a -> ServerT api m
676− hoistServerWithContext _ pc nt s = hoistServerWithContext (Proxy :: Proxy api) pc nt . s
678− route Proxy context subserver = route (Proxy :: Proxy api) context $
679− addBodyCheck subserver ctCheck bodyCheck
681− ctCheck :: DelayedIO (SourceIO chunk -> a)
682− -- TODO: do content-type check
683− ctCheck = return fromSourceIO
685− bodyCheck :: (SourceIO chunk -> a) -> DelayedIO a
686− bodyCheck fromRS = withRequest $ \req -> do
687− let mimeUnrender' = mimeUnrender (Proxy :: Proxy ctype) :: BL.ByteString -> Either String chunk
688− let framingUnrender' = framingUnrender (Proxy :: Proxy framing) mimeUnrender' :: SourceIO B.ByteString -> SourceIO chunk
689− let body = getRequestBodyChunk req
690− let rs = S.fromAction B.null body
691− let rs' = fromRS $ framingUnrender' rs
694-- | Make sure the incoming request starts with @"/path"@, strip it and
620-- | Make sure the incoming request starts with @"/path"@, strip it and
695-- pass the rest of the request path to @api@.
621-- pass the rest of the request path to @api@.
696instance (KnownSymbol path, HasServer api context) => HasServer (path :> api) context where
622instance (KnownSymbol path, HasServer api context) => HasServer (path :> api) context where
@@ -755,12 +681,12 @@ data EmptyServer = EmptyServer deriving (Typeable, Eq, Show, Bounded, Enum)
755emptyServer :: ServerT EmptyAPI m
681emptyServer :: ServerT EmptyAPI m
756emptyServer = Tagged EmptyServer
682emptyServer = Tagged EmptyServer
758−-- | The server for an `EmptyAPI` is `emptyServer`.
684+-- | The server for an `EmptyAPI` is `emptyAPIServer`.
760-- > type MyApi = "nothing" :> EmptyApi
686-- > type MyApi = "nothing" :> EmptyApi
762-- > server :: Server MyApi
688-- > server :: Server MyApi
763−-- > server = emptyServer
689+-- > server = emptyAPIServer
764instance HasServer EmptyAPI context where
690instance HasServer EmptyAPI context where
765 type ServerT EmptyAPI m = Tagged m EmptyServer
691 type ServerT EmptyAPI m = Tagged m EmptyServer
@@ -789,10 +715,7 @@ instance ( KnownSymbol realm
789-- * helpers
715-- * helpers
791ct_wildcard :: B.ByteString
717ct_wildcard :: B.ByteString
792−ct_wildcard = "*" <> "/" <> "*"
718+ct_wildcard = "*" <> "/" <> "*" -- Because CPP
794−getAcceptHeader :: Request -> AcceptHeader
795−getAcceptHeader = AcceptHeader . fromMaybe ct_wildcard . lookup hAccept . requestHeaders
797-- * General Authentication
720-- * General Authentication
@@ -817,15 +740,39 @@ instance (HasContextEntry context (NamedContext name subContext), HasServer subA
817 hoistServerWithContext _ _ nt s = hoistServerWithContext (Proxy :: Proxy subApi) (Proxy :: Proxy subContext) nt s
740 hoistServerWithContext _ _ nt s = hoistServerWithContext (Proxy :: Proxy subApi) (Proxy :: Proxy subContext) nt s
819-------------------------------------------------------------------------------
742-------------------------------------------------------------------------------
820−-- Custom type errors
743+-- TypeError helpers
821-------------------------------------------------------------------------------
744-------------------------------------------------------------------------------
823−-- Erroring instance for 'HasServer' when a combinator is not fully applied
746+#ifdef HAS_TYPE_ERROR
747+-- | This instance catches mistakes when there are non-saturated
748+-- type applications on LHS of ':>'.
750+-- >>> serve (Proxy :: Proxy (Capture "foo" :> Get '[JSON] Int)) (error "...")
752+-- ...Expected something of kind Symbol or *, got: k -> l on the LHS of ':>'.
753+-- ...Maybe you haven't applied enough arguments to
754+-- ...Capture' '[] "foo"
757+-- >>> undefined :: Server (Capture "foo" :> Get '[JSON] Int)
759+-- ...Expected something of kind Symbol or *, got: k -> l on the LHS of ':>'.
760+-- ...Maybe you haven't applied enough arguments to
761+-- ...Capture' '[] "foo"
824−instance TypeError (PartialApplication HasServer arr) => HasServer ((arr :: a -> b) :> sub) context
764+instance TypeError (HasServerArrowKindError arr) => HasServer ((arr :: k -> l) :> api) context
826− type ServerT (arr :> sub) _ = TypeError (PartialApplication (HasServer :: * -> [*] -> Constraint) arr)
766+ type ServerT (arr :> api) m = TypeError (HasServerArrowKindError arr)
827− route = error "unreachable"
767+ -- it doens't really matter what sub route we peak
768+ route _ _ _ = error "servant-server panic: impossible happened in HasServer (arr :> api)"
828− hoistServerWithContext _ _ _ _ = error "unreachable"
769+ hoistServerWithContext _ _ _ = id
771+-- Cannot have TypeError here, otherwise use of this symbol will error :)
772+type HasServerArrowKindError arr =
773+ 'Text "Expected something of kind Symbol or *, got: k -> l on the LHS of ':>'."
774+ ':$$: 'Text "Maybe you haven't applied enough arguments to"
830-- | This instance prevents from accidentally using '->' instead of ':>'
777-- | This instance prevents from accidentally using '->' instead of ':>'
@@ -859,103 +806,7 @@ type HasServerArrowTypeError a b =
859 ':$$: 'ShowType a
806 ':$$: 'ShowType a
860 ':$$: 'Text "and"
807 ':$$: 'Text "and"
861 ':$$: 'ShowType b
808 ':$$: 'ShowType b
863−-- Erroring instances for 'HasServer' for unknown API combinators
865−-- XXX: This omits the @context@ parameter, e.g.:
867−-- "There is no instance for HasServer (Bool :> …)". Do we care ?
868−instance {-# OVERLAPPABLE #-} TypeError (NoInstanceForSub HasServer ty) => HasServer (ty :> sub) context
870−instance {-# OVERLAPPABLE #-} TypeError (NoInstanceFor (HasServer api context)) => HasServer api context
872−-- | Ignore @'Fragment'@ in server handlers.
873−-- See <https://ietf.org/rfc/rfc2616.html#section-15.1.3> for more details.
877−-- > type MyApi = "books" :> Fragment Text :> Get '[JSON] [Book]
879−-- > server :: Server MyApi
880−-- > server = getBooks
881−-- > where getBooks :: Handler [Book]
882−-- > getBooks = ...return all books...
883−instance (AtLeastOneFragment api, FragmentUnique (Fragment a1 :> api), HasServer api context)
884− => HasServer (Fragment a1 :> api) context where
885− type ServerT (Fragment a1 :> api) m = ServerT api m
887− route _ = route (Proxy :: Proxy api)
889− hoistServerWithContext _ = hoistServerWithContext (Proxy :: Proxy api)
892-- >>> import Servant
812-- >>> import Servant
894−-- | A type that specifies that an API record contains a server implementation.
895−data AsServerT (m :: * -> *)
896−instance GenericMode (AsServerT m) where
897− type AsServerT m :- api = ServerT api m
899−type AsServer = AsServerT Handler
902−-- | Set of constraints required to convert to / from vanilla server types.
903−type GServerConstraints api m =
904− ( ToServant api (AsServerT m) ~ ServerT (ToServantApi api) m
905− , GServantProduct (Rep (api (AsServerT m)))
908−-- | This class is a necessary evil: in the implementation of 'HasServer' for
909−-- @'NamedRoutes' api@, we essentially need the quantified constraint @forall
910−-- m. 'GServerConstraints' m@ to hold.
912−-- We cannot require do that directly as the definition of 'GServerConstraints'
913−-- contains type family applications ('Rep' and 'ServerT'). The trick is to hide
914−-- those type family applications behind a typeclass providing evidence for
915−-- @'GServerConstraints' api m@ in the form of a dictionary, and require that
916−-- @forall m. 'GServer' api m@ instead.
918−-- Users shouldn't have to worry about this class, as the only possible instance
919−-- is provided in this module for all record APIs.
921−class GServer (api :: * -> *) (m :: * -> *) where
922− gServerProof :: Dict (GServerConstraints api m)
925− ( ToServant api (AsServerT m) ~ ServerT (ToServantApi api) m
926− , GServantProduct (Rep (api (AsServerT m)))
927− ) => GServer api m where
931− ( HasServer (ToServantApi api) context
932− , forall m. Generic (api (AsServerT m))
933− , forall m. GServer api m
934− ) => HasServer (NamedRoutes api) context where
936− type ServerT (NamedRoutes api) m = api (AsServerT m)
939− :: Proxy (NamedRoutes api)
941− -> Delayed env (api (AsServerT Handler))
943− route _ ctx delayed =
944− case gServerProof @api @Handler of
945− Dict -> route (Proxy @(ToServantApi api)) ctx (toServant <$> delayed)
947− hoistServerWithContext
948− :: forall m n. Proxy (NamedRoutes api)
950− -> (forall x. m x -> n x)
951− -> api (AsServerT m)
952− -> api (AsServerT n)
953− hoistServerWithContext _ pctx nat server =
954− case (gServerProof @api @m, gServerProof @api @n) of
956− fromServant servantSrvN
958− servantSrvM :: ServerT (ToServantApi api) m =
960− servantSrvN :: ServerT (ToServantApi api) n =
961− hoistServerWithContext (Proxy @(ToServantApi api)) pctx nat servantSrvM