From 32ba7fd735b965f2c2b4d919ec25042ca5389c9e Mon Sep 17 00:00:00 2001 From: metamuffin Date: Thu, 26 Jan 2023 07:30:22 +0100 Subject: set server header and refactor stream resp --- server/src/routes/mod.rs | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'server/src/routes/mod.rs') diff --git a/server/src/routes/mod.rs b/server/src/routes/mod.rs index cbc1b42..d1bf7fc 100644 --- a/server/src/routes/mod.rs +++ b/server/src/routes/mod.rs @@ -5,7 +5,7 @@ */ use crate::{database::Database, library::Library, routes::ui::error::MyResult, CONF}; use jellyremuxer::RemuxerContext; -use rocket::{catchers, config::SecretKey, fs::FileServer, get, routes, Build, Config, Rocket}; +use rocket::{catchers, config::SecretKey, fs::FileServer, get, routes, Build, Config, Rocket, fairing::AdHoc, http::Header}; use std::fs::File; use stream::r_stream; use ui::{ @@ -43,7 +43,10 @@ pub fn build_rocket( }) .manage(remuxer) .manage(library) - .manage(database) + .manage(database).attach(AdHoc::on_response("set server header", |_req,res| { + res.set_header(Header::new("server","jellything")); + Box::pin(async {}) + })) .register("/", catchers![r_catch]) .mount("/assets", FileServer::from(&CONF.asset_dir)) .mount( -- cgit v1.2.3-70-g09d2