smh-mongo-mcp-server/build/index.js

A Model Context Protocol server for MongoDB connections

import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import connectToMongoDB from "./mongo.js";
import { createServer } from "./server.js";
let mongoClient = null;
const main = async () => {
    const args = process.argv.slice(2);
    let connectionUrl = "";
    let readOnlyMode = process.env.MCP_MONGODB_READONLY === "true" || false;
    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 (!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);
    }
    console.log("connectionUrl", connectionUrl);
    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);
        mongoClient = client;
        if (!isConnected || !client || !db) {
            console.error("Failed to connect to MongoDB");
            process.exit(1);
        }
        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);
    }
};
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);
});