From 975f2325a0ee78f75e0ed2ab899d161868250e8d Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Fri, 24 Feb 2023 15:06:21 +0100
Subject: italic file type and hyperlink to source

---
 src/proxy.rs | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'src/proxy.rs')

diff --git a/src/proxy.rs b/src/proxy.rs
index 65fc5a8..aff1ae5 100644
--- a/src/proxy.rs
+++ b/src/proxy.rs
@@ -2,7 +2,7 @@ use crate::ServiceError;
 use http_body_util::{combinators::BoxBody, BodyExt};
 use hyper::{
     body::Incoming,
-    header::UPGRADE,
+    header::{SERVER, UPGRADE},
     http::{
         uri::{PathAndQuery, Scheme},
         HeaderValue,
@@ -67,8 +67,16 @@ pub async fn proxy_request(
             .map_err(ServiceError::Hyper)?
     };
 
-    resp.headers_mut()
-        .insert("server", HeaderValue::from_static("gnix"));
+    let server_header = resp.headers().get(SERVER).cloned();
+    resp.headers_mut().insert(
+        SERVER,
+        HeaderValue::from_str(&if let Some(o) = server_header {
+            format!("{} via gnix", o.to_str().unwrap())
+        } else {
+            format!("gnix")
+        })
+        .unwrap(),
+    );
 
     if do_upgrade {
         let on_upgrade_upstream = resp.extensions_mut().remove::<OnUpgrade>();
-- 
cgit v1.2.3-70-g09d2