mockserver-client 7.4.0

//! Breakpoint matcher registration, callback WebSocket client, and handler routing.
//!
//! This module provides the WebSocket callback stack for interactive breakpoints.
//! It connects to `/_mockserver_callback_websocket`, obtains a `clientId`, and
//! dispatches paused items to per-breakpoint-id handlers.

use base64::{engine::general_purpose::STANDARD as BASE64, Engine};
use serde::{Deserialize, Serialize};
use serde_json::Value;
use std::collections::HashMap;
use std::sync::{Arc, Mutex};
use tungstenite::{connect, Message};
use url::Url;

use crate::error::{Error, Result};
use crate::model::HttpRequest;

// ---------------------------------------------------------------------------
// Breakpoint phase constants
// ---------------------------------------------------------------------------

/// Breakpoint interception phase.
pub mod phase {
    pub const REQUEST: &str = "REQUEST";
    pub const RESPONSE: &str = "RESPONSE";
    pub const RESPONSE_STREAM: &str = "RESPONSE_STREAM";
    pub const INBOUND_STREAM: &str = "INBOUND_STREAM";
}

// ---------------------------------------------------------------------------
// Wire-contract DTOs
// ---------------------------------------------------------------------------

/// Registration request for a breakpoint matcher.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct BreakpointMatcherRegistration {
    pub http_request: HttpRequest,
    pub phases: Vec<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub client_id: Option<String>,
}

/// Response from registering a breakpoint matcher.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct BreakpointMatcherResponse {
    pub id: String,
    pub phases: Vec<String>,
}

/// An entry in the list of registered breakpoint matchers.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct BreakpointMatcherEntry {
    pub id: String,
    pub http_request: Value,
    pub phases: Vec<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub client_id: Option<String>,
}

/// Response from listing breakpoint matchers.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct BreakpointMatcherList {
    pub matchers: Vec<BreakpointMatcherEntry>,
}

/// A paused stream frame pushed by the server over the callback WebSocket.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct PausedStreamFrame {
    pub correlation_id: String,
    #[serde(default)]
    pub stream_id: String,
    #[serde(default)]
    pub sequence_number: i64,
    #[serde(default)]
    pub direction: String,
    #[serde(default)]
    pub phase: String,
    #[serde(default)]
    pub body: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub request_method: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub request_path: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub breakpoint_id: Option<String>,
}

impl PausedStreamFrame {
    /// Decode the Base64-encoded body to bytes.
    pub fn body_bytes(&self) -> std::result::Result<Vec<u8>, base64::DecodeError> {
        BASE64.decode(&self.body)
    }
}

/// Client-to-server reply for a stream frame decision.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct StreamFrameDecision {
    pub correlation_id: String,
    pub action: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub body: Option<String>,
}

impl StreamFrameDecision {
    /// Create a CONTINUE decision.
    pub fn continue_frame(correlation_id: impl Into<String>) -> Self {
        Self {
            correlation_id: correlation_id.into(),
            action: "CONTINUE".to_string(),
            body: None,
        }
    }

    /// Create a MODIFY decision with replacement bytes.
    pub fn modify(correlation_id: impl Into<String>, body: &[u8]) -> Self {
        Self {
            correlation_id: correlation_id.into(),
            action: "MODIFY".to_string(),
            body: Some(BASE64.encode(body)),
        }
    }

    /// Create a DROP decision.
    pub fn drop_frame(correlation_id: impl Into<String>) -> Self {
        Self {
            correlation_id: correlation_id.into(),
            action: "DROP".to_string(),
            body: None,
        }
    }

    /// Create an INJECT decision.
    pub fn inject(correlation_id: impl Into<String>, extra_body: &[u8]) -> Self {
        Self {
            correlation_id: correlation_id.into(),
            action: "INJECT".to_string(),
            body: Some(BASE64.encode(extra_body)),
        }
    }

    /// Create a CLOSE decision.
    pub fn close(correlation_id: impl Into<String>) -> Self {
        Self {
            correlation_id: correlation_id.into(),
            action: "CLOSE".to_string(),
            body: None,
        }
    }
}

