tambahan
This commit is contained in:
@@ -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;
|
||||
|
||||
Reference in New Issue
Block a user