use std::cell::RefCell; use std::marker::PhantomData; use std::rc::Rc; use actix_http::body::{Body, MessageBody}; use actix_http::{Extensions, PayloadStream}; use actix_server_config::ServerConfig; use actix_service::boxed::{self, BoxedNewService}; use actix_service::{ ApplyTransform, IntoNewService, IntoTransform, NewService, Transform, }; use futures::IntoFuture; use crate::app_service::{AppChain, AppEntry, AppInit, AppRouting, AppRoutingFactory}; use crate::resource::Resource; use crate::route::Route; use crate::service::{ HttpServiceFactory, ServiceFactory, ServiceFactoryWrapper, ServiceRequest, ServiceResponse, }; use crate::state::{State, StateFactory}; type HttpNewService

= BoxedNewService<(), ServiceRequest

, ServiceResponse, (), ()>; /// Application builder - structure that follows the builder pattern /// for building application instances. pub struct App where T: NewService>, { chain: T, extensions: Extensions, state: Vec>, host: String, _t: PhantomData<(P,)>, } impl App { /// Create application builder with empty state. Application can /// be configured with a builder-like pattern. pub fn new() -> Self { App { chain: AppChain, extensions: Extensions::new(), state: Vec::new(), host: "localhost:8080".to_string(), _t: PhantomData, } } } impl App where P: 'static, T: NewService< Request = ServiceRequest, Response = ServiceRequest

, Error = (), InitError = (), >, { /// Set application state. Applicatin state could be accessed /// by using `State` extractor where `T` is state type. /// /// **Note**: http server accepts an application factory rather than /// an application instance. Http server constructs an application /// instance for each thread, thus application state must be constructed /// multiple times. If you want to share state between different /// threads, a shared object should be used, e.g. `Arc`. Application /// state does not need to be `Send` or `Sync`. /// /// ```rust /// use std::cell::Cell; /// use actix_web::{web, App}; /// /// struct MyState { /// counter: Cell, /// } /// /// fn index(state: web::State) { /// state.counter.set(state.counter.get() + 1); /// } /// /// fn main() { /// let app = App::new() /// .state(MyState{ counter: Cell::new(0) }) /// .service( /// web::resource("/index.html").route( /// web::get().to(index))); /// } /// ``` pub fn state(mut self, state: S) -> Self { self.state.push(Box::new(State::new(state))); self } /// Set application state factory. This function is /// similar to `.state()` but it accepts state factory. State get /// constructed asynchronously during application initialization. pub fn state_factory(mut self, state: F) -> Self where F: Fn() -> Out + 'static, Out: IntoFuture + 'static, Out::Error: std::fmt::Debug, { self.state.push(Box::new(state)); self } /// Register a middleware. pub fn middleware( self, mw: F, ) -> AppRouter< T, P, B, impl NewService< Request = ServiceRequest

, Response = ServiceResponse, Error = (), InitError = (), >, > where M: Transform< AppRouting

, Request = ServiceRequest

, Response = ServiceResponse, Error = (), InitError = (), >, F: IntoTransform>, { let fref = Rc::new(RefCell::new(None)); let endpoint = ApplyTransform::new(mw, AppEntry::new(fref.clone())); AppRouter { endpoint, chain: self.chain, state: self.state, services: Vec::new(), default: None, factory_ref: fref, extensions: self.extensions, host: self.host, _t: PhantomData, } } /// Register a request modifier. It can modify any request parameters /// including payload stream. pub fn chain( self, chain: C, ) -> App< P1, impl NewService< Request = ServiceRequest, Response = ServiceRequest, Error = (), InitError = (), >, > where C: NewService< Request = ServiceRequest

, Response = ServiceRequest, Error = (), InitError = (), >, F: IntoNewService, { let chain = self.chain.and_then(chain.into_new_service()); App { chain, state: self.state, extensions: self.extensions, host: self.host, _t: PhantomData, } } /// Configure route for a specific path. /// /// This is a simplified version of the `App::service()` method. /// This method can not be could multiple times, in that case /// multiple resources with one route would be registered for same resource path. /// /// ```rust /// use actix_web::{web, App, HttpResponse}; /// /// fn index(data: web::Path<(String, String)>) -> &'static str { /// "Welcome!" /// } /// /// fn main() { /// let app = App::new() /// .route("/test1", web::get().to(index)) /// .route("/test2", web::post().to(|| HttpResponse::MethodNotAllowed())); /// } /// ``` pub fn route( self, path: &str, mut route: Route

, ) -> AppRouter> { self.service( Resource::new(path) .add_guards(route.take_guards()) .route(route), ) } /// Register http service. pub fn service(self, service: F) -> AppRouter> where F: HttpServiceFactory

