ellie/atuin
1
1
Fork 0
mirror of https://github.com/ellie/atuin.git synced 2024-09-19 08:57:16 +03:00
Code Issues Projects Releases Wiki Activity

A man is not dead while his name is still spoken (#1280)

Browse Source
This commit is contained in:
Chris Rose 2023-10-03 12:36:16 -07:00 committed by GitHub
parent 6f803b548b
commit 573ff69749
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
atuin-server/src/router.rs
View File

@ -2,7 +2,9 @@ use async_trait::async_trait;
use atuin_common::api::ErrorResponse;
use axum::{
extract::FromRequestParts,
response::IntoResponse,
http::Request,
middleware::Next,
response::{IntoResponse, Response},
routing::{delete, get, post},
Router,
};
@ -76,6 +78,18 @@ async fn teapot() -> impl IntoResponse {
(http::StatusCode::IM_A_TEAPOT, "🫖")
}
async fn clacks_overhead<B>(request: Request<B>, next: Next<B>) -> Response {
let mut response = next.run(request).await;
let gnu_terry_value = "GNU Terry Pratchett";
let gnu_terry_header = "X-Clacks-Overhead";
response
.headers_mut()
.insert(gnu_terry_header, gnu_terry_value.parse().unwrap());
response
}
#[derive(Clone)]
pub struct AppState<DB: Database> {
pub database: DB,
@ -107,5 +121,9 @@ pub fn router<DB: Database>(database: DB, settings: Settings<DB::Settings>) -> R
}
.fallback(teapot)
.with_state(AppState { database, settings })
.layer(ServiceBuilder::new().layer(TraceLayer::new_for_http()))
.layer(
ServiceBuilder::new()
.layer(axum::middleware::from_fn(clacks_overhead))
.layer(TraceLayer::new_for_http()),
)
}