From acf7e4213fc828ec1fd0db23a8222aeba41c2f4c Mon Sep 17 00:00:00 2001 From: bipproduction Date: Mon, 27 Oct 2025 02:53:50 +0800 Subject: [PATCH] tambahan --- src/server/routes/mcp_route.ts | 95 ++++++++-------------------------- 1 file changed, 21 insertions(+), 74 deletions(-) diff --git a/src/server/routes/mcp_route.ts b/src/server/routes/mcp_route.ts index 1fad24b..7b53740 100644 --- a/src/server/routes/mcp_route.ts +++ b/src/server/routes/mcp_route.ts @@ -1,18 +1,21 @@ import { Elysia } from "elysia"; export const MCPRoute = new Elysia() - .get("/mcp/:sessionId", async ({ params, set }) => { + .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"; - set.headers["Access-Control-Allow-Origin"] = "*"; const encoder = new TextEncoder(); const stream = new ReadableStream({ start(controller) { - // kirim pesan handshake awal - const initial = { + const send = (obj: any) => { + controller.enqueue(encoder.encode(`data: ${JSON.stringify(obj)}\n\n`)); + }; + + // Kirim pesan pembuka + send({ jsonrpc: "2.0", id: null, result: { @@ -21,82 +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(initial)}\n\n`)); - - // jaga stream tetap hidup setiap 10 detik + // Keep connection alive const interval = setInterval(() => { - controller.enqueue(encoder.encode(`: ping\n\n`)); // komentar SSE - }, 10000); + send({ ping: Date.now() }); + }, 15000); - // ketika stream ditutup - controller.enqueue(encoder.encode(`: stream ready\n\n`)); + // Simpan handle agar bisa dibersihkan nanti + (controller as any)._interval = interval; + }, + + cancel() { + // Ketika stream dibatalkan oleh client + console.log("🔌 Stream closed by client"); + // Bersihkan interval + clearInterval((this as any)._interval); }, }); return stream; - }) - - .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`, - }, - }; });