@unkey/api
Version:
Developer-friendly & type-safe Typescript SDK specifically catered to leverage *@unkey/api* API.
98 lines • 3.34 kB
JavaScript
/*
* Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
*/
import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import express from "express";
import { createConsoleLogger, } from "../../console-logger.js";
import { createMCPServer } from "../../server.js";
export async function main(flags) {
flags.env?.forEach(([key, value]) => {
process.env[key] = value;
});
switch (flags.transport) {
case "stdio":
await startStdio(flags);
break;
case "sse":
await startSSE(flags);
break;
default:
throw new Error(`Invalid transport: ${flags.transport}`);
}
}
async function startStdio(flags) {
const logger = createConsoleLogger(flags["log-level"]);
const transport = new StdioServerTransport();
const server = createMCPServer({
logger,
allowedTools: flags.tool,
scopes: flags.scope,
...{ rootKey: flags["root-key"] },
serverURL: flags["server-url"],
serverIdx: flags["server-index"],
});
await server.connect(transport);
const abort = async () => {
await server.close();
process.exit(0);
};
process.on("SIGTERM", abort);
process.on("SIGINT", abort);
}
async function startSSE(flags) {
const logger = createConsoleLogger(flags["log-level"]);
const app = express();
const mcpServer = createMCPServer({
logger,
allowedTools: flags.tool,
scopes: flags.scope,
...{ rootKey: flags["root-key"] },
serverURL: flags["server-url"],
serverIdx: flags["server-index"],
});
let transport;
const controller = new AbortController();
app.get("/sse", async (_req, res) => {
transport = new SSEServerTransport("/message", res);
await mcpServer.connect(transport);
mcpServer.server.onclose = async () => {
res.end();
};
});
app.post("/message", async (req, res) => {
if (!transport) {
throw new Error("Server transport not initialized");
}
await transport.handlePostMessage(req, res);
});
const httpServer = app.listen(flags.port, "0.0.0.0", () => {
const ha = httpServer.address();
const host = typeof ha === "string" ? ha : `${ha?.address}:${ha?.port}`;
logger.info("MCP HTTP server started", { host });
});
let closing = false;
controller.signal.addEventListener("abort", async () => {
if (closing) {
logger.info("Received second signal. Forcing shutdown.");
process.exit(1);
}
closing = true;
logger.info("Shutting down MCP server");
await mcpServer.close();
logger.info("Shutting down HTTP server");
const timer = setTimeout(() => {
logger.info("Forcing shutdown");
process.exit(1);
}, 5000);
httpServer.close(() => {
clearTimeout(timer);
logger.info("Graceful shutdown complete");
process.exit(0);
});
});
const abort = () => controller.abort();
process.on("SIGTERM", abort);
process.on("SIGINT", abort);
}
//# sourceMappingURL=impl.js.map