use serde::{Deserialize, Serialize};
use std::collections::HashMap;
#[derive(Debug, Serialize, Deserialize)]
pub struct McpRequest {
pub jsonrpc: String,
pub id: Option<serde_json::Value>,
pub method: String,
pub params: Option<serde_json::Value>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct McpResponse {
pub jsonrpc: String,
pub id: Option<serde_json::Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub result: Option<serde_json::Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub error: Option<McpError>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct McpError {
pub code: i32,
pub message: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub data: Option<serde_json::Value>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct Tool {
pub name: String,
pub description: String,
pub input_schema: serde_json::Value,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct ListToolsResult {
pub tools: Vec<Tool>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct CallToolParams {
pub name: String,
pub arguments: HashMap<String, serde_json::Value>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct CallToolResult {
pub content: Vec<Content>,
#[serde(skip_serializing_if = "Option::is_none")]
pub is_error: Option<bool>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct Content {
#[serde(rename = "type")]
pub content_type: String,
pub text: String,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct DatabaseInfo {
pub database_type: String,
pub version: String,
pub server_info: Option<String>,
pub connection_url: String,
pub current_user: Option<String>,
pub current_schema: Option<String>,
pub available_schemas: Option<Vec<String>>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct TableInfo {
pub table_name: String,
pub table_schema: Option<String>,
pub table_type: String,
pub columns: Vec<ColumnInfo>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct ColumnInfo {
pub column_name: String,
pub data_type: String,
pub is_nullable: bool,
pub column_default: Option<String>,
pub is_primary_key: bool,
pub is_auto_increment: bool,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct QueryResult {
pub rows_affected: Option<u64>,
pub rows: Vec<HashMap<String, serde_json::Value>>,
pub execution_time_ms: u128,
}
