rudof_mcp 0.2.20

RDF data shapes implementation in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use anyhow::{Result, anyhow};

use super::stdio_server::run_mcp_stdio;
use super::streamable_http::run_mcp_http;
use super::transport::TransportType;

#[derive(Clone, Debug)]
/// Configuration for the MCP server.
pub struct McpConfig {
    /// Transport type to use (Stdio or StreamableHTTP)
    pub transport: TransportType,
    /// Address to bind the HTTP server to (ignored for Stdio). Example: "127.0.0.1"
    pub bind_address: Option<String>,
    /// Port for HTTP transport (ignored for Stdio)
    pub port: Option<u16>,
    /// Endpoint route path for HTTP (ignored for Stdio). Example: "mcp"
    pub route_path: Option<String>,
    /// List of allowed networks/IPs for HTTP transport. Example: ["127.0.0.1", "192.168.1.0/24"]
    pub allowed_networks: Option<Vec<String>>,
}

impl Default for McpConfig {
    fn default() -> Self {
        Self {
            transport: TransportType::Stdio,
            bind_address: None,
            port: None,
            route_path: None,
            allowed_networks: None,
        }
    }
}

/// Entry point for running the MCP server with the specified transport.
/// This function routes to the appropriate transport implementation:
/// - TransportType::Stdio: Uses stdin/stdout for communication (for CLI tools, IDE extensions, etc.)
/// - TransportType::StreamableHTTP: Uses HTTP with SSE for communication (for web clients, network access, etc.)
///
/// # Arguments
/// * `transport` - The transport type to use
/// * `bind_address` - Address to bind the HTTP server to (ignored for Stdio)
/// Examples: "127.0.0.1" (localhost IPv4), "0.0.0.0" (all IPv4), "::1" (localhost IPv6)
/// * `port` - Port number for HTTP transport (ignored for Stdio)
/// * `route_path` - Route path for HTTP transport (ignored for Stdio)
/// * `allowed_networks` - Optional list of allowed IP addresses or CIDR ranges for HTTP transport .Examples: ["127.0.0.1",
/// "192.168.1.0/24", "10.0.0.0/8", "::1"]. If None or empty, defaults to localhost only
///
/// # Note
/// This function creates its own Tokio runtime. If you're already in an async context,
/// use `run_mcp_async` instead.
#[tokio::main]
pub async fn run_mcp(config: McpConfig) -> Result<()> {
    run_mcp_async(config).await
}

fn validate_http_config(config: &McpConfig) -> Result<(&str, u16, &str)> {
    let bind = config
        .bind_address
        .as_deref()
        .ok_or_else(|| anyhow!("bind_address is required when using StreamableHTTP transport"))?;
    let port = config
        .port
        .ok_or_else(|| anyhow!("port is required when using StreamableHTTP transport"))?;
    let route = config
        .route_path
        .as_deref()
        .ok_or_else(|| anyhow!("route_path is required when using StreamableHTTP transport"))?;
    Ok((bind, port, route))
}

/// Async version of `run_mcp` for use within an existing Tokio runtime.
pub async fn run_mcp_async(config: McpConfig) -> Result<()> {
    match config.transport {
        TransportType::Stdio => run_mcp_stdio().await,
        TransportType::StreamableHTTP => {
            let (bind, port, route) = validate_http_config(&config)?;
            run_mcp_http(bind, port, route, config.allowed_networks.clone()).await
        },
    }
}