mcp-stdio-proxy 0.2.9

MCP (Model Context Protocol) proxy server and CLI tool for protocol conversion and remote service access
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
81
82
83
84
85

use axum::{
    Router,
    extract::DefaultBodyLimit,
    routing::{delete, get, post},
};
use http::Method;
use tower_http::cors::{self, CorsLayer};

use crate::{
    AppError, AppState, DynamicRouterService,
    model::{GLOBAL_SSE_MCP_ROUTES_PREFIX, GLOBAL_STREAM_MCP_ROUTES_PREFIX, McpProtocol},
    server::handlers::check_mcp_is_status_handler,
};

use super::{
    get_health, get_ready,
    handlers::{
        add_route_handler, check_mcp_status_handler_sse, check_mcp_status_handler_stream,
        delete_route_handler, run_code_handler,
    },
    set_layer,
};

/// 获取路由
pub async fn get_router(state: AppState) -> Result<Router, AppError> {
    let health = Router::new()
        .route("/health", get(get_health))
        .route("/ready", get(get_ready));

    let cors = CorsLayer::new()
        // allow `GET` and `POST` when accessing the resource
        .allow_methods([
            Method::GET,
            Method::POST,
            Method::PATCH,
            Method::DELETE,
            Method::PUT,
        ])
        .allow_origin(cors::Any)
        .allow_headers(cors::Any);

    let api = Router::new()
        // .layer(from_fn_with_state(state.clone(), verify_token::<AppState>))
        //mcp sse 协议路由
        .route_service(
            &format!("{GLOBAL_SSE_MCP_ROUTES_PREFIX}/proxy/{{*path}}"),
            DynamicRouterService(McpProtocol::Sse),
        )
        .route(
            &format!("{GLOBAL_SSE_MCP_ROUTES_PREFIX}/add"),
            post(add_route_handler),
        )
        .route("/mcp/config/delete/{mcp_id}", delete(delete_route_handler))
        .route(
            "/mcp/check/status/{mcp_id}",
            get(check_mcp_is_status_handler),
        )
        .route(
            &format!("{GLOBAL_SSE_MCP_ROUTES_PREFIX}/check_status"),
            post(check_mcp_status_handler_sse),
        )
        //mcp stream 协议路由
        .route_service(
            &format!("{GLOBAL_STREAM_MCP_ROUTES_PREFIX}/proxy/{{*path}}"),
            DynamicRouterService(McpProtocol::Stream),
        )
        .route(
            &format!("{GLOBAL_STREAM_MCP_ROUTES_PREFIX}/check_status"),
            post(check_mcp_status_handler_stream),
        )
        .route("/api/run_code_with_log", post(run_code_handler))
        .layer(DefaultBodyLimit::max(20 * 1024 * 1024))
        .layer(tower_http::catch_panic::CatchPanicLayer::new())
        .layer(cors);

    // 创建基本路由
    let app: Router<AppState> = Router::new().merge(health).merge(api);
    // OpenAPI 文档 UI（Swagger UI + Scalar 双风格；无状态路由，with_state 前 merge）
    let app = app.merge(super::openapi::create_docs_router());

    // 添加状态
    let app = app.with_state(state.clone());
    let router = set_layer(app, state);
    Ok(router)
}