use indexmap::IndexMap;
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema)]
#[schemars(rename = "mcp.tool.Tool")]
pub struct Tool {
pub name: String,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub title: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub description: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub icons: Option<Vec<super::super::shared::Icon>>,
#[serde(rename = "inputSchema")]
pub input_schema: ToolSchemaObject,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
#[serde(rename = "outputSchema")]
pub output_schema: Option<ToolSchemaObject>,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub annotations: Option<super::ToolAnnotations>,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub execution: Option<super::ToolExecution>,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub _meta: Option<IndexMap<String, serde_json::Value>>,
}
impl Tool {
pub fn tool_key(&self, connection_tool_key: &str) -> String {
format!("{connection_tool_key}-{}", self.name)
}
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, JsonSchema)]
#[serde(rename_all = "lowercase")]
#[schemars(rename = "mcp.tool.ToolSchemaType")]
pub enum ToolSchemaType {
Object,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema)]
#[schemars(rename = "mcp.tool.ToolSchemaObject")]
pub struct ToolSchemaObject {
pub r#type: ToolSchemaType,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub properties: Option<IndexMap<String, serde_json::Value>>,
#[serde(skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub required: Option<Vec<String>>,
#[serde(flatten)]
pub extra: IndexMap<String, serde_json::Value>,
}