yume/juniper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Backport fix of Content-Type header parsing in juniper_axum (#1289, #1288)

Browse source 
Signed-off-by: Joe Grund <jgrund@whamcloud.io>
This commit is contained in:
Joe Grund 2024-10-28 12:54:21 +01:00 committed by tyranron
parent 6cf1bb30da
commit bb529cca97
No known key found for this signature in database
GPG key ID: 762E144FB230A4F0
2 changed files with 57 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
juniper_axum/CHANGELOG.md
View file

@ -6,6 +6,18 @@ All user visible changes to `juniper_axum` crate will be documented in this file
## master
### Fixed
- `Content-Type` header reading full value instead of just media type. ([#1289], [#1288])
[#1288]: /../../issues/1288
[#1289]: /../../pull/1289
## [0.1.0] · 2024-03-20
[0.1.0]: /../../tree/juniper_axum-v0.1.0/juniper_axum

56
juniper_axum/src/extract.rs
View file

@ -6,7 +6,7 @@ use axum::{
async_trait,
body::Body,
extract::{FromRequest, FromRequestParts, Query},
http::{HeaderValue, Method, Request, StatusCode},
http::{header, HeaderValue, Method, Request, StatusCode},
response::{IntoResponse as _, Response},
Json, RequestExt as _,
};
@ -85,7 +85,7 @@ where
async fn from_request(mut req: Request<Body>, state: &State) -> Result<Self, Self::Rejection> {
let content_type = req
.headers()
.get("content-type")
.get(header::CONTENT_TYPE)
.map(HeaderValue::to_str)
.transpose()
.map_err(|_| {
@ -122,7 +122,7 @@ where
.into_response()
})
}),
(&Method::POST, Some("application/json")) => {
(&Method::POST, Some(x)) if x.starts_with("application/json") => {
Json::<GraphQLBatchRequest<S>>::from_request(req, state)
.await
.map(|req| Self(req.0))
@ -130,14 +130,16 @@ where
(StatusCode::BAD_REQUEST, format!("Invalid JSON body: {e}")).into_response()
})
}
(&Method::POST, Some("application/graphql")) => String::from_request(req, state)
.await
.map(|body| {
Self(GraphQLBatchRequest::Single(GraphQLRequest::new(
body, None, None,
)))
})
.map_err(|_| (StatusCode::BAD_REQUEST, "Not valid UTF-8 body").into_response()),
(&Method::POST, Some(x)) if x.starts_with("application/graphql") => {
String::from_request(req, state)
.await
.map(|body| {
Self(GraphQLBatchRequest::Single(GraphQLRequest::new(
body, None, None,
)))
})
.map_err(|_| (StatusCode::BAD_REQUEST, "Not valid UTF-8 body").into_response())
}
(&Method::POST, _) => Err((
StatusCode::UNSUPPORTED_MEDIA_TYPE,
"`Content-Type` header is expected to be either `application/json` or \
@ -246,6 +248,22 @@ mod juniper_request_tests {
assert_eq!(do_from_request(req).await, expected);
}
#[tokio::test]
async fn from_json_post_request_with_charset() {
let req = Request::post("/")
.header("content-type", "application/json; charset=utf-8")
.body(Body::from(r#"{"query": "{ add(a: 2, b: 3) }"}"#))
.unwrap_or_else(|e| panic!("cannot build `Request`: {e}"));
let expected = JuniperRequest(GraphQLBatchRequest::Single(GraphQLRequest::new(
"{ add(a: 2, b: 3) }".to_string(),
None,
None,
)));
assert_eq!(do_from_request(req).await, expected);
}
#[tokio::test]
async fn from_graphql_post_request() {
let req = Request::post("/")
@ -262,6 +280,22 @@ mod juniper_request_tests {
assert_eq!(do_from_request(req).await, expected);
}
#[tokio::test]
async fn from_graphql_post_request_with_charset() {
let req = Request::post("/")
.header("content-type", "application/graphql; charset=utf-8")
.body(Body::from(r#"{ add(a: 2, b: 3) }"#))
.unwrap_or_else(|e| panic!("cannot build `Request`: {e}"));
let expected = JuniperRequest(GraphQLBatchRequest::Single(GraphQLRequest::new(
"{ add(a: 2, b: 3) }".to_string(),
None,
None,
)));
assert_eq!(do_from_request(req).await, expected);
}
/// Performs [`JuniperRequest::from_request()`].
async fn do_from_request(req: Request<Body>) -> JuniperRequest {
match JuniperRequest::from_request(req, &()).await {