// ---------------------------------------------------------------------------
// WebSocket message envelope
// ---------------------------------------------------------------------------

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct WsEnvelope {
    #[serde(rename = "type")]
    pub msg_type: String,
    pub value: String,
}

#[derive(Debug, Clone, Deserialize)]
#[serde(rename_all = "camelCase")]
struct WsClientIdDTO {
    client_id: String,
}

// ---------------------------------------------------------------------------
// Handler types
// ---------------------------------------------------------------------------

/// Handler for REQUEST phase breakpoints.
/// Receives the parsed request JSON. Return a Value that is either:
/// - An HttpRequest JSON object (continue/modify), or
/// - An HttpResponse JSON object with a "statusCode" field (abort).
///
/// Return None to auto-continue with the original request.
pub type BreakpointRequestHandler = Box<dyn Fn(Value) -> Option<Value> + Send + Sync>;

/// Handler for RESPONSE phase breakpoints.
/// Receives the request and response JSON objects.
/// Return a response Value, or None to auto-continue.
pub type BreakpointResponseHandler = Box<dyn Fn(Value, Value) -> Option<Value> + Send + Sync>;

/// Handler for stream frame breakpoints.
/// Receives the parsed PausedStreamFrame.
/// Return a StreamFrameDecision, or None to auto-continue.
pub type BreakpointStreamFrameHandler =
    Box<dyn Fn(&PausedStreamFrame) -> Option<StreamFrameDecision> + Send + Sync>;

/// Handler for object (closure) response callbacks.
///
/// Receives the matched request as JSON and returns the response JSON object.
/// Unlike breakpoint handlers this is mandatory (no auto-continue) — when a
/// request frame without a breakpoint id arrives, the registered closure
/// produces the response that MockServer returns to the caller.
///
/// Only `Send` is required (not `Sync`): the handler is stored behind a `Mutex`
/// and only ever invoked from the single WebSocket-read thread, so access is
/// already serialized. This keeps the public `mock_with_callback` bound to the
/// minimal `Fn(HttpRequest) -> HttpResponse + Send + 'static`.
pub type ObjectResponseHandler = Box<dyn Fn(Value) -> Value + Send>;

// ---------------------------------------------------------------------------
// BreakpointWebSocketClient
// ---------------------------------------------------------------------------

/// Internal WebSocket client for breakpoint callback resolution.
pub(crate) struct BreakpointWebSocketClient {
    pub(crate) client_id: String,
    socket: Arc<Mutex<tungstenite::WebSocket<tungstenite::stream::MaybeTlsStream<std::net::TcpStream>>>>,
    request_handlers: Arc<Mutex<HashMap<String, BreakpointRequestHandler>>>,
    response_handlers: Arc<Mutex<HashMap<String, BreakpointResponseHandler>>>,
    stream_frame_handlers: Arc<Mutex<HashMap<String, BreakpointStreamFrameHandler>>>,
    /// Single object/closure response callback for this WS client (request
    /// frames without an `X-MockServer-BreakpointId` header route here).
    object_response_handler: Arc<Mutex<Option<ObjectResponseHandler>>>,
    dead: Arc<std::sync::atomic::AtomicBool>,
    _read_thread: Option<std::thread::JoinHandle<()>>,
}

impl BreakpointWebSocketClient {
    /// Returns true if the read loop has exited (connection no longer usable).
    pub(crate) fn is_dead(&self) -> bool {
        self.dead.load(std::sync::atomic::Ordering::Relaxed)
    }

