use serde::{Deserialize, Serialize};
use serde_json::Value;
use std::io::Write;
#[derive(Debug, Deserialize)]
pub struct IncomingMessage {
#[allow(dead_code)] pub jsonrpc: String,
#[serde(default)]
pub id: Option<Value>,
#[serde(default)]
pub method: Option<String>,
#[serde(default)]
pub params: Option<Value>,
}
#[derive(Debug, Serialize)]
pub struct Response {
pub jsonrpc: &'static str,
#[serde(skip_serializing_if = "Option::is_none")]
pub result: Option<Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub error: Option<RpcError>,
pub id: Value,
}
#[derive(Debug, Serialize)]
pub struct RpcError {
pub code: i64,
pub message: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub data: Option<Value>,
}
pub fn success_response(id: Value, result: Value) -> Response {
Response {
jsonrpc: "2.0",
result: Some(result),
error: None,
id,
}
}
pub fn method_not_found(id: Value, method: &str) -> Response {
Response {
jsonrpc: "2.0",
result: None,
error: Some(RpcError {
code: -32601,
message: format!("Method not found: {method}"),
data: None,
}),
id,
}
}
pub fn parse_error() -> Response {
Response {
jsonrpc: "2.0",
result: None,
error: Some(RpcError {
code: -32700,
message: "Parse error".to_string(),
data: None,
}),
id: Value::Null,
}
}
pub fn send_response(stdout: &mut impl Write, response: &Response) {
match serde_json::to_string(response) {
Ok(json) => {
if let Err(e) = writeln!(stdout, "{json}") {
eprintln!("[mcp-server] Failed to write response: {e}");
}
if let Err(e) = stdout.flush() {
eprintln!("[mcp-server] Failed to flush stdout: {e}");
}
}
Err(e) => {
eprintln!("[mcp-server] Failed to serialize response: {e}");
}
}
}