2020-03-26 03:34:28 +00:00
|
|
|
//! A helper module that supports HTTP
|
|
|
|
|
2020-03-05 00:39:56 +00:00
|
|
|
mod graphiql_source;
|
2020-09-10 08:43:24 +00:00
|
|
|
mod multipart;
|
2020-03-05 00:39:56 +00:00
|
|
|
mod playground_source;
|
2020-04-14 01:53:17 +00:00
|
|
|
mod stream_body;
|
2020-03-05 00:39:56 +00:00
|
|
|
|
|
|
|
pub use graphiql_source::graphiql_source;
|
2020-09-10 08:43:24 +00:00
|
|
|
pub use multipart::{receive_multipart, MultipartOptions};
|
2020-06-06 01:55:53 +00:00
|
|
|
pub use playground_source::{playground_source, GraphQLPlaygroundConfig};
|
2020-04-14 01:53:17 +00:00
|
|
|
pub use stream_body::StreamBody;
|
2020-03-05 00:39:56 +00:00
|
|
|
|
2020-09-10 07:04:24 +00:00
|
|
|
use crate::{GQLQuery, ParseRequestError, Pos, QueryError, Variables};
|
|
|
|
use futures::io::AsyncRead;
|
2020-09-10 08:43:24 +00:00
|
|
|
use futures::AsyncReadExt;
|
2020-03-08 12:35:36 +00:00
|
|
|
use serde::ser::{SerializeMap, SerializeSeq};
|
2020-08-31 18:18:02 +00:00
|
|
|
use serde::{Deserialize, Serialize, Serializer};
|
2020-03-05 00:39:56 +00:00
|
|
|
|
2020-04-11 09:36:05 +00:00
|
|
|
/// Deserializable GraphQL Request object
|
2020-03-05 00:39:56 +00:00
|
|
|
#[derive(Deserialize, Clone, PartialEq, Debug)]
|
|
|
|
pub struct GQLRequest {
|
2020-03-20 03:56:08 +00:00
|
|
|
/// Query source
|
2020-03-05 00:39:56 +00:00
|
|
|
pub query: String,
|
2020-03-20 03:56:08 +00:00
|
|
|
|
|
|
|
/// Operation name for this query
|
2020-03-05 00:39:56 +00:00
|
|
|
#[serde(rename = "operationName")]
|
|
|
|
pub operation_name: Option<String>,
|
2020-03-20 03:56:08 +00:00
|
|
|
|
|
|
|
/// Variables for this query
|
2020-03-05 00:39:56 +00:00
|
|
|
pub variables: Option<serde_json::Value>,
|
|
|
|
}
|
|
|
|
|
2020-09-10 07:04:24 +00:00
|
|
|
/// Receive a GraphQL request from a content type and body.
|
|
|
|
pub async fn receive_body(
|
|
|
|
content_type: Option<impl AsRef<str>>,
|
|
|
|
body: impl AsyncRead,
|
|
|
|
opts: MultipartOptions,
|
2020-09-10 08:43:24 +00:00
|
|
|
) -> Result<Query, ParseRequestError> {
|
2020-09-10 07:04:24 +00:00
|
|
|
if let Some(Ok(boundary)) = content_type.map(multer::parse_boundary) {
|
|
|
|
receive_multipart(body, boundary, opts)
|
|
|
|
} else {
|
|
|
|
futures::pin_mut!(body);
|
|
|
|
let mut data = Vec::new();
|
2020-09-10 08:43:24 +00:00
|
|
|
body.read_to_end(&mut data)
|
|
|
|
.await
|
|
|
|
.map_err(ParseRequestError::Io)?;
|
2020-09-10 07:04:24 +00:00
|
|
|
Ok(GQLQuery::new_with_http_request(
|
|
|
|
serde_json::from_slice(&data).map_err(ParseRequestError::InvalidRequest)?,
|
|
|
|
))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-03-05 00:39:56 +00:00
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
use serde_json::json;
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_request() {
|
|
|
|
let request: GQLRequest = serde_json::from_value(json! ({
|
|
|
|
"query": "{ a b c }"
|
|
|
|
}))
|
|
|
|
.unwrap();
|
|
|
|
assert!(request.variables.is_none());
|
|
|
|
assert!(request.operation_name.is_none());
|
|
|
|
assert_eq!(request.query, "{ a b c }");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_request_with_operation_name() {
|
|
|
|
let request: GQLRequest = serde_json::from_value(json! ({
|
|
|
|
"query": "{ a b c }",
|
|
|
|
"operationName": "a"
|
|
|
|
}))
|
|
|
|
.unwrap();
|
|
|
|
assert!(request.variables.is_none());
|
|
|
|
assert_eq!(request.operation_name.as_deref(), Some("a"));
|
|
|
|
assert_eq!(request.query, "{ a b c }");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_request_with_variables() {
|
|
|
|
let request: GQLRequest = serde_json::from_value(json! ({
|
|
|
|
"query": "{ a b c }",
|
|
|
|
"variables": {
|
|
|
|
"v1": 100,
|
|
|
|
"v2": [1, 2, 3],
|
|
|
|
"v3": "str",
|
|
|
|
}
|
|
|
|
}))
|
|
|
|
.unwrap();
|
|
|
|
assert_eq!(
|
|
|
|
request.variables,
|
|
|
|
Some(json!({
|
|
|
|
"v1": 100,
|
|
|
|
"v2": [1, 2, 3],
|
|
|
|
"v3": "str",
|
|
|
|
}))
|
|
|
|
);
|
|
|
|
assert!(request.operation_name.is_none());
|
|
|
|
assert_eq!(request.query, "{ a b c }");
|
|
|
|
}
|
|
|
|
}
|