mockserver_client/

model.rs

//! Domain model types for the MockServer control-plane API.
//!
//! All types implement `Serialize`/`Deserialize` and use builder methods that
//! take `self` and return `Self`, enabling fluent construction.

use base64::{engine::general_purpose::STANDARD as BASE64, Engine};
use serde::{Deserialize, Serialize};
use std::collections::HashMap;

/// Free-form map used as a forward-compatibility safety net on the wire types
/// that model MockServer actions. Any JSON field the typed model does not yet
/// name is captured here (via `#[serde(flatten)]`) so it survives a
/// deserialize-then-serialize round-trip instead of being silently dropped.
///
/// An empty map contributes no keys when serialized, so a `flatten`ed `Extra`
/// is invisible on the wire unless the server actually sent unknown fields.
pub type Extra = serde_json::Map<String, serde_json::Value>;

/// Deserialize a MockServer `oneOf: [ <T>, [ <T> ] ]` field (a single object or
/// an array of objects) into a `Vec<T>`. Used by `beforeActions`, `afterActions`
/// and `capture`, which the server accepts in either shape.
fn one_or_many<'de, D, T>(deserializer: D) -> std::result::Result<Option<Vec<T>>, D::Error>
where
    D: serde::Deserializer<'de>,
    T: Deserialize<'de>,
{
    #[derive(Deserialize)]
    #[serde(untagged)]
    enum OneOrMany<T> {
        One(T),
        Many(Vec<T>),
    }
    let opt = Option::<OneOrMany<T>>::deserialize(deserializer)?;
    Ok(opt.map(|v| match v {
        OneOrMany::One(t) => vec![t],
        OneOrMany::Many(v) => v,
    }))
}

// ---------------------------------------------------------------------------
// ParameterValues
// ---------------------------------------------------------------------------

/// The value of a single key in a MockServer keyToMultiValue matcher (path
/// parameters, and in general query-string parameters / headers).
///
/// MockServer accepts two wire encodings for a key's value:
///   * the **plain** form — a list of exact-or-regex strings (`["42", "^\\d+$"]`),
///     modelled by [`Values`](Self::Values); and
///   * the **schema-matcher** form — a list of matcher objects
///     (`[{ "schema": { … } }]`, `[{ "not": true, "value": "x" }]`), or the
///     `{ "parameterStyle": …, "values": [ … ] }` object form — captured verbatim
///     by [`Matcher`](Self::Matcher) so no field is dropped on a round-trip.
///
/// The enum is `#[serde(untagged)]`: the plain string-array form deserialises to
/// [`Values`](Self::Values); anything else (a matcher-object array, an object, or
/// even a bare string) falls through to [`Matcher`](Self::Matcher).
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(untagged)]
pub enum ParameterValues {
    /// Plain multi-value form: a list of exact-or-regex string values.
    Values(Vec<String>),
    /// Schema / nottable / optional / parameter-style matcher form, kept verbatim.
    Matcher(serde_json::Value),
}

impl ParameterValues {
    /// Borrow the plain string values, if this is the [`Values`](Self::Values) form.
    pub fn as_values(&self) -> Option<&[String]> {
        match self {
            ParameterValues::Values(v) => Some(v),
            ParameterValues::Matcher(_) => None,
        }
    }
}

impl From<Vec<String>> for ParameterValues {
    fn from(values: Vec<String>) -> Self {
        ParameterValues::Values(values)
    }
}

// ---------------------------------------------------------------------------
// HttpRequest
// ---------------------------------------------------------------------------

/// Matcher for an HTTP request. Uses builder methods for fluent construction.
///
/// # Example
/// ```
/// use mockserver_client::HttpRequest;
///
/// let request = HttpRequest::new()
///     .method("POST")
///     .path("/api/users")
///     .header("Content-Type", "application/json")
///     .query_param("page", "1")
///     .body("{}");
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpRequest {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub method: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub path: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub query_string_parameters: Option<HashMap<String, Vec<String>>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub headers: Option<HashMap<String, Vec<String>>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub body: Option<Body>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub jwt: Option<Jwt>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub socket_address: Option<SocketAddress>,

    /// Negate the whole request matcher (`"not": true`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub not: Option<bool>,

    /// Match only requests received over TLS (`"secure"`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub secure: Option<bool>,

    /// Match only keep-alive requests (`"keepAlive"`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub keep_alive: Option<bool>,

    /// Match the request protocol (e.g. `"HTTP_1_1"`, `"HTTP_2"`, `"HTTP_3"`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub protocol: Option<String>,

    /// Path parameters (`/users/{id}` style), multiple values per key.
    ///
    /// Each key's value is a [`ParameterValues`], accepting both the plain
    /// string-list form and the schema/nottable matcher form.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub path_parameters: Option<HashMap<String, ParameterValues>>,

    /// Cookies to match (single value per name).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub cookies: Option<HashMap<String, String>>,

    /// Forward-compatibility catch-all for request fields the typed model does
    /// not yet name (e.g. `clientCertificate`, `localAddress`, `remoteAddress`).
    #[serde(flatten, default)]
    pub extra: Extra,
}

impl HttpRequest {
    /// Create a new empty request matcher.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the downstream socket address to connect to.
    ///
    /// Used by load-scenario steps (and forwarded/proxied requests) to direct
    /// the rendered request at a specific host/port/scheme rather than relying
    /// on the request's `Host` header.
    pub fn socket_address(mut self, socket_address: SocketAddress) -> Self {
        self.socket_address = Some(socket_address);
        self
    }

    /// Set the HTTP method to match.
    pub fn method(mut self, method: impl Into<String>) -> Self {
        self.method = Some(method.into());
        self
    }

    /// Set the path to match.
    pub fn path(mut self, path: impl Into<String>) -> Self {
        self.path = Some(path.into());
        self
    }

    /// Add a query string parameter (multiple values per key supported).
    pub fn query_param(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        let params = self
            .query_string_parameters
            .get_or_insert_with(HashMap::new);
        params.entry(key.into()).or_default().push(value.into());
        self
    }

    /// Add a header (multiple values per key supported).
    pub fn header(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        let headers = self.headers.get_or_insert_with(HashMap::new);
        headers.entry(key.into()).or_default().push(value.into());
        self
    }

    /// Set a plain string body matcher.
    pub fn body(mut self, body: impl Into<String>) -> Self {
        self.body = Some(Body::Plain(body.into()));
        self
    }

    /// Set a typed JSON body matcher.
    pub fn json_body(mut self, json: serde_json::Value) -> Self {
        self.body = Some(Body::Typed {
            body_type: "JSON".to_string(),
            json: json.to_string(),
        });
        self
    }

    /// Set a file body (type "FILE") with optional content type and template type.
    ///
    /// Use [`Body::file`] for richer construction if you need content type or
    /// template type set.
    pub fn file_body(mut self, file_path: impl Into<String>) -> Self {
        self.body = Some(Body::File {
            file_path: file_path.into(),
            content_type: None,
            template_type: None,
        });
        self
    }

    /// Set a pre-built [`Body`] value (use with [`Body::file`] for FILE bodies).
    pub fn body_value(mut self, body: Body) -> Self {
        self.body = Some(body);
        self
    }

    /// Set a JWT request matcher.
    ///
    /// Serialised under the `"jwt"` key alongside `method`/`path`/`headers`.
    ///
    /// # Example
    /// ```
    /// use mockserver_client::{HttpRequest, Jwt};
    ///
    /// let request = HttpRequest::new()
    ///     .method("GET")
    ///     .path("/secure")
    ///     .jwt(
    ///         Jwt::new()
    ///             .claim("sub", "user-123")
    ///             .claim("role", "!admin")
    ///             .issuer("https://issuer.example.com")
    ///             .algorithm("RS256"),
    ///     );
    /// ```
    pub fn jwt(mut self, jwt: Jwt) -> Self {
        self.jwt = Some(jwt);
        self
    }

    /// Negate the whole request matcher.
    pub fn not(mut self, not: bool) -> Self {
        self.not = Some(not);
        self
    }

    /// Match only requests received over TLS.
    pub fn secure(mut self, secure: bool) -> Self {
        self.secure = Some(secure);
        self
    }

    /// Match only keep-alive requests.
    pub fn keep_alive(mut self, keep_alive: bool) -> Self {
        self.keep_alive = Some(keep_alive);
        self
    }

    /// Match the request protocol (e.g. `"HTTP_1_1"`, `"HTTP_2"`).
    pub fn protocol(mut self, protocol: impl Into<String>) -> Self {
        self.protocol = Some(protocol.into());
        self
    }

    /// Add a plain path parameter value (multiple values per key supported).
    ///
    /// For schema/nottable matcher forms, set [`path_parameters`](Self::path_parameters)
    /// directly with a [`ParameterValues::Matcher`].
    pub fn path_param(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        let params = self.path_parameters.get_or_insert_with(HashMap::new);
        match params
            .entry(key.into())
            .or_insert_with(|| ParameterValues::Values(Vec::new()))
        {
            ParameterValues::Values(v) => v.push(value.into()),
            ParameterValues::Matcher(_) => {
                // Existing entry is a verbatim matcher form; leave it untouched
                // rather than silently coercing it to a plain value.
            }
        }
        self
    }

    /// Add a cookie to match (single value per name).
    pub fn cookie(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
        let cookies = self.cookies.get_or_insert_with(HashMap::new);
        cookies.insert(name.into(), value.into());
        self
    }
}

// ---------------------------------------------------------------------------
// Body
// ---------------------------------------------------------------------------

/// Request/response body — either a plain string, a typed object, or a file reference.
#[derive(Debug, Clone, PartialEq)]
pub enum Body {
    /// A plain string body.
    Plain(String),
    /// A typed body (e.g., JSON).
    Typed { body_type: String, json: String },
    /// A file body (`type: "FILE"`), with optional template evaluation.
    File {
        file_path: String,
        content_type: Option<String>,
        template_type: Option<String>,
    },
    /// An `ALL_OF` composite body matcher — every nested body matcher must match.
    ///
    /// Serialises to `{ "type": "ALL_OF", "bodyAllOf": [ <body>, ... ] }`,
    /// recursing through the normal [`Body`] serialisation for each sub-body.
    AllOf(Vec<Body>),
    /// A single-value typed body matcher whose value lives under a named key
    /// (e.g. `JSON_PATH` → `jsonPath`, `REGEX` → `regex`, `XPATH` → `xpath`).
    ///
    /// Serialises to `{ "type": <body_type>, <value_key>: <value> }`. Use the
    /// [`Body::json_path`] / [`Body::regex`] constructors for the common cases.
    Matcher {
        body_type: String,
        value_key: String,
        value: String,
    },
    /// Any typed body object captured verbatim as a JSON object — the
    /// forward-compatible catch-all for body matcher/value types that do not
    /// have a dedicated variant (`STRING`/`subString`, `XML`, `XML_SCHEMA`,
    /// `JSON_SCHEMA`, `PARAMETERS`, `BINARY`, `GRAPHQL`, `MULTIPART`, `WASM`,
    /// `JSON_RPC`, `FUZZY`, …). Serialises the map back exactly, so every body
    /// shape round-trips without silent field loss.
    Object(serde_json::Map<String, serde_json::Value>),
}

impl Body {
    /// Create a FILE body referencing a path on the server filesystem.
    ///
    /// # Example
    /// ```
    /// use mockserver_client::Body;
    ///
    /// let body = Body::file("/data/response.json")
    ///     .with_content_type("application/json")
    ///     .with_template_type("VELOCITY");
    /// ```
    pub fn file(file_path: impl Into<String>) -> Self {
        Body::File {
            file_path: file_path.into(),
            content_type: None,
            template_type: None,
        }
    }

    /// Set the content type on a FILE body. No-op on other variants.
    pub fn with_content_type(mut self, content_type: impl Into<String>) -> Self {
        if let Body::File {
            content_type: ref mut ct,
            ..
        } = self
        {
            *ct = Some(content_type.into());
        }
        self
    }

    /// Set the template type (e.g., "VELOCITY", "MUSTACHE") on a FILE body.
    /// No-op on other variants.
    pub fn with_template_type(mut self, template_type: impl Into<String>) -> Self {
        if let Body::File {
            template_type: ref mut tt,
            ..
        } = self
        {
            *tt = Some(template_type.into());
        }
        self
    }

    /// Create an `ALL_OF` composite body matcher — every nested body matcher
    /// must match for the request body to match.
    ///
    /// # Example
    /// ```
    /// use mockserver_client::Body;
    ///
    /// let body = Body::all_of(vec![
    ///     Body::json_path("$.name"),
    ///     Body::regex(".*active.*"),
    /// ]);
    /// ```
    pub fn all_of(bodies: Vec<Body>) -> Self {
        Body::AllOf(bodies)
    }

    /// Create a `JSON_PATH` body matcher.
    ///
    /// Serialises to `{ "type": "JSON_PATH", "jsonPath": <expression> }`.
    pub fn json_path(expression: impl Into<String>) -> Self {
        Body::Matcher {
            body_type: "JSON_PATH".to_string(),
            value_key: "jsonPath".to_string(),
            value: expression.into(),
        }
    }

    /// Create a `REGEX` body matcher.
    ///
    /// Serialises to `{ "type": "REGEX", "regex": <pattern> }`.
    pub fn regex(pattern: impl Into<String>) -> Self {
        Body::Matcher {
            body_type: "REGEX".to_string(),
            value_key: "regex".to_string(),
            value: pattern.into(),
        }
    }

    /// Create an `XPATH` body matcher (`{ "type": "XPATH", "xpath": <expr> }`).
    pub fn xpath(expression: impl Into<String>) -> Self {
        Body::Matcher {
            body_type: "XPATH".to_string(),
            value_key: "xpath".to_string(),
            value: expression.into(),
        }
    }

    /// Create a `STRING` body matcher. When `sub_string` is true the value need
    /// only be a substring of the request body.
    ///
    /// Serialises to `{ "type": "STRING", "string": <value>, "subString": <b> }`.
    pub fn string(value: impl Into<String>, sub_string: bool) -> Self {
        let mut map = serde_json::Map::new();
        map.insert("type".into(), serde_json::Value::from("STRING"));
        map.insert("string".into(), serde_json::Value::from(value.into()));
        map.insert("subString".into(), serde_json::Value::from(sub_string));
        Body::Object(map)
    }

    /// Create an `XML` body matcher (`{ "type": "XML", "xml": <value> }`).
    pub fn xml(value: impl Into<String>) -> Self {
        Self::single_object("XML", "xml", value.into())
    }

    /// Create an `XML_SCHEMA` body matcher
    /// (`{ "type": "XML_SCHEMA", "xmlSchema": <schema> }`).
    pub fn xml_schema(schema: impl Into<String>) -> Self {
        Self::single_object("XML_SCHEMA", "xmlSchema", schema.into())
    }

    /// Create a `JSON_SCHEMA` body matcher
    /// (`{ "type": "JSON_SCHEMA", "jsonSchema": <schema> }`).
    pub fn json_schema(schema: impl Into<String>) -> Self {
        Self::single_object("JSON_SCHEMA", "jsonSchema", schema.into())
    }

    /// Create a `PARAMETERS` (form/body parameter) matcher
    /// (`{ "type": "PARAMETERS", "parameters": { name: [values] } }`).
    pub fn parameters(parameters: HashMap<String, Vec<String>>) -> Self {
        let mut map = serde_json::Map::new();
        map.insert("type".into(), serde_json::Value::from("PARAMETERS"));
        map.insert(
            "parameters".into(),
            serde_json::to_value(parameters).unwrap_or(serde_json::Value::Null),
        );
        Body::Object(map)
    }

    /// Create a `BINARY` body matcher/value from raw bytes (base64-encoded on
    /// the wire as `base64Bytes`), with an optional content type.
    pub fn binary(data: impl AsRef<[u8]>, content_type: Option<String>) -> Self {
        let mut map = serde_json::Map::new();
        map.insert("type".into(), serde_json::Value::from("BINARY"));
        map.insert(
            "base64Bytes".into(),
            serde_json::Value::from(BASE64.encode(data.as_ref())),
        );
        if let Some(ct) = content_type {
            map.insert("contentType".into(), serde_json::Value::from(ct));
        }
        Body::Object(map)
    }

    /// Create a `GRAPHQL` body matcher (`{ "type": "GRAPHQL", "query": <query> }`).
    pub fn graphql(query: impl Into<String>) -> Self {
        let mut map = serde_json::Map::new();
        map.insert("type".into(), serde_json::Value::from("GRAPHQL"));
        map.insert("query".into(), serde_json::Value::from(query.into()));
        Body::Object(map)
    }

    /// Create a `WASM` custom-rule body matcher from a pre-built JSON object.
    ///
    /// Captured verbatim, so any current or future WASM matcher fields survive
    /// a round-trip.
    pub fn wasm(object: serde_json::Map<String, serde_json::Value>) -> Self {
        Body::Object(object)
    }

    /// Build a `Body::Object` from a raw JSON object — the escape hatch for any
    /// body type not covered by a dedicated constructor.
    pub fn object(object: serde_json::Map<String, serde_json::Value>) -> Self {
        Body::Object(object)
    }

    fn single_object(body_type: &str, key: &str, value: String) -> Self {
        let mut map = serde_json::Map::new();
        map.insert("type".into(), serde_json::Value::from(body_type));
        map.insert(key.into(), serde_json::Value::from(value));
        Body::Object(map)
    }
}

