tambahan

src/server/routes/mcp_route.ts

@@ -1,15 +1,21 @@
 import { Elysia } from "elysia";
 
 export const MCPRoute = new Elysia()
-  .get("/mcp/:sessionId", ({ params }) => {
-    const encoder = new TextEncoder();
+  .get("/mcp/:sessionId", ({ set, params }) => {
+    set.headers["Content-Type"] = "text/event-stream; charset=utf-8";
+    set.headers["Cache-Control"] = "no-cache";
+    set.headers["Connection"] = "keep-alive";
 
-    let interval: Timer | null = null;
+    const encoder = new TextEncoder();
 
     const stream = new ReadableStream({
       start(controller) {
-        // kirim event awal (flush langsung)
-        const init = {
+        const send = (obj: any) => {
+          controller.enqueue(encoder.encode(`data: ${JSON.stringify(obj)}\n\n`));
+        };
+
+        // Kirim pesan pembuka
+        send({
           jsonrpc: "2.0",
           id: null,
           result: {
@@ -18,94 +24,26 @@ export const MCPRoute = new Elysia()
               "tools/list": true,
               "tools/call": true,
             },
-            status: `MCP session ${params.sessionId} aktif`,
+            status: `MCP Stream aktif untuk session ${params.sessionId}`,
           },
-        };
+        });
 
-        controller.enqueue(
-          encoder.encode(`data: ${JSON.stringify(init)}\n\n`)
-        );
+        // Keep connection alive
+        const interval = setInterval(() => {
+          send({ ping: Date.now() });
+        }, 15000);
 
-        // SSE heartbeat
-        interval = setInterval(() => {
-          controller.enqueue(encoder.encode(`: ping ${Date.now()}\n\n`));
-        }, 10000);
+        // Simpan handle agar bisa dibersihkan nanti
+        (controller as any)._interval = interval;
       },
+
       cancel() {
-        if (interval) clearInterval(interval);
-        console.log(`[SSE] koneksi session ${params.sessionId} ditutup`);
+        // Ketika stream dibatalkan oleh client
+        console.log("🔌 Stream closed by client");
+        // Bersihkan interval
+        clearInterval((this as any)._interval);
       },
     });
 
-    // gunakan Response manual agar Bun flush data pertama
-    return new Response(stream, {
-      headers: {
-        "Content-Type": "text/event-stream; charset=utf-8",
-        "Cache-Control": "no-cache, no-transform",
-        Connection: "keep-alive",
-        "Access-Control-Allow-Origin": "*",
-      },
-    });
-  })
-
-  .post("/mcp", async ({ body, set }) => {
-    set.headers["Content-Type"] = "application/json; charset=utf-8";
-    const { id, method, params } = body as any;
-
-    if (method === "tools/list") {
-      return {
-        jsonrpc: "2.0",
-        id,
-        result: {
-          tools: [
-            {
-              name: "pengajuan-pembuatan-ktp",
-              description:
-                "untuk melakukan pengajuan pembuatan ktp\nmembutuhkan :\n- jenis\n- name\n- deskripsi",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  JSON: { type: "object" },
-                },
-                required: ["JSON"],
-                additionalProperties: true,
-                $schema: "http://json-schema.org/draft-07/schema#",
-              },
-            },
-            {
-              name: "pengetahuan_malik_kurosaki",
-              description: "penjelasan tentang malik kurosaki",
-              inputSchema: {
-                type: "object",
-                properties: {
-                  input: { type: "string" },
-                },
-                additionalProperties: true,
-                $schema: "http://json-schema.org/draft-07/schema#",
-              },
-            },
-          ],
-        },
-      };
-    }
-
-    if (method === "tools/call") {
-      const { tool, arguments: args } = params;
-      if (tool === "pengajuan-pembuatan-ktp") {
-        return {
-          jsonrpc: "2.0",
-          id,
-          result: { message: "Berhasil menerima pengajuan KTP", data: args },
-        };
-      }
-    }
-
-    return {
-      jsonrpc: "2.0",
-      id,
-      error: {
-        code: -32601,
-        message: `Method ${method} tidak dikenali`,
-      },
-    };
+    return stream;
   });