xevion/Pac-Man
1
0
Fork 0
mirror of https://github.com/Xevion/Pac-Man.git synced 2025-12-13 10:12:19 -06:00
Code Issues Packages Projects Releases Wiki Activity

feat: add Server header middleware, bump version to v0.4.0

Browse Source
This commit is contained in:
Ryan Walters
2025-09-17 12:37:12 -05:00
parent 916428fe76
commit 1be59f474d
4 changed files with 29 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
pacman-server/src/main.rs
View File

@@ -19,6 +19,9 @@ use tokio::signal::unix::{signal, SignalKind};
use tokio::sync::{watch, Notify};
use tracing::{info, trace, warn};
// Constant value for the Server header: "<crate>/<version>"
const SERVER_HEADER_VALUE: &str = concat!(env!("CARGO_PKG_NAME"), "/", env!("CARGO_PKG_VERSION"));
#[tokio::main]
async fn main() {
// Load environment variables
@@ -60,7 +63,8 @@ async fn main() {
.route("/logout", get(routes::logout_handler))
.route("/profile", get(routes::profile_handler))
.with_state(app_state.clone())
.layer(CookieLayer::default());
.layer(CookieLayer::default())
.layer(axum::middleware::from_fn(inject_server_header));
info!(%addr, "Starting HTTP server bind");
let listener = tokio::net::TcpListener::bind(addr).await.unwrap();
@@ -172,3 +176,15 @@ async fn shutdown_signal() -> Instant {
_ = sigterm => { Instant::now() }
}
}
async fn inject_server_header(
req: axum::http::Request<axum::body::Body>,
next: axum::middleware::Next,
) -> Result<axum::response::Response, axum::http::StatusCode> {
let mut res = next.run(req).await;
res.headers_mut().insert(
axum::http::header::SERVER,
axum::http::HeaderValue::from_static(SERVER_HEADER_VALUE),
);
Ok(res)
}