/**
* Entry point for the Azure Storage MCP server.
* Supports both stdio (VS Code Copilot) and HTTP (testing) transports.
*
* Usage:
* npx tsx main.ts --stdio # stdio transport (for VS Code)
* npx tsx main.ts # HTTP transport on port 3001
*/
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
import cors from "cors";
import express from "express";
import type { Request, Response } from "express";
import { createServer } from "./server.js";
async function startStdioServer(): Promise<void> {
const server = createServer();
const transport = new StdioServerTransport();
await server.connect(transport);
console.error("Azure Storage MCP server running on stdio");
}
async function startHttpServer(): Promise<void> {
const port = parseInt(process.env.PORT ?? "3001", 10);
const app = express();
app.use(cors());
app.use(express.json());
app.all("/mcp", async (req: Request, res: Response) => {
const server = createServer();
const transport = new StreamableHTTPServerTransport({
sessionIdGenerator: undefined,
});
res.on("close", () => {
transport.close().catch(() => {});
server.close().catch(() => {});
});
try {
await server.connect(transport);
await transport.handleRequest(req, res, req.body);
} catch (error) {
console.error("MCP error:", error);
if (!res.headersSent) {
res.status(500).json({
jsonrpc: "2.0",
error: { code: -32603, message: "Internal server error" },
id: null,
});
}
}
});
app.listen(port, () => {
console.log(
`Azure Storage MCP server listening on http://localhost:${port}/mcp`,
);
});
}
async function main(): Promise<void> {
if (process.argv.includes("--stdio")) {
await startStdioServer();
} else {
await startHttpServer();
}
}
main().catch((error) => {
console.error("Fatal error:", error);
process.exit(1);
});
process.on("SIGINT", () => process.exit(0));
process.on("SIGTERM", () => process.exit(0));
