#54 Add do_use_guard option

src/resource_files.rs

@@ -8,7 +8,7 @@ use actix_web::{
         header::{self, ContentType},
         Method, StatusCode,
     },
-    HttpMessage, HttpRequest, HttpResponse, ResponseError,
+    HttpMessage, HttpRequest, HttpResponse, ResponseError, guard::{Guard, GuardContext},
 };
 use derive_more::{Deref, Display, Error};
 use futures_util::future::{ok, FutureExt, LocalBoxFuture, Ready};
@@ -38,8 +38,10 @@ use std::{collections::HashMap, ops::Deref, rc::Rc};
 /// }
 /// ```
 #[allow(clippy::needless_doctest_main)]
+#[derive(Clone)]
 pub struct ResourceFiles {
     not_resolve_defaults: bool,
+    use_guard: bool,
     not_found_resolves_to: Option<String>,
     inner: Rc<ResourceFilesInner>,
 }
@@ -61,6 +63,7 @@ impl ResourceFiles {
             inner: Rc::new(inner),
             not_resolve_defaults: false,
             not_found_resolves_to: None,
+            use_guard: false,
         }
     }
 
@@ -85,6 +88,14 @@ impl ResourceFiles {
     pub fn resolve_not_found_to_root(self) -> Self {
         self.resolve_not_found_to(INDEX_HTML)
     }
+
+    /// Use guard to check if this request should be handled.
+    /// Can be useful if you want to serve static files from root, or another path that should also be used by other handlers.
+    /// By default guard is not used.
+    pub fn do_use_guard(mut self) -> Self {
+        self.use_guard = true;
+        self
+    }
 }
 
 impl Deref for ResourceFiles {
@@ -95,6 +106,17 @@ impl Deref for ResourceFiles {
     }
 }
 
+impl Guard for ResourceFiles {
+    fn check(&self, ctx: &GuardContext<'_>) -> bool {
+        for filename in self.inner.files.keys() {
+            if *filename == ctx.head().uri.path() {
+                return true;
+            }
+        }
+        false
+    }
+}
+
 impl HttpServiceFactory for ResourceFiles {
     fn register(self, config: &mut AppService) {
         let prefix = self.path.trim_start_matches('/');
@@ -103,7 +125,12 @@ impl HttpServiceFactory for ResourceFiles {
         } else {
             ResourceDef::prefix(prefix)
         };
-        config.register_service(rdef, None, self, None)
+        let guards: Option<Vec<Box<dyn Guard>>> = if self.use_guard {
+            Some(vec!(Box::new(self.clone())))
+        } else {
+            None
+        };
+        config.register_service(rdef, guards, self, None);
     }
 }
 
@@ -133,7 +160,7 @@ pub struct ResourceFilesService {
     inner: Rc<ResourceFilesInner>,
 }
 
-impl<'a> Service<ServiceRequest> for ResourceFilesService {
+impl Service<ServiceRequest> for ResourceFilesService {
     type Response = ServiceResponse;
     type Error = Error;
     type Future = Ready<Result<Self::Response, Self::Error>>;