180 lines
4.5 KiB
Rust
180 lines
4.5 KiB
Rust
use failure::*;
|
|
|
|
use crate::api::schema::*;
|
|
use serde_json::{Value};
|
|
use std::collections::HashMap;
|
|
use std::sync::Arc;
|
|
|
|
use hyper::{Body, Response};
|
|
use hyper::rt::Future;
|
|
use hyper::http::request::Parts;
|
|
|
|
pub type BoxFut = Box<Future<Item = Response<Body>, Error = failure::Error> + Send>;
|
|
|
|
type ApiHandlerFn = fn(Value, &ApiMethod) -> Result<Value, Error>;
|
|
|
|
type ApiAsyncHandlerFn = fn(Parts, Body, Value, &ApiAsyncMethod) -> Result<BoxFut, Error>;
|
|
|
|
pub struct ApiMethod {
|
|
pub parameters: ObjectSchema,
|
|
pub returns: Arc<Schema>,
|
|
pub handler: ApiHandlerFn,
|
|
}
|
|
|
|
impl ApiMethod {
|
|
|
|
pub fn new(handler: ApiHandlerFn, parameters: ObjectSchema) -> Self {
|
|
Self {
|
|
parameters,
|
|
handler,
|
|
returns: Arc::new(Schema::Null),
|
|
}
|
|
}
|
|
|
|
pub fn returns<S: Into<Arc<Schema>>>(mut self, schema: S) -> Self {
|
|
|
|
self.returns = schema.into();
|
|
|
|
self
|
|
}
|
|
}
|
|
|
|
pub struct ApiAsyncMethod {
|
|
pub parameters: ObjectSchema,
|
|
pub returns: Arc<Schema>,
|
|
pub handler: ApiAsyncHandlerFn,
|
|
}
|
|
|
|
impl ApiAsyncMethod {
|
|
|
|
pub fn new(handler: ApiAsyncHandlerFn, parameters: ObjectSchema) -> Self {
|
|
Self {
|
|
parameters,
|
|
handler,
|
|
returns: Arc::new(Schema::Null),
|
|
}
|
|
}
|
|
|
|
pub fn returns<S: Into<Arc<Schema>>>(mut self, schema: S) -> Self {
|
|
|
|
self.returns = schema.into();
|
|
|
|
self
|
|
}
|
|
}
|
|
|
|
pub enum SubRoute {
|
|
None,
|
|
Hash(HashMap<String, Router>),
|
|
MatchAll { router: Box<Router>, param_name: String },
|
|
}
|
|
|
|
pub enum MethodDefinition {
|
|
None,
|
|
Simple(ApiMethod),
|
|
Async(ApiAsyncMethod),
|
|
}
|
|
|
|
pub struct Router {
|
|
pub get: MethodDefinition,
|
|
pub put: MethodDefinition,
|
|
pub post: MethodDefinition,
|
|
pub delete: MethodDefinition,
|
|
pub subroute: SubRoute,
|
|
}
|
|
|
|
impl Router {
|
|
|
|
pub fn new() -> Self {
|
|
Self {
|
|
get: MethodDefinition::None,
|
|
put: MethodDefinition::None,
|
|
post: MethodDefinition::None,
|
|
delete: MethodDefinition::None,
|
|
subroute: SubRoute::None
|
|
}
|
|
}
|
|
|
|
pub fn subdir<S: Into<String>>(mut self, subdir: S, router: Router) -> Self {
|
|
if let SubRoute::None = self.subroute {
|
|
self.subroute = SubRoute::Hash(HashMap::new());
|
|
}
|
|
match self.subroute {
|
|
SubRoute::Hash(ref mut map) => {
|
|
map.insert(subdir.into(), router);
|
|
}
|
|
_ => panic!("unexpected subroute type"),
|
|
}
|
|
self
|
|
}
|
|
|
|
pub fn subdirs(mut self, map: HashMap<String, Router>) -> Self {
|
|
self.subroute = SubRoute::Hash(map);
|
|
self
|
|
}
|
|
|
|
pub fn match_all<S: Into<String>>(mut self, param_name: S, router: Router) -> Self {
|
|
if let SubRoute::None = self.subroute {
|
|
self.subroute = SubRoute::MatchAll { router: Box::new(router), param_name: param_name.into() };
|
|
} else {
|
|
panic!("unexpected subroute type");
|
|
}
|
|
self
|
|
}
|
|
|
|
pub fn get(mut self, m: ApiMethod) -> Self {
|
|
self.get = MethodDefinition::Simple(m);
|
|
self
|
|
}
|
|
|
|
pub fn put(mut self, m: ApiMethod) -> Self {
|
|
self.put = MethodDefinition::Simple(m);
|
|
self
|
|
}
|
|
|
|
pub fn post(mut self, m: ApiMethod) -> Self {
|
|
self.post = MethodDefinition::Simple(m);
|
|
self
|
|
}
|
|
|
|
pub fn upload(mut self, m: ApiAsyncMethod) -> Self {
|
|
self.post = MethodDefinition::Async(m);
|
|
self
|
|
}
|
|
|
|
pub fn download(mut self, m: ApiAsyncMethod) -> Self {
|
|
self.get = MethodDefinition::Async(m);
|
|
self
|
|
}
|
|
|
|
|
|
pub fn delete(mut self, m: ApiMethod) -> Self {
|
|
self.delete = MethodDefinition::Simple(m);
|
|
self
|
|
}
|
|
|
|
pub fn find_route(&self, components: &[&str], uri_param: &mut HashMap<String, String>) -> Option<&Router> {
|
|
|
|
if components.len() == 0 { return Some(self); };
|
|
|
|
let (dir, rest) = (components[0], &components[1..]);
|
|
|
|
match self.subroute {
|
|
SubRoute::None => {},
|
|
SubRoute::Hash(ref dirmap) => {
|
|
if let Some(ref router) = dirmap.get(dir) {
|
|
println!("FOUND SUBDIR {}", dir);
|
|
return router.find_route(rest, uri_param);
|
|
}
|
|
}
|
|
SubRoute::MatchAll { ref router, ref param_name } => {
|
|
println!("URI PARAM {} = {}", param_name, dir); // fixme: store somewhere
|
|
uri_param.insert(param_name.clone(), dir.into());
|
|
return router.find_route(rest, uri_param);
|
|
},
|
|
}
|
|
|
|
None
|
|
}
|
|
}
|