actix-extras/src/application.rs

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

use error::UriGenerationError;
use handler::{Reply, RouteHandler};
use resource::Resource;
use recognizer::{RouteRecognizer, check_pattern, PatternElement};
use httprequest::HttpRequest;
use channel::{HttpHandler, IntoHttpHandler};
use pipeline::Pipeline;
use middlewares::Middleware;

pub struct Router<S>(Rc<RouteRecognizer<Resource<S>>>);

impl<S: 'static> Router<S> {
    pub fn new(prefix: String, map: HashMap<String, Resource<S>>) -> Router<S>
    {
        let mut resources = Vec::new();
        for (path, resource) in map {
            resources.push((path, resource.get_name(), resource))
        }

        Router(Rc::new(RouteRecognizer::new(prefix, resources)))
    }

    pub fn has_route(&self, path: &str) -> bool {
        self.0.recognize(path).is_some()
    }

    pub fn resource_path<'a, U>(&self, prefix: &str, name: &str, elements: U)
                                -> Result<String, UriGenerationError>
        where U: IntoIterator<Item=&'a str>
    {
        if let Some(pattern) = self.0.get_pattern(name) {
            let mut iter = elements.into_iter();
            let mut vec = vec![prefix];
            for el in pattern.elements() {
                match *el {
                    PatternElement::Str(ref s) => vec.push(s),
                    PatternElement::Var(_) => {
                        if let Some(val) = iter.next() {
                            vec.push(val)
                        } else {
                            return Err(UriGenerationError::NotEnoughElements)
                        }
                    }
                }
            }
            let s = vec.join("/").to_owned();
            Ok(s)
        } else {
            Err(UriGenerationError::ResourceNotFound)
        }
    }
}

/// Application
pub struct HttpApplication<S> {
    state: Rc<S>,
    prefix: String,
    default: Resource<S>,
    router: Router<S>,
    middlewares: Rc<Vec<Box<Middleware>>>,
}

impl<S: 'static> HttpApplication<S> {

    fn run(&self, req: HttpRequest) -> Reply {
        let mut req = req.with_state(Rc::clone(&self.state));

        if let Some((params, h)) = self.router.0.recognize(req.path()) {
            if let Some(params) = params {
                req.set_match_info(params);
                req.set_prefix(self.router.0.prefix());
            }
            h.handle(req)
        } else {
            self.default.handle(req)
        }
    }
}

impl<S: 'static> HttpHandler for HttpApplication<S> {

    fn handle(&self, req: HttpRequest) -> Result<Pipeline, HttpRequest> {
        if req.path().starts_with(&self.prefix) {
            Ok(Pipeline::new(req, Rc::clone(&self.middlewares),
                             &|req: HttpRequest| self.run(req)))
        } else {
            Err(req)
        }
    }
}

struct ApplicationParts<S> {
    state: S,
    prefix: String,
    default: Resource<S>,
    resources: HashMap<String, Resource<S>>,
    middlewares: Vec<Box<Middleware>>,
}

/// Structure that follows the builder pattern for building `Application` structs.
pub struct Application<S=()> {
    parts: Option<ApplicationParts<S>>,
}

impl Application<()> {