impl Serialize for Body {
    fn serialize<S>(&self, serializer: S) -> std::result::Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        match self {
            Body::Plain(s) => serializer.serialize_str(s),
            Body::Typed { body_type, json } => {
                use serde::ser::SerializeMap;
                let mut map = serializer.serialize_map(Some(2))?;
                map.serialize_entry("type", body_type)?;
                map.serialize_entry("json", json)?;
                map.end()
            }
            Body::File {
                file_path,
                content_type,
                template_type,
            } => {
                use serde::ser::SerializeMap;
                let count = 2
                    + content_type.as_ref().map_or(0, |_| 1)
                    + template_type.as_ref().map_or(0, |_| 1);
                let mut map = serializer.serialize_map(Some(count))?;
                map.serialize_entry("type", "FILE")?;
                map.serialize_entry("filePath", file_path)?;
                if let Some(ct) = content_type {
                    map.serialize_entry("contentType", ct)?;
                }
                if let Some(tt) = template_type {
                    map.serialize_entry("templateType", tt)?;
                }
                map.end()
            }
            Body::AllOf(bodies) => {
                use serde::ser::SerializeMap;
                let mut map = serializer.serialize_map(Some(2))?;
                map.serialize_entry("type", "ALL_OF")?;
                map.serialize_entry("bodyAllOf", bodies)?;
                map.end()
            }
            Body::Matcher {
                body_type,
                value_key,
                value,
            } => {
                use serde::ser::SerializeMap;
                let mut map = serializer.serialize_map(Some(2))?;
                map.serialize_entry("type", body_type)?;
                map.serialize_entry(value_key.as_str(), value)?;
                map.end()
            }
            Body::Object(object) => object.serialize(serializer),
        }
    }
}

/// Map a single-value matcher `body_type` to its wire value-key and extract the
/// string value from the deserialised object (e.g. `JSON_PATH` → `jsonPath`).
fn matcher_key_value(
    body_type: &str,
    map: &serde_json::Map<String, serde_json::Value>,
) -> Option<(String, String)> {
    let key = match body_type {
        "JSON_PATH" => "jsonPath",
        "REGEX" => "regex",
        "XPATH" => "xpath",
        _ => return None,
    };
    map.get(key)
        .and_then(|v| v.as_str())
        .map(|s| (key.to_string(), s.to_string()))
}

impl<'de> Deserialize<'de> for Body {
    fn deserialize<D>(deserializer: D) -> std::result::Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        use serde_json::Value;
        let v = Value::deserialize(deserializer)?;
        match v {
            Value::String(s) => Ok(Body::Plain(s)),
            Value::Object(map) => {
                let body_type = map
                    .get("type")
                    .and_then(|v| v.as_str())
                    .unwrap_or("JSON")
                    .to_string();
                if body_type == "FILE" {
                    let file_path = map
                        .get("filePath")
                        .and_then(|v| v.as_str())
                        .unwrap_or("")
                        .to_string();
                    let content_type = map
                        .get("contentType")
                        .and_then(|v| v.as_str())
                        .map(|s| s.to_string());
                    let template_type = map
                        .get("templateType")
                        .and_then(|v| v.as_str())
                        .map(|s| s.to_string());
                    Ok(Body::File {
                        file_path,
                        content_type,
                        template_type,
                    })
                } else if body_type == "ALL_OF" {
                    let bodies = map
                        .get("bodyAllOf")
                        .and_then(|v| v.as_array())
                        .map(|arr| {
                            arr.iter()
                                .cloned()
                                .map(serde_json::from_value)
                                .collect::<std::result::Result<Vec<Body>, _>>()
                        })
                        .transpose()
                        .map_err(serde::de::Error::custom)?
                        .unwrap_or_default();
                    Ok(Body::AllOf(bodies))
                } else if map.len() == 2 && matcher_key_value(&body_type, &map).is_some() {
                    // Only the bare `{ "type": <T>, <valueKey>: <v> }` shape uses
                    // the dedicated Matcher variant. A matcher carrying extra keys
                    // (not, optional, matchType, contentType, …) falls through to
                    // the verbatim Object variant so those fields are not dropped.
                    let (value_key, value) = matcher_key_value(&body_type, &map)
                        .expect("matcher_key_value checked above");
                    Ok(Body::Matcher {
                        body_type,
                        value_key,
                        value,
                    })
                } else if body_type == "JSON"
                    && map.len() == 2
                    && map.get("json").is_some_and(|v| v.is_string())
                {
                    // Preserve the dedicated typed-JSON representation, but only
                    // for the bare `{ "type": "JSON", "json": ... }` shape — a
                    // JSON body carrying extra keys (matchType, contentType,
                    // not, optional, …) falls through to the verbatim Object
                    // variant so those fields are not dropped.
                    let json = map
                        .get("json")
                        .and_then(|v| v.as_str())
                        .unwrap_or("")
                        .to_string();
                    Ok(Body::Typed { body_type, json })
                } else {
                    // Any other typed body object (STRING, XML, XML_SCHEMA,
                    // JSON_SCHEMA, PARAMETERS, BINARY, GRAPHQL, MULTIPART, WASM,
                    // JSON_RPC, FUZZY, an inline JSON object literal, …) is kept
                    // verbatim so no field is silently dropped.
                    Ok(Body::Object(map))
                }
            }
            _ => Ok(Body::Plain(v.to_string())),
        }
    }
}

// ---------------------------------------------------------------------------
// Jwt
// ---------------------------------------------------------------------------

/// JWT request matcher — matches a JSON Web Token carried on the request.
///
/// Serialised under the request's `"jwt"` key. Each entry in [`claims`](Self::claims)
/// is a claim name mapped to an exact-or-regex string; a leading `!` negates the
/// match. The optional [`issuer`](Self::issuer), [`audience`](Self::audience),
/// [`algorithm`](Self::algorithm), [`header`](Self::header) and
/// [`scheme`](Self::scheme) fields are omitted from the wire form when unset.
///
/// # Example
/// ```
/// use mockserver_client::Jwt;
///
/// let jwt = Jwt::new()
///     .claim("sub", "user-123")
///     .claim("role", "!admin")
///     .claim("email", "^.+@example.com$")
///     .issuer("https://issuer.example.com")
///     .audience("my-api")
///     .algorithm("RS256")
///     .header("authorization")
///     .scheme("Bearer");
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct Jwt {
    /// Claim name → exact-or-regex value (leading `!` negates).
    pub claims: HashMap<String, String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub issuer: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub audience: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub algorithm: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub header: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub scheme: Option<String>,
}

impl Jwt {
    /// Create a new empty JWT matcher (no claims, no constraints).
    pub fn new() -> Self {
        Self::default()
    }

    /// Add a claim constraint. The value is an exact-or-regex string; a leading
    /// `!` negates the match.
    pub fn claim(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
        self.claims.insert(name.into(), value.into());
        self
    }

    /// Replace the full claims map.
    pub fn claims(mut self, claims: HashMap<String, String>) -> Self {
        self.claims = claims;
        self
    }

    /// Require the `iss` (issuer) claim to equal the given value.
    pub fn issuer(mut self, issuer: impl Into<String>) -> Self {
        self.issuer = Some(issuer.into());
        self
    }

    /// Require the `aud` (audience) claim to equal the given value.
    pub fn audience(mut self, audience: impl Into<String>) -> Self {
        self.audience = Some(audience.into());
        self
    }

    /// Require the token to be signed with the given algorithm (e.g. "RS256").
    pub fn algorithm(mut self, algorithm: impl Into<String>) -> Self {
        self.algorithm = Some(algorithm.into());
        self
    }

    /// Set the request header the token is carried in (default "authorization").
    pub fn header(mut self, header: impl Into<String>) -> Self {
        self.header = Some(header.into());
        self
    }

    /// Set the auth scheme prefix stripped from the header value (e.g. "Bearer").
    pub fn scheme(mut self, scheme: impl Into<String>) -> Self {
        self.scheme = Some(scheme.into());
        self
    }
}

// ---------------------------------------------------------------------------
// HttpResponse
// ---------------------------------------------------------------------------

/// Builder for an HTTP response action.
///
/// # Example
/// ```
/// use mockserver_client::HttpResponse;
///
/// let response = HttpResponse::new()
///     .status_code(201)
///     .header("Location", "/api/users/42")
///     .body("{\"id\": 42}");
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub status_code: Option<u16>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub headers: Option<HashMap<String, Vec<String>>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub body: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    /// Response cookies (single value per name; emitted as `Set-Cookie`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub cookies: Option<HashMap<String, String>>,

    /// HTTP reason phrase (e.g. `"Not Found"`); overrides the default for the
    /// status code.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub reason_phrase: Option<String>,

    /// A status-code range to respond with a random status from (e.g. `"2xx"`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub status_code_range: Option<String>,

    /// Response trailers (HTTP/2 trailing headers), multiple values per key.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub trailers: Option<HashMap<String, Vec<String>>>,

    /// Generate the response body from an inline/JSON-schema string.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub generate_from_schema: Option<String>,

    /// Connection-level options (chunking, content-length, socket close, …).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub connection_options: Option<ConnectionOptions>,

    /// Fail the first N requests then recover (circuit-breaker style).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub recover_after: Option<RecoverAfter>,

    /// Mark this response as the primary action of a composite expectation.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,

    /// Forward-compatibility catch-all for response fields the typed model does
    /// not yet name.
    #[serde(flatten, default)]
    pub extra: Extra,
}

impl HttpResponse {
    /// Create a new empty response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the HTTP status code.
    pub fn status_code(mut self, code: u16) -> Self {
        self.status_code = Some(code);
        self
    }

    /// Add a response header.
    pub fn header(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        let headers = self.headers.get_or_insert_with(HashMap::new);
        headers.entry(key.into()).or_default().push(value.into());
        self
    }

    /// Set the response body as a string.
    pub fn body(mut self, body: impl Into<String>) -> Self {
        self.body = Some(body.into());
        self
    }

    /// Set a response delay.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }

    /// Add a response cookie (single value per name).
    pub fn cookie(mut self, name: impl Into<String>, value: impl Into<String>) -> Self {
        let cookies = self.cookies.get_or_insert_with(HashMap::new);
        cookies.insert(name.into(), value.into());
        self
    }

    /// Set the HTTP reason phrase (e.g. `"Not Found"`).
    pub fn reason_phrase(mut self, reason_phrase: impl Into<String>) -> Self {
        self.reason_phrase = Some(reason_phrase.into());
        self
    }

    /// Set a status-code range to respond with a random status from (e.g. `"2xx"`).
    pub fn status_code_range(mut self, range: impl Into<String>) -> Self {
        self.status_code_range = Some(range.into());
        self
    }

    /// Add a response trailer (HTTP/2 trailing header), multiple values per key.
    pub fn trailer(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        let trailers = self.trailers.get_or_insert_with(HashMap::new);
        trailers.entry(key.into()).or_default().push(value.into());
        self
    }

    /// Generate the response body from a schema string.
    pub fn generate_from_schema(mut self, schema: impl Into<String>) -> Self {
        self.generate_from_schema = Some(schema.into());
        self
    }

    /// Set connection-level options.
    pub fn connection_options(mut self, options: ConnectionOptions) -> Self {
        self.connection_options = Some(options);
        self
    }

    /// Set a recover-after (fail-first-N) policy.
    pub fn recover_after(mut self, recover_after: RecoverAfter) -> Self {
        self.recover_after = Some(recover_after);
        self
    }

    /// Mark this response as the primary action of a composite expectation.
    pub fn primary(mut self, primary: bool) -> Self {
        self.primary = Some(primary);
        self
    }
}

// ---------------------------------------------------------------------------
// HttpTemplate (response or forward)
// ---------------------------------------------------------------------------

/// Template action — evaluate a response or forward template (Velocity, Mustache, etc.).
///
/// Used as `httpResponseTemplate` or `httpForwardTemplate` in an expectation.
///
/// # Example
/// ```
/// use mockserver_client::HttpTemplate;
///
/// let tmpl = HttpTemplate::new("VELOCITY", "{ \"statusCode\": 200 }")
///     .template_file("/path/to/template.vm");
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpTemplate {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub template_type: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub template: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub template_file: Option<String>,
}

impl HttpTemplate {
    /// Create a template action with the given type and inline template body.
    pub fn new(template_type: impl Into<String>, template: impl Into<String>) -> Self {
        Self {
            template_type: Some(template_type.into()),
            template: Some(template.into()),
            template_file: None,
        }
    }

    /// Create a template action that loads from a file path.
    pub fn from_file(template_type: impl Into<String>, file_path: impl Into<String>) -> Self {
        Self {
            template_type: Some(template_type.into()),
            template: None,
            template_file: Some(file_path.into()),
        }
    }

    /// Set the template type (e.g., "VELOCITY", "MUSTACHE").
    pub fn template_type(mut self, template_type: impl Into<String>) -> Self {
        self.template_type = Some(template_type.into());
        self
    }

    /// Set the inline template body.
    pub fn template(mut self, template: impl Into<String>) -> Self {
        self.template = Some(template.into());
        self
    }

    /// Set the template file path (alternative to inline template).
    pub fn template_file(mut self, file_path: impl Into<String>) -> Self {
        self.template_file = Some(file_path.into());
        self
    }
}

// ---------------------------------------------------------------------------
// HttpForward
// ---------------------------------------------------------------------------

/// Forward action — proxy the matched request to another host.
///
/// # Example
/// ```
/// use mockserver_client::HttpForward;
///
/// let forward = HttpForward::new("backend.local", 8080);
/// ```
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpForward {
    pub host: String,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub port: Option<u16>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub scheme: Option<String>,

    /// Delay applied before the request is forwarded.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl HttpForward {
    /// Create a forward action to the given host and port.
    pub fn new(host: impl Into<String>, port: u16) -> Self {
        Self {
            host: host.into(),
            port: Some(port),
            scheme: None,
            delay: None,
        }
    }

    /// Set the scheme (HTTP or HTTPS).
    pub fn scheme(mut self, scheme: impl Into<String>) -> Self {
        self.scheme = Some(scheme.into());
        self
    }

    /// Set a delay applied before the request is forwarded.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

// ---------------------------------------------------------------------------
// HttpClassCallback
// ---------------------------------------------------------------------------

/// Class callback action — delegates the response (or forward) to a server-side
/// class that implements MockServer's callback interface.
///
/// This is a purely declarative (REST-only) callback: no WebSocket is involved.
/// The named class must be on the MockServer server's classpath. Serialized as
/// `httpResponseClassCallback` or `httpForwardClassCallback` in an expectation.
///
/// # Example
/// ```
/// use mockserver_client::HttpClassCallback;
///
/// let cb = HttpClassCallback::new("com.example.MyCallback").primary(true);
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpClassCallback {
    pub callback_class: String,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,
}

impl HttpClassCallback {
    /// Create a class callback referencing the fully-qualified class name of a
    /// server-side callback implementation.
    pub fn new(callback_class: impl Into<String>) -> Self {
        Self {
            callback_class: callback_class.into(),
            delay: None,
            primary: None,
        }
    }

    /// Set a delay applied before the callback runs.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }

    /// Mark this callback as primary (kept on the primary event-loop thread).
    pub fn primary(mut self, primary: bool) -> Self {
        self.primary = Some(primary);
        self
    }
}

// ---------------------------------------------------------------------------
// HttpObjectCallback
// ---------------------------------------------------------------------------

/// Object (closure) callback action — delegates the response (or forward) to a
/// client-side closure invoked over the callback WebSocket.
///
/// The `client_id` is the id assigned by MockServer when the client opens the
/// callback WebSocket (`/_mockserver_callback_websocket`). When a request
/// matches, the server pushes it over that socket and the client's registered
/// closure produces the response. Serialized as `httpResponseObjectCallback` or
/// `httpForwardObjectCallback` in an expectation.
///
/// Most users do not construct this directly — use
/// [`MockServerClient::mock_with_callback`](crate::MockServerClient::mock_with_callback),
/// which opens the shared WebSocket, registers the closure, and wires up the
/// `client_id` automatically.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpObjectCallback {
    pub client_id: String,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_callback: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,
}

impl HttpObjectCallback {
    /// Create an object callback bound to the given callback-WebSocket client id.
    pub fn new(client_id: impl Into<String>) -> Self {
        Self {
            client_id: client_id.into(),
            response_callback: None,
            delay: None,
            primary: None,
        }
    }

    /// Set whether the callback also receives the response (forward + response form).
    pub fn response_callback(mut self, response_callback: bool) -> Self {
        self.response_callback = Some(response_callback);
        self
    }

    /// Set a delay applied before the callback runs.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }

    /// Mark this callback as primary (kept on the primary event-loop thread).
    pub fn primary(mut self, primary: bool) -> Self {
        self.primary = Some(primary);
        self
    }
}

// ---------------------------------------------------------------------------
// HttpError
// ---------------------------------------------------------------------------

/// Error action — return a connection-level error to the caller.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpError {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub drop_connection: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_bytes: Option<String>,

    /// Delay applied before the error is returned.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl HttpError {
    /// Create a new error action.
    pub fn new() -> Self {
        Self::default()
    }

    /// Drop the connection without a response.
    pub fn drop_connection(mut self, drop: bool) -> Self {
        self.drop_connection = Some(drop);
        self
    }

    /// Send arbitrary bytes then close.
    pub fn response_bytes(mut self, bytes: impl Into<String>) -> Self {
        self.response_bytes = Some(bytes.into());
        self
    }

    /// Set a delay applied before the error is returned.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

