2020-03-26 03:34:28 +00:00
|
|
|
//! A helper module that supports HTTP
|
|
|
|
|
2020-09-17 08:39:55 +00:00
|
|
|
mod batch;
|
2020-03-05 00:39:56 +00:00
|
|
|
mod graphiql_source;
|
2020-09-14 18:38:41 +00:00
|
|
|
#[cfg(feature = "multipart")]
|
2020-09-10 08:43:24 +00:00
|
|
|
mod multipart;
|
2020-03-05 00:39:56 +00:00
|
|
|
mod playground_source;
|
2020-09-15 12:07:59 +00:00
|
|
|
mod websocket;
|
2020-03-05 00:39:56 +00:00
|
|
|
|
2020-09-17 08:39:55 +00:00
|
|
|
#[cfg(feature = "multipart")]
|
|
|
|
pub use batch::receive_batch_body;
|
|
|
|
pub use batch::receive_batch_json;
|
2020-03-05 00:39:56 +00:00
|
|
|
pub use graphiql_source::graphiql_source;
|
2020-09-14 18:38:41 +00:00
|
|
|
#[cfg(feature = "multipart")]
|
2020-09-17 08:39:55 +00:00
|
|
|
pub use multipart::MultipartOptions;
|
2020-06-06 01:55:53 +00:00
|
|
|
pub use playground_source::{playground_source, GraphQLPlaygroundConfig};
|
2020-09-17 18:22:54 +00:00
|
|
|
pub use websocket::WebSocket;
|
2020-03-05 00:39:56 +00:00
|
|
|
|
2020-09-12 16:07:46 +00:00
|
|
|
use crate::{ParseRequestError, Request};
|
2020-09-10 07:04:24 +00:00
|
|
|
use futures::io::AsyncRead;
|
2020-09-10 08:43:24 +00:00
|
|
|
use futures::AsyncReadExt;
|
2020-09-10 08:54:38 +00:00
|
|
|
|
2020-09-14 19:16:41 +00:00
|
|
|
#[cfg(feature = "multipart")]
|
2020-09-15 18:32:13 +00:00
|
|
|
#[cfg_attr(feature = "nightly", doc(cfg(feature = "multipart")))]
|
2020-09-17 08:39:55 +00:00
|
|
|
/// Receive a GraphQL request from a content type and body.
|
2020-09-10 07:04:24 +00:00
|
|
|
pub async fn receive_body(
|
|
|
|
content_type: Option<impl AsRef<str>>,
|
2020-09-13 10:49:07 +00:00
|
|
|
body: impl AsyncRead + Send + 'static,
|
2020-09-10 07:04:24 +00:00
|
|
|
opts: MultipartOptions,
|
2020-09-10 08:54:38 +00:00
|
|
|
) -> Result<Request, ParseRequestError> {
|
2020-09-10 07:04:24 +00:00
|
|
|
if let Some(Ok(boundary)) = content_type.map(multer::parse_boundary) {
|
2020-09-17 08:39:55 +00:00
|
|
|
multipart::receive_multipart(body, boundary, opts).await
|
2020-09-14 19:16:41 +00:00
|
|
|
} else {
|
|
|
|
receive_json(body).await
|
2020-03-05 00:39:56 +00:00
|
|
|
}
|
2020-09-14 19:16:41 +00:00
|
|
|
}
|
2020-09-14 18:38:41 +00:00
|
|
|
|
2020-09-14 19:16:41 +00:00
|
|
|
/// Receive a GraphQL request from a body as JSON.
|
2020-09-14 19:20:15 +00:00
|
|
|
pub async fn receive_json(
|
|
|
|
body: impl AsyncRead + Send + 'static,
|
|
|
|
) -> Result<Request, ParseRequestError> {
|
2020-09-14 18:38:41 +00:00
|
|
|
let mut data = Vec::new();
|
|
|
|
futures::pin_mut!(body);
|
|
|
|
body.read_to_end(&mut data)
|
|
|
|
.await
|
|
|
|
.map_err(ParseRequestError::Io)?;
|
|
|
|
Ok(serde_json::from_slice::<Request>(&data).map_err(ParseRequestError::InvalidRequest)?)
|
2020-03-05 00:39:56 +00:00
|
|
|
}
|