custom_requests.rs

use schemars::JsonSchema;
use serde::{Deserialize, Serialize};

/// Schema descriptor for a single custom method, produced by the
/// `#[custom_methods]` macro's generated `custom_method_schemas()` function.
///
/// `params_schema` / `response_schema` hold `$ref` pointers or inline schemas
/// produced by `SchemaGenerator::subschema_for`. All referenced types are
/// collected in the generator's `$defs` map.
///
/// `params_type_name` / `response_type_name` carry the Rust struct name so the
/// binary can key `$defs` entries and annotate them with `x-method` / `x-side`.
#[derive(Debug, Serialize)]
pub struct CustomMethodSchema {
    pub method: String,
    pub params_schema: Option<schemars::Schema>,
    pub params_type_name: Option<String>,
    pub response_schema: Option<schemars::Schema>,
    pub response_type_name: Option<String>,
}

/// Add an extension to an active session.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct AddExtensionRequest {
    pub session_id: String,
    /// Extension configuration (see ExtensionConfig variants: Stdio, StreamableHttp, Builtin, Platform).
    pub config: serde_json::Value,
}

/// Remove an extension from an active session.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct RemoveExtensionRequest {
    pub session_id: String,
    pub name: String,
}

/// List all tools available in a session.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct GetToolsRequest {
    pub session_id: String,
}

#[derive(Debug, Serialize, JsonSchema)]
pub struct GetToolsResponse {
    /// Array of tool info objects with `name`, `description`, `parameters`, and optional `permission`.
    pub tools: Vec<serde_json::Value>,
}

/// Read a resource from an extension.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct ReadResourceRequest {
    pub session_id: String,
    pub uri: String,
    pub extension_name: String,
}

#[derive(Debug, Serialize, JsonSchema)]
pub struct ReadResourceResponse {
    /// The resource result from the extension (MCP ReadResourceResult).
    pub result: serde_json::Value,
}

/// Update the working directory for a session.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct UpdateWorkingDirRequest {
    pub session_id: String,
    pub working_dir: String,
}

/// Get a session by ID.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct GetSessionRequest {
    pub session_id: String,
    #[serde(default)]
    pub include_messages: bool,
}

/// Get a session response.
#[derive(Debug, Serialize, JsonSchema)]
pub struct GetSessionResponse {
    /// The session object with id, name, working_dir, timestamps, tokens, etc.
    pub session: serde_json::Value,
}

/// Delete a session.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct DeleteSessionRequest {
    pub session_id: String,
}

/// Export a session as a JSON string.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct ExportSessionRequest {
    pub session_id: String,
}

#[derive(Debug, Serialize, JsonSchema)]
pub struct ExportSessionResponse {
    pub data: String,
}

/// Import a session from a JSON string.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct ImportSessionRequest {
    pub data: String,
}

#[derive(Debug, Serialize, JsonSchema)]
pub struct ImportSessionResponse {
    /// The imported session object.
    pub session: serde_json::Value,
}

/// List configured extensions and any warnings.
#[derive(Debug, Serialize, JsonSchema)]
pub struct GetExtensionsResponse {
    /// Array of ExtensionEntry objects with `enabled` flag and config details.
    pub extensions: Vec<serde_json::Value>,
    pub warnings: Vec<String>,
}

/// Apply system prompt instructions to an active session.
/// Equivalent to POST /agent/update_from_session in the HTTP API.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct SetSessionInstructionsRequest {
    pub session_id: String,
    /// Instructions to prepend to the agent's system prompt (e.g. channel context, persona).
    pub instructions: String,
}

/// Health check.
#[derive(Debug, Deserialize, JsonSchema)]
pub struct HealthRequest {}

#[derive(Debug, Serialize, JsonSchema)]
pub struct HealthResponse {
    pub status: String,
}

/// Empty success response for operations that return no data.
#[derive(Debug, Serialize, JsonSchema)]
pub struct EmptyResponse {}