module Network.HTTP.Toolkit.Request (
RequestPath
, Request(..)
, readRequest
, readRequestWithLimit
, parseRequestLine
, sendRequest
, formatRequestLine
, determineRequestBodyType
) where
import Control.Applicative
import Control.Exception
import Data.Maybe
import Data.Foldable
import Data.Traversable
import Data.ByteString (ByteString)
import qualified Data.ByteString.Char8 as B
import Network.HTTP.Types
import Network.HTTP.Toolkit.Error
import Network.HTTP.Toolkit.InputStream
import Network.HTTP.Toolkit.Header
import Network.HTTP.Toolkit.Body
type RequestPath = ByteString
data Request a = Request {
requestMethod :: Method
, requestPath :: RequestPath
, requestHeaders :: [Header]
, requestBody :: a
} deriving (Eq, Show, Functor, Foldable, Traversable)
readRequest :: Bool -> InputStream -> IO (Request BodyReader)
readRequest = readRequestWithLimit defaultHeaderSizeLimit
readRequestWithLimit :: Limit -> Bool -> InputStream -> IO (Request BodyReader)
readRequestWithLimit limit raw c = do
(startLine, headers) <- readMessageHeader limit c
(method, path) <- parseRequestLine_ startLine
Request method path headers <$> makeBodyReader raw (determineRequestBodyType headers) c
parseRequestLine_ :: ByteString -> IO (Method, RequestPath)
parseRequestLine_ input = maybe (throwIO $ InvalidRequestLine input) return (parseRequestLine input)
parseRequestLine :: ByteString -> Maybe (Method, RequestPath)
parseRequestLine input = case B.words input of
method : path : _ -> Just (method, path)
_ -> Nothing
determineRequestBodyType :: [Header] -> BodyType
determineRequestBodyType = fromMaybe None . bodyTypeFromHeaders
formatRequestLine :: Method -> RequestPath -> ByteString
formatRequestLine method path = B.unwords [method, path, "HTTP/1.1"]
sendRequest :: (ByteString -> IO ()) -> Request BodyReader -> IO ()
sendRequest send (Request method path headers body) = do
sendHeader send (formatRequestLine method path) headers
sendBody send body