rust acp client for extension methods (#8227)

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>

crates/goose-sdk/src/custom_requests.rs

@@ -0,0 +1,155 @@
+use sacp::{JsonRpcRequest, JsonRpcResponse};
+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, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/extensions/add", response = EmptyResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct AddExtensionRequest {
+    pub session_id: String,
+    /// Extension configuration (see ExtensionConfig variants: Stdio, StreamableHttp, Builtin, Platform).
+    #[serde(default)]
+    pub config: serde_json::Value,
+}
+
+/// Remove an extension from an active session.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/extensions/remove", response = EmptyResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct RemoveExtensionRequest {
+    pub session_id: String,
+    pub name: String,
+}
+
+/// List all tools available in a session.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/tools", response = GetToolsResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct GetToolsRequest {
+    pub session_id: String,
+}
+
+/// Tools response.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcResponse)]
+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, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/resource/read", response = ReadResourceResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct ReadResourceRequest {
+    pub session_id: String,
+    pub uri: String,
+    pub extension_name: String,
+}
+
+/// Resource read response.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcResponse)]
+pub struct ReadResourceResponse {
+    /// The resource result from the extension (MCP ReadResourceResult).
+    #[serde(default)]
+    pub result: serde_json::Value,
+}
+
+/// Update the working directory for a session.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/working_dir/update", response = EmptyResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct UpdateWorkingDirRequest {
+    pub session_id: String,
+    pub working_dir: String,
+}
+
+/// Get a session by ID.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "session/get", response = GetSessionResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct GetSessionRequest {
+    pub session_id: String,
+    #[serde(default)]
+    pub include_messages: bool,
+}
+
+/// Get a session response.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcResponse)]
+pub struct GetSessionResponse {
+    /// The session object with id, name, working_dir, timestamps, tokens, etc.
+    #[serde(default)]
+    pub session: serde_json::Value,
+}
+
+/// Delete a session.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "session/delete", response = EmptyResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct DeleteSessionRequest {
+    pub session_id: String,
+}
+
+/// Export a session as a JSON string.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/session/export", response = ExportSessionResponse)]
+#[serde(rename_all = "camelCase")]
+pub struct ExportSessionRequest {
+    pub session_id: String,
+}
+
+/// Export session response.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcResponse)]
+pub struct ExportSessionResponse {
+    pub data: String,
+}
+
+/// Import a session from a JSON string.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/session/import", response = ImportSessionResponse)]
+pub struct ImportSessionRequest {
+    pub data: String,
+}
+
+/// Import session response.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcResponse)]
+pub struct ImportSessionResponse {
+    /// The imported session object.
+    #[serde(default)]
+    pub session: serde_json::Value,
+}
+
+/// List configured extensions and any warnings.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcRequest)]
+#[request(method = "_goose/config/extensions", response = GetExtensionsResponse)]
+pub struct GetExtensionsRequest {}
+
+/// List configured extensions and any warnings.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcResponse)]
+pub struct GetExtensionsResponse {
+    /// Array of ExtensionEntry objects with `enabled` flag and config details.
+    pub extensions: Vec<serde_json::Value>,
+    pub warnings: Vec<String>,
+}
+
+/// Empty success response for operations that return no data.
+#[derive(Debug, Default, Clone, Serialize, Deserialize, JsonSchema, JsonRpcResponse)]
+pub struct EmptyResponse {}

crates/goose-sdk/src/lib.rs

@@ -0,0 +1 @@
+pub mod custom_requests;