    /// Connect to the callback WebSocket and obtain a clientId.
    pub(crate) fn connect(base_url: &str) -> Result<Self> {
        let parsed = Url::parse(base_url)
            .map_err(|e| Error::InvalidRequest(format!("bad base URL: {e}")))?;

        let ws_scheme = if parsed.scheme() == "https" { "wss" } else { "ws" };
        let host = parsed.host_str().unwrap_or("localhost");
        let port = parsed.port().unwrap_or(if parsed.scheme() == "https" { 443 } else { 80 });
        let path = parsed.path().trim_end_matches('/');
        let ws_url = format!("{ws_scheme}://{host}:{port}{path}/_mockserver_callback_websocket");

        // NOTE: tungstenite's blocking `connect` does not expose a handshake timeout.
        // A connect timeout would require manually creating a TcpStream with
        // `TcpStream::connect_timeout`, then upgrading via `tungstenite::client`,
        // which adds complexity and fragility. Leaving this as a deferred follow-up;
        // the OS-level TCP connect timeout (typically ~60-120s) applies instead.
        let (mut socket, _response) = connect(&ws_url)
            .map_err(|e| Error::InvalidRequest(format!("WebSocket connect failed: {e}")))?;

        // Read registration reply
        let reg_msg = socket
            .read()
            .map_err(|e| Error::InvalidRequest(format!("WebSocket read registration: {e}")))?;

        let reg_text = match reg_msg {
            Message::Text(t) => t,
            _ => return Err(Error::InvalidRequest("Expected text registration message".into())),
        };

        let envelope: WsEnvelope = serde_json::from_str(&reg_text)?;
        let client_id = if envelope.msg_type
            == "org.mockserver.serialization.model.WebSocketClientIdDTO"
        {
            let dto: WsClientIdDTO = serde_json::from_str(&envelope.value)?;
            dto.client_id
        } else {
            return Err(Error::InvalidRequest(format!(
                "Unexpected registration type: {}",
                envelope.msg_type
            )));
        };

        let request_handlers: Arc<Mutex<HashMap<String, BreakpointRequestHandler>>> =
            Arc::new(Mutex::new(HashMap::new()));
        let response_handlers: Arc<Mutex<HashMap<String, BreakpointResponseHandler>>> =
            Arc::new(Mutex::new(HashMap::new()));
        let stream_frame_handlers: Arc<Mutex<HashMap<String, BreakpointStreamFrameHandler>>> =
            Arc::new(Mutex::new(HashMap::new()));
        let object_response_handler: Arc<Mutex<Option<ObjectResponseHandler>>> =
            Arc::new(Mutex::new(None));

        let socket = Arc::new(Mutex::new(socket));
        let dead = Arc::new(std::sync::atomic::AtomicBool::new(false));

        // Start read loop in a background thread
        let rh = Arc::clone(&request_handlers);
        let resh = Arc::clone(&response_handlers);
        let sfh = Arc::clone(&stream_frame_handlers);
        let orh = Arc::clone(&object_response_handler);
        let sock = Arc::clone(&socket);
        let dead_flag = Arc::clone(&dead);

        let read_thread = std::thread::spawn(move || {
            read_loop(sock, rh, resh, sfh, orh, dead_flag);
        });

        Ok(Self {
            client_id,
            socket,
            request_handlers,
            response_handlers,
            stream_frame_handlers,
            object_response_handler,
            dead,
            _read_thread: Some(read_thread),
        })
    }

    pub(crate) fn set_request_handler(&self, breakpoint_id: &str, handler: BreakpointRequestHandler) {
        self.request_handlers
            .lock()
            .unwrap()
            .insert(breakpoint_id.to_string(), handler);
    }

    pub(crate) fn set_response_handler(&self, breakpoint_id: &str, handler: BreakpointResponseHandler) {
        self.response_handlers
            .lock()
            .unwrap()
            .insert(breakpoint_id.to_string(), handler);
    }

    pub(crate) fn set_stream_frame_handler(
        &self,
        breakpoint_id: &str,
        handler: BreakpointStreamFrameHandler,
    ) {
        self.stream_frame_handlers
            .lock()
            .unwrap()
            .insert(breakpoint_id.to_string(), handler);
    }

    /// Register (or replace) the single object/closure response callback.
    pub(crate) fn set_object_response_handler(&self, handler: ObjectResponseHandler) {
        *self.object_response_handler.lock().unwrap() = Some(handler);
    }

    pub(crate) fn remove_handlers(&self, breakpoint_id: &str) {
        self.request_handlers.lock().unwrap().remove(breakpoint_id);
        self.response_handlers.lock().unwrap().remove(breakpoint_id);
        self.stream_frame_handlers
            .lock()
            .unwrap()
            .remove(breakpoint_id);
    }

