tambahan
This commit is contained in:
bipproduction
@@ -2,227 +2,242 @@
|
|||||||
import { Elysia, t } from "elysia";
|
import { Elysia, t } from "elysia";
|
||||||
|
|
||||||
function createStream(handler: (controller: ReadableStreamDefaultController) => Promise<void>) {
|
function createStream(handler: (controller: ReadableStreamDefaultController) => Promise<void>) {
|
||||||
return new ReadableStream({
|
return new ReadableStream({
|
||||||
async start(controller) {
|
async start(controller) {
|
||||||
try {
|
try {
|
||||||
controller.enqueue(":\n\n"); // Heartbeat awal
|
controller.enqueue(":\n\n"); // Heartbeat awal
|
||||||
await handler(controller);
|
await handler(controller);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
controller.enqueue(
|
controller.enqueue(
|
||||||
JSON.stringify({
|
JSON.stringify({
|
||||||
jsonrpc: "2.0",
|
jsonrpc: "2.0",
|
||||||
error: { code: -32000, message: "Internal Server Error" },
|
error: { code: -32000, message: "Internal Server Error" },
|
||||||
}) + "\n"
|
}) + "\n"
|
||||||
);
|
);
|
||||||
} finally {
|
} finally {
|
||||||
try {
|
try {
|
||||||
controller.close();
|
controller.close();
|
||||||
} catch {}
|
} catch { }
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
cancel() {
|
cancel() {
|
||||||
console.log("🔴 Client closed the connection");
|
console.log("🔴 Client closed the connection");
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
export const MCPRoute = new Elysia({
|
export const MCPRoute = new Elysia({
|
||||||
prefix: "/mcp",
|
prefix: "/mcp",
|
||||||
tags: ["mcp"],
|
tags: ["mcp"],
|
||||||
})
|
})
|
||||||
/**
|
.get("/", ({ set }) => {
|
||||||
* ✅ GET /mcp/:id — streaming response untuk n8n + curl
|
set.headers["Content-Type"] = "application/json; charset=utf-8";
|
||||||
*/
|
return {
|
||||||
.get("/:id", ({ params, query, set }) => {
|
|
||||||
const id = params.id ?? 1;
|
|
||||||
const method = query.method as string | undefined;
|
|
||||||
|
|
||||||
set.headers["Content-Type"] = "application/json; charset=utf-8";
|
|
||||||
set.headers["Cache-Control"] = "no-cache, no-transform";
|
|
||||||
set.headers["Connection"] = "keep-alive";
|
|
||||||
set.headers["Transfer-Encoding"] = "chunked";
|
|
||||||
set.headers["X-Accel-Buffering"] = "no";
|
|
||||||
|
|
||||||
const stream = createStream(async (controller) => {
|
|
||||||
// Jika tidak ada method → status server
|
|
||||||
if (!method) {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
jsonrpc: "2.0",
|
||||||
id,
|
id: 1,
|
||||||
result: {
|
result: {
|
||||||
status: "MCP Server Ready",
|
|
||||||
message: "Use ?method=mcp/version or ?method=tools/list",
|
|
||||||
availableMethods: ["mcp/version", "tools/list", "tools/sayHello"],
|
|
||||||
},
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// mcp/version
|
|
||||||
if (method === "mcp/version") {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
result: {
|
|
||||||
protocol: "2024-11-05",
|
|
||||||
capabilities: { "tools/list": true, "tools/call": true },
|
|
||||||
},
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// tools/list
|
|
||||||
if (method === "tools/list") {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
result: [
|
|
||||||
{
|
|
||||||
name: "sayHello",
|
|
||||||
description: "Greets a user",
|
|
||||||
inputSchema: {
|
|
||||||
type: "object",
|
|
||||||
properties: { name: { type: "string", description: "Your name" } },
|
|
||||||
required: ["name"],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// tools/sayHello (progress + result)
|
|
||||||
if (method === "tools/sayHello") {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({ jsonrpc: "2.0", id, result: { status: "Processing..." } }) + "\n"
|
|
||||||
);
|
|
||||||
await Bun.sleep(500);
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
result: { message: `Hello ${query?.name || "User"}` },
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Unknown method
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
error: { code: -32601, message: `Method "${method}" not found` },
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
return new Response(stream);
|
|
||||||
})
|
|
||||||
/**
|
|
||||||
* ✅ POST /mcp — JSON-RPC versi body
|
|
||||||
*/
|
|
||||||
.post(
|
|
||||||
"/",
|
|
||||||
async ({ body, set }) => {
|
|
||||||
const { id, method, params } = body as any;
|
|
||||||
|
|
||||||
set.headers["Content-Type"] = "application/json; charset=utf-8";
|
|
||||||
set.headers["Connection"] = "keep-alive";
|
|
||||||
set.headers["Transfer-Encoding"] = "chunked";
|
|
||||||
set.headers["X-Accel-Buffering"] = "no";
|
|
||||||
|
|
||||||
const stream = createStream(async (controller) => {
|
|
||||||
if (!method) {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
error: { code: -32600, message: "Method required" },
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (method === "mcp/version") {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
result: {
|
|
||||||
protocol: "2024-11-05",
|
protocol: "2024-11-05",
|
||||||
capabilities: { "tools/list": true, "tools/call": true },
|
capabilities: {
|
||||||
},
|
"tools/list": true,
|
||||||
}) + "\n"
|
"tools/call": true,
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (method === "tools/list") {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
result: [
|
|
||||||
{
|
|
||||||
name: "sayHello",
|
|
||||||
description: "Greets a user",
|
|
||||||
inputSchema: {
|
|
||||||
type: "object",
|
|
||||||
properties: { name: { type: "string" } },
|
|
||||||
required: ["name"],
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
],
|
status: "MCP Server Ready",
|
||||||
}) + "\n"
|
},
|
||||||
);
|
};
|
||||||
return;
|
})
|
||||||
|
/**
|
||||||
|
* ✅ GET /mcp/:id — streaming response untuk n8n + curl
|
||||||
|
*/
|
||||||
|
.get("/:id", ({ params, query, set }) => {
|
||||||
|
const id = params.id ?? 1;
|
||||||
|
const method = query.method as string | undefined;
|
||||||
|
|
||||||
|
set.headers["Content-Type"] = "application/json; charset=utf-8";
|
||||||
|
set.headers["Cache-Control"] = "no-cache, no-transform";
|
||||||
|
set.headers["Connection"] = "keep-alive";
|
||||||
|
set.headers["Transfer-Encoding"] = "chunked";
|
||||||
|
set.headers["X-Accel-Buffering"] = "no";
|
||||||
|
|
||||||
|
const stream = createStream(async (controller) => {
|
||||||
|
// Jika tidak ada method → status server
|
||||||
|
if (!method) {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: {
|
||||||
|
status: "MCP Server Ready",
|
||||||
|
message: "Use ?method=mcp/version or ?method=tools/list",
|
||||||
|
availableMethods: ["mcp/version", "tools/list", "tools/sayHello"],
|
||||||
|
},
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// mcp/version
|
||||||
|
if (method === "mcp/version") {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: {
|
||||||
|
protocol: "2024-11-05",
|
||||||
|
capabilities: { "tools/list": true, "tools/call": true },
|
||||||
|
},
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// tools/list
|
||||||
|
if (method === "tools/list") {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: [
|
||||||
|
{
|
||||||
|
name: "sayHello",
|
||||||
|
description: "Greets a user",
|
||||||
|
inputSchema: {
|
||||||
|
type: "object",
|
||||||
|
properties: { name: { type: "string", description: "Your name" } },
|
||||||
|
required: ["name"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// tools/sayHello (progress + result)
|
||||||
|
if (method === "tools/sayHello") {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({ jsonrpc: "2.0", id, result: { status: "Processing..." } }) + "\n"
|
||||||
|
);
|
||||||
|
await Bun.sleep(500);
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: { message: `Hello ${query?.name || "User"}` },
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unknown method
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
error: { code: -32601, message: `Method "${method}" not found` },
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
return new Response(stream);
|
||||||
|
})
|
||||||
|
/**
|
||||||
|
* ✅ POST /mcp — JSON-RPC versi body
|
||||||
|
*/
|
||||||
|
.post(
|
||||||
|
"/",
|
||||||
|
async ({ body, set }) => {
|
||||||
|
const { id, method, params } = body as any;
|
||||||
|
|
||||||
|
set.headers["Content-Type"] = "application/json; charset=utf-8";
|
||||||
|
set.headers["Connection"] = "keep-alive";
|
||||||
|
set.headers["Transfer-Encoding"] = "chunked";
|
||||||
|
set.headers["X-Accel-Buffering"] = "no";
|
||||||
|
|
||||||
|
const stream = createStream(async (controller) => {
|
||||||
|
if (!method) {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
error: { code: -32600, message: "Method required" },
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (method === "mcp/version") {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: {
|
||||||
|
protocol: "2024-11-05",
|
||||||
|
capabilities: { "tools/list": true, "tools/call": true },
|
||||||
|
},
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (method === "tools/list") {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: [
|
||||||
|
{
|
||||||
|
name: "sayHello",
|
||||||
|
description: "Greets a user",
|
||||||
|
inputSchema: {
|
||||||
|
type: "object",
|
||||||
|
properties: { name: { type: "string" } },
|
||||||
|
required: ["name"],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (method === "tools/sayHello") {
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: { status: "Processing..." },
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
await Bun.sleep(500);
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
result: { message: `Hello ${params?.name || "User"}` },
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
controller.enqueue(
|
||||||
|
JSON.stringify({
|
||||||
|
jsonrpc: "2.0",
|
||||||
|
id,
|
||||||
|
error: { code: -32601, message: `Method "${method}" not found` },
|
||||||
|
}) + "\n"
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
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()])),
|
||||||
|
}),
|
||||||
}
|
}
|
||||||
|
);
|
||||||
if (method === "tools/sayHello") {
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
result: { status: "Processing..." },
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
await Bun.sleep(500);
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
result: { message: `Hello ${params?.name || "User"}` },
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
controller.enqueue(
|
|
||||||
JSON.stringify({
|
|
||||||
jsonrpc: "2.0",
|
|
||||||
id,
|
|
||||||
error: { code: -32601, message: `Method "${method}" not found` },
|
|
||||||
}) + "\n"
|
|
||||||
);
|
|
||||||
});
|
|
||||||
|
|
||||||
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()])),
|
|
||||||
}),
|
|
||||||
}
|
|
||||||
);
|
|
||||||
|
|
||||||
export default MCPRoute;
|
export default MCPRoute;
|
||||||
|
|||||||
Reference in New Issue
Block a user