This commit is contained in:
bipproduction
2025-10-26 21:42:37 +08:00
parent 948bf14312
commit a0434c3c32

View File

@@ -1,106 +1,91 @@
import { Elysia, t } from "elysia";
export const MCPRoute = new Elysia({
prefix: "/mcp-server",
tags: ["mcp-server"],
prefix: "/mcp-server",
tags: ["mcp-server"],
})
.post("/mcp", ({ body, set }) => {
const { id, method, params } = body as any;
set.headers["Content-Type"] = "application/json; charset=utf-8";
set.headers["Transfer-Encoding"] = "chunked";
set.headers["Connection"] = "keep-alive";
// ✅ 1. GET untuk handshake n8n (wajib)
.get("/mcp", ({ set }) => {
set.headers["Content-Type"] = "application/json";
return {
jsonrpc: "2.0",
result: {
protocol: "2024-11-05",
capabilities: {
"tools/list": true,
"tools/call": true,
}
}
};
})
// ✅ Streaming Response
const stream = new ReadableStream({
async start(controller) {
// tools/list
if (method === "tools/list") {
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
result: [
{
name: "sayHello",
description: "Greets user",
inputSchema: {
type: "object",
properties: { name: { type: "string" } },
},
},
],
}) + "\n"
);
// ❌ Jangan tutup langsung, beri delay agar n8n sempat membaca
await Bun.sleep(200);
controller.close();
return;
// ✅ 2. POST untuk komunikasi streaming (tools/list, tools/call)
.post("/mcp", ({ body, set }) => {
const { id, method, params } = body as any;
set.headers["Content-Type"] = "application/json; charset=utf-8";
set.headers["Transfer-Encoding"] = "chunked";
set.headers["Connection"] = "keep-alive";
const stream = new ReadableStream({
async start(controller) {
if (method === "tools/list") {
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
result: [
{
name: "sayHello",
description: "Greets user",
inputSchema: {
type: "object",
properties: { name: { type: "string" } }
}
}
]
}) + "\n"
);
controller.close();
return;
}
// tools/sayHello → streaming progress
if (method === "tools/sayHello") {
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
result: { status: "Processing..." },
}) + "\n"
);
await Bun.sleep(500);
if (method === "tools/call" && params?.name === "sayHello") {
controller.enqueue(
JSON.stringify({ jsonrpc: "2.0", id, result: { status: "Processing..." } }) + "\n"
);
await Bun.sleep(300);
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
result: { message: `Hello ${params?.arguments?.name || "User"}!` }
}) + "\n"
);
controller.close();
return;
}
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
result: { message: `Hello ${params?.name || "User"}` },
}) + "\n"
);
await Bun.sleep(200);
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
error: { code: -32601, message: `Method ${method} not found` }
}) + "\n"
);
controller.close();
}
});
controller.close();
return;
}
if (method === "mcp/version") {
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
result: {
protocol: "2024-11-05", // versi MCP baru
capabilities: {
"tools/list": true,
"tools/call": true,
},
},
}) + "\n"
);
controller.close();
return;
}
// Method tidak dikenal
controller.enqueue(
JSON.stringify({
jsonrpc: "2.0",
id,
error: { code: -32601, message: `Method ${method} not found` },
}) + "\n"
);
await Bun.sleep(200);
controller.close();
},
});
return new Response(stream);
}, {
body: t.Object({
jsonrpc: t.Optional(t.String()),
method: t.String(),
params: t.Optional(t.Record(t.String(), t.Any())),
id: t.Optional(t.Union([t.String(), t.Number()])),
}),
});
return new Response(stream);
}, {
body: t.Object({
jsonrpc: t.Optional(t.String()),
method: t.String(),
params: t.Optional(t.Any()),
id: t.Optional(t.Union([t.String(), t.Number()])),
}),
});
export default MCPRoute;