// ---------------------------------------------------------------------------
// HttpSseResponse (Server-Sent Events)
// ---------------------------------------------------------------------------

/// A single Server-Sent Event in an [`HttpSseResponse`].
///
/// Maps to the `events[]` entries of the `httpSseResponse` wire shape.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct SseEvent {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub event: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub data: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub id: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub retry: Option<u32>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl SseEvent {
    /// Create a new empty SSE event.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the `event:` field (event type/name).
    pub fn event(mut self, event: impl Into<String>) -> Self {
        self.event = Some(event.into());
        self
    }

    /// Set the `data:` payload.
    pub fn data(mut self, data: impl Into<String>) -> Self {
        self.data = Some(data.into());
        self
    }

    /// Set the `id:` field.
    pub fn id(mut self, id: impl Into<String>) -> Self {
        self.id = Some(id.into());
        self
    }

    /// Set the `retry:` reconnection time in milliseconds.
    pub fn retry(mut self, retry: u32) -> Self {
        self.retry = Some(retry);
        self
    }

    /// Set a delay before this event is emitted.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

/// Builder for a Server-Sent Events (SSE) streaming response action.
///
/// Serialized as the `httpSseResponse` action in an expectation.
///
/// # Example
/// ```
/// use mockserver_client::{HttpSseResponse, SseEvent};
///
/// let sse = HttpSseResponse::new()
///     .status_code(200)
///     .header("Content-Type", "text/event-stream")
///     .event(SseEvent::new().event("message").data("hello").id("1"))
///     .close_connection(true);
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpSseResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub status_code: Option<u16>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub headers: Option<HashMap<String, Vec<String>>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub events: Option<Vec<SseEvent>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub close_connection: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl HttpSseResponse {
    /// Create a new empty SSE response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the HTTP status code.
    pub fn status_code(mut self, code: u16) -> Self {
        self.status_code = Some(code);
        self
    }

    /// Add a response header.
    pub fn header(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        let headers = self.headers.get_or_insert_with(HashMap::new);
        headers.entry(key.into()).or_default().push(value.into());
        self
    }

    /// Append an SSE event to the stream.
    pub fn event(mut self, event: SseEvent) -> Self {
        self.events.get_or_insert_with(Vec::new).push(event);
        self
    }

    /// Replace all SSE events.
    pub fn events(mut self, events: Vec<SseEvent>) -> Self {
        self.events = Some(events);
        self
    }

    /// Whether to close the connection after emitting all events.
    pub fn close_connection(mut self, close: bool) -> Self {
        self.close_connection = Some(close);
        self
    }

    /// Set a delay before the response starts.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

// ---------------------------------------------------------------------------
// HttpWebSocketResponse
// ---------------------------------------------------------------------------

/// A single WebSocket message in an [`HttpWebSocketResponse`].
///
/// Either `text` or `binary` should be set. Binary data is base64-encoded
/// on the wire (the schema declares `binary` as `format: byte`).
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct WebSocketMessage {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub text: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub binary: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl WebSocketMessage {
    /// Create a text WebSocket message.
    pub fn text(text: impl Into<String>) -> Self {
        Self {
            text: Some(text.into()),
            binary: None,
            delay: None,
        }
    }

    /// Create a binary WebSocket message from raw bytes (base64-encoded on the wire).
    pub fn binary(data: impl AsRef<[u8]>) -> Self {
        Self {
            text: None,
            binary: Some(BASE64.encode(data.as_ref())),
            delay: None,
        }
    }

    /// Create a binary WebSocket message from an already base64-encoded string.
    pub fn binary_base64(base64: impl Into<String>) -> Self {
        Self {
            text: None,
            binary: Some(base64.into()),
            delay: None,
        }
    }

    /// Set a delay before this message is sent.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

/// A per-incoming-frame response rule inside an [`HttpWebSocketResponse::matchers`].
///
/// When an incoming WebSocket frame matches this rule (by `frame_type` and/or
/// `text_matcher`), the paired [`responses`](Self::responses) are sent back.
/// Unknown fields are captured in [`extra`](Self::extra) so the shape round-trips
/// without loss.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct WebSocketMatcher {
    /// Frame type to match: `"TEXT"`, `"BINARY"`, `"PING"`, `"PONG"` or `"ANY"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub frame_type: Option<String>,

    /// Exact-or-regex matcher applied to a text frame's payload.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub text_matcher: Option<String>,

    /// Messages sent in reply when an incoming frame matches this rule.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub responses: Option<Vec<WebSocketMessage>>,

    /// Forward-compatibility catch-all for matcher fields not yet named.
    #[serde(flatten, default)]
    pub extra: Extra,
}

impl WebSocketMatcher {
    /// Create a new empty matcher rule.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the frame type to match (`"TEXT"`, `"BINARY"`, `"PING"`, `"PONG"`, `"ANY"`).
    pub fn frame_type(mut self, frame_type: impl Into<String>) -> Self {
        self.frame_type = Some(frame_type.into());
        self
    }

    /// Set an exact-or-regex matcher for a text frame's payload.
    pub fn text_matcher(mut self, text_matcher: impl Into<String>) -> Self {
        self.text_matcher = Some(text_matcher.into());
        self
    }

    /// Append a reply message sent when an incoming frame matches this rule.
    pub fn response(mut self, response: WebSocketMessage) -> Self {
        self.responses.get_or_insert_with(Vec::new).push(response);
        self
    }

    /// Replace all reply messages.
    pub fn responses(mut self, responses: Vec<WebSocketMessage>) -> Self {
        self.responses = Some(responses);
        self
    }
}

/// Builder for a WebSocket streaming response action.
///
/// Serialized as the `httpWebSocketResponse` action in an expectation.
///
/// # Example
/// ```
/// use mockserver_client::{HttpWebSocketResponse, WebSocketMessage};
///
/// let ws = HttpWebSocketResponse::new()
///     .subprotocol("chat")
///     .message(WebSocketMessage::text("hello"))
///     .message(WebSocketMessage::binary([0x01, 0x02]))
///     .close_connection(true);
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpWebSocketResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub subprotocol: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub messages: Option<Vec<WebSocketMessage>>,

    /// Per-incoming-frame response rules; when set, an incoming frame matching a
    /// rule triggers that rule's `responses`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub matchers: Option<Vec<WebSocketMatcher>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub close_connection: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl HttpWebSocketResponse {
    /// Create a new empty WebSocket response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Append an incoming-frame matcher rule.
    pub fn matcher(mut self, matcher: WebSocketMatcher) -> Self {
        self.matchers.get_or_insert_with(Vec::new).push(matcher);
        self
    }

    /// Replace all incoming-frame matcher rules.
    pub fn matchers(mut self, matchers: Vec<WebSocketMatcher>) -> Self {
        self.matchers = Some(matchers);
        self
    }

    /// Set the negotiated subprotocol.
    pub fn subprotocol(mut self, subprotocol: impl Into<String>) -> Self {
        self.subprotocol = Some(subprotocol.into());
        self
    }

    /// Append a WebSocket message to send.
    pub fn message(mut self, message: WebSocketMessage) -> Self {
        self.messages.get_or_insert_with(Vec::new).push(message);
        self
    }

    /// Replace all WebSocket messages.
    pub fn messages(mut self, messages: Vec<WebSocketMessage>) -> Self {
        self.messages = Some(messages);
        self
    }

    /// Whether to close the connection after emitting all messages.
    pub fn close_connection(mut self, close: bool) -> Self {
        self.close_connection = Some(close);
        self
    }

    /// Set a delay before the response starts.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

// ---------------------------------------------------------------------------
// DnsResponse
// ---------------------------------------------------------------------------

/// A single DNS resource record in a [`DnsResponse`].
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct DnsRecord {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,

    #[serde(rename = "type", skip_serializing_if = "Option::is_none")]
    pub record_type: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub dns_class: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub ttl: Option<u32>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub value: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub priority: Option<u32>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub weight: Option<u32>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub port: Option<u16>,
}

impl DnsRecord {
    /// Create a new empty DNS record.
    pub fn new() -> Self {
        Self::default()
    }

    /// Create an `A` (IPv4 address) record.
    pub fn a(name: impl Into<String>, ip: impl Into<String>) -> Self {
        Self::new().name(name).record_type("A").value(ip)
    }

    /// Create an `AAAA` (IPv6 address) record.
    pub fn aaaa(name: impl Into<String>, ip: impl Into<String>) -> Self {
        Self::new().name(name).record_type("AAAA").value(ip)
    }

    /// Create a `CNAME` record.
    pub fn cname(name: impl Into<String>, target: impl Into<String>) -> Self {
        Self::new().name(name).record_type("CNAME").value(target)
    }

    /// Create a `TXT` record.
    pub fn txt(name: impl Into<String>, text: impl Into<String>) -> Self {
        Self::new().name(name).record_type("TXT").value(text)
    }

    /// Set the record name.
    pub fn name(mut self, name: impl Into<String>) -> Self {
        self.name = Some(name.into());
        self
    }

    /// Set the record type (e.g. "A", "AAAA", "CNAME", "MX", "SRV", "TXT", "PTR").
    pub fn record_type(mut self, record_type: impl Into<String>) -> Self {
        self.record_type = Some(record_type.into());
        self
    }

    /// Set the DNS class (e.g. "IN", "CH", "HS", "ANY").
    pub fn dns_class(mut self, dns_class: impl Into<String>) -> Self {
        self.dns_class = Some(dns_class.into());
        self
    }

    /// Set the time-to-live in seconds.
    pub fn ttl(mut self, ttl: u32) -> Self {
        self.ttl = Some(ttl);
        self
    }

    /// Set the record value (address, target, text, etc.).
    pub fn value(mut self, value: impl Into<String>) -> Self {
        self.value = Some(value.into());
        self
    }

    /// Set the priority (MX/SRV).
    pub fn priority(mut self, priority: u32) -> Self {
        self.priority = Some(priority);
        self
    }

    /// Set the weight (SRV).
    pub fn weight(mut self, weight: u32) -> Self {
        self.weight = Some(weight);
        self
    }

    /// Set the port (SRV).
    pub fn port(mut self, port: u16) -> Self {
        self.port = Some(port);
        self
    }
}

/// Builder for a DNS response action.
///
/// Serialized as the `dnsResponse` action in an expectation.
///
/// # Example
/// ```
/// use mockserver_client::{DnsResponse, DnsRecord};
///
/// let dns = DnsResponse::new()
///     .response_code("NOERROR")
///     .answer_record(DnsRecord::a("example.com", "1.2.3.4").ttl(300));
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct DnsResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub answer_records: Option<Vec<DnsRecord>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub authority_records: Option<Vec<DnsRecord>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub additional_records: Option<Vec<DnsRecord>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_code: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl DnsResponse {
    /// Create a new empty DNS response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Append an answer-section record.
    pub fn answer_record(mut self, record: DnsRecord) -> Self {
        self.answer_records
            .get_or_insert_with(Vec::new)
            .push(record);
        self
    }

    /// Replace all answer-section records.
    pub fn answer_records(mut self, records: Vec<DnsRecord>) -> Self {
        self.answer_records = Some(records);
        self
    }

    /// Append an authority-section record.
    pub fn authority_record(mut self, record: DnsRecord) -> Self {
        self.authority_records
            .get_or_insert_with(Vec::new)
            .push(record);
        self
    }

    /// Append an additional-section record.
    pub fn additional_record(mut self, record: DnsRecord) -> Self {
        self.additional_records
            .get_or_insert_with(Vec::new)
            .push(record);
        self
    }

    /// Set the DNS response code (e.g. "NOERROR", "NXDOMAIN", "SERVFAIL").
    pub fn response_code(mut self, code: impl Into<String>) -> Self {
        self.response_code = Some(code.into());
        self
    }

    /// Set a delay before the response is returned.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

// ---------------------------------------------------------------------------
// BinaryResponse
// ---------------------------------------------------------------------------

/// Builder for a raw binary response action.
///
/// Serialized as the `binaryResponse` action in an expectation. The binary
/// payload is base64-encoded on the wire (the schema declares `binaryData`
/// as a string).
///
/// # Example
/// ```
/// use mockserver_client::BinaryResponse;
///
/// let resp = BinaryResponse::from_bytes([0xDE, 0xAD, 0xBE, 0xEF]);
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct BinaryResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub binary_data: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl BinaryResponse {
    /// Create a new empty binary response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Create a binary response from raw bytes (base64-encoded on the wire).
    pub fn from_bytes(data: impl AsRef<[u8]>) -> Self {
        Self {
            binary_data: Some(BASE64.encode(data.as_ref())),
            delay: None,
        }
    }

    /// Create a binary response from an already base64-encoded string.
    pub fn from_base64(base64: impl Into<String>) -> Self {
        Self {
            binary_data: Some(base64.into()),
            delay: None,
        }
    }

    /// Set the binary payload from raw bytes (base64-encoded on the wire).
    pub fn binary_data(mut self, data: impl AsRef<[u8]>) -> Self {
        self.binary_data = Some(BASE64.encode(data.as_ref()));
        self
    }

    /// Set a delay before the response is returned.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

// ---------------------------------------------------------------------------
// GrpcStreamResponse
// ---------------------------------------------------------------------------

/// A single gRPC stream message in a [`GrpcStreamResponse`].
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct GrpcStreamMessage {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub json: Option<String>,

    /// Template engine used to render the message (`"VELOCITY"`, `"JAVASCRIPT"`
    /// or `"MUSTACHE"`); when unset the `json` is sent verbatim.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub template_type: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl GrpcStreamMessage {
    /// Create a gRPC stream message from a JSON-encoded protobuf message string.
    pub fn json(json: impl Into<String>) -> Self {
        Self {
            json: Some(json.into()),
            template_type: None,
            delay: None,
        }
    }

    /// Set the template engine used to render this message.
    pub fn template_type(mut self, template_type: impl Into<String>) -> Self {
        self.template_type = Some(template_type.into());
        self
    }

    /// Set a delay before this message is sent.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

/// Builder for a gRPC streaming response action.
///
/// Serialized as the `grpcStreamResponse` action in an expectation.
///
/// # Example
/// ```
/// use mockserver_client::{GrpcStreamResponse, GrpcStreamMessage};
///
/// let grpc = GrpcStreamResponse::new()
///     .status_name("OK")
///     .message(GrpcStreamMessage::json("{\"id\":1}"))
///     .close_connection(true);
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct GrpcStreamResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub status_name: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub status_message: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub headers: Option<HashMap<String, Vec<String>>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub messages: Option<Vec<GrpcStreamMessage>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub close_connection: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,
}

impl GrpcStreamResponse {
    /// Create a new empty gRPC stream response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the gRPC status name (e.g. "OK", "NOT_FOUND").
    pub fn status_name(mut self, status_name: impl Into<String>) -> Self {
        self.status_name = Some(status_name.into());
        self
    }

    /// Set the gRPC status message.
    pub fn status_message(mut self, status_message: impl Into<String>) -> Self {
        self.status_message = Some(status_message.into());
        self
    }

    /// Add a response header (gRPC metadata).
    pub fn header(mut self, key: impl Into<String>, value: impl Into<String>) -> Self {
        let headers = self.headers.get_or_insert_with(HashMap::new);
        headers.entry(key.into()).or_default().push(value.into());
        self
    }

    /// Append a gRPC stream message.
    pub fn message(mut self, message: GrpcStreamMessage) -> Self {
        self.messages.get_or_insert_with(Vec::new).push(message);
        self
    }

    /// Replace all gRPC stream messages.
    pub fn messages(mut self, messages: Vec<GrpcStreamMessage>) -> Self {
        self.messages = Some(messages);
        self
    }

    /// Whether to close the stream after emitting all messages.
    pub fn close_connection(mut self, close: bool) -> Self {
        self.close_connection = Some(close);
        self
    }

    /// Set a delay before the response starts.
    pub fn delay(mut self, delay: Delay) -> Self {
        self.delay = Some(delay);
        self
    }
}

// ---------------------------------------------------------------------------
// OpenApiExpectation
// ---------------------------------------------------------------------------

/// An OpenAPI specification import — registers matchers and example responses
/// for the operations in an OpenAPI/Swagger spec.
///
/// Sent via `PUT /mockserver/openapi`. The spec may be a URL, a filesystem
/// path (`file://...`), a classpath resource, or an inline JSON/YAML payload.
///
/// # Example
/// ```
/// use mockserver_client::OpenApiExpectation;
///
/// let expectation = OpenApiExpectation::new(
///     "https://example.com/petstore.yaml",
/// )
/// .operation("listPets", "200")
/// .operation("showPetById", "200");
/// ```
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct OpenApiExpectation {
    pub spec_url_or_payload: String,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub operations_and_responses: Option<HashMap<String, String>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub context_path_prefix: Option<String>,
}

impl OpenApiExpectation {
    /// Create an OpenAPI import from a spec URL, file path, classpath resource,
    /// or inline JSON/YAML payload.
    pub fn new(spec_url_or_payload: impl Into<String>) -> Self {
        Self {
            spec_url_or_payload: spec_url_or_payload.into(),
            operations_and_responses: None,
            context_path_prefix: None,
        }
    }

    /// Map an `operationId` to the status code (or example name) to respond with.
    ///
    /// When no operations are specified, MockServer creates example responses
    /// for every operation in the spec.
    pub fn operation(
        mut self,
        operation_id: impl Into<String>,
        status_code: impl Into<String>,
    ) -> Self {
        self.operations_and_responses
            .get_or_insert_with(HashMap::new)
            .insert(operation_id.into(), status_code.into());
        self
    }

    /// Replace the full operations-to-responses map.
    pub fn operations_and_responses(mut self, map: HashMap<String, String>) -> Self {
        self.operations_and_responses = Some(map);
        self
    }

    /// Set a context-path prefix to prepend to every generated matcher path.
    pub fn context_path_prefix(mut self, prefix: impl Into<String>) -> Self {
        self.context_path_prefix = Some(prefix.into());
        self
    }
}

// ---------------------------------------------------------------------------
// Delay
// ---------------------------------------------------------------------------

/// A time delay (e.g., for response delays).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct Delay {
    pub time_unit: String,
    pub value: u64,
}

impl Delay {
    /// Create a delay in milliseconds.
    pub fn milliseconds(value: u64) -> Self {
        Self {
            time_unit: "MILLISECONDS".to_string(),
            value,
        }
    }

    /// Create a delay in seconds.
    pub fn seconds(value: u64) -> Self {
        Self {
            time_unit: "SECONDS".to_string(),
            value,
        }
    }
}

// ---------------------------------------------------------------------------
// Times
// ---------------------------------------------------------------------------

/// How many times an expectation should be matched.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct Times {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub remaining_times: Option<u32>,

    #[serde(default)]
    pub unlimited: bool,
}

impl Times {
    /// Match unlimited times.
    pub fn unlimited() -> Self {
        Self {
            remaining_times: None,
            unlimited: true,
        }
    }

    /// Match exactly `n` times.
    pub fn exactly(n: u32) -> Self {
        Self {
            remaining_times: Some(n),
            unlimited: false,
        }
    }

    /// Match once.
    pub fn once() -> Self {
        Self::exactly(1)
    }
}

// ---------------------------------------------------------------------------
// TimeToLive
// ---------------------------------------------------------------------------

/// How long an expectation remains active.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct TimeToLive {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub time_unit: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub time_to_live: Option<u64>,

    #[serde(default)]
    pub unlimited: bool,
}

impl TimeToLive {
    /// Unlimited TTL (never expires).
    pub fn unlimited() -> Self {
        Self {
            time_unit: None,
            time_to_live: None,
            unlimited: true,
        }
    }

    /// Expire after the given number of seconds.
    pub fn seconds(seconds: u64) -> Self {
        Self {
            time_unit: Some("SECONDS".to_string()),
            time_to_live: Some(seconds),
            unlimited: false,
        }
    }

    /// Expire after the given number of milliseconds.
    pub fn milliseconds(millis: u64) -> Self {
        Self {
            time_unit: Some("MILLISECONDS".to_string()),
            time_to_live: Some(millis),
            unlimited: false,
        }
    }
}

// ---------------------------------------------------------------------------
// VerificationTimes
// ---------------------------------------------------------------------------

/// Verification constraints — how many times a request must have been received.
///
/// On the wire both `atLeast` and `atMost` are ALWAYS sent, using `-1` to mean
/// "unbounded". The MockServer server deserializes these into primitive `int`
/// fields, so an omitted bound defaults to `0` server-side — which would turn
/// `at_least(n)` into an impossible `between(n, 0)` constraint. Emitting the
/// explicit `-1` sentinel (matching the Java client) avoids that.
#[derive(Debug, Clone, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct VerificationTimes {
    pub at_least: Option<u32>,
    pub at_most: Option<u32>,
}

