ohttp-gateway 0.1.1

A OHTTP Gateway server, meant to run between a OHTTP Relay and a target web service.
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use crate::AppState;
use axum::{
    extract::State,
    http::{header, HeaderName, StatusCode},
    response::{IntoResponse, Response},
};
use chrono::Utc;
use tracing::info;

/// Handler for /ohttp-keys endpoint
/// Returns key configurations in the standard OHTTP format
pub async fn get_ohttp_keys(State(state): State<AppState>) -> Result<Response, StatusCode> {
    state.metrics.key_requests_total.inc();

    match state.key_manager.get_encoded_config().await {
        Ok(config_bytes) => {
            info!("Serving {} bytes of key configurations", config_bytes.len());

            // Calculate cache duration based on rotation interval
            let max_age = calculate_cache_max_age(&state);

            Ok((
                StatusCode::OK,
                [
                    (header::CONTENT_TYPE, "application/ohttp-keys"),
                    (header::CACHE_CONTROL, &format!("public, max-age={max_age}")),
                    (HeaderName::from_static("x-content-type-options"), "nosniff"),
                ],
                config_bytes,
            )
                .into_response())
        }
        Err(e) => {
            tracing::error!("Failed to encode key config: {e}");
            Err(StatusCode::INTERNAL_SERVER_ERROR)
        }
    }
}

/// Legacy endpoint for backward compatibility
/// Some older clients may still use /ohttp-configs
pub async fn get_legacy_ohttp_configs(
    State(state): State<AppState>,
) -> Result<Response, StatusCode> {
    // Just forward to the main handler
    get_ohttp_keys(State(state)).await
}

/// Calculate appropriate cache duration for key configurations
fn calculate_cache_max_age(state: &AppState) -> u64 {
    // Cache for 10% of rotation interval, minimum 1 hour, maximum 24 hours
    let ten_percent = state.config.key_rotation_interval.as_secs() / 10;
    let one_hour = 3600;
    let twenty_four_hours = 86400;

    ten_percent.max(one_hour).min(twenty_four_hours)
}

/// Health check endpoint specifically for key management
pub async fn key_health_check(State(state): State<AppState>) -> impl IntoResponse {
    let stats = state.key_manager.get_stats().await;

    let health_status = if stats.active_keys > 0 && stats.expired_keys == 0 {
        "healthy"
    } else if stats.active_keys > 0 {
        "degraded"
    } else {
        "unhealthy"
    };

    axum::Json(serde_json::json!({
        "status": health_status,
        "timestamp": Utc::now().to_rfc3339(),
        "key_stats": {
            "active_key_id": stats.active_key_id,
            "total_keys": stats.total_keys,
            "active_keys": stats.active_keys,
            "expired_keys": stats.expired_keys,
            "rotation_enabled": stats.auto_rotation_enabled,
            "rotation_interval_hours": stats.rotation_interval.as_secs() / 3600,
        }
    }))
}