    pub(crate) fn clear_handlers(&self) {
        self.request_handlers.lock().unwrap().clear();
        self.response_handlers.lock().unwrap().clear();
        self.stream_frame_handlers.lock().unwrap().clear();
    }

    pub(crate) fn close(&self) {
        if let Ok(mut sock) = self.socket.lock() {
            let _ = sock.close(None);
        }
    }
}

// ---------------------------------------------------------------------------
// Read loop (runs on a background thread)
// ---------------------------------------------------------------------------

/// Drop guard that sets the `dead` flag when the read loop exits for any reason.
struct ReadLoopDeadGuard(Arc<std::sync::atomic::AtomicBool>);
impl Drop for ReadLoopDeadGuard {
    fn drop(&mut self) {
        self.0.store(true, std::sync::atomic::Ordering::Relaxed);
    }
}

#[allow(clippy::too_many_arguments)]
fn read_loop(
    socket: Arc<Mutex<tungstenite::WebSocket<tungstenite::stream::MaybeTlsStream<std::net::TcpStream>>>>,
    request_handlers: Arc<Mutex<HashMap<String, BreakpointRequestHandler>>>,
    response_handlers: Arc<Mutex<HashMap<String, BreakpointResponseHandler>>>,
    stream_frame_handlers: Arc<Mutex<HashMap<String, BreakpointStreamFrameHandler>>>,
    object_response_handler: Arc<Mutex<Option<ObjectResponseHandler>>>,
    dead: Arc<std::sync::atomic::AtomicBool>,
) {
    let _guard = ReadLoopDeadGuard(dead);
    loop {
        let msg = {
            let mut sock = match socket.lock() {
                Ok(s) => s,
                Err(e) => {
                    eprintln!("mockserver: breakpoint ws read loop terminated: mutex poisoned: {e}");
                    return;
                }
            };
            match sock.read() {
                Ok(m) => m,
                Err(e) => {
                    eprintln!("mockserver: breakpoint ws read loop terminated: {e}");
                    return;
                }
            }
        };

        let text = match msg {
            Message::Text(t) => t,
            Message::Close(_) => {
                eprintln!("mockserver: breakpoint ws read loop terminated: received close frame");
                return;
            }
            _ => continue,
        };

        let envelope: WsEnvelope = match serde_json::from_str(&text) {
            Ok(e) => e,
            Err(_) => continue,
        };

        match envelope.msg_type.as_str() {
            "org.mockserver.model.HttpRequest" => {
                handle_request(
                    &envelope.value,
                    &request_handlers,
                    &object_response_handler,
                    &socket,
                );
            }
            "org.mockserver.model.HttpRequestAndHttpResponse" => {
                handle_response(
                    &envelope.value,
                    &response_handlers,
                    &socket,
                );
            }
            "org.mockserver.serialization.model.PausedStreamFrameDTO" => {
                handle_stream_frame(
                    &envelope.value,
                    &stream_frame_handlers,
                    &socket,
                );
            }
            "org.mockserver.serialization.model.WebSocketClientIdDTO" => {
                // Already handled during connect
            }
            _ => {}
        }
    }
}

fn handle_request(
    value_json: &str,
    handlers: &Arc<Mutex<HashMap<String, BreakpointRequestHandler>>>,
    object_response_handler: &Arc<Mutex<Option<ObjectResponseHandler>>>,
    socket: &Arc<Mutex<tungstenite::WebSocket<tungstenite::stream::MaybeTlsStream<std::net::TcpStream>>>>,
) {
    // A request frame WITHOUT a breakpoint id is an object-callback invocation;
    // route it to the registered closure. A frame WITH a breakpoint id is a
    // paused-breakpoint request and stays on the breakpoint path.
    if let Some((type_name, result)) = route_object_callback(value_json, object_response_handler) {
        send_envelope(socket, &type_name, &result);
        return;
    }
    if let Some((type_name, result)) = route_request(value_json, handlers) {
        send_envelope(socket, &type_name, &result);
    }
}