impl Serialize for VerificationTimes {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        use serde::ser::SerializeStruct;
        let mut state = serializer.serialize_struct("VerificationTimes", 2)?;
        state.serialize_field("atLeast", &self.at_least.map_or(-1_i64, i64::from))?;
        state.serialize_field("atMost", &self.at_most.map_or(-1_i64, i64::from))?;
        state.end()
    }
}

impl VerificationTimes {
    /// Require at least `n` matching requests.
    pub fn at_least(n: u32) -> Self {
        Self {
            at_least: Some(n),
            at_most: None,
        }
    }

    /// Require at most `n` matching requests.
    pub fn at_most(n: u32) -> Self {
        Self {
            at_least: None,
            at_most: Some(n),
        }
    }

    /// Require exactly `n` matching requests.
    pub fn exactly(n: u32) -> Self {
        Self {
            at_least: Some(n),
            at_most: Some(n),
        }
    }

    /// Require between `min` and `max` matching requests (inclusive).
    pub fn between(min: u32, max: u32) -> Self {
        Self {
            at_least: Some(min),
            at_most: Some(max),
        }
    }
}

// ---------------------------------------------------------------------------
// Stateful scenarios
// ---------------------------------------------------------------------------

/// How MockServer selects which of an expectation's multiple `http_responses`
/// to return on each match. Maps to the `responseMode` field.
///
/// - `Sequential` (default) — cycle through the responses in order.
/// - `Random` — pick a response uniformly at random.
/// - `Weighted` — pick a response weighted by the index-aligned
///   [`Expectation::response_weights`].
/// - `Switch` — return the same response for [`Expectation::switch_after`]
///   matches before advancing to the next.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum ResponseMode {
    /// Cycle through the responses in order (default).
    Sequential,
    /// Pick a response uniformly at random.
    Random,
    /// Pick a response weighted by [`Expectation::response_weights`].
    Weighted,
    /// Return each response for [`Expectation::switch_after`] matches before advancing.
    Switch,
}

/// The protocol event that triggers a [`CrossProtocolScenario`] state
/// transition. Maps to the `trigger` field.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum CrossProtocolTrigger {
    /// A DNS query is observed.
    DnsQuery,
    /// A WebSocket connection is established.
    WebsocketConnect,
    /// A gRPC request is observed.
    GrpcRequest,
    /// An HTTP request is observed.
    HttpRequest,
}

/// A cross-protocol scenario correlation: when a protocol event matching
/// [`trigger`](Self::trigger) (and optionally [`match_pattern`](Self::match_pattern))
/// is observed, the named scenario is advanced to [`target_state`](Self::target_state).
///
/// Maps to entries of the `crossProtocolScenarios` array.
///
/// # Example
/// ```
/// use mockserver_client::{CrossProtocolScenario, CrossProtocolTrigger};
///
/// let scenario = CrossProtocolScenario::new(
///     CrossProtocolTrigger::DnsQuery,
///     "Deploy",
///     "DnsObserved",
/// )
/// .match_pattern("api.example.com");
/// ```
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct CrossProtocolScenario {
    pub trigger: CrossProtocolTrigger,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub match_pattern: Option<String>,

    pub scenario_name: String,

    pub target_state: String,
}

impl CrossProtocolScenario {
    /// Create a cross-protocol scenario for the given trigger that advances
    /// `scenario_name` to `target_state` when an event fires.
    pub fn new(
        trigger: CrossProtocolTrigger,
        scenario_name: impl Into<String>,
        target_state: impl Into<String>,
    ) -> Self {
        Self {
            trigger,
            match_pattern: None,
            scenario_name: scenario_name.into(),
            target_state: target_state.into(),
        }
    }

    /// Set the substring filter on the event identifier (omit to match all).
    pub fn match_pattern(mut self, pattern: impl Into<String>) -> Self {
        self.match_pattern = Some(pattern.into());
        self
    }
}

// ---------------------------------------------------------------------------
// ConnectionOptions / RecoverAfter
// ---------------------------------------------------------------------------

/// Connection-level options for an [`HttpResponse`] — control content-length,
/// chunking, keep-alive and socket-close behaviour.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ConnectionOptions {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub suppress_content_length_header: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub content_length_header_override: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub suppress_connection_header: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub chunk_size: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub chunk_delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub keep_alive_override: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub close_socket: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub close_socket_delay: Option<Delay>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl ConnectionOptions {
    /// Create a new empty set of connection options.
    pub fn new() -> Self {
        Self::default()
    }
}

/// Circuit-breaker style policy on an [`HttpResponse`]: fail the first
/// `fail_times` requests with `fail_response`, then serve the real response.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct RecoverAfter {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub fail_times: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub fail_response: Option<serde_json::Value>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub idempotency_header: Option<String>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl RecoverAfter {
    /// Create a new empty recover-after policy.
    pub fn new() -> Self {
        Self::default()
    }
}

// ---------------------------------------------------------------------------
// RateLimit
// ---------------------------------------------------------------------------

/// Declarative, protocol-agnostic rate limit / quota attached to an expectation.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct RateLimit {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,

    /// `"fixed_window"` (default) or `"token_bucket"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub algorithm: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub limit: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub window_millis: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub burst: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub refill_per_second: Option<f64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub error_status: Option<i32>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub retry_after: Option<String>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl RateLimit {
    /// Create a new empty rate limit.
    pub fn new() -> Self {
        Self::default()
    }

    /// Create a `fixed_window` rate limit of `limit` requests per `window_millis`.
    pub fn fixed_window(limit: i64, window_millis: i64) -> Self {
        Self {
            algorithm: Some("fixed_window".to_string()),
            limit: Some(limit),
            window_millis: Some(window_millis),
            ..Default::default()
        }
    }

    /// Create a `token_bucket` rate limit with `burst` capacity refilled at
    /// `refill_per_second` tokens per second.
    pub fn token_bucket(burst: i64, refill_per_second: f64) -> Self {
        Self {
            algorithm: Some("token_bucket".to_string()),
            burst: Some(burst),
            refill_per_second: Some(refill_per_second),
            ..Default::default()
        }
    }
}

// ---------------------------------------------------------------------------
// HttpForwardWithFallback / HttpForwardValidateAction / HttpOverrideForwardedRequest
// ---------------------------------------------------------------------------

/// Forward action that falls back to a canned response when the upstream fails
/// (serialised as `httpForwardWithFallback`).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpForwardWithFallback {
    pub http_forward: HttpForward,

    pub fallback_response: HttpResponse,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub fallback_on_status_codes: Option<Vec<i32>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub fallback_on_timeout: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl HttpForwardWithFallback {
    /// Create a forward-with-fallback action.
    pub fn new(http_forward: HttpForward, fallback_response: HttpResponse) -> Self {
        Self {
            http_forward,
            fallback_response,
            fallback_on_status_codes: None,
            fallback_on_timeout: None,
            delay: None,
            primary: None,
            extra: Extra::new(),
        }
    }

    /// Fall back when the upstream returns any of these status codes.
    pub fn fallback_on_status_codes(mut self, codes: Vec<i32>) -> Self {
        self.fallback_on_status_codes = Some(codes);
        self
    }

    /// Fall back when the upstream request times out.
    pub fn fallback_on_timeout(mut self, fallback: bool) -> Self {
        self.fallback_on_timeout = Some(fallback);
        self
    }
}

/// Forward action that also validates request/response against an OpenAPI spec
/// (serialised as `httpForwardValidateAction`).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpForwardValidateAction {
    pub spec_url_or_payload: String,

    pub host: String,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub port: Option<u16>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub scheme: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub validate_request: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub validate_response: Option<bool>,

    /// `"STRICT"` or `"LOG_ONLY"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub validation_mode: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl HttpForwardValidateAction {
    /// Create a forward-and-validate action against the given spec and host.
    pub fn new(spec_url_or_payload: impl Into<String>, host: impl Into<String>) -> Self {
        Self {
            spec_url_or_payload: spec_url_or_payload.into(),
            host: host.into(),
            port: None,
            scheme: None,
            validate_request: None,
            validate_response: None,
            validation_mode: None,
            delay: None,
            primary: None,
            extra: Extra::new(),
        }
    }
}

/// Override the forwarded request and/or response (serialised as
/// `httpOverrideForwardedRequest`).
///
/// Covers both wire shapes accepted by the server: the modern
/// `requestOverride`/`requestModifier`/`responseOverride`/`responseModifier`
/// form and the legacy `httpRequest`/`httpResponse` form. The `requestModifier`
/// and `responseModifier` sub-objects are kept as free-form JSON
/// ([`serde_json::Value`]) — they round-trip exactly, and the `extra` catch-all
/// preserves any other field.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpOverrideForwardedRequest {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub request_override: Option<HttpRequest>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub request_modifier: Option<serde_json::Value>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_override: Option<HttpResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_modifier: Option<serde_json::Value>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_template: Option<HttpTemplate>,

    /// Legacy shape: request to forward.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_request: Option<HttpRequest>,

    /// Legacy shape: response to return.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_response: Option<HttpResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl HttpOverrideForwardedRequest {
    /// Create a new empty override action.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the request override.
    pub fn request_override(mut self, request: HttpRequest) -> Self {
        self.request_override = Some(request);
        self
    }

    /// Set the response override.
    pub fn response_override(mut self, response: HttpResponse) -> Self {
        self.response_override = Some(response);
        self
    }
}

// ---------------------------------------------------------------------------
// ExpectationAction (before/after) / CaptureRule / ExpectationStep
// ---------------------------------------------------------------------------

/// A side-effect action run before (`beforeActions`) or after (`afterActions`)
/// an expectation's main action fires: an out-of-band request, or a class/object
/// callback.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ExpectationAction {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_request: Option<HttpRequest>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_class_callback: Option<HttpClassCallback>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_object_callback: Option<HttpObjectCallback>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub blocking: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub timeout: Option<Delay>,

    /// `"FAIL_FAST"` or `"BEST_EFFORT"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub failure_policy: Option<String>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl ExpectationAction {
    /// Create a before/after action that fires an out-of-band HTTP request.
    pub fn request(request: HttpRequest) -> Self {
        Self {
            http_request: Some(request),
            ..Default::default()
        }
    }

    /// Create a before/after action that invokes a server-side class callback.
    pub fn class_callback(callback: HttpClassCallback) -> Self {
        Self {
            http_class_callback: Some(callback),
            ..Default::default()
        }
    }
}

/// A capture rule (`capture`) — extract a value from the matched request and
/// bind it into scenario/template state under `into`.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct CaptureRule {
    /// One of `jsonPath`, `xpath`, `header`, `queryStringParameter`, `cookie`,
    /// `pathParameter`.
    pub source: String,

    pub expression: String,

    pub into: String,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl CaptureRule {
    /// Create a capture rule binding `expression` (evaluated against `source`)
    /// into the variable `into`.
    pub fn new(
        source: impl Into<String>,
        expression: impl Into<String>,
        into: impl Into<String>,
    ) -> Self {
        Self {
            source: source.into(),
            expression: expression.into(),
            into: into.into(),
            extra: Extra::new(),
        }
    }
}

/// One step of a multi-step expectation (`steps`) — used to script a sequence of
/// responder/side-effect actions for a single match.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ExpectationStep {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_request: Option<HttpRequest>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_class_callback: Option<HttpClassCallback>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_object_callback: Option<HttpObjectCallback>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_forward: Option<HttpForward>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_override_forwarded_request: Option<HttpOverrideForwardedRequest>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_response: Option<HttpResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_error: Option<HttpError>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub responder: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub blocking: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub timeout: Option<Delay>,

    /// `"FAIL_FAST"` or `"BEST_EFFORT"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub failure_policy: Option<String>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl ExpectationStep {
    /// Create a new empty step.
    pub fn new() -> Self {
        Self::default()
    }

    /// Create a step whose responder action is the given response.
    pub fn response(response: HttpResponse) -> Self {
        Self {
            http_response: Some(response),
            responder: Some(true),
            ..Default::default()
        }
    }
}

// ---------------------------------------------------------------------------
// GrpcBidiResponse
// ---------------------------------------------------------------------------

/// A single message in a [`GrpcBidiResponse`] (or one of its rule responses).
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct GrpcBidiMessage {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub json: Option<String>,

    /// `"VELOCITY"`, `"JAVASCRIPT"` or `"MUSTACHE"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub template_type: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl GrpcBidiMessage {
    /// Create a bidi message from a JSON-encoded protobuf message string.
    pub fn json(json: impl Into<String>) -> Self {
        Self {
            json: Some(json.into()),
            ..Default::default()
        }
    }
}

/// A request-keyed rule in a [`GrpcBidiResponse`] — when an incoming message
/// matches `match_json`, the paired `responses` are sent.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct GrpcBidiRule {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub match_json: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub responses: Option<Vec<GrpcBidiMessage>>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

/// A gRPC bidirectional-streaming response action (`grpcBidiResponse`).
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct GrpcBidiResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub status_name: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub status_message: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub headers: Option<HashMap<String, Vec<String>>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub messages: Option<Vec<GrpcBidiMessage>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub rules: Option<Vec<GrpcBidiRule>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub close_connection: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl GrpcBidiResponse {
    /// Create a new empty gRPC bidi response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Append a streamed message.
    pub fn message(mut self, message: GrpcBidiMessage) -> Self {
        self.messages.get_or_insert_with(Vec::new).push(message);
        self
    }

    /// Append a request-keyed rule.
    pub fn rule(mut self, rule: GrpcBidiRule) -> Self {
        self.rules.get_or_insert_with(Vec::new).push(rule);
        self
    }
}

