handle trailing slashes (#788)

mirage · Jan 4, 2023 · 5c5a51e · 5c5a51e
1 parent 2954572
commit 5c5a51e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/lib/middleware.ml b/lib/middleware.ml
@@ -26,4 +26,14 @@ struct
           ~headers:(Dream.all_headers response)
           (Dream.status response)
     | _ -> handler request
+
+  let no_trailing_slash next_handler request =
+    let target = Dream.target request in
+    match target with
+    | "/" -> next_handler request
+    | _ ->
+        let length = String.length target in
+        if target.[length - 1] = '/' then
+          Dream.redirect request (String.sub target 0 (length - 1))
+        else next_handler request
 end
diff --git a/lib/mirageio.ml b/lib/mirageio.ml
@@ -174,7 +174,10 @@ struct
     let router = Dream.router routes
   end
 
-  let router = Dream.logger @@ Middleware.head @@ Router.router
+  let router =
+    Dream.logger @@ Middleware.no_trailing_slash @@ Middleware.head
+    @@ Router.router
+
   let http ?(port = 80) stack = Dream.http ~port (Stack.tcp stack) router
 
   let https ?(port = 443) ?tls stack =