pmcp-server 0.2.3

//! Protocol compliance testing tool.
//!
//! Wraps `mcp_tester::ServerTester::run_compliance_tests()` to let users
//! test any MCP server for protocol compliance via a single tool call.

use async_trait::async_trait;
use pmcp::types::ToolInfo;
use pmcp::ToolHandler;
use serde::Deserialize;
use serde_json::{json, Value};

use super::{create_tester, default_timeout, internal_err};

/// Input parameters for the `test_check` tool.
#[derive(Deserialize)]
struct TestCheckInput {
    /// MCP server URL to test.
    url: String,
    /// Enable strict compliance mode (warnings become failures).
    #[serde(default)]
    strict: bool,
    /// Timeout in seconds (default: 30).
    #[serde(default = "default_timeout")]
    timeout: u64,
}

/// Protocol compliance testing tool.
///
/// Connects to a remote MCP server and runs the full compliance test suite,
/// returning a structured `TestReport` as JSON.
pub struct TestCheckTool;

#[async_trait]
impl ToolHandler for TestCheckTool {
    async fn handle(&self, args: Value, _extra: pmcp::RequestHandlerExtra) -> pmcp::Result<Value> {
        let params: TestCheckInput = serde_json::from_value(args)
            .map_err(|e| pmcp::Error::validation(format!("Invalid arguments: {e}")))?;

        let mut tester = create_tester(&params.url, params.timeout)?;

        let report = tester
            .run_conformance_tests(params.strict, None)
            .await
            .map_err(internal_err)?;

        serde_json::to_value(&report).map_err(internal_err)
    }

    fn metadata(&self) -> Option<ToolInfo> {
        Some(ToolInfo::new(
            "test_check",
            Some("Run MCP protocol compliance checks against a remote server".to_string()),
            json!({
                "type": "object",
                "properties": {
                    "url": {
                        "type": "string",
                        "description": "MCP server URL to test"
                    },
                    "strict": {
                        "type": "boolean",
                        "description": "Enable strict compliance mode (warnings become failures)",
                        "default": false
                    },
                    "timeout": {
                        "type": "integer",
                        "description": "Timeout in seconds",
                        "default": 30
                    }
                },
                "required": ["url"]
            }),
        ))
    }
}