sse endpoint

2026-05-22 09:47:20 +02:00
parent 78b674fa1c
commit ca1315bfaf
3 changed files with 31 additions and 2 deletions
@@ -24,3 +24,5 @@ docker build -t server .
  - requires `X-Device-ID: <device-id>`
 - `POST /mcp` for MCP/JSON-RPC requests
  - typically uses `Authorization: Bearer <token>`
+- `GET /mcp` for MCP server-sent events
+  - provides a minimal SSE stream for MCP clients that probe or require it
@@ -6,10 +6,34 @@ use protocol::{
    ApiResponse, InitializeResult, JSONRPC_VERSION, JsonRpcRequest, jsonrpc_error,
    jsonrpc_notification_ok, jsonrpc_ok,
 };
-use rocket::{State, http::Status, post, serde::json::Json};
+use rocket::{
+    Shutdown, State, get,
+    http::Status,
+    post,
+    response::stream::{Event, EventStream},
+    serde::json::Json,
+};
 use serde_json::Value;
 use tools::{handle_tool_call, tool_definitions};

+const MCP_MESSAGE_ENDPOINT_EVENT: &str = "endpoint";
+
+#[get("/mcp")]
+pub fn sse(shutdown: Shutdown) -> EventStream![] {
+    EventStream! {
+        yield Event::json(&serde_json::json!({ "path": "/mcp" })).event(MCP_MESSAGE_ENDPOINT_EVENT);
+
+        loop {
+            tokio::select! {
+                _ = shutdown.clone() => break,
+                _ = tokio::time::sleep(std::time::Duration::from_secs(15)) => {
+                    yield Event::comment("keepalive");
+                }
+            }
+        }
+    }
+}
+
 #[post("/mcp", data = "<body>")]
 pub async fn route(
    body: Json<JsonRpcRequest>,
@@ -33,7 +33,10 @@ pub async fn apply_schema(database: &SqlitePool) -> Result<(), sqlx::Error> {
 pub fn build_rocket(database: SqlitePool) -> rocket::Rocket<rocket::Build> {
    rocket::build()
        .manage(build_state(database))
-        .mount("/", routes![transport::socket::connect, api::mcp::route])
+        .mount(
+            "/",
+            routes![transport::socket::connect, api::mcp::route, api::mcp::sse],
+        )
        .register("/", rocket::catchers![api::catchers::default_catcher])
 }