disablable server header

1 files changed, 15 insertions, 13 deletions

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)
 }