// ---------------------------------------------------------------------------
// HttpLlmResponse
// ---------------------------------------------------------------------------

/// A single tool call in an [`LlmCompletion`].
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LlmToolCall {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub id: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub arguments: Option<String>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

/// Token-usage accounting for an [`LlmCompletion`].
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LlmUsage {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub input_tokens: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub output_tokens: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub cached_input_tokens: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub cache_creation_tokens: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub reasoning_tokens: Option<i64>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

/// Streaming timing model (physics) for an [`LlmCompletion`].
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LlmStreamingPhysics {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub time_to_first_token: Option<Delay>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub tokens_per_second: Option<i32>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub jitter: Option<f64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub seed: Option<i64>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub subword_streaming: Option<bool>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

/// The chat/text completion of an [`HttpLlmResponse`].
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LlmCompletion {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub text: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub tool_calls: Option<Vec<LlmToolCall>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub stop_reason: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub usage: Option<LlmUsage>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub streaming: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub output_schema: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub enforce_output_schema: Option<bool>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub tool_choice: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub reasoning_text: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub reasoning_signature: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub model: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub streaming_physics: Option<LlmStreamingPhysics>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl LlmCompletion {
    /// Create a completion with the given assistant text.
    pub fn text(text: impl Into<String>) -> Self {
        Self {
            text: Some(text.into()),
            ..Default::default()
        }
    }
}

/// The LLM response action (`httpLlmResponse`). Only the completion, embedding,
/// rerank, moderation and content-filter sub-objects are commonly set; each is
/// optional and every unknown/nested field is preserved via its own `extra`
/// catch-all so full LLM configs round-trip.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpLlmResponse {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub delay: Option<Delay>,

    /// `"ANTHROPIC"`, `"OPENAI"`, `"OPENAI_RESPONSES"`, `"GEMINI"`, `"BEDROCK"`,
    /// `"AZURE_OPENAI"`, `"OLLAMA"`, `"COHERE"`, `"VOYAGE"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub provider: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub model: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub completion: Option<LlmCompletion>,

    /// Embedding-response config (kept free-form; round-trips verbatim).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub embedding: Option<serde_json::Value>,

    /// Rerank-response config (kept free-form; round-trips verbatim).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rerank: Option<serde_json::Value>,

    /// Moderation-response config (kept free-form; round-trips verbatim).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub moderation: Option<serde_json::Value>,

    /// Content-filter config (kept free-form; round-trips verbatim).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub content_filter: Option<serde_json::Value>,

    /// Conversation-matching predicates (kept free-form; round-trips verbatim).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub conversation_predicates: Option<serde_json::Value>,

    /// LLM-specific chaos config (kept free-form; round-trips verbatim).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub chaos: Option<serde_json::Value>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub primary: Option<bool>,

    #[serde(flatten, default)]
    pub extra: Extra,
}

impl HttpLlmResponse {
    /// Create a new empty LLM response.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the provider (e.g. `"ANTHROPIC"`, `"OPENAI"`).
    pub fn provider(mut self, provider: impl Into<String>) -> Self {
        self.provider = Some(provider.into());
        self
    }

    /// Set the model name.
    pub fn model(mut self, model: impl Into<String>) -> Self {
        self.model = Some(model.into());
        self
    }

    /// Set the completion.
    pub fn completion(mut self, completion: LlmCompletion) -> Self {
        self.completion = Some(completion);
        self
    }
}

// ---------------------------------------------------------------------------
// Expectation
// ---------------------------------------------------------------------------

/// A full expectation combining a request matcher with an action.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct Expectation {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub id: Option<String>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub priority: Option<i32>,

    /// Match only a percentage (0–100) of otherwise-matching requests.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub percentage: Option<i32>,

    /// Declarative HTTP chaos / fault-injection profile.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub chaos: Option<HttpChaosProfile>,

    /// Declarative, protocol-agnostic rate limit / quota.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rate_limit: Option<RateLimit>,

    /// Request matcher. Optional because a `steps`-only (or side-effect-only)
    /// expectation carries no top-level request. Omitted from the wire form when
    /// `None`; an explicitly empty [`HttpRequest`] still serialises as `{}`.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub http_request: Option<HttpRequest>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_response: Option<HttpResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_forward: Option<HttpForward>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_response_template: Option<HttpTemplate>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_forward_template: Option<HttpTemplate>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_error: Option<HttpError>,

    /// Class callback that produces the response (serialized as `httpResponseClassCallback`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_response_class_callback: Option<HttpClassCallback>,

    /// Class callback that produces the request to forward (serialized as `httpForwardClassCallback`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_forward_class_callback: Option<HttpClassCallback>,

    /// Object/closure callback that produces the response (serialized as `httpResponseObjectCallback`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_response_object_callback: Option<HttpObjectCallback>,

    /// Object/closure callback that produces the request to forward (serialized as `httpForwardObjectCallback`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_forward_object_callback: Option<HttpObjectCallback>,

    /// Override the forwarded request/response (`httpOverrideForwardedRequest`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_override_forwarded_request: Option<HttpOverrideForwardedRequest>,

    /// Forward and validate against an OpenAPI spec (`httpForwardValidateAction`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_forward_validate_action: Option<HttpForwardValidateAction>,

    /// Forward with a fallback response on failure (`httpForwardWithFallback`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_forward_with_fallback: Option<HttpForwardWithFallback>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_sse_response: Option<HttpSseResponse>,

    /// LLM response action (`httpLlmResponse`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_llm_response: Option<HttpLlmResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_web_socket_response: Option<HttpWebSocketResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub dns_response: Option<DnsResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub binary_response: Option<BinaryResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub grpc_stream_response: Option<GrpcStreamResponse>,

    /// gRPC bidirectional-streaming response action (`grpcBidiResponse`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub grpc_bidi_response: Option<GrpcBidiResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub times: Option<Times>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub time_to_live: Option<TimeToLive>,

    /// Name of the state-machine this expectation participates in.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub scenario_name: Option<String>,

    /// State the scenario must be in for this expectation to match.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub scenario_state: Option<String>,

    /// State the scenario transitions to after this expectation matches.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub new_scenario_state: Option<String>,

    /// Multiple responses; takes priority over the singular [`Expectation::http_response`].
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_responses: Option<Vec<HttpResponse>>,

    /// How a response is selected from [`Expectation::http_responses`].
    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_mode: Option<ResponseMode>,

    /// Index-aligned relative weights for [`ResponseMode::Weighted`].
    #[serde(skip_serializing_if = "Option::is_none")]
    pub response_weights: Option<Vec<i32>>,

    /// Requests per response block before advancing under [`ResponseMode::Switch`] (default 1).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub switch_after: Option<i32>,

    /// Cross-protocol scenario correlations that advance scenario state on protocol events.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub cross_protocol_scenarios: Option<Vec<CrossProtocolScenario>>,

    /// Side-effect actions run before the main action fires (`beforeActions`).
    ///
    /// The server accepts a single object or an array; this client accepts both
    /// on the wire and always serialises an array.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "one_or_many",
        default
    )]
    pub before_actions: Option<Vec<ExpectationAction>>,

    /// Side-effect actions run after the main action fires (`afterActions`).
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "one_or_many",
        default
    )]
    pub after_actions: Option<Vec<ExpectationAction>>,

    /// Capture rules that bind request values into scenario/template state.
    #[serde(
        skip_serializing_if = "Option::is_none",
        deserialize_with = "one_or_many",
        default
    )]
    pub capture: Option<Vec<CaptureRule>>,

    /// Optional namespace (tenant) this expectation belongs to.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub namespace: Option<String>,

    /// Multi-step script for a single match (`steps`).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub steps: Option<Vec<ExpectationStep>>,

    /// Creation timestamp (set by the server; round-tripped when present).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub timestamp: Option<String>,

    /// Forward-compatibility catch-all for any expectation field the typed model
    /// does not yet name, so unknown fields survive a round-trip instead of
    /// being silently dropped.
    #[serde(flatten, default)]
    pub extra: Extra,
}

impl Expectation {
    /// Create a new expectation with the given request matcher.
    pub fn new(request: HttpRequest) -> Self {
        Self {
            http_request: Some(request),
            ..Default::default()
        }
    }

    /// Set the expectation ID (for upsert semantics).
    pub fn id(mut self, id: impl Into<String>) -> Self {
        self.id = Some(id.into());
        self
    }

    /// Set the priority (higher = matched first).
    pub fn priority(mut self, priority: i32) -> Self {
        self.priority = Some(priority);
        self
    }

    /// Set a response action.
    pub fn respond(mut self, response: HttpResponse) -> Self {
        self.http_response = Some(response);
        self
    }

    /// Set a forward action.
    pub fn forward(mut self, forward: HttpForward) -> Self {
        self.http_forward = Some(forward);
        self
    }

    /// Set a response template action.
    pub fn respond_template(mut self, template: HttpTemplate) -> Self {
        self.http_response_template = Some(template);
        self
    }

    /// Set a forward template action.
    pub fn forward_template(mut self, template: HttpTemplate) -> Self {
        self.http_forward_template = Some(template);
        self
    }

    /// Set an error action.
    pub fn error(mut self, error: HttpError) -> Self {
        self.http_error = Some(error);
        self
    }

    /// Respond via a server-side class callback (`httpResponseClassCallback`).
    ///
    /// The named class must implement MockServer's callback interface and be on
    /// the server's classpath. Convenience over building an [`HttpClassCallback`]
    /// directly; use [`respond_class_callback`](Self::respond_class_callback) for
    /// the full builder (delay, primary).
    pub fn respond_with_class_callback(mut self, callback_class: impl Into<String>) -> Self {
        self.http_response_class_callback = Some(HttpClassCallback::new(callback_class));
        self
    }

    /// Respond via a pre-built [`HttpClassCallback`] (`httpResponseClassCallback`).
    pub fn respond_class_callback(mut self, callback: HttpClassCallback) -> Self {
        self.http_response_class_callback = Some(callback);
        self
    }

    /// Forward via a server-side class callback (`httpForwardClassCallback`).
    pub fn forward_with_class_callback(mut self, callback_class: impl Into<String>) -> Self {
        self.http_forward_class_callback = Some(HttpClassCallback::new(callback_class));
        self
    }

    /// Forward via a pre-built [`HttpClassCallback`] (`httpForwardClassCallback`).
    pub fn forward_class_callback(mut self, callback: HttpClassCallback) -> Self {
        self.http_forward_class_callback = Some(callback);
        self
    }

    /// Respond via an object/closure callback (`httpResponseObjectCallback`).
    ///
    /// Most users should call
    /// [`MockServerClient::mock_with_callback`](crate::MockServerClient::mock_with_callback)
    /// instead, which opens the callback WebSocket, registers the closure, and
    /// fills in the `client_id` automatically.
    pub fn respond_object_callback(mut self, callback: HttpObjectCallback) -> Self {
        self.http_response_object_callback = Some(callback);
        self
    }

    /// Forward via an object/closure callback (`httpForwardObjectCallback`).
    pub fn forward_object_callback(mut self, callback: HttpObjectCallback) -> Self {
        self.http_forward_object_callback = Some(callback);
        self
    }

    /// Set a Server-Sent Events (SSE) response action.
    pub fn respond_sse(mut self, sse: HttpSseResponse) -> Self {
        self.http_sse_response = Some(sse);
        self
    }

    /// Set a WebSocket response action.
    pub fn respond_web_socket(mut self, ws: HttpWebSocketResponse) -> Self {
        self.http_web_socket_response = Some(ws);
        self
    }

    /// Set a DNS response action.
    pub fn respond_dns(mut self, dns: DnsResponse) -> Self {
        self.dns_response = Some(dns);
        self
    }

    /// Set a raw binary response action.
    pub fn respond_binary(mut self, binary: BinaryResponse) -> Self {
        self.binary_response = Some(binary);
        self
    }

    /// Set a gRPC streaming response action.
    pub fn respond_grpc_stream(mut self, grpc: GrpcStreamResponse) -> Self {
        self.grpc_stream_response = Some(grpc);
        self
    }

    /// Set the number of times this expectation matches.
    pub fn times(mut self, times: Times) -> Self {
        self.times = Some(times);
        self
    }

    /// Set the time-to-live.
    pub fn time_to_live(mut self, ttl: TimeToLive) -> Self {
        self.time_to_live = Some(ttl);
        self
    }

    /// Set the scenario (state-machine) name this expectation participates in.
    pub fn scenario_name(mut self, name: impl Into<String>) -> Self {
        self.scenario_name = Some(name.into());
        self
    }

    /// Set the state the scenario must be in for this expectation to match.
    pub fn scenario_state(mut self, state: impl Into<String>) -> Self {
        self.scenario_state = Some(state.into());
        self
    }

    /// Set the state the scenario transitions to after this expectation matches.
    pub fn new_scenario_state(mut self, state: impl Into<String>) -> Self {
        self.new_scenario_state = Some(state.into());
        self
    }

    /// Append a response to the multiple-responses list (`http_responses`).
    ///
    /// When set, `http_responses` takes priority over the singular
    /// [`respond`](Self::respond) action.
    pub fn respond_with(mut self, response: HttpResponse) -> Self {
        self.http_responses
            .get_or_insert_with(Vec::new)
            .push(response);
        self
    }

    /// Replace all multiple responses (`http_responses`).
    pub fn http_responses(mut self, responses: Vec<HttpResponse>) -> Self {
        self.http_responses = Some(responses);
        self
    }

    /// Set how a response is selected from `http_responses`.
    pub fn response_mode(mut self, mode: ResponseMode) -> Self {
        self.response_mode = Some(mode);
        self
    }

    /// Set the index-aligned relative weights for [`ResponseMode::Weighted`].
    pub fn response_weights(mut self, weights: Vec<i32>) -> Self {
        self.response_weights = Some(weights);
        self
    }

    /// Set the number of requests per response block before advancing under
    /// [`ResponseMode::Switch`].
    pub fn switch_after(mut self, switch_after: i32) -> Self {
        self.switch_after = Some(switch_after);
        self
    }

    /// Append a [`CrossProtocolScenario`] correlation.
    pub fn cross_protocol_scenario(mut self, scenario: CrossProtocolScenario) -> Self {
        self.cross_protocol_scenarios
            .get_or_insert_with(Vec::new)
            .push(scenario);
        self
    }

    /// Replace all cross-protocol scenario correlations.
    pub fn cross_protocol_scenarios(mut self, scenarios: Vec<CrossProtocolScenario>) -> Self {
        self.cross_protocol_scenarios = Some(scenarios);
        self
    }

    /// Match only a percentage (0–100) of otherwise-matching requests.
    pub fn percentage(mut self, percentage: i32) -> Self {
        self.percentage = Some(percentage);
        self
    }

    /// Attach a declarative HTTP chaos / fault-injection profile.
    pub fn chaos(mut self, chaos: HttpChaosProfile) -> Self {
        self.chaos = Some(chaos);
        self
    }

    /// Attach a declarative rate limit / quota.
    pub fn rate_limit(mut self, rate_limit: RateLimit) -> Self {
        self.rate_limit = Some(rate_limit);
        self
    }

    /// Override the forwarded request/response (`httpOverrideForwardedRequest`).
    pub fn override_forwarded_request(
        mut self,
        override_request: HttpOverrideForwardedRequest,
    ) -> Self {
        self.http_override_forwarded_request = Some(override_request);
        self
    }

    /// Forward and validate against an OpenAPI spec (`httpForwardValidateAction`).
    pub fn forward_validate(mut self, action: HttpForwardValidateAction) -> Self {
        self.http_forward_validate_action = Some(action);
        self
    }

    /// Forward with a fallback response on failure (`httpForwardWithFallback`).
    pub fn forward_with_fallback(mut self, action: HttpForwardWithFallback) -> Self {
        self.http_forward_with_fallback = Some(action);
        self
    }

    /// Set an LLM response action (`httpLlmResponse`).
    pub fn respond_llm(mut self, llm: HttpLlmResponse) -> Self {
        self.http_llm_response = Some(llm);
        self
    }

    /// Set a gRPC bidirectional-streaming response action (`grpcBidiResponse`).
    pub fn respond_grpc_bidi(mut self, grpc: GrpcBidiResponse) -> Self {
        self.grpc_bidi_response = Some(grpc);
        self
    }

    /// Append a before-action (`beforeActions`).
    pub fn before_action(mut self, action: ExpectationAction) -> Self {
        self.before_actions
            .get_or_insert_with(Vec::new)
            .push(action);
        self
    }

    /// Append an after-action (`afterActions`).
    pub fn after_action(mut self, action: ExpectationAction) -> Self {
        self.after_actions.get_or_insert_with(Vec::new).push(action);
        self
    }

    /// Append a capture rule (`capture`).
    pub fn capture_rule(mut self, rule: CaptureRule) -> Self {
        self.capture.get_or_insert_with(Vec::new).push(rule);
        self
    }

    /// Set the namespace (tenant) this expectation belongs to.
    pub fn namespace(mut self, namespace: impl Into<String>) -> Self {
        self.namespace = Some(namespace.into());
        self
    }

