refactor

2026-05-22 08:24:17 +02:00
parent d893e4c609
commit 749638b75c
16 changed files with 770 additions and 782 deletions
@@ -0,0 +1,54 @@
+mod protocol;
+mod tools;
+
+use crate::{api::auth::MaybeBearerToken, app::state::AppState};
+use protocol::{
+    ApiResponse, InitializeResult, JSONRPC_VERSION, jsonrpc_error, jsonrpc_notification_ok,
+    jsonrpc_ok,
+};
+use rocket::{State, http::Status, post, serde::json::Json};
+use serde_json::Value;
+use tools::{handle_tool_call, tool_definitions};
+
+pub use protocol::JsonRpcRequest;
+
+#[post("/mcp", data = "<body>")]
+pub async fn route(
+    body: Json<JsonRpcRequest>,
+    token: MaybeBearerToken,
+    state: &State<AppState>,
+) -> ApiResponse {
+    let request = body.into_inner();
+
+    if request.jsonrpc != JSONRPC_VERSION {
+        return jsonrpc_error(
+            Status::BadRequest,
+            request.id,
+            -32600,
+            "Invalid JSON-RPC version",
+        );
+    }
+
+    match request.method.as_str() {
+        "initialize" => jsonrpc_ok(request.id, InitializeResult::new()),
+        "notifications/initialized" => jsonrpc_notification_ok(),
+        "tools/list" => match token.0 {
+            Some(_) => jsonrpc_ok(request.id, tool_definitions()),
+            None => missing_authorization(request.id),
+        },
+        "tools/call" => {
+            handle_tool_call(request.id, request.params, token.0.as_deref(), state).await
+        }
+        _ if request.id.is_none() => jsonrpc_notification_ok(),
+        _ => jsonrpc_error(Status::BadRequest, request.id, -32601, "Method not found"),
+    }
+}
+
+fn missing_authorization(id: Option<Value>) -> ApiResponse {
+    jsonrpc_error(
+        Status::Unauthorized,
+        id,
+        -32001,
+        "Missing Authorization header",
+    )
+}