From 1300089bc4cdab99a5ce2f5eda58cccf086a7e64 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Sun, 30 Mar 2025 06:01:19 +0200
Subject: disablable server header

---
 src/main.rs | 28 +++++++++++++++-------------
 1 file changed, 15 insertions(+), 13 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index af9c30b..7fb07ca 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -403,19 +403,21 @@ async fn service(
     };
     let mut resp = config.handler.handle(&mut context, request).await?;
 
-    let server_header = resp.headers().get(SERVER).cloned();
-    resp.headers_mut().insert(
-        SERVER,
-        if let Some(o) = server_header {
-            HeaderValue::from_str(&format!(
-                "{} via gnix",
-                o.to_str().ok().unwrap_or("invalid")
-            ))
-            .unwrap()
-        } else {
-            HeaderValue::from_static("gnix")
-        },
-    );
+    if !config.disable_server_header {
+        let server_header = resp.headers().get(SERVER).cloned();
+        resp.headers_mut().insert(
+            SERVER,
+            if let Some(o) = server_header {
+                HeaderValue::from_str(&format!(
+                    "{} via gnix",
+                    o.to_str().ok().unwrap_or("invalid")
+                ))
+                .unwrap()
+            } else {
+                HeaderValue::from_static("gnix")
+            },
+        );
+    }
 
     Ok(resp)
 }
-- 
cgit v1.2.3-70-g09d2