tibba-middleware 0.2.2

middleware for tibba
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

// Copyright 2026 Tree xie.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use super::LOG_TARGET;
use axum::extract::Request;
use axum::middleware::Next;
use axum::response::Response;
use axum_extra::extract::cookie::CookieJar;
use scopeguard::defer;
use std::sync::Arc;
use tibba_state::{CTX, Context};
use tibba_util::{
    get_device_id_from_cookie, set_header_if_not_exist, set_no_cache_if_not_exist, uuid,
};
use tracing::debug;

/// Entry middleware that sets up request context and handles response headers
///
/// This middleware:
/// 1. Extracts device ID from cookies
/// 2. Generates a unique trace ID
/// 3. Creates and manages request context
/// 4. Sets response headers for caching and tracing
///
/// The middleware ensures each request has:
/// - A unique trace ID for request tracking
/// - Proper cache control headers
/// - Access to device identification
/// - Request-scoped context
pub async fn entry(jar: CookieJar, mut req: Request, next: Next) -> Response {
    // Log middleware entry
    debug!(target: LOG_TARGET, "--> entry");
    // Ensure exit logging happens even if processing panics
    defer!(debug!(target: LOG_TARGET, "<-- entry"););

    // Extract device ID from cookies for user/device tracking
    let device_id = get_device_id_from_cookie(&jar).unwrap_or_default();
    // Generate unique trace ID for request tracking
    // Create new context with device and trace information
    let ctx = Arc::new(Context::new(device_id, uuid()));

    req.extensions_mut().insert(ctx.clone());

    // Process request within context scope
    let mut res = CTX.scope(ctx.clone(), async { next.run(req).await }).await;

    // Add response headers
    let headers = res.headers_mut();
    // Ensure cache control headers are set
    set_no_cache_if_not_exist(headers);
    // Add trace ID header for request tracking
    let _ = set_header_if_not_exist(headers, "X-Trace-Id", &ctx.trace_id);

    res
}