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.ByteString (ByteString)
import qualified Data.ByteString.Char8 as B
import Network.HTTP.Types
import Network.HTTP.Toolkit.Error
import Network.HTTP.Toolkit.Connection
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)
readRequest :: Connection -> IO (Request BodyReader)
readRequest = readRequestWithLimit defaultHeaderSizeLimit
readRequestWithLimit :: Limit -> Connection -> IO (Request BodyReader)
readRequestWithLimit limit c = do
(startLine, headers) <- readMessageHeader limit c
(method, path) <- parseRequestLine_ startLine
Request method path headers <$> makeBodyReader c (determineRequestBodyType headers)
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