    /// Append a step to the multi-step script (`steps`).
    pub fn step(mut self, step: ExpectationStep) -> Self {
        self.steps.get_or_insert_with(Vec::new).push(step);
        self
    }

    /// Replace all steps (`steps`).
    pub fn steps(mut self, steps: Vec<ExpectationStep>) -> Self {
        self.steps = Some(steps);
        self
    }
}

// ---------------------------------------------------------------------------
// Verification
// ---------------------------------------------------------------------------

/// A verification request sent to MockServer.
///
/// At least one of `http_request` or `http_response` must be set.
/// `http_response` uses the same [`HttpResponse`] type as expectations —
/// the server matches against the recorded response's status code, headers,
/// and body.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct Verification {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_request: Option<HttpRequest>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_response: Option<HttpResponse>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub times: Option<VerificationTimes>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub maximum_number_of_request_to_return_in_verification_failure: Option<u32>,
}

/// A verification sequence request.
///
/// `http_responses` is index-aligned with `http_requests` — each entry
/// constrains the response that must have been returned for the
/// corresponding request.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct VerificationSequence {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_requests: Option<Vec<HttpRequest>>,

    #[serde(skip_serializing_if = "Option::is_none")]
    pub http_responses: Option<Vec<HttpResponse>>,
}

// ---------------------------------------------------------------------------
// Ports
// ---------------------------------------------------------------------------

/// Port list (used by status and bind endpoints).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct Ports {
    pub ports: Vec<u16>,
}

// ---------------------------------------------------------------------------
// Scenario state
// ---------------------------------------------------------------------------

/// A scenario and its current state, as returned by the scenario REST
/// endpoints (`GET /mockserver/scenario` and `GET /mockserver/scenario/{name}`).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct ScenarioState {
    /// The scenario (state-machine) name.
    pub scenario_name: String,
    /// The scenario's current state.
    pub current_state: String,
}

/// Wrapper for the `GET /mockserver/scenario` list response shape
/// (`{"scenarios":[{"scenarioName","currentState"}]}`).
#[derive(Debug, Clone, Deserialize)]
pub(crate) struct ScenarioList {
    #[serde(default)]
    pub scenarios: Vec<ScenarioState>,
}

// ---------------------------------------------------------------------------
// Retrieve types
// ---------------------------------------------------------------------------

/// The type of data to retrieve from MockServer.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RetrieveType {
    /// Recorded inbound requests.
    Requests,
    /// Active (live) expectations.
    ActiveExpectations,
    /// Recorded expectations (from proxy mode).
    RecordedExpectations,
    /// Log messages.
    Logs,
    /// Request/response pairs.
    RequestResponses,
}

impl RetrieveType {
    /// The query parameter value for this type.
    pub fn as_str(&self) -> &'static str {
        match self {
            RetrieveType::Requests => "REQUESTS",
            RetrieveType::ActiveExpectations => "ACTIVE_EXPECTATIONS",
            RetrieveType::RecordedExpectations => "RECORDED_EXPECTATIONS",
            RetrieveType::Logs => "LOGS",
            RetrieveType::RequestResponses => "REQUEST_RESPONSES",
        }
    }
}

/// The response format for retrieve calls.
///
/// In addition to JSON and log-entry formats, MockServer can return the
/// retrieved expectations as SDK setup code (the builder code that recreates
/// the expectations) in a range of languages.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RetrieveFormat {
    Json,
    LogEntries,
    Java,
    JavaScript,
    Python,
    Go,
    CSharp,
    Ruby,
    Rust,
    Php,
}

impl RetrieveFormat {
    /// The query parameter value for this format.
    pub fn as_str(&self) -> &'static str {
        match self {
            RetrieveFormat::Json => "JSON",
            RetrieveFormat::LogEntries => "LOG_ENTRIES",
            RetrieveFormat::Java => "JAVA",
            RetrieveFormat::JavaScript => "JAVASCRIPT",
            RetrieveFormat::Python => "PYTHON",
            RetrieveFormat::Go => "GO",
            RetrieveFormat::CSharp => "CSHARP",
            RetrieveFormat::Ruby => "RUBY",
            RetrieveFormat::Rust => "RUST",
            RetrieveFormat::Php => "PHP",
        }
    }
}

/// The type of data to clear from MockServer.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ClearType {
    All,
    Log,
    Expectations,
}

impl ClearType {
    /// The query parameter value for this type.
    pub fn as_str(&self) -> &'static str {
        match self {
            ClearType::All => "ALL",
            ClearType::Log => "LOG",
            ClearType::Expectations => "EXPECTATIONS",
        }
    }
}

// ---------------------------------------------------------------------------
// Pact verification result
// ---------------------------------------------------------------------------

/// Outcome of a Pact contract verification (`PUT /mockserver/pact/verify`).
///
/// The server replies `202 ACCEPTED` when every interaction in the contract
/// matched an active expectation, or `406 NOT_ACCEPTABLE` when verification
/// failed — in both cases the body is the same verification report JSON.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct PactVerification {
    /// `true` when verification passed (`202`), `false` when it failed (`406`).
    pub passed: bool,
    /// The verification report JSON returned by the server (verbatim).
    pub report: String,
}

// ---------------------------------------------------------------------------
// Operating mode
// ---------------------------------------------------------------------------

/// High-level operating mode for MockServer (set via `PUT /mockserver/mode`,
/// read via `GET /mockserver/mode`).
///
/// Each mode packages the common record / replay / pass-through workflows into a
/// single switch (a convenience over `attemptToProxyIfNoMatchingExpectation`):
///
/// * [`MockMode::Simulate`] — match expectations and return mocks; unmatched
///   requests get a `404`. This is the default (proxy-on-no-match disabled).
/// * [`MockMode::Spy`] — match expectations and return mocks, but forward
///   unmatched requests to the real upstream so they are served live and recorded
///   (proxy-on-no-match enabled).
/// * [`MockMode::Capture`] — forward and record; with no expectations defined this
///   captures all traffic. Backed by the same proxy flag as [`MockMode::Spy`].
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum MockMode {
    /// Match expectations; unmatched requests get a `404` (default).
    Simulate,
    /// Match expectations; unmatched requests forwarded to the upstream and recorded.
    Spy,
    /// Forward and record all traffic.
    Capture,
}

impl MockMode {
    /// The wire value for this mode (the `mode` query parameter / JSON field).
    pub fn as_str(&self) -> &'static str {
        match self {
            MockMode::Simulate => "SIMULATE",
            MockMode::Spy => "SPY",
            MockMode::Capture => "CAPTURE",
        }
    }

    /// Whether, in this mode, a request matching no expectation is proxied to its
    /// upstream (and thereby recorded) rather than answered with a `404`.
    pub fn proxy_unmatched_requests(&self) -> bool {
        !matches!(self, MockMode::Simulate)
    }
}

impl std::fmt::Display for MockMode {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str(self.as_str())
    }
}

impl std::str::FromStr for MockMode {
    type Err = String;

    /// Parse a mode name case-insensitively (matches the server's
    /// `MockMode.parse`). Returns an error message for blank/unknown values.
    fn from_str(value: &str) -> std::result::Result<Self, Self::Err> {
        match value.trim().to_uppercase().as_str() {
            "" => Err("mode is required (one of SIMULATE, SPY, CAPTURE)".to_string()),
            "SIMULATE" => Ok(MockMode::Simulate),
            "SPY" => Ok(MockMode::Spy),
            "CAPTURE" => Ok(MockMode::Capture),
            other => Err(format!(
                "unknown mode '{other}' (expected one of SIMULATE, SPY, CAPTURE)"
            )),
        }
    }
}

// ---------------------------------------------------------------------------
// gRPC descriptor management
// ---------------------------------------------------------------------------

/// A single gRPC method registered from an uploaded descriptor set.
///
/// Returned by [`MockServerClient::retrieve_grpc_services`] as part of a
/// [`GrpcService`]. Maps to the `methods[]` entries of the
/// `PUT /mockserver/grpc/services` wire shape.
///
/// [`MockServerClient::retrieve_grpc_services`]: crate::MockServerClient::retrieve_grpc_services
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct GrpcMethod {
    /// The simple method name (e.g. `SayHello`).
    pub name: String,

    /// Fully-qualified name of the request message type.
    pub input_type: String,

    /// Fully-qualified name of the response message type.
    pub output_type: String,

    /// Whether the method uses client-side streaming.
    pub client_streaming: bool,

    /// Whether the method uses server-side streaming.
    pub server_streaming: bool,
}

/// A gRPC service registered from an uploaded descriptor set.
///
/// Returned by [`MockServerClient::retrieve_grpc_services`]. Maps to the
/// top-level entries of the `PUT /mockserver/grpc/services` wire shape.
///
/// [`MockServerClient::retrieve_grpc_services`]: crate::MockServerClient::retrieve_grpc_services
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct GrpcService {
    /// Fully-qualified service name (e.g. `helloworld.Greeter`).
    pub name: String,

    /// The methods declared by this service.
    pub methods: Vec<GrpcMethod>,
}

// ---------------------------------------------------------------------------
// SocketAddress
// ---------------------------------------------------------------------------

/// A downstream socket address (host / port / scheme) to direct a request at.
///
/// Maps to MockServer's `SocketAddress` model. Used by load-scenario steps to
/// target a specific upstream rather than relying on the `Host` header.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct SocketAddress {
    /// The downstream host name or IP.
    pub host: String,

    /// The downstream port.
    pub port: u16,

    /// The scheme to connect with — `"HTTP"` or `"HTTPS"`. Defaults to `"HTTP"`
    /// on the server when omitted.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub scheme: Option<String>,
}

impl SocketAddress {
    /// Create a plain HTTP socket address.
    pub fn new(host: impl Into<String>, port: u16) -> Self {
        Self {
            host: host.into(),
            port,
            scheme: None,
        }
    }

    /// Set the scheme (`"HTTP"` or `"HTTPS"`).
    pub fn scheme(mut self, scheme: impl Into<String>) -> Self {
        self.scheme = Some(scheme.into());
        self
    }

    /// Convenience: an HTTPS socket address.
    pub fn https(host: impl Into<String>, port: u16) -> Self {
        Self::new(host, port).scheme("HTTPS")
    }
}

// ---------------------------------------------------------------------------
// Load scenario registry (PUT/GET/DELETE /mockserver/loadScenario[/...])
// ---------------------------------------------------------------------------

/// The interpolation curve used to ramp a value (virtual users or arrival
/// rate) from a start setpoint to an end setpoint across a ramp [`LoadStage`].
/// Maps to the `RampCurve` schema. Only meaningful for ramp stages; ignored for
/// holds and pauses.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum RampCurve {
    /// Constant slope.
    Linear,
    /// Ease-in: slow then fast.
    Quadratic,
    /// A steeper ease-in.
    Exponential,
}

/// The kind of a [`LoadStage`].
///
/// - `Vu` — closed model: hold or ramp the number of concurrent virtual users.
/// - `Rate` — open model: hold or ramp an arrival rate in iterations/second.
/// - `Pause` — drive no load for the duration.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadStageType {
    /// Closed model — hold or ramp concurrent virtual users.
    Vu,
    /// Open model — hold or ramp an arrival rate in iterations/second.
    Rate,
    /// Drive no load for the duration.
    Pause,
}

/// One stage of a [`LoadProfile`]: a contiguous slice of the run holding or
/// ramping a setpoint for `duration_millis`. Stages run in sequence. Maps to the
/// `LoadStage` schema.
///
/// Use the constructors [`LoadStage::vu_hold`], [`LoadStage::vu_ramp`],
/// [`LoadStage::rate_hold`], [`LoadStage::rate_ramp`] and [`LoadStage::pause`]
/// rather than building the struct directly so only the relevant fields are set
/// (and therefore serialized).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadStage {
    /// The kind of stage — `VU`, `RATE` or `PAUSE`.
    #[serde(rename = "type")]
    pub stage_type: LoadStageType,

    /// How long this stage runs in milliseconds (> 0).
    pub duration_millis: u64,

    /// Ramp shape (ramp stages only); omitted for holds and pauses.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub curve: Option<RampCurve>,

    /// VU hold: the number of virtual users to hold for the stage.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub vus: Option<u32>,

    /// VU ramp: virtual users at the start of the ramp.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub start_vus: Option<u32>,

    /// VU ramp: virtual users at the end of the ramp.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub end_vus: Option<u32>,

    /// RATE hold: arrival rate to hold, in iterations per second.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rate: Option<f64>,

    /// RATE ramp: arrival rate at the start of the ramp, in iterations/second.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub start_rate: Option<f64>,

    /// RATE ramp: arrival rate at the end of the ramp, in iterations/second.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub end_rate: Option<f64>,

    /// RATE stage only: optional cap on the auto-scaling virtual-user pool.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub max_vus: Option<u32>,
}

impl LoadStage {
    fn base(stage_type: LoadStageType, duration_millis: u64) -> Self {
        Self {
            stage_type,
            duration_millis,
            curve: None,
            vus: None,
            start_vus: None,
            end_vus: None,
            rate: None,
            start_rate: None,
            end_rate: None,
            max_vus: None,
        }
    }

    /// A VU stage holding `vus` virtual users for `duration_millis`.
    pub fn vu_hold(vus: u32, duration_millis: u64) -> Self {
        let mut stage = Self::base(LoadStageType::Vu, duration_millis);
        stage.vus = Some(vus);
        stage
    }

    /// A VU stage ramping from `start_vus` to `end_vus` over `duration_millis`
    /// along `curve`.
    pub fn vu_ramp(start_vus: u32, end_vus: u32, duration_millis: u64, curve: RampCurve) -> Self {
        let mut stage = Self::base(LoadStageType::Vu, duration_millis);
        stage.start_vus = Some(start_vus);
        stage.end_vus = Some(end_vus);
        stage.curve = Some(curve);
        stage
    }

    /// A RATE stage holding `rate` iterations/second for `duration_millis`.
    pub fn rate_hold(rate: f64, duration_millis: u64) -> Self {
        let mut stage = Self::base(LoadStageType::Rate, duration_millis);
        stage.rate = Some(rate);
        stage
    }

    /// A RATE stage ramping from `start_rate` to `end_rate` iterations/second
    /// over `duration_millis` along `curve`.
    pub fn rate_ramp(
        start_rate: f64,
        end_rate: f64,
        duration_millis: u64,
        curve: RampCurve,
    ) -> Self {
        let mut stage = Self::base(LoadStageType::Rate, duration_millis);
        stage.start_rate = Some(start_rate);
        stage.end_rate = Some(end_rate);
        stage.curve = Some(curve);
        stage
    }

    /// A PAUSE stage that drives no load for `duration_millis`.
    pub fn pause(duration_millis: u64) -> Self {
        Self::base(LoadStageType::Pause, duration_millis)
    }

    /// Cap the auto-scaling virtual-user pool for this RATE stage.
    pub fn max_vus(mut self, max_vus: u32) -> Self {
        self.max_vus = Some(max_vus);
        self
    }
}

/// A named load shape that expands server-side into ordinary [`LoadStage`]s.
/// Maps to the `LoadShapeType` schema.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadShapeType {
    /// Ramp up, hold the peak, ramp back down, with an optional recovery hold.
    Spike,
    /// A flight of pure-hold steps, each one "step" higher.
    Stairs,
    /// Ramp 0 to target then hold.
    RampHold,
}

/// What a [`LoadShape`] drives. Maps to the `LoadShapeMetric` schema.
///
/// - `Vu` — concurrent virtual users (closed model).
/// - `Rate` — arrival rate in iterations/second (open model).
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadShapeMetric {
    /// Concurrent virtual users (closed model).
    Vu,
    /// Arrival rate in iterations/second (open model).
    Rate,
}

/// A declarative named load shape that expands into ordinary [`LoadStage`]s.
/// Maps to the `LoadShape` schema. Only the parameters its `type` needs are
/// read; the rest are ignored. Use a shape OR an explicit `stages` list, not
/// both.
///
/// Use the constructors [`LoadShape::spike`], [`LoadShape::stairs`] and
/// [`LoadShape::ramp_hold`] so only the relevant fields are set (and therefore
/// serialized).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadShape {
    /// The named shape — `SPIKE`, `STAIRS` or `RAMP_HOLD`.
    #[serde(rename = "type")]
    pub shape_type: LoadShapeType,

    /// What the shape drives — `VU` (default) or `RATE`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub metric: Option<LoadShapeMetric>,

    /// Ramp interpolation curve used by the shape's ramps.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub curve: Option<RampCurve>,

    /// SPIKE: the level held before and after the spike.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub baseline: Option<f64>,

    /// SPIKE: the level held at the top of the spike.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub peak: Option<f64>,

    /// SPIKE: duration of the baseline to peak ramp.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub ramp_up_millis: Option<u64>,

    /// SPIKE: duration to hold at the peak; RAMP_HOLD: duration to hold at the
    /// target.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub hold_millis: Option<u64>,

    /// SPIKE: duration of the peak to baseline ramp.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub ramp_down_millis: Option<u64>,

    /// SPIKE (optional): duration to hold at baseline after the down ramp.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub recovery_hold_millis: Option<u64>,

    /// STAIRS: the level of the first step.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub start: Option<f64>,

    /// STAIRS: how much each step rises above the previous one.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub step: Option<f64>,

    /// STAIRS: the number of steps.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub steps: Option<u32>,

    /// STAIRS: how long each step holds at its level.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub step_duration_millis: Option<u64>,

    /// RAMP_HOLD: the level ramped up to (from 0) and then held.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub target: Option<f64>,

    /// RAMP_HOLD: duration of the 0 to target ramp.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub ramp_millis: Option<u64>,
}