fn handle_response(
    value_json: &str,
    handlers: &Arc<Mutex<HashMap<String, BreakpointResponseHandler>>>,
    socket: &Arc<Mutex<tungstenite::WebSocket<tungstenite::stream::MaybeTlsStream<std::net::TcpStream>>>>,
) {
    if let Some((type_name, result)) = route_response(value_json, handlers) {
        send_envelope(socket, &type_name, &result);
    }
}

fn handle_stream_frame(
    value_json: &str,
    handlers: &Arc<Mutex<HashMap<String, BreakpointStreamFrameHandler>>>,
    socket: &Arc<Mutex<tungstenite::WebSocket<tungstenite::stream::MaybeTlsStream<std::net::TcpStream>>>>,
) {
    if let Some((type_name, decision)) = route_stream_frame(value_json, handlers) {
        send_envelope(socket, &type_name, &decision);
    }
}

// ---------------------------------------------------------------------------
// Pure routing functions (testable without a live WebSocket)
// ---------------------------------------------------------------------------

/// Route a REQUEST-phase message: dispatch to the per-breakpoint-id handler,
/// auto-continue on missing handler or panic. Returns (type_name, reply_value).
pub fn route_request(
    value_json: &str,
    handlers: &Arc<Mutex<HashMap<String, BreakpointRequestHandler>>>,
) -> Option<(String, Value)> {
    let request: Value = serde_json::from_str(value_json).ok()?;

    let correlation_id = extract_header(&request, "WebSocketCorrelationId");
    let breakpoint_id = extract_header(&request, "X-MockServer-BreakpointId");

    let mut result: Option<Value> = None;
    if let Some(bp_id) = &breakpoint_id {
        if let Ok(h) = handlers.lock() {
            if let Some(handler) = h.get(bp_id) {
                result = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
                    handler(request.clone())
                }))
                .unwrap_or(None);
            }
        }
    }

    // Auto-continue with original request if no result
    let mut result = result.unwrap_or_else(|| request.clone());

    // Determine type: statusCode present => HttpResponse (abort)
    let type_name = if result.get("statusCode").is_some() {
        "org.mockserver.model.HttpResponse"
    } else {
        "org.mockserver.model.HttpRequest"
    };

    if let Some(corr) = &correlation_id {
        set_header(&mut result, "WebSocketCorrelationId", corr);
    }

    Some((type_name.to_string(), result))
}

/// Route an object-callback request frame to the registered closure.
///
/// Object-callback frames are `org.mockserver.model.HttpRequest` messages that
/// carry a `WebSocketCorrelationId` but **no** `X-MockServer-BreakpointId`
/// header (that header marks a paused-breakpoint request, handled elsewhere).
///
/// Returns `None` when the frame is a breakpoint request (has a breakpoint id)
/// or when no object handler is registered — in both cases the caller falls
/// through to the breakpoint request path. On a match, invokes the closure
/// (auto-continuing with a passthrough response on panic), echoes the
/// `WebSocketCorrelationId` onto the reply, and returns
/// `("org.mockserver.model.HttpResponse", response_json)`.
pub fn route_object_callback(
    value_json: &str,
    handler: &Arc<Mutex<Option<ObjectResponseHandler>>>,
) -> Option<(String, Value)> {
    let request: Value = serde_json::from_str(value_json).ok()?;

    // Breakpoint requests carry an explicit breakpoint id — never object-route them.
    if extract_header(&request, "X-MockServer-BreakpointId").is_some() {
        return None;
    }

    let correlation_id = extract_header(&request, "WebSocketCorrelationId");

    // No handler registered: let the breakpoint path handle it (it auto-continues).
    let guard = handler.lock().ok()?;
    let handler = guard.as_ref()?;

    let mut result = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
        handler(request.clone())
    }))
    .unwrap_or_else(|_| serde_json::json!({}));

    if let Some(corr) = &correlation_id {
        set_header(&mut result, "WebSocketCorrelationId", corr);
    }

    Some(("org.mockserver.model.HttpResponse".to_string(), result))
}

