support accept header mime types

gotham_restful/src/routing.rs

@@ -16,11 +16,15 @@ use gotham::{
 	handler::{HandlerFuture, IntoHandlerError},
 	helpers::http::response::create_response,
 	pipeline::chain::PipelineHandleChain,
-	router::builder::*,
+	router::{
+		builder::*,
+		non_match::RouteNonMatch,
+		route::matcher::{AcceptHeaderRouteMatcher, RouteMatcher}
+	},
 	state::{FromState, State}
 };
 use hyper::Body;
-use mime::APPLICATION_JSON;
+use mime::{Mime, APPLICATION_JSON};
 use serde::de::DeserializeOwned;
 use std::panic::RefUnwindSafe;
 
@@ -109,7 +113,7 @@ where
 	F : FnOnce(&mut State) -> R,
 	R : ResourceResult
 {
-	let res = get_result(&mut state).to_json();
+	let res = get_result(&mut state).to_response();
 	match res {
 		Ok((status, body)) => {
 			let res = create_response(&state, status, APPLICATION_JSON, body);
@@ -148,7 +152,7 @@ where
 				}
 			};
 
-			let res = get_result(&mut state, body).to_json();
+			let res = get_result(&mut state, body).to_response();
 			match res {
 				Ok((status, body)) => {
 					let res = create_response(&state, status, APPLICATION_JSON, body);
@@ -246,6 +250,33 @@ where
 	to_handler_future(state, |state| Handler::delete(state, id))
 }
 
+#[derive(Clone)]
+struct MaybeMatchAcceptHeader
+{
+	matcher : Option<AcceptHeaderRouteMatcher>
+}
+
+impl RouteMatcher for MaybeMatchAcceptHeader
+{
+	fn is_match(&self, state : &State) -> Result<(), RouteNonMatch>
+	{
+		match &self.matcher {
+			Some(matcher) => matcher.is_match(state),
+			None => Ok(())
+		}
+	}
+}
+
+impl From<Option<Vec<Mime>>> for MaybeMatchAcceptHeader
+{
+	fn from(types : Option<Vec<Mime>>) -> Self
+	{
+		Self {
+			matcher: types.map(AcceptHeaderRouteMatcher::new)
+		}
+	}
+}
+
 macro_rules! implDrawResourceRoutes {
 	($implType:ident) => {
 		
@@ -289,7 +320,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceReadAll<Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.get(&self.1)
+					.extend_route_matcher(matcher)
 					.to(|state| read_all_handler::<Handler, Res>(state));
 			}
 
@@ -299,7 +332,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceRead<ID, Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.get(&format!("{}/:id", self.1))
+					.extend_route_matcher(matcher)
 					.with_path_extractor::<PathExtractor<ID>>()
 					.to(|state| read_handler::<Handler, ID, Res>(state));
 			}
@@ -310,7 +345,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceSearch<Query, Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.get(&format!("{}/search", self.1))
+					.extend_route_matcher(matcher)
 					.with_query_string_extractor::<Query>()
 					.to(|state| search_handler::<Handler, Query, Res>(state));
 			}
@@ -321,7 +358,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceCreate<Body, Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.post(&self.1)
+					.extend_route_matcher(matcher)
 					.to(|state| create_handler::<Handler, Body, Res>(state));
 			}
 
@@ -331,7 +370,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceUpdateAll<Body, Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.put(&self.1)
+					.extend_route_matcher(matcher)
 					.to(|state| update_all_handler::<Handler, Body, Res>(state));
 			}
 
@@ -342,7 +383,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceUpdate<ID, Body, Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.put(&format!("{}/:id", self.1))
+					.extend_route_matcher(matcher)
 					.with_path_extractor::<PathExtractor<ID>>()
 					.to(|state| update_handler::<Handler, ID, Body, Res>(state));
 			}
@@ -352,7 +395,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceDeleteAll<Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.delete(&self.1)
+					.extend_route_matcher(matcher)
 					.to(|state| delete_all_handler::<Handler, Res>(state));
 			}
 
@@ -362,7 +407,9 @@ macro_rules! implDrawResourceRoutes {
 				Res : ResourceResult,
 				Handler : ResourceDelete<ID, Res>
 			{
+				let matcher : MaybeMatchAcceptHeader = Res::accepted_types().into();
 				self.0.delete(&format!("{}/:id", self.1))
+					.extend_route_matcher(matcher)
 					.with_path_extractor::<PathExtractor<ID>>()
 					.to(|state| delete_handler::<Handler, ID, Res>(state));
 			}