mcp-mongo-server/build/index.js

A Model Context Protocol server for MongoDB connections

#!/usr/bin/env node
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { connectToMongoDB } from "./mongo.js";
import { createServer } from "./server.js";
// Declare a client variable in the global scope for cleanup handlers
let mongoClient = 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);
});