impl LoadShape {
    fn base(shape_type: LoadShapeType) -> Self {
        Self {
            shape_type,
            metric: None,
            curve: None,
            baseline: None,
            peak: None,
            ramp_up_millis: None,
            hold_millis: None,
            ramp_down_millis: None,
            recovery_hold_millis: None,
            start: None,
            step: None,
            steps: None,
            step_duration_millis: None,
            target: None,
            ramp_millis: None,
        }
    }

    /// A SPIKE shape: ramp `baseline` to `peak` over `ramp_up_millis`, hold for
    /// `hold_millis`, then ramp back down over `ramp_down_millis`.
    pub fn spike(
        baseline: f64,
        peak: f64,
        ramp_up_millis: u64,
        hold_millis: u64,
        ramp_down_millis: u64,
    ) -> Self {
        let mut shape = Self::base(LoadShapeType::Spike);
        shape.baseline = Some(baseline);
        shape.peak = Some(peak);
        shape.ramp_up_millis = Some(ramp_up_millis);
        shape.hold_millis = Some(hold_millis);
        shape.ramp_down_millis = Some(ramp_down_millis);
        shape
    }

    /// A STAIRS shape: `steps` pure-hold steps, the first at `start` and each
    /// rising by `step`, every step holding for `step_duration_millis`.
    pub fn stairs(start: f64, step: f64, steps: u32, step_duration_millis: u64) -> Self {
        let mut shape = Self::base(LoadShapeType::Stairs);
        shape.start = Some(start);
        shape.step = Some(step);
        shape.steps = Some(steps);
        shape.step_duration_millis = Some(step_duration_millis);
        shape
    }

    /// A RAMP_HOLD shape: ramp from 0 to `target` over `ramp_millis`, then hold
    /// for `hold_millis`.
    pub fn ramp_hold(target: f64, ramp_millis: u64, hold_millis: u64) -> Self {
        let mut shape = Self::base(LoadShapeType::RampHold);
        shape.target = Some(target);
        shape.ramp_millis = Some(ramp_millis);
        shape.hold_millis = Some(hold_millis);
        shape
    }

    /// Set what the shape drives (`VU` or `RATE`).
    pub fn metric(mut self, metric: LoadShapeMetric) -> Self {
        self.metric = Some(metric);
        self
    }

    /// Set the ramp interpolation curve.
    pub fn curve(mut self, curve: RampCurve) -> Self {
        self.curve = Some(curve);
        self
    }

    /// SPIKE only: hold at baseline for `recovery_hold_millis` after the down
    /// ramp.
    pub fn recovery_hold_millis(mut self, recovery_hold_millis: u64) -> Self {
        self.recovery_hold_millis = Some(recovery_hold_millis);
        self
    }
}

/// The per-run metric a [`LoadThreshold`] evaluates. Maps to the threshold
/// `metric` enum.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadThresholdMetric {
    /// 50th-percentile latency in milliseconds.
    LatencyP50,
    /// 95th-percentile latency in milliseconds.
    LatencyP95,
    /// 99th-percentile latency in milliseconds.
    LatencyP99,
    /// 99.9th-percentile latency in milliseconds.
    LatencyP999,
    /// Failed / requests, as a 0.0-1.0 fraction.
    ErrorRate,
    /// Throughput in requests/second over the run's elapsed time.
    ThroughputRps,
}

/// How a [`LoadThreshold`]'s observed value is compared to its threshold. Maps
/// to the threshold `comparator` enum.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadComparator {
    /// observed < threshold.
    LessThan,
    /// observed <= threshold.
    LessThanOrEqual,
    /// observed > threshold.
    GreaterThan,
    /// observed >= threshold.
    GreaterThanOrEqual,
}

/// An in-run pass/fail threshold for a load scenario: a per-run metric compared
/// against a value. All thresholds must hold for the run verdict to be PASS
/// (logical AND). Maps to the `LoadThreshold` schema.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadThreshold {
    /// The per-run metric to evaluate.
    pub metric: LoadThresholdMetric,

    /// How the observed per-run value is compared to the threshold.
    pub comparator: LoadComparator,

    /// The threshold value (milliseconds for latency metrics, a 0.0-1.0
    /// fraction for `ERROR_RATE`, requests/second for `THROUGHPUT_RPS`).
    pub threshold: f64,
}

impl LoadThreshold {
    /// Create a threshold comparing `metric` to `threshold` using `comparator`.
    pub fn new(metric: LoadThresholdMetric, comparator: LoadComparator, threshold: f64) -> Self {
        Self {
            metric,
            comparator,
            threshold,
        }
    }
}

/// How a [`LoadPacing`] target iteration cycle is derived from its value. Maps
/// to the pacing `mode` enum.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadPacingMode {
    /// No pacing (immediate reschedule).
    None,
    /// `value` is the target cycle in milliseconds.
    ConstantPacing,
    /// `value` is the target iterations/second per VU (cycle = 1000 / value ms).
    ConstantThroughput,
}

/// Adaptive iteration pacing (think-time) for a load scenario: a target
/// per-virtual-user iteration cycle time. Applies only to the closed-model VU
/// loop. Maps to the `LoadPacing` schema.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadPacing {
    /// How the target iteration cycle is derived from `value`.
    pub mode: LoadPacingMode,

    /// For `CONSTANT_PACING` the target cycle in milliseconds; for
    /// `CONSTANT_THROUGHPUT` the target iterations/second per VU. Must be > 0
    /// when `mode` is not `NONE`.
    pub value: f64,
}

impl LoadPacing {
    /// Create a pacing rule with the given mode and value.
    pub fn new(mode: LoadPacingMode, value: f64) -> Self {
        Self { mode, value }
    }

    /// `CONSTANT_PACING`: target a per-VU iteration cycle of `cycle_millis`.
    pub fn constant_pacing(cycle_millis: f64) -> Self {
        Self::new(LoadPacingMode::ConstantPacing, cycle_millis)
    }

    /// `CONSTANT_THROUGHPUT`: target `iterations_per_second` per VU.
    pub fn constant_throughput(iterations_per_second: f64) -> Self {
        Self::new(LoadPacingMode::ConstantThroughput, iterations_per_second)
    }
}

/// The format of a [`LoadFeeder`]'s raw `data`. Maps to the feeder `format`
/// enum.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadFeederFormat {
    /// CSV: first line is the header row.
    Csv,
    /// JSON: an array of flat objects.
    Json,
}

/// How a [`LoadFeeder`] selects a row each iteration. Maps to the feeder
/// `strategy` enum.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadFeederStrategy {
    /// Cycle rows and never exhaust (default).
    Circular,
    /// Pick a uniformly random row each iteration.
    Random,
    /// Use each row once in order; COMPLETES the run when exhausted.
    Sequential,
}

/// Parameterized test data (a data feeder) for a load scenario: an inline
/// dataset from which one row is selected per iteration and exposed to the
/// iteration's templates as `$iteration.data.<column>`. Supply EITHER `rows`
/// (the primary form) OR `data` + `format`. Maps to the `LoadFeeder` schema.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadFeeder {
    /// Inline dataset: a list of column-name to value maps, one per row.
    #[serde(skip_serializing_if = "Vec::is_empty")]
    pub rows: Vec<HashMap<String, String>>,

    /// Optional raw inline dataset parsed server-side into rows per `format`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub data: Option<String>,

    /// The format of `data` (required when `data` is set).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub format: Option<LoadFeederFormat>,

    /// How a row is chosen each iteration.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub strategy: Option<LoadFeederStrategy>,
}

impl LoadFeeder {
    /// A feeder from an inline list of column-name to value rows.
    pub fn rows(rows: Vec<HashMap<String, String>>) -> Self {
        Self {
            rows,
            ..Self::default()
        }
    }

    /// A feeder from raw inline `data` parsed server-side as `format`.
    pub fn data(data: impl Into<String>, format: LoadFeederFormat) -> Self {
        Self {
            data: Some(data.into()),
            format: Some(format),
            ..Self::default()
        }
    }

    /// Set the row-selection strategy.
    pub fn strategy(mut self, strategy: LoadFeederStrategy) -> Self {
        self.strategy = Some(strategy);
        self
    }
}

/// Where a [`LoadCapture`] extracts its value from. Maps to the capture
/// `source` enum.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadCaptureSource {
    /// A JSONPath over the response body.
    BodyJsonpath,
    /// A response header value.
    Header,
    /// A regex over the response body string (capture group 1).
    BodyRegex,
}

/// A declarative cross-step capture / correlation rule: extracts a value from a
/// step's response and binds it to a variable name a later step in the same
/// iteration can reference via `$iteration.captured.<name>`. Best-effort. Maps
/// to the `LoadCapture` schema.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadCapture {
    /// The variable name later steps reference.
    pub name: String,

    /// Where to extract from.
    pub source: LoadCaptureSource,

    /// The JSONPath, header name, or regex driving the extraction.
    pub expression: String,

    /// Optional fallback value bound when extraction yields nothing.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub default_value: Option<String>,
}

impl LoadCapture {
    /// Create a capture binding `name` to the value extracted from `source` via
    /// `expression`.
    pub fn new(
        name: impl Into<String>,
        source: LoadCaptureSource,
        expression: impl Into<String>,
    ) -> Self {
        Self {
            name: name.into(),
            source,
            expression: expression.into(),
            default_value: None,
        }
    }

    /// Set the fallback value bound to the variable on no match.
    pub fn default_value(mut self, default_value: impl Into<String>) -> Self {
        self.default_value = Some(default_value.into());
        self
    }
}

/// How each iteration of a load scenario selects which steps to run. Maps to
/// the `stepSelection` enum.
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum LoadStepSelection {
    /// Run ALL steps in declared order (a multi-step user journey).
    Sequential,
    /// Run exactly ONE step per iteration chosen at random by weight.
    Weighted,
}

/// The load profile of a load scenario: EITHER an ordered list of [`LoadStage`]s
/// run in sequence, OR a single named [`LoadShape`] that expands into stages.
/// Maps to the `LoadProfile` schema.
///
/// Use [`LoadProfile::of`] to build from a list of stages, the convenience
/// constructors [`LoadProfile::constant`] / [`LoadProfile::linear`] for a single
/// VU stage, or [`LoadProfile::shaped`] for a named shape.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadProfile {
    /// Ordered stages run one after another. Omitted (empty) when a `shape` is
    /// used.
    #[serde(skip_serializing_if = "Vec::is_empty")]
    pub stages: Vec<LoadStage>,

    /// A named shape that expands server-side into stages. Use a shape OR
    /// `stages`, not both.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub shape: Option<LoadShape>,
}

impl LoadProfile {
    /// A profile from an explicit list of stages.
    pub fn of(stages: Vec<LoadStage>) -> Self {
        Self {
            stages,
            shape: None,
        }
    }

    /// A profile from a single named [`LoadShape`].
    pub fn shaped(shape: LoadShape) -> Self {
        Self {
            stages: Vec::new(),
            shape: Some(shape),
        }
    }

    /// A single VU stage holding `vus` virtual users for `duration_millis`.
    pub fn constant(vus: u32, duration_millis: u64) -> Self {
        Self::of(vec![LoadStage::vu_hold(vus, duration_millis)])
    }

    /// A single linear VU ramp from `start_vus` to `end_vus` over
    /// `duration_millis`.
    pub fn linear(start_vus: u32, end_vus: u32, duration_millis: u64) -> Self {
        Self::of(vec![LoadStage::vu_ramp(
            start_vus,
            end_vus,
            duration_millis,
            RampCurve::Linear,
        )])
    }

    /// Append a stage and return the profile.
    pub fn add_stage(mut self, stage: LoadStage) -> Self {
        self.stages.push(stage);
        self
    }
}

/// A single templated request step in a load scenario. Maps to the `LoadStep`
/// schema.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadStep {
    /// The templated request to fire each iteration.
    pub request: HttpRequest,

    /// Optional inter-step pause (a [`Delay`]).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub think_time: Option<Delay>,

    /// Optional cross-step capture rules applied to this step's response. Each
    /// binds an extracted value to a variable name visible to SUBSEQUENT steps
    /// in the same iteration.
    #[serde(skip_serializing_if = "Vec::is_empty")]
    pub captures: Vec<LoadCapture>,

    /// Relative selection weight, used only when the scenario's
    /// `stepSelection` is `WEIGHTED`. Must be > 0 when `WEIGHTED`; ignored
    /// under the default `SEQUENTIAL` mode.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub weight: Option<f64>,
}

impl LoadStep {
    /// Create a step from a request matcher/template.
    pub fn new(request: HttpRequest) -> Self {
        Self {
            request,
            think_time: None,
            captures: Vec::new(),
            weight: None,
        }
    }

    /// Set the inter-step pause.
    pub fn think_time(mut self, delay: Delay) -> Self {
        self.think_time = Some(delay);
        self
    }

    /// Append a cross-step capture rule applied to this step's response.
    pub fn capture(mut self, capture: LoadCapture) -> Self {
        self.captures.push(capture);
        self
    }

    /// Set the relative selection weight (used only under `WEIGHTED`
    /// `stepSelection`).
    pub fn weight(mut self, weight: f64) -> Self {
        self.weight = Some(weight);
        self
    }
}

/// An API-driven load scenario: ordered templated steps driven at a target
/// concurrency. Maps to the `LoadScenario` schema (the body of
/// `PUT /mockserver/loadScenario`, which registers the scenario in the
/// registry without running it). The unique [`name`](LoadScenario::name) is the
/// registry key used by `start`/`stop` and the per-scenario `GET`/`DELETE`
/// endpoints.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct LoadScenario {
    /// Human-readable scenario name.
    pub name: String,

    /// Template engine for per-iteration rendering — `"VELOCITY"` (default) or
    /// `"MUSTACHE"`. (JavaScript is rejected for load steps.)
    #[serde(skip_serializing_if = "Option::is_none")]
    pub template_type: Option<String>,

    /// Optional hard cap on the total number of requests dispatched.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub max_requests: Option<u64>,

    /// Optional delay (milliseconds) applied between a `start` request being
    /// accepted and the scenario actually beginning to drive load. Honoured by
    /// `PUT /mockserver/loadScenario/start`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub start_delay_millis: Option<u64>,

    /// Optional in-run pass/fail thresholds; the run carries a PASS verdict iff
    /// all hold, FAIL otherwise. Empty/omitted means no verdict is computed.
    #[serde(skip_serializing_if = "Vec::is_empty")]
    pub thresholds: Vec<LoadThreshold>,

    /// When true, a FAIL verdict aborts the run early. Default false (omitted).
    #[serde(skip_serializing_if = "std::ops::Not::not")]
    pub abort_on_fail: bool,

    /// Suppress `abort_on_fail` for the first N milliseconds of the run so noisy
    /// startup samples cannot trigger a premature abort.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub abort_grace_millis: Option<u64>,

    /// Optional adaptive iteration pacing (closed-model VU loop only).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub pacing: Option<LoadPacing>,

    /// Optional parameterized test data (a data feeder).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub feeder: Option<LoadFeeder>,

    /// How each iteration selects which steps to run — `SEQUENTIAL` (default,
    /// omitted) or `WEIGHTED`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub step_selection: Option<LoadStepSelection>,

    /// The ramp profile.
    pub profile: LoadProfile,

    /// Ordered list of request steps fired in sequence each iteration (max 50).
    pub steps: Vec<LoadStep>,
}

impl LoadScenario {
    /// Create a scenario with the given name, profile and steps.
    pub fn new(name: impl Into<String>, profile: LoadProfile, steps: Vec<LoadStep>) -> Self {
        Self {
            name: name.into(),
            template_type: None,
            max_requests: None,
            start_delay_millis: None,
            thresholds: Vec::new(),
            abort_on_fail: false,
            abort_grace_millis: None,
            pacing: None,
            feeder: None,
            step_selection: None,
            profile,
            steps,
        }
    }

    /// Add an in-run pass/fail threshold.
    pub fn threshold(mut self, threshold: LoadThreshold) -> Self {
        self.thresholds.push(threshold);
        self
    }

    /// Set whether a FAIL verdict aborts the run early.
    pub fn abort_on_fail(mut self, abort_on_fail: bool) -> Self {
        self.abort_on_fail = abort_on_fail;
        self
    }

    /// Set the abort grace window (milliseconds) for `abort_on_fail`.
    pub fn abort_grace_millis(mut self, abort_grace_millis: u64) -> Self {
        self.abort_grace_millis = Some(abort_grace_millis);
        self
    }

    /// Set the adaptive iteration pacing.
    pub fn pacing(mut self, pacing: LoadPacing) -> Self {
        self.pacing = Some(pacing);
        self
    }

    /// Set the parameterized test data feeder.
    pub fn feeder(mut self, feeder: LoadFeeder) -> Self {
        self.feeder = Some(feeder);
        self
    }

    /// Set how each iteration selects which steps to run.
    pub fn step_selection(mut self, step_selection: LoadStepSelection) -> Self {
        self.step_selection = Some(step_selection);
        self
    }

