From c5e8c1b710aad692f2baef654261bfa11a70d11e Mon Sep 17 00:00:00 2001
From: Josh Leeb-du Toit <josh.leebdutoit@gmail.com>
Date: Thu, 21 Jun 2018 18:17:27 +1000
Subject: [PATCH] Propagate default resources to underlying scopes

---
 src/application.rs | 24 ++++++++++++++++--------
 src/handler.rs     | 11 +++++++++++
 src/scope.rs       |  8 ++++++++
 3 files changed, 35 insertions(+), 8 deletions(-)

diff --git a/src/application.rs b/src/application.rs
index 93008b3d2..b22fe8bb4 100644
--- a/src/application.rs
+++ b/src/application.rs
@@ -29,7 +29,7 @@ pub struct HttpApplication<S = ()> {
 #[doc(hidden)]
 pub struct Inner<S> {
     prefix: usize,
-    default: ResourceHandler<S>,
+    default: Rc<RefCell<ResourceHandler<S>>>,
     encoding: ContentEncoding,
     resources: Vec<ResourceHandler<S>>,
     handlers: Vec<PrefixHandlerType<S>>,
@@ -51,7 +51,7 @@ impl<S: 'static> PipelineHandler<S> for Inner<S> {
         match htype {
             HandlerType::Normal(idx) => match self.resources[idx].handle(req) {
                 Ok(result) => result,
-                Err(req) => match self.default.handle(req) {
+                Err(req) => match self.default.borrow_mut().handle(req) {
                     Ok(result) => result,
                     Err(_) => AsyncResult::ok(HttpResponse::new(StatusCode::NOT_FOUND)),
                 },
@@ -60,7 +60,7 @@ impl<S: 'static> PipelineHandler<S> for Inner<S> {
                 PrefixHandlerType::Handler(_, ref mut hnd) => hnd.handle(req),
                 PrefixHandlerType::Scope(_, ref mut hnd, _) => hnd.handle(req),
             },
-            HandlerType::Default => match self.default.handle(req) {
+            HandlerType::Default => match self.default.borrow_mut().handle(req) {
                 Ok(result) => result,
                 Err(_) => AsyncResult::ok(HttpResponse::new(StatusCode::NOT_FOUND)),
             },
@@ -172,7 +172,7 @@ struct ApplicationParts<S> {
     state: S,
     prefix: String,
     settings: ServerSettings,
-    default: ResourceHandler<S>,
+    default: Rc<RefCell<ResourceHandler<S>>>,
     resources: Vec<(Resource, Option<ResourceHandler<S>>)>,
     handlers: Vec<PrefixHandlerType<S>>,
     external: HashMap<String, Resource>,
@@ -223,7 +223,7 @@ where
                 state,
                 prefix: "/".to_owned(),
                 settings: ServerSettings::default(),
-                default: ResourceHandler::default_not_found(),
+                default: Rc::new(RefCell::new(ResourceHandler::default_not_found())),
                 resources: Vec::new(),
                 handlers: Vec::new(),
                 external: HashMap::new(),
@@ -473,7 +473,7 @@ where
     {
         {
             let parts = self.parts.as_mut().expect("Use after finish");
-            f(&mut parts.default);
+            f(&mut parts.default.borrow_mut());
         }
         self
     }
@@ -614,7 +614,7 @@ where
 
     /// Finish application configuration and create `HttpHandler` object.
     pub fn finish(&mut self) -> HttpApplication<S> {
-        let parts = self.parts.take().expect("Use after finish");
+        let mut parts = self.parts.take().expect("Use after finish");
         let prefix = parts.prefix.trim().trim_right_matches('/');
         let (prefix, prefix_len) = if prefix.is_empty() {
             ("/".to_owned(), 0)
@@ -627,11 +627,19 @@ where
             resources.push((pattern, None));
         }
 
+        for ref mut handler in parts.handlers.iter_mut() {
+            if let PrefixHandlerType::Scope(_, ref mut route_handler, _) = handler {
+                if !route_handler.has_default_resource() {
+                    route_handler.default_resource(Rc::clone(&parts.default));
+                }
+            };
+        }
+
         let (router, resources) = Router::new(&prefix, parts.settings, resources);
 
         let inner = Rc::new(RefCell::new(Inner {
             prefix: prefix_len,
-            default: parts.default,
+            default: Rc::clone(&parts.default),
             encoding: parts.encoding,
             handlers: parts.handlers,
             resources,
diff --git a/src/handler.rs b/src/handler.rs
index d330e0716..4428ce83f 100644
--- a/src/handler.rs
+++ b/src/handler.rs
@@ -1,5 +1,7 @@
+use std::cell::RefCell;
 use std::marker::PhantomData;
 use std::ops::Deref;
+use std::rc::Rc;
 
 use futures::future::{err, ok, Future};
 use futures::{Async, Poll};
@@ -8,6 +10,7 @@ use error::Error;
 use http::StatusCode;
 use httprequest::HttpRequest;
 use httpresponse::HttpResponse;
+use resource::ResourceHandler;
 
 /// Trait defines object that could be registered as route handler
 #[allow(unused_variables)]
@@ -403,6 +406,14 @@ where
 // /// Trait defines object that could be registered as resource route
 pub(crate) trait RouteHandler<S>: 'static {
     fn handle(&mut self, req: HttpRequest<S>) -> AsyncResult<HttpResponse>;
+
+    fn has_default_resource(&self) -> bool {
+        false
+    }
+
+    fn default_resource(&mut self, default: Rc<RefCell<ResourceHandler<S>>>) {
+        unimplemented!()
+    }
 }
 
 /// Route handler wrapper for Handler
diff --git a/src/scope.rs b/src/scope.rs
index 6cc4929e1..70fb17287 100644
--- a/src/scope.rs
+++ b/src/scope.rs
@@ -405,6 +405,14 @@ impl<S: 'static> RouteHandler<S> for Scope<S> {
             unimplemented!()
         }
     }
+
+    fn has_default_resource(&self) -> bool {
+        self.default.is_some()
+    }
+
+    fn default_resource(&mut self, default: ScopeResource<S>) {
+        self.default = Some(default);
+    }
 }
 
 struct Wrapper<S: 'static> {