/// Route a RESPONSE-phase message: dispatch to the per-breakpoint-id handler,
/// auto-continue on missing handler or panic. Returns (type_name, reply_value).
pub fn route_response(
    value_json: &str,
    handlers: &Arc<Mutex<HashMap<String, BreakpointResponseHandler>>>,
) -> Option<(String, Value)> {
    let req_and_resp: Value = serde_json::from_str(value_json).ok()?;

    let http_request = req_and_resp.get("httpRequest").cloned().unwrap_or(Value::Null);
    let http_response = req_and_resp.get("httpResponse").cloned().unwrap_or(Value::Null);

    let correlation_id = extract_header(&http_request, "WebSocketCorrelationId");
    let breakpoint_id = extract_header(&http_request, "X-MockServer-BreakpointId");

    let mut result: Option<Value> = None;
    if let Some(bp_id) = &breakpoint_id {
        if let Ok(h) = handlers.lock() {
            if let Some(handler) = h.get(bp_id) {
                result = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
                    handler(http_request.clone(), http_response.clone())
                }))
                .unwrap_or(None);
            }
        }
    }

    let mut result = result.unwrap_or(http_response);

    if let Some(corr) = &correlation_id {
        set_header(&mut result, "WebSocketCorrelationId", corr);
    }

    Some(("org.mockserver.model.HttpResponse".to_string(), result))
}

/// Route a stream-frame message: dispatch to the per-breakpoint-id handler,
/// auto-continue on missing handler or panic. Returns (type_name, decision).
pub fn route_stream_frame(
    value_json: &str,
    handlers: &Arc<Mutex<HashMap<String, BreakpointStreamFrameHandler>>>,
) -> Option<(String, StreamFrameDecision)> {
    let frame: PausedStreamFrame = serde_json::from_str(value_json).ok()?;

    let mut decision: Option<StreamFrameDecision> = None;
    if let Some(bp_id) = &frame.breakpoint_id {
        if let Ok(h) = handlers.lock() {
            if let Some(handler) = h.get(bp_id) {
                decision = std::panic::catch_unwind(std::panic::AssertUnwindSafe(|| {
                    handler(&frame)
                }))
                .unwrap_or(None);
            }
        }
    }

    let mut decision = decision.unwrap_or_else(|| StreamFrameDecision::continue_frame(&frame.correlation_id));
    decision.correlation_id = frame.correlation_id;

    Some((
        "org.mockserver.serialization.model.StreamFrameDecisionDTO".to_string(),
        decision,
    ))
}

fn send_envelope<T: Serialize>(
    socket: &Arc<Mutex<tungstenite::WebSocket<tungstenite::stream::MaybeTlsStream<std::net::TcpStream>>>>,
    type_name: &str,
    value: &T,
) {
    let value_json = match serde_json::to_string(value) {
        Ok(j) => j,
        Err(_) => return,
    };
    let envelope = WsEnvelope {
        msg_type: type_name.to_string(),
        value: value_json,
    };
    let env_json = match serde_json::to_string(&envelope) {
        Ok(j) => j,
        Err(_) => return,
    };
    if let Ok(mut sock) = socket.lock() {
        let _ = sock.send(Message::Text(env_json));
    }
}

// ---------------------------------------------------------------------------
// Header helpers
// ---------------------------------------------------------------------------

/// Extract the first value of a header from a JSON request object.
pub fn extract_header(obj: &Value, name: &str) -> Option<String> {
    let headers = obj.get("headers")?.as_object()?;
    let name_lower = name.to_lowercase();
    for (k, v) in headers {
        if k.to_lowercase() == name_lower {
            if let Some(arr) = v.as_array() {
                return arr.first()?.as_str().map(|s| s.to_string());
            }
            return v.as_str().map(|s| s.to_string());
        }
    }
    None
}

/// Set a header value on a JSON request/response object.
pub fn set_header(obj: &mut Value, name: &str, value: &str) {
    if let Some(obj_map) = obj.as_object_mut() {
        let headers = obj_map
            .entry("headers")
            .or_insert_with(|| Value::Object(serde_json::Map::new()));
        if let Some(h) = headers.as_object_mut() {
            h.insert(
                name.to_string(),
                serde_json::json!([value]),
            );
        }
    }
}