index.ts•2.85 kB
#!/usr/bin/env node
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { connectToMongoDB } from "./mongo.js";
import { createServer } from "./server.js";
import { MongoClient } from "mongodb";
// Declare a client variable in the global scope for cleanup handlers
let mongoClient: MongoClient | null = null; // TODO: Fix Typescript error
/**
* Start the server using stdio transport and initialize MongoDB connection.
*/
async function main() {
const args = process.argv.slice(2);
// Default to environment variables
let connectionUrl = "";
let readOnlyMode = process.env.MCP_MONGODB_READONLY === "true" || false;
// Parse command line arguments (these take precedence)
for (let i = 0; i < args.length; i++) {
if (args[i] === "--read-only" || args[i] === "-r") {
readOnlyMode = true;
} else if (!connectionUrl) {
connectionUrl = args[i];
}
}
// If no connection URL from command line, use environment variable
if (!connectionUrl) {
connectionUrl = process.env.MCP_MONGODB_URI || "";
}
if (!connectionUrl) {
console.error(
"Please provide a MongoDB connection URL via command-line argument or MCP_MONGODB_URI environment variable",
);
console.error("Usage: command <mongodb-url> [--read-only|-r]");
console.error(
" or: MCP_MONGODB_URI=<mongodb-url> [MCP_MONGODB_READONLY=true] command",
);
process.exit(1);
}
// Ensure connection URL has the correct prefix
if (
!connectionUrl.startsWith("mongodb://") &&
!connectionUrl.startsWith("mongodb+srv://")
) {
console.error(
"Invalid MongoDB connection URL. URL must start with 'mongodb://' or 'mongodb+srv://'",
);
process.exit(1);
}
try {
const { client, db, isConnected, isReadOnlyMode } = await connectToMongoDB(
connectionUrl,
readOnlyMode,
);
// Store client in global variable for cleanup
mongoClient = client;
if (!isConnected || !client || !db) {
console.error("Failed to connect to MongoDB");
process.exit(1);
}
// Pass db instead of client to createServer
const server = createServer(client, db, isReadOnlyMode);
const transport = new StdioServerTransport();
await server.connect(transport);
console.warn("Server connected successfully");
} catch (error) {
console.error("Failed to connect to MongoDB:", error);
if (mongoClient) {
await mongoClient.close();
}
process.exit(1);
}
}
// Handle cleanup
process.on("SIGINT", async () => {
if (mongoClient) {
await mongoClient.close();
}
process.exit(0);
});
process.on("SIGTERM", async () => {
if (mongoClient) {
await mongoClient.close();
}
process.exit(0);
});
main().catch((error) => {
console.error("Server error:", error);
process.exit(1);
});