registry set server header

1 files changed, 5 insertions, 1 deletions

diff --git a/server/registry/src/main.rs b/server/registry/src/main.rs
index 5bb7a0a3..30fb8d66 100644
--- a/server/registry/src/main.rs
+++ b/server/registry/src/main.rs
@@ -26,7 +26,7 @@ use list::{generate_html_list, generate_json_list, r_list};
 use lobby::lobby_wrapper;
 use log::{error, info};
 use register::r_register;
-use rocket::{get, routes, shield::Shield, Config};
+use rocket::{fairing::AdHoc, get, http::Header, routes, shield::Shield, Config};
 use std::{
     cmp::Reverse,
     collections::HashMap,
@@ -73,6 +73,10 @@ fn main() {
                 })
                 .attach(Shield::new())
                 .manage(registry)
+                .attach(AdHoc::on_response("set server header", |_req, res| {
+                    res.set_header(Header::new("server", "hurrycurry-registry"));
+                    Box::pin(async {})
+                }))
                 .mount("/", routes![r_index, r_list, r_register])
                 .ignite()
                 .await