Mark final chunk as final

Network/HTTP/Semantics/Client.hs

@@ -30,6 +30,7 @@ module Network.HTTP.Semantics.Client (
     responseHeaders,
     responseBodySize,
     getResponseBodyChunk,
+    getResponseBodyChunk',
     getResponseTrailers,
 
     -- * Aux
@@ -133,7 +134,11 @@ responseBodySize (Response rsp) = inpObjBodySize rsp
 -- | Reading a chunk of the response body.
 --   An empty 'ByteString' returned when finished.
 getResponseBodyChunk :: Response -> IO ByteString
-getResponseBodyChunk (Response rsp) = inpObjBody rsp
+getResponseBodyChunk = fmap fst . getResponseBodyChunk'
+
+-- | Generalization of 'getResponseBodyChunk' which also returns if the 'ByteString' is the final one
+getResponseBodyChunk' :: Response -> IO (ByteString, Bool)
+getResponseBodyChunk' (Response rsp) = inpObjBody rsp
 
 -- | Reading response trailers.
 --   This function must be called after 'getResponseBodyChunk'

Network/HTTP/Semantics/Server.hs

@@ -15,6 +15,7 @@ module Network.HTTP.Semantics.Server (
     requestHeaders,
     requestBodySize,
     getRequestBodyChunk,
+    getRequestBodyChunk',
     getRequestTrailers,
 
     -- * Aux
@@ -125,7 +126,11 @@ requestBodySize (Request req) = inpObjBodySize req
 -- | Reading a chunk of the request body.
 --   An empty 'ByteString' returned when finished.
 getRequestBodyChunk :: Request -> IO ByteString
-getRequestBodyChunk (Request req) = inpObjBody req
+getRequestBodyChunk = fmap fst . getRequestBodyChunk'
+
+-- | Generalization of 'getRequestBodyChunk' which also returns if the 'ByteString' is the final one
+getRequestBodyChunk' :: Request -> IO (ByteString, Bool)
+getRequestBodyChunk' (Request req) = inpObjBody req
 
 -- | Reading request trailers.
 --   This function must be called after 'getRequestBodyChunk'

Network/HTTP/Semantics/Types.hs

@@ -47,7 +47,7 @@ type Path = ByteString
 
 ----------------------------------------------------------------
 
-type InpBody = IO ByteString
+type InpBody = IO (ByteString, Bool)
 
 data OutBody
     = OutBodyNone