actix-extras/src/resource.rs

use std::rc::Rc;
use std::marker::PhantomData;
use std::collections::HashMap;

use actix::Actor;
use http::Method;

use task::Task;
use route::{Route, Payload, RouteHandler};
use context::HttpContext;
use httpcodes::HTTPMethodNotAllowed;
use httpmessage::{HttpRequest, HttpResponse, IntoHttpResponse};

/// Resource
pub struct HttpResource<S=()> {
    state: PhantomData<S>,
    routes: HashMap<Method, Box<RouteHandler<S>>>,
    default: Box<RouteHandler<S>>,
}

impl<S> Default for HttpResource<S> {
    fn default() -> Self {
        HttpResource {
            state: PhantomData,
            routes: HashMap::new(),
            default: Box::new(HTTPMethodNotAllowed)}
    }
}


impl<S> HttpResource<S> where S: 'static {

    /// Register handler for specified method.
    pub fn handler<H>(&mut self, method: Method, handler: H) -> &mut Self
        where H: RouteHandler<S>
    {
        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<H>(&mut self, handler: H) -> &mut Self
        where H: RouteHandler<S>
    {
        self.default = Box::new(handler);
        self
    }

    /// Handler for `GET` method.
    pub fn get<A>(&mut self) -> &mut Self where A: Route<State=S>
    {
        self.handler(Method::GET, A::factory())
    }

    /// Handler for `POST` method.
    pub fn post<A>(&mut self) -> &mut Self where A: Route<State=S>
    {
        self.handler(Method::POST, A::factory())
    }

    /// Handler for `PUR` method.
    pub fn put<A>(&mut self) -> &mut Self where A: Route<State=S>
    {
        self.handler(Method::PUT, A::factory())
    }

    /// Handler for `METHOD` method.
    pub fn delete<A>(&mut self) -> &mut Self where A: Route<State=S>
    {
        self.handler(Method::DELETE, A::factory())
    }
}


impl<S: 'static> RouteHandler<S> for HttpResource<S> {

    fn handle(&self, req: HttpRequest, payload: Option<Payload>, state: Rc<S>) -> 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 HttpMessageItem<A> where A: Actor<Context=HttpContext<A>> + Route {
    Message(HttpResponse),
    Actor(A),
}

pub struct HttpMessage<A: Actor<Context=HttpContext<A>> + Route> (HttpMessageItem<A>);

impl<A> HttpMessage<A> where A: Actor<Context=HttpContext<A>> + Route
{
    /// Create async response
    pub fn stream(act: A) -> Self {
        HttpMessage(HttpMessageItem::Actor(act))
    }

    /// Create response with empty body
    pub fn reply<I>(req: HttpRequest, msg: I) -> Self
        where I: IntoHttpResponse
    {
        HttpMessage(HttpMessageItem::Message(msg.into_response(req)))
    }

    pub(crate) fn into(self, mut ctx: HttpContext<A>) -> Task {
        match self.0 {
            HttpMessageItem::Message(msg) => {
                Task::reply(msg)
            },
            HttpMessageItem::Actor(act) => {
                ctx.set_actor(act);
                Task::with_stream(ctx)
            }
        }
    }
}