This commit is contained in:
bipproduction
2025-10-27 02:54:33 +08:00
parent acf7e4213f
commit fa9d06a5ea

View File

@@ -1,21 +1,18 @@
import { Elysia } from "elysia";
export const MCPRoute = new Elysia()
.get("/mcp/:sessionId", ({ set, params }) => {
.get("/mcp/:sessionId", async ({ params, set }) => {
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) {
const send = (obj: any) => {
controller.enqueue(encoder.encode(`data: ${JSON.stringify(obj)}\n\n`));
};
// Kirim pesan pembuka
send({
// kirim pesan handshake awal
const initial = {
jsonrpc: "2.0",
id: null,
result: {
@@ -24,26 +21,82 @@ export const MCPRoute = new Elysia()
"tools/list": true,
"tools/call": true,
},
status: `MCP Stream aktif untuk session ${params.sessionId}`,
status: `MCP session ${params.sessionId} aktif`,
},
});
};
// Keep connection alive
controller.enqueue(encoder.encode(`data: ${JSON.stringify(initial)}\n\n`));
// jaga stream tetap hidup setiap 10 detik
const interval = setInterval(() => {
send({ ping: Date.now() });
}, 15000);
controller.enqueue(encoder.encode(`: ping\n\n`)); // komentar SSE
}, 10000);
// 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);
// ketika stream ditutup
controller.enqueue(encoder.encode(`: stream ready\n\n`));
},
});
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`,
},
};
});