link-assistant-router 1.4.2

//! HTTP mapping for token admission failures.

use axum::http::StatusCode;
use axum::response::Response;

pub fn budget_error_response(error: &crate::token::TokenError) -> Response {
    match error {
        crate::token::TokenError::LimitExceeded(_)
        | crate::token::TokenError::TokenLimitExceeded(_)
        | crate::token::TokenError::RateLimitExceeded => crate::api_error::error_response(
            StatusCode::TOO_MANY_REQUESTS,
            "rate_limit_error",
            // The client-facing wording, which carries the counts and says
            // what to do about them; `Display` is for the server's own log
            // (issue #355).
            &error.client_message(),
        ),
        crate::token::TokenError::Storage(_) => crate::api_error::error_response(
            StatusCode::INTERNAL_SERVER_ERROR,
            "storage_error",
            &error.to_string(),
        ),
        _ => crate::api_error::error_response(
            StatusCode::INTERNAL_SERVER_ERROR,
            "api_error",
            &error.to_string(),
        ),
    }
}