    /// Create application with empty state. Application can
    /// be configured with builder-like pattern.
    ///
    /// This method accepts path prefix for which it should serve requests.
    pub fn new<T: Into<String>>(prefix: T) -> Application<()> {
        Application {
            parts: Some(ApplicationParts {
                state: (),
                prefix: prefix.into(),
                default: Resource::default_not_found(),
                resources: HashMap::new(),
                middlewares: Vec::new(),
            })
        }
    }
}

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

    /// Create application with specific state. Application can be
    /// configured with builder-like pattern.
    ///
    /// State is shared with all reousrces within same application and could be
    /// accessed with `HttpRequest::state()` method.
    pub fn with_state<T: Into<String>>(prefix: T, state: S) -> Application<S> {
        Application {
            parts: Some(ApplicationParts {
                state: state,
                prefix: prefix.into(),
                default: Resource::default_not_found(),
                resources: HashMap::new(),
                middlewares: Vec::new(),
            })
        }
    }

    /// Configure resource for specific path.
    ///
    /// Resource may have variable path also. For instance, a resource with
    /// the path */a/{name}/c* would match all incoming requests with paths
    /// such as */a/b/c*, */a/1/c*, and */a/etc/c*.
    ///
    /// A variable part is specified in the form `{identifier}`, where
    /// the identifier can be used later in a request handler to access the matched
    /// value for that part. This is done by looking up the identifier
    /// in the `Params` object returned by `HttpRequest.match_info()` method.
    ///
    /// By default, each part matches the regular expression `[^{}/]+`.
    ///
    /// You can also specify a custom regex in the form `{identifier:regex}`:
    ///
    /// For instance, to route Get requests on any route matching `/users/{userid}/{friend}` and
    /// store userid and friend in the exposed Params object:
    ///
    /// ```rust
    /// # extern crate actix_web;
    /// use actix_web::*;
    ///
    /// fn main() {
    ///     let app = Application::new("/")
    ///         .resource("/test", |r| {
    ///              r.method(Method::GET).f(|_| httpcodes::HTTPOk);
    ///              r.method(Method::HEAD).f(|_| httpcodes::HTTPMethodNotAllowed);
    ///         })
    ///         .finish();
    /// }
    /// ```
    pub fn resource<F, P: Into<String>>(&mut self, path: P, f: F) -> &mut Self
        where F: FnOnce(&mut Resource<S>) + 'static
    {
        {
            let parts = self.parts.as_mut().expect("Use after finish");

            // add resource
            let path = path.into();
            if !parts.resources.contains_key(&path) {
                check_pattern(&path);
                parts.resources.insert(path.clone(), Resource::default());
            }
            f(parts.resources.get_mut(&path).unwrap());
        }
        self
    }

    /// Default resource is used if no match route could be found.
    pub fn default_resource<F>(&mut self, f: F) -> &mut Self
        where F: FnOnce(&mut Resource<S>) + 'static
    {
        {
            let parts = self.parts.as_mut().expect("Use after finish");
            f(&mut parts.default);
        }
        self
    }

    /// Register a middleware
    pub fn middleware<T>(&mut self, mw: T) -> &mut Self
        where T: Middleware + 'static
    {
        self.parts.as_mut().expect("Use after finish")
            .middlewares.push(Box::new(mw));
        self
    }

    /// Finish application configuration and create HttpHandler object
    pub fn finish(&mut self) -> HttpApplication<S> {
        let parts = self.parts.take().expect("Use after finish");
        let prefix = if parts.prefix.ends_with('/') {
            parts.prefix
        } else {
            parts.prefix + "/"
        };
        HttpApplication {
            state: Rc::new(parts.state),
            prefix: prefix.clone(),
            default: parts.default,
            router: Router::new(prefix, parts.resources),
            middlewares: Rc::new(parts.middlewares),
        }
    }
}

impl<S: 'static> IntoHttpHandler for Application<S> {
    type Handler = HttpApplication<S>;

    fn into_handler(mut self) -> HttpApplication<S> {
        self.finish()
    }
}

impl<'a, S: 'static> IntoHttpHandler for &'a mut Application<S> {
    type Handler = HttpApplication<S>;

    fn into_handler(self) -> HttpApplication<S> {
        self.finish()
    }
}

#[doc(hidden)]
impl<S: 'static> Iterator for Application<S> {
    type Item = HttpApplication<S>;

    fn next(&mut self) -> Option<Self::Item> {
        if self.parts.is_some() {
            Some(self.finish())
        } else {
            None
        }
    }
}


#[cfg(test)]
mod tests {
    use std::str::FromStr;
    use http::{Method, Version, Uri, HeaderMap, StatusCode};
    use super::*;
    use handler::ReplyItem;
    use httprequest::HttpRequest;
    use httpresponse::HttpResponse;
    use payload::Payload;
    use httpcodes;

    impl Reply {
        fn msg(self) -> Option<HttpResponse> {
            match self.into() {
                ReplyItem::Message(resp) => Some(resp),
                _ => None,
            }
        }
    }

    #[test]
    fn test_default_resource() {
        let app = Application::new("/")
            .resource("/test", |r| r.h(httpcodes::HTTPOk))
            .finish();

        let req = HttpRequest::new(
            Method::GET, Uri::from_str("/test").unwrap(),
            Version::HTTP_11, HeaderMap::new(), Payload::empty());
        let resp = app.run(req).msg().unwrap();
        assert_eq!(resp.status(), StatusCode::OK);

        let req = HttpRequest::new(
            Method::GET, Uri::from_str("/blah").unwrap(),
            Version::HTTP_11, HeaderMap::new(), Payload::empty());
        let resp = app.run(req).msg().unwrap();
        assert_eq!(resp.status(), StatusCode::NOT_FOUND);

        let app = Application::new("/")
            .default_resource(|r| r.h(httpcodes::HTTPMethodNotAllowed))
            .finish();
        let req = HttpRequest::new(
            Method::GET, Uri::from_str("/blah").unwrap(),
            Version::HTTP_11, HeaderMap::new(), Payload::empty());
        let resp = app.run(req).msg().unwrap();
        assert_eq!(resp.status(), StatusCode::METHOD_NOT_ALLOWED);
    }
}