use rmcp::{
    RmcpError,
    model::CallToolRequestParam,
    service::ServiceExt,
    transport::{ConfigureCommandExt, TokioChildProcess},
};
use tokio::process::Command;

/**
 * Complete Rust MCP Client Example
 * 
 * This client demonstrates how to:
 * 1. Connect to an MCP server using child process transport
 * 2. List available tools
 * 3. Call various calculator tools
 * 4. Handle responses from the server
 */

#[tokio::main]
async fn main() -> Result<(), RmcpError> {
    // Assume the server is a sibling project named "calculator-server" in the same directory
    let server_dir = std::path::Path::new(env!("CARGO_MANIFEST_DIR"))
        .parent()
        .expect("failed to locate workspace root")
        .join("calculator-server");

    let client = ()
        .serve(
            TokioChildProcess::new(Command::new("cargo").configure(|cmd| {
                cmd.arg("run").current_dir(server_dir);
            }))
            .map_err(RmcpError::transport_creation::<TokioChildProcess>)?,
        )
        .await?;

    // Initialize
    let server_info = client.peer_info();
    println!("Server info: {:?}", server_info);

    // List tools
    let tools = client.list_tools(Default::default()).await?;
    println!("Available tools: {:?}", tools);

    // Call add tool with arguments = {"a": 3, "b": 2}
    let a = 3;
    let b = 2;
    let tool_result = client
        .call_tool(CallToolRequestParam {
            name: "add".into(),
            arguments: serde_json::json!({ "a": a, "b": b }).as_object().cloned(),
        })
        .await?;
    println!("Result of {:?} + {:?}: {:?}", a, b, tool_result);

    client.cancel().await?;
    Ok(())
}