static: Extract lookup(path) method (rust-lang#109)

Turbo87 · web-flow · commit a9f63259888d · 2022-12-04T16:06:04.000+01:00
This allows us to lookup specific files without having to mutate the `path` of the request.
diff --git a/conduit-static/src/lib.rs b/conduit-static/src/lib.rs
@@ -15,11 +15,8 @@ impl Static {
             path: path.as_ref().to_path_buf(),
         }
     }
-}
 
-impl Handler for Static {
-    fn call(&self, request: &mut dyn RequestExt) -> HandlerResult {
-        let request_path = request.path();
+    pub fn lookup(&self, request_path: &str) -> HandlerResult {
         let request_path = request_path.strip_prefix('/').unwrap_or(request_path);
         if request_path.contains("..") {
             return Ok(not_found());
@@ -47,6 +44,12 @@ impl Handler for Static {
     }
 }
 
+impl Handler for Static {
+    fn call(&self, request: &mut dyn RequestExt) -> HandlerResult {
+        self.lookup(request.path())
+    }
+}
+
 fn not_found() -> Response<Body> {
     Response::builder()
         .status(StatusCode::NOT_FOUND)
@@ -85,6 +88,25 @@ mod tests {
         assert_eq!(*res.into_cow(), b"[package]"[..]);
     }
 
+    #[test]
+    fn test_lookup() {
+        let td = TempDir::new("conduit-static").unwrap();
+        let root = td.path();
+        let handler = Static::new(root);
+        File::create(&root.join("Cargo.toml"))
+            .unwrap()
+            .write_all(b"[package]")
+            .unwrap();
+
+        let res = handler.lookup("Cargo.toml").expect("No response");
+        assert_eq!(
+            res.headers().get(header::CONTENT_TYPE).unwrap(),
+            "application/toml"
+        );
+        assert_eq!(res.headers().get(header::CONTENT_LENGTH).unwrap(), "9");
+        assert_eq!(*res.into_cow(), b"[package]"[..]);
+    }
+
     #[test]
     fn test_mime_types() {
         let td = TempDir::new("conduit-static").unwrap();
