1 files changed, 64 insertions, 0 deletions
diff --git a/src/hyper_body.rs b/src/hyper_body.rs
new file mode 100644
index 0000000..c6824fb
--- /dev/null
+++ b/src/hyper_body.rs
@@ -0,0 +1,64 @@
+//! Extends `hyper::Body` with [`SputnikBody`].
+use hyper::http::{self, response::Builder};
+use async_trait::async_trait;
+use serde::de::DeserializeOwned;
+
+use crate::response::EmptyBuilder;
+
+impl EmptyBuilder<hyper::Body> for Builder {
+    fn empty(self) -> http::Result<http::response::Response<hyper::Body>> {
+        self.body(hyper::Body::empty())
+    }
+}
+
+/// Adds deserialization methods to [`hyper::Body`].
+#[async_trait]
+pub trait SputnikBody {
+    /// Parses a `application/x-www-form-urlencoded` request body into a given struct.
+    async fn into_form<T: DeserializeOwned>(self) -> Result<T, FormError>;
+
+    /// Attempts to deserialize the request body as JSON.
+    #[cfg(feature = "hyper_body_json")]
+    #[cfg_attr(docsrs, doc(cfg(feature = "hyper_body_json")))]
+    async fn into_json<T: DeserializeOwned>(self) -> Result<T, JsonError>;
+}
+
+#[async_trait]
+impl SputnikBody for hyper::Body {
+
+    async fn into_form<T: DeserializeOwned>(self) -> Result<T, FormError> {
+        let full_body = hyper::body::to_bytes(self).await.map_err(BodyError)?;
+        Ok(serde_urlencoded::from_bytes::<T>(&full_body)?)
+    }
+
+    #[cfg(feature = "hyper_body_json")]
+    #[cfg_attr(docsrs, doc(cfg(feature = "hyper_body_json")))]
+    async fn into_json<T: DeserializeOwned>(self) -> Result<T, JsonError> {
+        let full_body = hyper::body::to_bytes(self).await.map_err(BodyError)?;
+        Ok(serde_json::from_slice::<T>(&full_body)?)
+    }
+}
+
+#[derive(thiserror::Error, Debug)]
+#[error("failed to read body")]
+pub struct BodyError(pub hyper::Error);
+
+#[derive(thiserror::Error, Debug)]
+pub enum FormError {
+    #[error("{0}")]
+    Body(#[from] BodyError),
+
+    #[error("form deserialize error: {0}")]
+    Deserialize(#[from] serde_urlencoded::de::Error),
+}
+
+#[cfg(feature = "hyper_body_json")]
+#[cfg_attr(docsrs, doc(cfg(feature = "hyper_body_json")))]
+#[derive(thiserror::Error, Debug)]
+pub enum JsonError {
+    #[error("{0}")]
+    Body(#[from] BodyError),
+
+    #[error("json deserialize error: {0}")]
+    Deserialize(#[from] serde_json::Error),
+}
+\ No newline at end of file