mcp-stdio-proxy 0.2.9

//! Protocol Detection (Re-export Layer)
//!
//! This module re-exports protocol detection functions and provides
//! a convenient combined detection function.
//!
//! Detection logic: If SSE detected → Sse, else → Stream (default fallback)

// Re-export the detection functions
pub use mcp_sse_proxy::is_sse_with_headers;

use crate::model::McpProtocol;
use anyhow::Result;
use log::info;
use std::collections::HashMap;

/// Automatically detect the MCP service protocol type, with optional custom headers
///
/// Detection logic:
/// 1. First try to detect SSE protocol (GET /sse returns text/event-stream)
/// 2. If not SSE, default to Streamable HTTP (modern MCP standard)
///
/// # Arguments
///
/// * `url` - The URL to detect
/// * `headers` - Optional custom headers to include in the detection request
///
/// # Returns
///
/// Returns the detected protocol type (Sse or Stream)
pub async fn detect_mcp_protocol_with_headers(
    url: &str,
    headers: Option<&HashMap<String, String>>,
) -> Result<McpProtocol> {
    info!(
        "Start automatically detecting MCP service protocol: {}",
        url
    );

    // Try SSE first
    if is_sse_with_headers(url, headers).await {
        info!("SSE protocol detected: {}", url);
        return Ok(McpProtocol::Sse);
    }

    // Default to Streamable HTTP (modern MCP standard)
    info!("Default uses Streamable HTTP protocol: {}", url);
    Ok(McpProtocol::Stream)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    async fn test_detect_invalid_url() {
        // Invalid URL should default to Stream
        let result = detect_mcp_protocol_with_headers("not-a-url", None).await;
        assert!(result.is_ok());
        assert_eq!(result.unwrap(), McpProtocol::Stream);
    }

    #[tokio::test]
    async fn test_detect_nonexistent_server() {
        // Non-existent server should default to Stream
        let result = detect_mcp_protocol_with_headers("http://localhost:99999/mcp", None).await;
        assert!(result.is_ok());
        assert_eq!(result.unwrap(), McpProtocol::Stream);
    }

    #[tokio::test]
    async fn test_detect_with_headers_nonexistent_server() {
        let mut headers = HashMap::new();
        headers.insert("Authorization".to_string(), "Bearer test-token".to_string());
        let result =
            detect_mcp_protocol_with_headers("http://localhost:99999/mcp", Some(&headers)).await;
        assert!(result.is_ok());
        assert_eq!(result.unwrap(), McpProtocol::Stream);
    }
}