2019-09-26 17:24:40 +02:00
|
|
|
use crate::{IndexResource, Resource, ResourceResult};
|
|
|
|
use futures::future::{err, ok};
|
|
|
|
use gotham::{
|
|
|
|
handler::{HandlerFuture, IntoHandlerError},
|
|
|
|
helpers::http::response::create_response,
|
|
|
|
pipeline::chain::PipelineHandleChain,
|
|
|
|
router::builder::*,
|
|
|
|
state::State
|
|
|
|
};
|
|
|
|
use mime::APPLICATION_JSON;
|
|
|
|
use serde::Serialize;
|
|
|
|
use std::panic::RefUnwindSafe;
|
|
|
|
|
2019-09-26 17:42:28 +02:00
|
|
|
/// This trait adds the `resource` method to gotham's routing. It allows you to register
|
|
|
|
/// any RESTful `Resource` with a path.
|
|
|
|
pub trait DrawResources
|
|
|
|
{
|
|
|
|
fn resource<R : Resource, T : ToString>(&mut self, path : T);
|
|
|
|
}
|
|
|
|
|
|
|
|
/// This trait allows to draw routes within an resource. Use this only inside the
|
|
|
|
/// `Resource::setup` method.
|
2019-09-26 17:24:40 +02:00
|
|
|
pub trait DrawResourceRoutes
|
|
|
|
{
|
|
|
|
fn index<R : Serialize, E : Serialize, Res : ResourceResult<R, E>, IR : IndexResource<R, E, Res>>(&mut self);
|
|
|
|
}
|
|
|
|
|
|
|
|
fn to_handler_future<R, E, Res>(state : State, r : Res) -> Box<HandlerFuture>
|
|
|
|
where
|
|
|
|
R : Serialize,
|
|
|
|
E : Serialize,
|
|
|
|
Res : ResourceResult<R, E>
|
|
|
|
{
|
|
|
|
let (status, res) = r.to_result();
|
|
|
|
let json = match res {
|
|
|
|
Ok(json) => serde_json::to_string(&json),
|
|
|
|
Err(json) => serde_json::to_string(&json)
|
|
|
|
};
|
|
|
|
match json {
|
|
|
|
Ok(body) => {
|
|
|
|
let res = create_response(&state, status, APPLICATION_JSON, body);
|
|
|
|
Box::new(ok((state, res)))
|
|
|
|
},
|
|
|
|
Err(e) => Box::new(err((state, e.into_handler_error())))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fn index_handler<R, E, Res, IR>(mut state : State) -> Box<HandlerFuture>
|
|
|
|
where
|
|
|
|
R : Serialize,
|
|
|
|
E : Serialize,
|
|
|
|
Res : ResourceResult<R, E>,
|
|
|
|
IR : IndexResource<R, E, Res>
|
|
|
|
{
|
|
|
|
let res = IR::index(&mut state);
|
|
|
|
to_handler_future(state, res)
|
|
|
|
}
|
|
|
|
|
|
|
|
macro_rules! implDrawResourceRoutes {
|
|
|
|
($implType:ident) => {
|
2019-09-26 17:42:28 +02:00
|
|
|
impl<'a, C, P> DrawResources for $implType<'a, C, P>
|
|
|
|
where
|
|
|
|
C : PipelineHandleChain<P> + Copy + Send + Sync + 'static,
|
|
|
|
P : RefUnwindSafe + Send + Sync + 'static
|
|
|
|
{
|
|
|
|
fn resource<R : Resource, T : ToString>(&mut self, path : T)
|
|
|
|
{
|
|
|
|
R::setup();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-09-26 17:24:40 +02:00
|
|
|
impl<'a, C, P> DrawResourceRoutes for ($implType<'a, C, P>, String)
|
|
|
|
where
|
|
|
|
C : PipelineHandleChain<P> + Copy + Send + Sync + 'static,
|
|
|
|
P : RefUnwindSafe + Send + Sync + 'static
|
|
|
|
{
|
|
|
|
/// Register an `IndexResource` with this resource.
|
|
|
|
fn index<R : Serialize, E : Serialize, Res : ResourceResult<R, E>, IR : IndexResource<R, E, Res>>(&mut self)
|
|
|
|
{
|
|
|
|
self.0.get(&self.1).to(|state| index_handler::<R, E, Res, IR>(state));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
implDrawResourceRoutes!(RouterBuilder);
|
|
|
|
implDrawResourceRoutes!(ScopeBuilder);
|