use std::rc::Rc; use std::marker::PhantomData; use std::collections::HashMap; use actix::Actor; use http::Method; use task::Task; use route::{Route, RouteHandler}; use payload::Payload; use context::HttpContext; use httpcodes::HTTPMethodNotAllowed; use httpmessage::{HttpRequest, HttpResponse}; /// Http resource /// /// `Resource` is an entry in route table which corresponds to requested URL. /// /// Resource in turn has at least one route. /// Route corresponds to handling HTTP method by calling route handler. /// /// ```rust,ignore /// /// struct MyRoute; /// /// fn main() { /// let mut routes = RoutingMap::default(); /// /// routes /// .add_resource("/") /// .post::(); /// } pub struct Resource { state: PhantomData, routes: HashMap>>, default: Box>, } impl Default for Resource { fn default() -> Self { Resource { state: PhantomData, routes: HashMap::new(), default: Box::new(HTTPMethodNotAllowed)} } } impl Resource where S: 'static { /// Register handler for specified method. pub fn handler(&mut self, method: Method, handler: H) -> &mut Self where H: RouteHandler { self.routes.insert(method, Box::new(handler)); self } /// Default handler is used if no matched route found. /// By default `HTTPMethodNotAllowed` is used. pub fn default_handler(&mut self, handler: H) -> &mut Self where H: RouteHandler { self.default = Box::new(handler); self } /// Handler for `GET` method. pub fn get(&mut self) -> &mut Self where A: Actor> + Route { self.handler(Method::GET, A::factory()) } /// Handler for `POST` method. pub fn post(&mut self) -> &mut Self where A: Actor> + Route { self.handler(Method::POST, A::factory()) } /// Handler for `PUR` method. pub fn put(&mut self) -> &mut Self where A: Actor> + Route { self.handler(Method::PUT, A::factory()) } /// Handler for `METHOD` method. pub fn delete(&mut self) -> &mut Self where A: Actor> + Route { self.handler(Method::DELETE, A::factory()) } } impl RouteHandler for Resource { fn handle(&self, req: HttpRequest, payload: Payload, state: Rc) -> Task { if let Some(handler) = self.routes.get(req.method()) { handler.handle(req, payload, state) } else { self.default.handle(req, payload, state) } } } #[cfg_attr(feature="cargo-clippy", allow(large_enum_variant))] enum ReplyItem where A: Actor + Route { Message(HttpRequest, HttpResponse), Actor(A), } /// Represents response process. pub struct Reply (ReplyItem); impl Reply where A: Actor + Route { /// Create async response pub fn stream(act: A) -> Self { Reply(ReplyItem::Actor(act)) } /// Send response pub fn reply>(req: HttpRequest, response: R) -> Self { Reply(ReplyItem::Message(req, response.into())) } pub fn into(self, mut ctx: HttpContext) -> Task where A: Actor> { match self.0 { ReplyItem::Message(req, msg) => { Task::reply(req, msg) }, ReplyItem::Actor(act) => { ctx.set_actor(act); Task::with_stream(ctx) } } } }