obscura-server 0.6.2

A server for relaying secure messages using the Signal Protocol
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

use crate::api::AppState;
use axum::body::Body;
use axum::extract::State;
use axum::http::{Request, StatusCode};
use axum::middleware::Next;
use axum::response::Response;

/// Middleware to log rate limit events and add Retry-After headers.
///
/// # Panics
/// Panics if the `x-ratelimit-after` header value cannot be parsed as a string.
pub(crate) async fn log_rate_limit_events(State(state): State<AppState>, req: Request<Body>, next: Next) -> Response {
    let mut response = next.run(req).await;

    let status = response.status();
    let ratelimit_after = if status == StatusCode::TOO_MANY_REQUESTS {
        response.headers().get("x-ratelimit-after").and_then(|v| v.to_str().ok().map(ToString::to_string))
    } else {
        None
    };

    state.rate_limit_service.log_decision(status, ratelimit_after.clone());

    if status == StatusCode::TOO_MANY_REQUESTS
        && let Some(after) = ratelimit_after
    {
        response.headers_mut().insert("retry-after", after.parse().expect("Failed to parse retry-after header value"));
    }

    response
}