This commit is contained in:
bipproduction
2025-10-26 20:59:23 +08:00
parent 92ca73006a
commit 953a43039f

View File

@@ -1,67 +1,72 @@
import Elysia from "elysia"; import { Elysia } from "elysia";
import { t } from "elysia"; import { t } from "elysia";
export const MCPRoute = new Elysia({ export const MCPRoute = new Elysia({ prefix: "/mcp-server" })
prefix: "/mcp-server", .post("/mcp", async ({ body, set }) => {
tags: ["mcp-server"],
})
.post("/mcp", ({ body }) => {
const { id, method, params } = body; const { id, method, params } = body;
set.headers['Content-Type'] = 'application/json';
set.headers['Transfer-Encoding'] = 'chunked'; // ✅ Streaming-required for n8n
// ==== TOOLS EXECUTION ==== // ---- STREAMING RESPONSE CONSTRUCTION ----
const stream = new ReadableStream({
async start(controller) {
// Jika tools/sayHello → kirim stream
if (method === "tools/sayHello") { if (method === "tools/sayHello") {
controller.enqueue(JSON.stringify({
return {
jsonrpc: "2.0", jsonrpc: "2.0",
id, id,
result: { result: { message: "Processing..." }
message: `Hello from MCP Server`, }) + "\n"); // kirim chunk pertama
},
}; await Bun.sleep(500); // contoh delay
controller.enqueue(JSON.stringify({
jsonrpc: "2.0",
id,
result: { message: `Hello ` }
}) + "\n");
controller.close();
} }
// ==== LIST ALL TOOLS ==== // Jika tools/list → kirim langsung tapi tetap stream
if (method === "tools/list") { else if (method === "tools/list") {
return { controller.enqueue(JSON.stringify({
jsonrpc: "2.0", jsonrpc: "2.0",
id, id,
result: [ result: [
{ {
name: "sayHello", name: "sayHello",
description: "Greets a user with a name", description: "Greets user",
inputSchema: { inputSchema: {
type: "object", type: "object",
properties: { properties: { name: { type: "string" } },
name: { type: "string" }, }
}, }
required: ["name"], ]
}, }));
}, controller.close();
],
};
} }
// ==== IF METHOD NOT FOUND ==== // Jika tidak ada method
return { else {
controller.enqueue(JSON.stringify({
jsonrpc: "2.0", jsonrpc: "2.0",
id, id,
error: { error: { code: -32601, message: `Method ${method} not found` }
code: -32601, }));
message: `Method '${method}' not found`, controller.close();
}, }
}; }
});
return new Response(stream);
}, { }, {
// ✅ Body JSON-RPC 2.0 schema (params & id bisa optional)
body: t.Object({ body: t.Object({
jsonrpc: t.Optional(t.String()), jsonrpc: t.Optional(t.String()),
method: t.String(), method: t.String(),
params: t.Optional(t.Record(t.String(), t.Any())), params: t.Optional(t.Record(t.String(), t.Any())),
id: t.Optional(t.Union([t.String(), t.Number()])), id: t.Optional(t.Union([t.String(), t.Number()])),
}), })
detail: {
summary: "MCP Server Endpoint",
description: "Handle MCP JSON-RPC requests for tools and discovery.",
},
}); });
export default MCPRoute; export default MCPRoute;