    /// Set the template engine (`"VELOCITY"` or `"MUSTACHE"`).
    pub fn template_type(mut self, template_type: impl Into<String>) -> Self {
        self.template_type = Some(template_type.into());
        self
    }

    /// Set the hard cap on total requests dispatched.
    pub fn max_requests(mut self, max_requests: u64) -> Self {
        self.max_requests = Some(max_requests);
        self
    }

    /// Set the delay (milliseconds) before the scenario begins driving load
    /// once started.
    pub fn start_delay_millis(mut self, start_delay_millis: u64) -> Self {
        self.start_delay_millis = Some(start_delay_millis);
        self
    }
}

// ---------------------------------------------------------------------------
// SLO verdicts (PUT /mockserver/verifySLO)
// ---------------------------------------------------------------------------

/// A single service-level objective over the recorded SLI samples. Maps to the
/// `SloObjective` schema.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct SloObjective {
    /// The indicator to evaluate — one of `LATENCY_P50`, `LATENCY_P95`,
    /// `LATENCY_P99`, `ERROR_RATE`.
    pub sli: String,

    /// How the observed value is compared to the threshold — one of
    /// `LESS_THAN`, `LESS_THAN_OR_EQUAL`, `GREATER_THAN`,
    /// `GREATER_THAN_OR_EQUAL`.
    pub comparator: String,

    /// The objective threshold (milliseconds for latency SLIs, a 0.0–1.0
    /// fraction for `ERROR_RATE`).
    pub threshold: f64,

    /// Which recorded traffic to evaluate — `"FORWARD"` (default) or
    /// `"INBOUND"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub scope: Option<String>,
}

impl SloObjective {
    /// Create an objective.
    pub fn new(sli: impl Into<String>, comparator: impl Into<String>, threshold: f64) -> Self {
        Self {
            sli: sli.into(),
            comparator: comparator.into(),
            threshold,
            scope: None,
        }
    }

    /// Set the evaluation scope (`"FORWARD"` or `"INBOUND"`).
    pub fn scope(mut self, scope: impl Into<String>) -> Self {
        self.scope = Some(scope.into());
        self
    }
}

/// The time window of an SLO evaluation. Maps to the `SloCriteria.window`
/// object.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct SloWindow {
    /// `"LOOKBACK"` (default) or `"EXPLICIT"`.
    #[serde(rename = "type", skip_serializing_if = "Option::is_none")]
    pub window_type: Option<String>,

    /// LOOKBACK: window length ending now.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub lookback_millis: Option<u64>,

    /// EXPLICIT: window start in epoch milliseconds.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub from_epoch_millis: Option<u64>,

    /// EXPLICIT: window end in epoch milliseconds.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub to_epoch_millis: Option<u64>,
}

impl SloWindow {
    /// A LOOKBACK window of `millis` ending now.
    pub fn lookback(millis: u64) -> Self {
        Self {
            window_type: Some("LOOKBACK".to_string()),
            lookback_millis: Some(millis),
            ..Default::default()
        }
    }

    /// An EXPLICIT window between two epoch-millisecond bounds.
    pub fn explicit(from_epoch_millis: u64, to_epoch_millis: u64) -> Self {
        Self {
            window_type: Some("EXPLICIT".to_string()),
            from_epoch_millis: Some(from_epoch_millis),
            to_epoch_millis: Some(to_epoch_millis),
            ..Default::default()
        }
    }
}

/// A named set of service-level objectives over a time window. Maps to the
/// `SloCriteria` schema (the body of `PUT /mockserver/verifySLO`).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct SloCriteria {
    /// Human-readable criteria name, echoed back in the verdict.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,

    /// The time window to evaluate over.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub window: Option<SloWindow>,

    /// Minimum samples required in the window; below this the verdict is
    /// INCONCLUSIVE.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub minimum_sample_count: Option<u64>,

    /// Optional list of upstream hosts to restrict the evaluation to.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub upstream_hosts: Option<Vec<String>>,

    /// The objectives (the verdict is the logical AND of all of them).
    pub objectives: Vec<SloObjective>,
}

impl SloCriteria {
    /// Create criteria from a set of objectives.
    pub fn new(objectives: Vec<SloObjective>) -> Self {
        Self {
            name: None,
            window: None,
            minimum_sample_count: None,
            upstream_hosts: None,
            objectives,
        }
    }

    /// Set the criteria name.
    pub fn name(mut self, name: impl Into<String>) -> Self {
        self.name = Some(name.into());
        self
    }

    /// Set the evaluation window.
    pub fn window(mut self, window: SloWindow) -> Self {
        self.window = Some(window);
        self
    }

    /// Set the minimum sample count.
    pub fn minimum_sample_count(mut self, count: u64) -> Self {
        self.minimum_sample_count = Some(count);
        self
    }

    /// Restrict the evaluation to the given upstream hosts.
    pub fn upstream_hosts(mut self, hosts: Vec<String>) -> Self {
        self.upstream_hosts = Some(hosts);
        self
    }
}

/// The evaluated result of a single objective. Maps to the `SloObjectiveResult`
/// schema.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct SloObjectiveResult {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub sli: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub comparator: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub threshold: Option<f64>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub observed_value: Option<f64>,
    /// `PASS`, `FAIL` or `INCONCLUSIVE`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub result: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub detail: Option<String>,
}

/// The overall verdict of an SLO evaluation. Maps to the `SloVerdict` schema —
/// the response of `PUT /mockserver/verifySLO`.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct SloVerdict {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,
    /// `PASS`, `FAIL` or `INCONCLUSIVE`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub result: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub window_from_epoch_millis: Option<u64>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub window_to_epoch_millis: Option<u64>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub sample_count: Option<u64>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub objective_results: Vec<SloObjectiveResult>,
}

impl SloVerdict {
    /// Whether the overall verdict is `PASS`.
    pub fn is_pass(&self) -> bool {
        self.result.as_deref() == Some("PASS")
    }

    /// Whether the overall verdict is `FAIL`.
    pub fn is_fail(&self) -> bool {
        self.result.as_deref() == Some("FAIL")
    }

    /// Whether the overall verdict is `INCONCLUSIVE`.
    pub fn is_inconclusive(&self) -> bool {
        self.result.as_deref() == Some("INCONCLUSIVE")
    }
}

// ---------------------------------------------------------------------------
// Preemption (PUT/GET/DELETE /mockserver/preemption)
// ---------------------------------------------------------------------------

/// Preemption simulation parameters (all fields optional). Maps to the
/// `PreemptionRequest` schema (the body of `PUT /mockserver/preemption`).
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct PreemptionRequest {
    /// How draining is signalled — `"reject503"`, `"goaway"` or `"both"`
    /// (default).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub mode: Option<String>,

    /// How long in-flight requests are allowed to drain (clamped server-side).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub drain_millis: Option<u64>,

    /// Auto-uncordon after this many milliseconds (dead-man's switch); `0`
    /// (default) means no auto-uncordon.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub ttl_millis: Option<u64>,

    /// HTTP/2 GOAWAY `last_stream_id` to advertise; `-1` (default) lets the
    /// server choose.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub last_stream_id: Option<i64>,
}

impl PreemptionRequest {
    /// An empty request (server defaults: mode "both", default drain, no TTL).
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the signalling mode (`"reject503"`, `"goaway"` or `"both"`).
    pub fn mode(mut self, mode: impl Into<String>) -> Self {
        self.mode = Some(mode.into());
        self
    }

    /// Set the drain window in milliseconds.
    pub fn drain_millis(mut self, millis: u64) -> Self {
        self.drain_millis = Some(millis);
        self
    }

    /// Set the auto-uncordon TTL in milliseconds.
    pub fn ttl_millis(mut self, millis: u64) -> Self {
        self.ttl_millis = Some(millis);
        self
    }

    /// Set the HTTP/2 GOAWAY `last_stream_id` to advertise.
    pub fn last_stream_id(mut self, id: i64) -> Self {
        self.last_stream_id = Some(id);
        self
    }
}

/// The current cordon/drain status of the server. Maps to the
/// `PreemptionStatus` schema — the response of `PUT`/`GET /mockserver/preemption`.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "camelCase")]
pub struct PreemptionStatus {
    /// `"inactive"`, `"draining"` or `"drained"`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub state: Option<String>,

    /// Number of requests currently in flight.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub in_flight: Option<u64>,

    /// Milliseconds left in the drain window.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub drain_remaining_millis: Option<u64>,

    /// Active signalling mode (omitted when inactive).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub mode: Option<String>,
}

// ---------------------------------------------------------------------------
// Service chaos (PUT /mockserver/serviceChaos)
// ---------------------------------------------------------------------------

/// An HTTP chaos / fault-injection profile for a host or expectation. Maps to
/// the `HttpChaosProfile` schema. Captures the commonly-used fields; the model
/// carries an `extra` map for any additional server-supported keys.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct HttpChaosProfile {
    /// HTTP error status code to return instead of the real response.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub error_status: Option<u16>,

    /// Probability (0.0–1.0) that a request triggers the error.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub error_probability: Option<f64>,

    /// Injected latency (a [`Delay`]).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub latency: Option<Delay>,

    /// When true, drops the TCP connection without responding.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub connection_drop: Option<bool>,

    /// Fixed seed for deterministic probabilistic outcomes.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub seed: Option<i64>,

    /// Literal `Retry-After` header value returned with an injected error.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub retry_after: Option<String>,

    /// Probability (0.0–1.0) of dropping the TCP connection without responding.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub drop_connection_probability: Option<f64>,

    /// Let the first N requests succeed before chaos becomes active.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub succeed_first: Option<i64>,

    /// Number of requests to fail once chaos is active.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub fail_request_count: Option<i64>,

    /// Time-based outage: chaos activates this many ms after first match.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub outage_after_millis: Option<i64>,

    /// Time-based outage: chaos stays active this many ms then self-heals.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub outage_duration_millis: Option<i64>,

    /// Keep only this leading fraction (0.0–1.0) of the response body.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub truncate_body_at_fraction: Option<f64>,

    /// Corrupt the response body so it fails to parse.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub malformed_body: Option<bool>,

    /// Dribble the response body in chunks of this many bytes.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub slow_response_chunk_size: Option<i64>,

    /// Delay between slow-response chunks.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub slow_response_chunk_delay: Option<Delay>,

    /// Shared quota counter key.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub quota_name: Option<String>,

    /// Max requests allowed per quota window.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub quota_limit: Option<i64>,

    /// Quota fixed-window length in milliseconds.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub quota_window_millis: Option<i64>,

    /// Status returned when the quota is exceeded (default 429).
    #[serde(skip_serializing_if = "Option::is_none")]
    pub quota_error_status: Option<u16>,

    /// Ramp error/drop probabilities linearly over this many ms from first match.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub degradation_ramp_millis: Option<i64>,

    /// Rewrite the response body as a GraphQL error envelope.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub graphql_errors: Option<bool>,

    /// Message in `errors[0].message` of the GraphQL error envelope.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub graphql_error_message: Option<String>,

    /// Value for `errors[0].extensions.code`.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub graphql_error_code: Option<String>,

    /// Whether `data` is null (default true) in the GraphQL error envelope.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub graphql_nullify_data: Option<bool>,

    /// Any additional fields the server supports that are not modelled above.
    #[serde(flatten)]
    pub extra: HashMap<String, serde_json::Value>,
}

impl HttpChaosProfile {
    /// Create an empty chaos profile.
    pub fn new() -> Self {
        Self::default()
    }

    /// Set the error status code returned on fault.
    pub fn error_status(mut self, status: u16) -> Self {
        self.error_status = Some(status);
        self
    }

    /// Set the probability (0.0–1.0) of triggering the error.
    pub fn error_probability(mut self, probability: f64) -> Self {
        self.error_probability = Some(probability);
        self
    }

    /// Set the injected latency.
    pub fn latency(mut self, latency: Delay) -> Self {
        self.latency = Some(latency);
        self
    }

    /// Drop the TCP connection without responding.
    pub fn connection_drop(mut self, drop: bool) -> Self {
        self.connection_drop = Some(drop);
        self
    }

    /// Set the deterministic seed.
    pub fn seed(mut self, seed: i64) -> Self {
        self.seed = Some(seed);
        self
    }
}

// ---------------------------------------------------------------------------
// Chaos experiment (PUT /mockserver/chaosExperiment)
// ---------------------------------------------------------------------------

/// A single stage of a chaos experiment. Maps to a `ChaosExperiment.stages[]`
/// entry.
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ChaosStage {
    /// How long this stage runs before advancing (max 86_400_000 = 24h).
    pub duration_millis: u64,

    /// Map of host -> chaos profile to apply during this stage.
    pub profiles: HashMap<String, HttpChaosProfile>,
}

impl ChaosStage {
    /// Create a stage running for `duration_millis`.
    pub fn new(duration_millis: u64) -> Self {
        Self {
            duration_millis,
            profiles: HashMap::new(),
        }
    }

    /// Add a host -> chaos profile to apply during the stage.
    pub fn profile(mut self, host: impl Into<String>, profile: HttpChaosProfile) -> Self {
        self.profiles.insert(host.into(), profile);
        self
    }
}

/// A scheduled multi-stage chaos experiment definition. Maps to the
/// `ChaosExperiment` schema (the body of `PUT /mockserver/chaosExperiment`).
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(rename_all = "camelCase")]
pub struct ChaosExperiment {
    /// Human-readable experiment name.
    #[serde(skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,

    /// Whether to loop back to stage 0 after the last stage completes (default
    /// false). Serialized as `loop` on the wire.
    #[serde(rename = "loop", skip_serializing_if = "Option::is_none")]
    pub loop_back: Option<bool>,

    /// The ordered sequence of stages.
    pub stages: Vec<ChaosStage>,
}

impl ChaosExperiment {
    /// Create an experiment from an ordered list of stages.
    pub fn new(stages: Vec<ChaosStage>) -> Self {
        Self {
            name: None,
            loop_back: None,
            stages,
        }
    }

    /// Set the experiment name.
    pub fn name(mut self, name: impl Into<String>) -> Self {
        self.name = Some(name.into());
        self
    }

    /// Set whether the experiment loops back to the first stage.
    pub fn loop_back(mut self, loop_back: bool) -> Self {
        self.loop_back = Some(loop_back);
        self
    }
}

// ---------------------------------------------------------------------------
// Tests
// ---------------------------------------------------------------------------

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_grpc_services_deserialize_from_server_wire_shape() {
        // Mirrors the JSON array produced by `PUT /mockserver/grpc/services`
        // in mockserver-core HttpState.java (camelCase keys, full type names).
        let wire = r#"[
            {
                "name": "helloworld.Greeter",
                "methods": [
                    {
                        "name": "SayHello",
                        "inputType": "helloworld.HelloRequest",
                        "outputType": "helloworld.HelloReply",
                        "clientStreaming": false,
                        "serverStreaming": false
                    },
                    {
                        "name": "LotsOfReplies",
                        "inputType": "helloworld.HelloRequest",
                        "outputType": "helloworld.HelloReply",
                        "clientStreaming": false,
                        "serverStreaming": true
                    }
                ]
            }
        ]"#;

        let services: Vec<GrpcService> = serde_json::from_str(wire).unwrap();
        assert_eq!(services.len(), 1);
        let svc = &services[0];
        assert_eq!(svc.name, "helloworld.Greeter");
        assert_eq!(svc.methods.len(), 2);

        let unary = &svc.methods[0];
        assert_eq!(unary.name, "SayHello");
        assert_eq!(unary.input_type, "helloworld.HelloRequest");
        assert_eq!(unary.output_type, "helloworld.HelloReply");
        assert!(!unary.client_streaming);
        assert!(!unary.server_streaming);

        let server_stream = &svc.methods[1];
        assert_eq!(server_stream.name, "LotsOfReplies");
        assert!(!server_stream.client_streaming);
        assert!(server_stream.server_streaming);
    }

    #[test]
    fn test_grpc_method_serializes_with_camel_case_keys() {
        let method = GrpcMethod {
            name: "BidiChat".into(),
            input_type: "chat.Message".into(),
            output_type: "chat.Message".into(),
            client_streaming: true,
            server_streaming: true,
        };
        let value = serde_json::to_value(&method).unwrap();
        assert_eq!(value["name"], "BidiChat");
        assert_eq!(value["inputType"], "chat.Message");
        assert_eq!(value["outputType"], "chat.Message");
        assert_eq!(value["clientStreaming"], true);
        assert_eq!(value["serverStreaming"], true);
    }

    #[test]
    fn test_grpc_services_empty_array() {
        let services: Vec<GrpcService> = serde_json::from_str("[]").unwrap();
        assert!(services.is_empty());
    }

    #[test]
    fn test_grpc_service_round_trips() {
        let original = GrpcService {
            name: "helloworld.Greeter".into(),
            methods: vec![GrpcMethod {
                name: "SayHello".into(),
                input_type: "helloworld.HelloRequest".into(),
                output_type: "helloworld.HelloReply".into(),
                client_streaming: false,
                server_streaming: false,
            }],
        };
        let json = serde_json::to_string(&original).unwrap();
        let parsed: GrpcService = serde_json::from_str(&json).unwrap();
        assert_eq!(original, parsed);
    }
}