62 lines
1.9 KiB
Rust
62 lines
1.9 KiB
Rust
use axum::headers::{AccessControlMaxAge, HeaderMapExt};
|
|
use axum::http::header::CONTENT_SECURITY_POLICY;
|
|
use axum::http::{Request, StatusCode};
|
|
use axum::middleware::{from_fn, Next};
|
|
use axum::Router;
|
|
use hyper::Response;
|
|
use std::time::Duration;
|
|
use tower::util::MapResponseLayer;
|
|
use tower_http::services::{ServeDir, ServeFile};
|
|
|
|
pub fn static_serve_sw(file: &str) -> Router {
|
|
Router::new()
|
|
.nest_service("", ServeFile::new(file))
|
|
.layer(MapResponseLayer::new(|mut res: Response<_>| {
|
|
res.headers_mut()
|
|
.typed_insert(AccessControlMaxAge::from(Duration::from_secs(60 * 10)));
|
|
|
|
res
|
|
}))
|
|
}
|
|
|
|
pub fn static_serve(dir: &str) -> Router {
|
|
let serve_dir = ServeDir::new(dir);
|
|
Router::new()
|
|
.nest_service("", serve_dir)
|
|
.layer(MapResponseLayer::new(|mut res: Response<_>| {
|
|
res.headers_mut()
|
|
.typed_insert(AccessControlMaxAge::from(Duration::from_secs(
|
|
60 * 60 * 24 * 7,
|
|
)));
|
|
|
|
res
|
|
}))
|
|
}
|
|
|
|
pub fn static_serve_svg(dir: &str) -> Router {
|
|
let serve_dir = ServeDir::new(dir);
|
|
Router::new()
|
|
.nest_service("", serve_dir)
|
|
.layer(from_fn(|req: Request<_>, next: Next<_>| async {
|
|
if !req.uri().path().ends_with(".svg") {
|
|
return Err(StatusCode::NOT_FOUND);
|
|
}
|
|
|
|
Ok(next.run(req).await)
|
|
}))
|
|
.layer(MapResponseLayer::new(|mut res: Response<_>| {
|
|
let headers = res.headers_mut();
|
|
headers.typed_insert(AccessControlMaxAge::from(Duration::from_secs(
|
|
60 * 60 * 24 * 7,
|
|
)));
|
|
headers.insert(
|
|
CONTENT_SECURITY_POLICY,
|
|
"default-src 'none'; style-src 'unsafe-inline'"
|
|
.parse()
|
|
.unwrap(),
|
|
);
|
|
|
|
res
|
|
}))
|
|
}
|