Use http::Response for constructing replies

1 files changed, 14 insertions, 11 deletions

diff --git a/src/main.rs b/src/main.rs
index 6669791..e3ac9eb 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -51,20 +51,23 @@ async fn main() {
 
       let (theme, res) = lfm_embed::theming::render_theme(q.theme.as_deref(), &data);
       log::debug!(target: "lfm_embed::server::user", "Using theme {theme}");
+
       match res {
         Err(status) =>
-          Box::new(warp::reply::with_status(warp::reply::html("<h1>Internal Server Error.</h1>"), status))
-            as Box<dyn warp::reply::Reply>,
+          http::Response::builder()
+            .status(status)
+            .header("Content-Type", "text/html")
+            .body("<h1>Internal Server Error.</h1><p>Occurred when templating, check the logs.</p>".into())
+            .unwrap(),
+
         Ok(contents) =>
-          Box::new(warp::reply::with_header(
-            warp::reply::with_header(
-              warp::reply::with_status(
-                warp::reply::html(
-                  contents
-                ), status
-              ), "Refresh", dur.as_secs()
-            ), "X-Selected-Theme", theme
-         )) as Box<dyn warp::reply::Reply>
+          http::Response::builder()
+            .status(status)
+            .header("Refresh", dur.as_secs())
+            .header("X-Selected-Theme", theme)
+            .header("Content-Type", "text/html")
+            .body(contents)
+            .unwrap()
       }
     });