+ 'static, { let fref = Rc::new(RefCell::new(None)); AppRouter { chain: self.chain, default: None, endpoint: AppEntry::new(fref.clone()), factory_ref: fref, extensions: self.extensions, state: self.state, host: self.host, services: vec![Box::new(ServiceFactoryWrapper::new(service))], _t: PhantomData, } } /// Set server host name. /// /// Host name is used by application router aa a hostname for url /// generation. Check [ConnectionInfo](./dev/struct.ConnectionInfo. /// html#method.host) documentation for more information. /// /// By default host name is set to a "localhost" value. pub fn hostname(mut self, val: &str) -> Self { self.host = val.to_owned(); self } } /// Application router builder - Structure that follows the builder pattern /// for building application instances. pub struct AppRouter { chain: C, endpoint: T, services: Vec>>, default: Option>>, factory_ref: Rc>>>, extensions: Extensions, state: Vec>, host: String, _t: PhantomData<(P, B)>, } impl AppRouter where P: 'static, B: MessageBody, T: NewService< Request = ServiceRequest

, Response = ServiceResponse, Error = (), InitError = (), >, { /// Configure route for a specific path. /// /// This is a simplified version of the `App::service()` method. /// This method can not be could multiple times, in that case /// multiple resources with one route would be registered for same resource path. /// /// ```rust /// use actix_web::{web, App, HttpResponse}; /// /// fn index(data: web::Path<(String, String)>) -> &'static str { /// "Welcome!" /// } /// /// fn main() { /// let app = App::new() /// .route("/test1", web::get().to(index)) /// .route("/test2", web::post().to(|| HttpResponse::MethodNotAllowed())); /// } /// ``` pub fn route(self, path: &str, mut route: Route

) -> Self { self.service( Resource::new(path) .add_guards(route.take_guards()) .route(route), ) } /// Register http service. /// /// Http service is any type that implements `HttpServiceFactory` trait. /// /// Actix web provides several services implementations: /// /// * *Resource* is an entry in route table which corresponds to requested URL. /// * *Scope* is a set of resources with common root path. /// * "StaticFiles" is a service for static files support pub fn service(mut self, factory: F) -> Self where F: HttpServiceFactory

+ 'static, { self.services .push(Box::new(ServiceFactoryWrapper::new(factory))); self } /// Register a middleware. pub fn middleware( self, mw: F, ) -> AppRouter< C, P, B1, impl NewService< Request = ServiceRequest

, Response = ServiceResponse, Error = (), InitError = (), >, > where M: Transform< T::Service, Request = ServiceRequest

, Response = ServiceResponse, Error = (), InitError = (), >, B1: MessageBody, F: IntoTransform, { let endpoint = ApplyTransform::new(mw, self.endpoint); AppRouter { endpoint, chain: self.chain, state: self.state, services: self.services, default: self.default, factory_ref: self.factory_ref, extensions: self.extensions, host: self.host, _t: PhantomData, } } /// Default resource to be used if no matching route could be found. /// /// Default resource works with resources only and does not work with /// custom services. pub fn default_resource(mut self, f: F) -> Self where F: FnOnce(Resource

) -> Resource, U: NewService< Request = ServiceRequest

, Response = ServiceResponse, Error = (), InitError = (), > + 'static, { // create and configure default resource self.default = Some(Rc::new(boxed::new_service( f(Resource::new("")).into_new_service().map_init_err(|_| ()), ))); self } /// Register an external resource. /// /// External resources are useful for URL generation purposes only /// and are never considered for matching at request time. Calls to /// `HttpRequest::url_for()` will work as expected. /// /// ```rust,ignore /// # extern crate actix_web; /// use actix_web::{App, HttpRequest, HttpResponse, Result}; /// /// fn index(req: &HttpRequest) -> Result { /// let url = req.url_for("youtube", &["oHg5SJYRHA0"])?; /// assert_eq!(url.as_str(), "https://youtube.com/watch/oHg5SJYRHA0"); /// Ok(HttpResponse::Ok().into()) /// } /// /// fn main() { /// let app = App::new() /// .resource("/index.html", |r| r.get().f(index)) /// .external_resource("youtube", "https://youtube.com/watch/{video_id}") /// .finish(); /// } /// ``` pub fn external_resource(self, _name: N, _url: U) -> Self where N: AsRef, U: AsRef, { // self.parts // .as_mut() // .expect("Use after finish") // .router // .register_external(name.as_ref(), ResourceDef::external(url.as_ref())); self } } impl IntoNewService, ServerConfig> for AppRouter where T: NewService< Request = ServiceRequest

