mockforge-http 0.3.126

HTTP/REST protocol support for MockForge
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174

//! Consistency middleware for HTTP
//!
//! This middleware ensures HTTP requests/responses use the unified state
//! from the consistency engine (persona, scenario, reality level, etc.)

use crate::consistency::HttpAdapter;
use axum::{body::Body, extract::Request, http::Response, middleware::Next};
use mockforge_core::consistency::ConsistencyEngine;
use mockforge_core::request_logger::RealityTraceMetadata;
use std::sync::Arc;
use tracing::debug;

/// Consistency middleware state
#[derive(Clone)]
pub struct ConsistencyMiddlewareState {
    /// Consistency engine
    pub engine: Arc<ConsistencyEngine>,
    /// HTTP adapter
    pub adapter: Arc<HttpAdapter>,
    /// X-Ray state for request context storage (optional)
    pub xray_state: Option<Arc<crate::handlers::xray::XRayState>>,
}

/// Consistency middleware
///
/// This middleware:
/// 1. Extracts workspace ID from request (header, query param, or default)
/// 2. Gets unified state from consistency engine
/// 3. Inserts state into request extensions for handlers to use
/// 4. Ensures responses reflect the unified state
pub async fn consistency_middleware(req: Request, next: Next) -> Response<Body> {
    // Extract workspace ID from request
    // Priority: X-MockForge-Workspace header > query param > default
    let workspace_id = req
        .headers()
        .get("X-MockForge-Workspace")
        .and_then(|h| h.to_str().ok())
        .map(|s| s.to_string())
        .or_else(|| {
            req.uri().query().and_then(|q| {
                q.split('&').find_map(|pair| {
                    let mut parts = pair.splitn(2, '=');
                    if parts.next() == Some("workspace") {
                        parts.next().and_then(|v| {
                            urlencoding::decode(v).ok().map(|decoded| decoded.to_string())
                        })
                    } else {
                        None
                    }
                })
            })
        })
        .unwrap_or_else(|| "default".to_string());

    // Get state from extensions (set by router)
    let state = req.extensions().get::<ConsistencyMiddlewareState>();

    if let Some(state) = state {
        // Get unified state for workspace
        if let Some(unified_state) = state.engine.get_state(&workspace_id).await {
            // Extract values for headers before moving unified_state
            let persona_id = unified_state.active_persona.as_ref().map(|p| p.id.clone());
            let scenario_id = unified_state.active_scenario.clone();
            let reality_level = unified_state.reality_level.value();
            let reality_ratio = unified_state.reality_continuum_ratio;
            // Note: ChaosScenario is now serde_json::Value, so we extract the name field
            let chaos_rules: Vec<String> = unified_state
                .active_chaos_rules
                .iter()
                .filter_map(|r| r.get("name").and_then(|v| v.as_str()).map(|s| s.to_string()))
                .collect();
            let request_id = uuid::Uuid::new_v4().to_string();

            // Build reality trace metadata from unified state
            // Use the path from the request URI for path-specific blend ratio calculation
            let path = req.uri().path();

            // Record reality continuum usage if blend ratio > 0
            if reality_ratio > 0.0 {
                mockforge_core::pillar_tracking::record_reality_usage(
                    Some(workspace_id.clone()),
                    None,
                    "blended_reality_ratio",
                    serde_json::json!({
                        "ratio": reality_ratio,
                        "path": path
                    }),
                )
                .await;
            }

            // Record chaos usage if chaos rules are active
            if !chaos_rules.is_empty() {
                mockforge_core::pillar_tracking::record_reality_usage(
                    Some(workspace_id.clone()),
                    None,
                    "chaos_enabled",
                    serde_json::json!({
                        "rules": chaos_rules,
                        "count": chaos_rules.len()
                    }),
                )
                .await;
            }
            let reality_metadata =
                RealityTraceMetadata::from_unified_state(&unified_state, reality_ratio, path);

            // Store request context snapshot if X-Ray state is available
            if let Some(xray_state) = &state.xray_state {
                // Clone unified_state before moving it
                let unified_state_clone = unified_state.clone();
                let request_id_clone = request_id.clone();
                let workspace_id_clone = workspace_id.clone();

                // Store snapshot asynchronously (don't block request processing)
                let xray_state_clone = xray_state.clone();
                tokio::spawn(async move {
                    crate::handlers::xray::store_request_context(
                        &xray_state_clone,
                        request_id_clone,
                        workspace_id_clone,
                        &unified_state_clone,
                    )
                    .await;
                });
            }

            // Insert unified state and reality metadata into request extensions for handlers
            let mut req = req;
            req.extensions_mut().insert(unified_state);
            req.extensions_mut().insert(reality_metadata);

            // Continue with request processing
            let mut response = next.run(req).await;

            // Add X-Ray headers to response for browser extension
            // Use .parse().ok() to safely handle dynamic values that may contain invalid header characters
            if let Ok(value) = workspace_id.parse() {
                response.headers_mut().insert("X-MockForge-Workspace", value);
            }
            if let Ok(value) = request_id.parse() {
                response.headers_mut().insert("X-MockForge-Request-ID", value);
            }
            if let Some(ref persona_id) = persona_id {
                if let Ok(value) = persona_id.parse() {
                    response.headers_mut().insert("X-MockForge-Persona", value);
                }
            }
            if let Some(ref scenario_id) = scenario_id {
                if let Ok(value) = scenario_id.parse() {
                    response.headers_mut().insert("X-MockForge-Scenario", value);
                }
            }
            if let Ok(value) = reality_level.to_string().parse() {
                response.headers_mut().insert("X-MockForge-Reality-Level", value);
            }
            if let Ok(value) = reality_ratio.to_string().parse() {
                response.headers_mut().insert("X-MockForge-Reality-Ratio", value);
            }
            if !chaos_rules.is_empty() {
                if let Ok(value) = chaos_rules.join(",").parse() {
                    response.headers_mut().insert("X-MockForge-Chaos-Rules", value);
                }
            }

            return response;
        } else {
            debug!("No unified state found for workspace {}", workspace_id);
        }
    }

    // Continue without unified state if not available
    next.run(req).await
}