This commit is contained in:
bipproduction
2025-10-26 21:12:05 +08:00
parent 8838cc3152
commit 8b30810b38

View File

@@ -1,62 +1,71 @@
import { Elysia } from "elysia"; import { Elysia, t } from "elysia";
import { t } from "elysia";
export const MCPRoute = new Elysia({ prefix: "/mcp-server", tags: ["mcp-server"] }) export const MCPRoute = new Elysia({
.all("/mcp", async ({ body, set }) => { prefix: "/api/mcp-server", // ✅ Sesuaikan dengan endpoint n8n
const { id, method, params } = body; tags: ["mcp-server"],
set.headers['Content-Type'] = 'application/json'; })
set.headers['Transfer-Encoding'] = 'chunked'; // ✅ Streaming-required for n8n .post("/mcp", ({ body, set }) => {
const { id, method, params } = body as any;
set.headers["Content-Type"] = "application/json";
set.headers["Transfer-Encoding"] = "chunked";
set.headers["Connection"] = "keep-alive";
// ---- STREAMING RESPONSE CONSTRUCTION ----
const stream = new ReadableStream({ const stream = new ReadableStream({
async start(controller) { async start(controller) {
// Jika tools/sayHello → kirim stream // tools.list → kirim daftar tools
if (method === "tools/sayHello") { if (method === "tools/list") {
controller.enqueue(JSON.stringify({ controller.enqueue(
jsonrpc: "2.0", JSON.stringify({
id, jsonrpc: "2.0",
result: { message: "Processing..." } id,
}) + "\n"); // kirim chunk pertama result: [
{
await Bun.sleep(500); // contoh delay name: "sayHello",
controller.enqueue(JSON.stringify({ description: "Greets user",
jsonrpc: "2.0", inputSchema: {
id, type: "object",
result: { message: `Hello ` } properties: { name: { type: "string" } },
}) + "\n"); },
},
],
}) + "\n"
);
controller.close(); controller.close();
} }
// Jika tools/list → kirim langsung tapi tetap stream // tools.sayHello → streaming bertahap
else if (method === "tools/list") { else if (method === "tools/sayHello") {
controller.enqueue(JSON.stringify({ controller.enqueue(
jsonrpc: "2.0", JSON.stringify({
id, jsonrpc: "2.0",
result: [ id,
{ result: { message: "Processing..." },
name: "sayHello", }) + "\n"
description: "Greets user", );
inputSchema: { await Bun.sleep(500);
type: "object", controller.enqueue(
properties: { name: { type: "string" } }, JSON.stringify({
} jsonrpc: "2.0",
} id,
] result: { message: `Hello ${params?.name || ""}` },
})); }) + "\n"
);
controller.close(); controller.close();
} }
// Jika tidak ada method // Method tidak ditemukan
else { else {
controller.enqueue(JSON.stringify({ controller.enqueue(
jsonrpc: "2.0", JSON.stringify({
id, jsonrpc: "2.0",
error: { code: -32601, message: `Method ${method} not found` } id,
})); error: { code: -32601, message: `Method ${method} not found` },
}) + "\n"
);
controller.close(); controller.close();
} }
} },
}); });
return new Response(stream); return new Response(stream);
@@ -66,7 +75,7 @@ export const MCPRoute = new Elysia({ prefix: "/mcp-server", tags: ["mcp-server"]
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()])),
}) }),
}); });
export default MCPRoute; export default MCPRoute;