, Response = ServiceResponse, Error = (), InitError = (), >, C: NewService< Request = ServiceRequest, Response = ServiceRequest

, Error = (), InitError = (), >, { fn into_new_service(self) -> AppInit { AppInit { chain: self.chain, state: self.state, endpoint: self.endpoint, services: RefCell::new(self.services), default: self.default, factory_ref: self.factory_ref, extensions: Rc::new(RefCell::new(Rc::new(self.extensions))), } } } #[cfg(test)] mod tests { use actix_service::Service; use super::*; use crate::http::{Method, StatusCode}; use crate::test::{block_on, init_service, TestRequest}; use crate::{web, HttpResponse}; #[test] fn test_default_resource() { let mut srv = init_service( App::new().service(web::resource("/test").to(|| HttpResponse::Ok())), ); let req = TestRequest::with_uri("/test").to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::OK); let req = TestRequest::with_uri("/blah").to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::NOT_FOUND); let mut srv = init_service( App::new() .service(web::resource("/test").to(|| HttpResponse::Ok())) .service( web::resource("/test2") .default_resource(|r| r.to(|| HttpResponse::Created())) .route(web::get().to(|| HttpResponse::Ok())), ) .default_resource(|r| r.to(|| HttpResponse::MethodNotAllowed())), ); let req = TestRequest::with_uri("/blah").to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::METHOD_NOT_ALLOWED); let req = TestRequest::with_uri("/test2").to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::OK); let req = TestRequest::with_uri("/test2") .method(Method::POST) .to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::CREATED); } #[test] fn test_state() { let mut srv = init_service(App::new().state(10usize).service( web::resource("/").to(|_: web::State| HttpResponse::Ok()), )); let req = TestRequest::default().to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::OK); let mut srv = init_service(App::new().state(10u32).service( web::resource("/").to(|_: web::State| HttpResponse::Ok()), )); let req = TestRequest::default().to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::INTERNAL_SERVER_ERROR); } #[test] fn test_state_factory() { let mut srv = init_service(App::new().state_factory(|| Ok::<_, ()>(10usize)).service( web::resource("/").to(|_: web::State| HttpResponse::Ok()), )); let req = TestRequest::default().to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::OK); let mut srv = init_service(App::new().state_factory(|| Ok::<_, ()>(10u32)).service( web::resource("/").to(|_: web::State| HttpResponse::Ok()), )); let req = TestRequest::default().to_request(); let resp = block_on(srv.call(req)).unwrap(); assert_eq!(resp.status(), StatusCode::INTERNAL_SERVER_ERROR); } // #[test] // fn test_handler() { // let app = App::new() // .handler("/test", |_: &_| HttpResponse::Ok()) // .finish(); // let req = TestRequest::with_uri("/test").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::OK); // let req = TestRequest::with_uri("/test/").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::OK); // let req = TestRequest::with_uri("/test/app").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::OK); // let req = TestRequest::with_uri("/testapp").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND); // let req = TestRequest::with_uri("/blah").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND); // } // #[test] // fn test_handler2() { // let app = App::new() // .handler("test", |_: &_| HttpResponse::Ok()) // .finish(); // let req = TestRequest::with_uri("/test").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::OK); // let req = TestRequest::with_uri("/test/").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::OK); // let req = TestRequest::with_uri("/test/app").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::OK); // let req = TestRequest::with_uri("/testapp").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND); // let req = TestRequest::with_uri("/blah").request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND); // } // #[test] // fn test_route() { // let app = App::new() // .route("/test", Method::GET, |_: HttpRequest| HttpResponse::Ok()) // .route("/test", Method::POST, |_: HttpRequest| { // HttpResponse::Created() // }) // .finish(); // let req = TestRequest::with_uri("/test").method(Method::GET).request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::OK); // let req = TestRequest::with_uri("/test") // .method(Method::POST) // .request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::CREATED); // let req = TestRequest::with_uri("/test") // .method(Method::HEAD) // .request(); // let resp = app.run(req); // assert_eq!(resp.as_msg().status(), StatusCode::NOT_FOUND); // } }