Static: don't set Content-Type on error responses

Fixes #88.
aantron · Jul 2, 2021 · 809835d · 809835d
1 parent cd8464e
commit 809835d
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/middleware/static.ml b/src/middleware/static.ml
@@ -76,12 +76,19 @@ let static ?(loader = from_filesystem) local_root = fun request ->
 
       let%lwt response = loader local_root path request in
 
-      (* TODO Can use a concise helper here. *)
       let response =
         if Dream.has_header "Content-Type" response then
           response
         else
-          Dream.add_header "Content-Type" (Magic_mime.lookup path) response
+          match Dream.status response with
+          | `OK
+          | `Non_Authoritative_Information
+          | `No_Content
+          | `Reset_Content
+          | `Partial_Content ->
+            Dream.add_header "Content-Type" (Magic_mime.lookup path) response
+          | _ ->
+            response
       in
 
       Lwt.return response