Some handy extensions to work with HTTPClientResponse body

Author: makoni

Sources/AsyncHTTPClient/AsyncAwait/HTTPClientResponse.swift

@@ -138,6 +138,19 @@ extension HTTPClientResponse {
     }
 }
 
+@available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
+extension HTTPClientResponse {
+	/// Response body as `ByteBuffer`.
+	public var bytes: ByteBuffer {
+		get async throws {
+			let expectedBytes = headers
+				.first(name: "content-length")
+				.flatMap(Int.init) ?? 1024 * 1024
+			return try await body.collect(upTo: expectedBytes)
+		}
+	}
+}
+
 @available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
 @usableFromInline
 typealias TransactionBody = NIOThrowingAsyncSequenceProducer<

Sources/AsyncHTTPClient/FoundationExtensions.swift

@@ -64,3 +64,14 @@ extension HTTPClient.Body {
         return self.bytes(data)
     }
 }
+
+@available(macOS 10.15, iOS 13.0, watchOS 6.0, tvOS 13.0, *)
+extension HTTPClientResponse {
+	/// Response body as `Data`.
+	public var data: Data? {
+		get async throws {
+			var bytes = try await bytes
+			return bytes.readData(length: bytes.readableBytes)
+		}
+	}
+}