use std::{future::Future, pin::Pin};
use hyper::http::{self, response::Builder};
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())
}
}
pub trait SputnikBody {
fn into_form<T: DeserializeOwned>(
self,
) -> Pin<Box<dyn Future<Output = Result<T, FormError>> + Send + Sync>>;
#[cfg(feature = "hyper_body_json")]
#[cfg_attr(docsrs, doc(cfg(feature = "hyper_body_json")))]
fn into_json<T: DeserializeOwned>(
self,
) -> Pin<Box<dyn Future<Output = Result<T, JsonError>> + Send + Sync>>;
}
impl SputnikBody for hyper::Body {
fn into_form<T: DeserializeOwned>(
self,
) -> Pin<Box<dyn Future<Output = Result<T, FormError>> + Send + Sync>> {
Box::pin(async move {
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")))]
fn into_json<T: DeserializeOwned>(
self,
) -> Pin<Box<dyn Future<Output = Result<T, JsonError>> + Send + Sync>> {
Box::pin(async move {
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),
}