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; pub type BoxFut = Box, Error = failure::Error> + Send>; type ApiHandlerFn = fn(Value, &ApiMethod) -> Result; type ApiUploadHandlerFn = fn(hyper::Body, Value, &ApiUploadMethod) -> BoxFut; pub struct ApiMethod { pub parameters: ObjectSchema, pub returns: Arc, pub handler: ApiHandlerFn, } impl ApiMethod { pub fn new(handler: ApiHandlerFn, parameters: ObjectSchema) -> Self { Self { parameters, handler, returns: Arc::new(Schema::Null), } } pub fn returns>>(mut self, schema: S) -> Self { self.returns = schema.into(); self } } pub struct ApiUploadMethod { pub parameters: ObjectSchema, pub returns: Arc, pub handler: ApiUploadHandlerFn, } impl ApiUploadMethod { pub fn new(handler: ApiUploadHandlerFn, parameters: ObjectSchema) -> Self { Self { parameters, handler, returns: Arc::new(Schema::Null), } } pub fn returns>>(mut self, schema: S) -> Self { self.returns = schema.into(); self } } pub enum SubRoute { None, Hash(HashMap), MatchAll { router: Box, param_name: String }, } pub enum MethodDefinition { None, Simple(ApiMethod), Upload(ApiUploadMethod), } 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>(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) -> Self { self.subroute = SubRoute::Hash(map); self } pub fn match_all>(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: ApiUploadMethod) -> Self { self.post = MethodDefinition::Upload(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) -> 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 } }