mongodb-mcp-server 0.1.3 → 0.3.0

package/dist/{telemetry → esm/telemetry}/telemetry.js

@@ -1,50 +1,32 @@
-import logger, { LogId } from "../logger.js";
+import { LogId } from "../common/logger.js";
 import { MACHINE_METADATA } from "./constants.js";
 import { EventCache } from "./eventCache.js";
-import nodeMachineId from "node-machine-id";
-import { getDeviceId } from "@mongodb-js/device-id";
-export const DEVICE_ID_TIMEOUT = 3000;
+import { detectContainerEnv } from "../helpers/container.js";
 export class Telemetry {
-    constructor(session, userConfig, commonProperties, { eventCache, getRawMachineId }) {
+    constructor(session, userConfig, commonProperties, { eventCache, deviceId }) {
         this.session = session;
         this.userConfig = userConfig;
         this.commonProperties = commonProperties;
         this.isBufferingEvents = true;
-        this.deviceIdAbortController = new AbortController();
         this.eventCache = eventCache;
-        this.getRawMachineId = getRawMachineId;
+        this.deviceId = deviceId;
     }
-    static create(session, userConfig, { commonProperties = { ...MACHINE_METADATA }, eventCache = EventCache.getInstance(), getRawMachineId = () => nodeMachineId.machineId(true), } = {}) {
-        const instance = new Telemetry(session, userConfig, commonProperties, { eventCache, getRawMachineId });
-        void instance.start();
+    static create(session, userConfig, deviceId, { commonProperties = { ...MACHINE_METADATA }, eventCache = EventCache.getInstance(), } = {}) {
+        const instance = new Telemetry(session, userConfig, commonProperties, { eventCache, deviceId });
+        void instance.setup();
         return instance;
     }
-    async start() {
+    async setup() {
         if (!this.isTelemetryEnabled()) {
             return;
         }
-        this.deviceIdPromise = getDeviceId({
-            getMachineId: () => this.getRawMachineId(),
-            onError: (reason, error) => {
-                switch (reason) {
-                    case "resolutionError":
-                        logger.debug(LogId.telemetryDeviceIdFailure, "telemetry", String(error));
-                        break;
-                    case "timeout":
-                        logger.debug(LogId.telemetryDeviceIdTimeout, "telemetry", "Device ID retrieval timed out");
-                        break;
-                    case "abort":
-                        // No need to log in the case of aborts
-                        break;
-                }
-            },
-            abortSignal: this.deviceIdAbortController.signal,
-        });
-        this.commonProperties.device_id = await this.deviceIdPromise;
+        this.setupPromise = Promise.all([this.deviceId.get(), detectContainerEnv()]);
+        const [deviceIdValue, containerEnv] = await this.setupPromise;
+        this.commonProperties.device_id = deviceIdValue;
+        this.commonProperties.is_container_env = containerEnv;
         this.isBufferingEvents = false;
     }
     async close() {
-        this.deviceIdAbortController.abort();
         this.isBufferingEvents = false;
         await this.emitEvents(this.eventCache.getEvents());
     }
@@ -55,13 +37,23 @@ export class Telemetry {
     async emitEvents(events) {
         try {
             if (!this.isTelemetryEnabled()) {
-                logger.info(LogId.telemetryEmitFailure, "telemetry", `Telemetry is disabled.`);
+                this.session.logger.info({
+                    id: LogId.telemetryEmitFailure,
+                    context: "telemetry",
+                    message: "Telemetry is disabled.",
+                    noRedaction: true,
+                });
                 return;
             }
             await this.emit(events);
         }
         catch {
-            logger.debug(LogId.telemetryEmitFailure, "telemetry", `Error emitting telemetry events.`);
+            this.session.logger.debug({
+                id: LogId.telemetryEmitFailure,
+                context: "telemetry",
+                message: "Error emitting telemetry events.",
+                noRedaction: true,
+            });
         }
     }
     /**
@@ -71,8 +63,9 @@ export class Telemetry {
     getCommonProperties() {
         return {
             ...this.commonProperties,
-            mcp_client_version: this.session.agentRunner?.version,
-            mcp_client_name: this.session.agentRunner?.name,
+            transport: this.userConfig.transport,
+            mcp_client_version: this.session.mcpClient?.version,
+            mcp_client_name: this.session.mcpClient?.name,
             session_id: this.session.sessionId,
             config_atlas_auth: this.session.apiClient.hasCredentials() ? "true" : "false",
             config_connection_string: this.userConfig.connectionString ? "true" : "false",
@@ -104,14 +97,26 @@ export class Telemetry {
         }
         const cachedEvents = this.eventCache.getEvents();
         const allEvents = [...cachedEvents, ...events];
-        logger.debug(LogId.telemetryEmitStart, "telemetry", `Attempting to send ${allEvents.length} events (${cachedEvents.length} cached)`);
+        this.session.logger.debug({
+            id: LogId.telemetryEmitStart,
+            context: "telemetry",
+            message: `Attempting to send ${allEvents.length} events (${cachedEvents.length} cached)`,
+        });
         const result = await this.sendEvents(this.session.apiClient, allEvents);
         if (result.success) {
             this.eventCache.clearEvents();
-            logger.debug(LogId.telemetryEmitSuccess, "telemetry", `Sent ${allEvents.length} events successfully: ${JSON.stringify(allEvents, null, 2)}`);
+            this.session.logger.debug({
+                id: LogId.telemetryEmitSuccess,
+                context: "telemetry",
+                message: `Sent ${allEvents.length} events successfully: ${JSON.stringify(allEvents, null, 2)}`,
+            });
             return;
         }
-        logger.debug(LogId.telemetryEmitFailure, "telemetry", `Error sending event to client: ${result.error instanceof Error ? result.error.message : String(result.error)}`);
+        this.session.logger.debug({
+            id: LogId.telemetryEmitFailure,
+            context: "telemetry",
+            message: `Error sending event to client: ${result.error instanceof Error ? result.error.message : String(result.error)}`,
+        });
         this.eventCache.appendEvents(events);
     }
     /**

package/dist/esm/telemetry/telemetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.js","sourceRoot":"","sources":["../../../src/telemetry/telemetry.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAE5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAQ7D,MAAM,OAAO,SAAS;IAOlB,YACqB,OAAgB,EAChB,UAAsB,EACtB,gBAAkC,EACnD,EAAE,UAAU,EAAE,QAAQ,EAAkD;QAHvD,YAAO,GAAP,OAAO,CAAS;QAChB,eAAU,GAAV,UAAU,CAAY;QACtB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAT/C,sBAAiB,GAAY,IAAI,CAAC;QAYtC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,MAAM,CACT,OAAgB,EAChB,UAAsB,EACtB,QAAkB,EAClB,EACI,gBAAgB,GAAG,EAAE,GAAG,gBAAgB,EAAE,EAC1C,UAAU,GAAG,UAAU,CAAC,WAAW,EAAE,MAIrC,EAAE;QAEN,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;QAEhG,KAAK,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,OAAO,QAAQ,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,KAAK;QACf,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;YAC7B,OAAO;QACX,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;QAC7E,MAAM,CAAC,aAAa,EAAE,YAAY,CAAC,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC;QAE9D,IAAI,CAAC,gBAAgB,CAAC,SAAS,GAAG,aAAa,CAAC;QAChD,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,GAAG,YAAY,CAAC;QAEtD,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,KAAK;QACd,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC;QAC/B,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC;IACvD,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,UAAU,CAAC,MAAmB;QACvC,IAAI,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;gBAC7B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;oBACrB,EAAE,EAAE,KAAK,CAAC,oBAAoB;oBAC9B,OAAO,EAAE,WAAW;oBACpB,OAAO,EAAE,wBAAwB;oBACjC,WAAW,EAAE,IAAI;iBACpB,CAAC,CAAC;gBACH,OAAO;YACX,CAAC;YAED,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC;QAAC,MAAM,CAAC;YACL,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;gBACtB,EAAE,EAAE,KAAK,CAAC,oBAAoB;gBAC9B,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,kCAAkC;gBAC3C,WAAW,EAAE,IAAI;aACpB,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAED;;;OAGG;IACI,mBAAmB;QACtB,OAAO;YACH,GAAG,IAAI,CAAC,gBAAgB;YACxB,SAAS,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS;YACpC,kBAAkB,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO;YACnD,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI;YAC7C,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;YAClC,iBAAiB,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;YAC7E,wBAAwB,EAAE,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;SAChF,CAAC;IACN,CAAC;IAED;;;;;;OAMG;IACI,kBAAkB;QACrB,sDAAsD;QACtD,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;YAC3C,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,MAAM,UAAU,GAAG,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC;QACjD,OAAO,CAAC,UAAU,CAAC;IACvB,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,IAAI,CAAC,MAAmB;QAClC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YACrC,OAAO;QACX,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;QACjD,MAAM,SAAS,GAAG,CAAC,GAAG,YAAY,EAAE,GAAG,MAAM,CAAC,CAAC;QAE/C,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YACtB,EAAE,EAAE,KAAK,CAAC,kBAAkB;YAC5B,OAAO,EAAE,WAAW;YACpB,OAAO,EAAE,sBAAsB,SAAS,CAAC,MAAM,YAAY,YAAY,CAAC,MAAM,UAAU;SAC3F,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACxE,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;YAC9B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;gBACtB,EAAE,EAAE,KAAK,CAAC,oBAAoB;gBAC9B,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,QAAQ,SAAS,CAAC,MAAM,yBAAyB,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE;aACjG,CAAC,CAAC;YACH,OAAO;QACX,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YACtB,EAAE,EAAE,KAAK,CAAC,oBAAoB;YAC9B,OAAO,EAAE,WAAW;YACpB,OAAO,EAAE,kCAAkC,MAAM,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;SAC3H,CAAC,CAAC;QACH,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,UAAU,CAAC,MAAiB,EAAE,MAAmB;QAC3D,IAAI,CAAC;YACD,MAAM,MAAM,CAAC,UAAU,CACnB,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBACnB,GAAG,KAAK;gBACR,UAAU,EAAE,EAAE,GAAG,IAAI,CAAC,mBAAmB,EAAE,EAAE,GAAG,KAAK,CAAC,UAAU,EAAE;aACrE,CAAC,CAAC,CACN,CAAC;YACF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO;gBACH,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;aACnE,CAAC;QACN,CAAC;IACL,CAAC;CACJ"}

package/dist/esm/telemetry/types.d.ts

@@ -0,0 +1,70 @@
+/**
+ * Result type constants for telemetry events
+ */
+export type TelemetryResult = "success" | "failure";
+export type ServerCommand = "start" | "stop";
+export type TelemetryBoolSet = "true" | "false";
+/**
+ * Base interface for all events
+ */
+export type TelemetryEvent<T> = {
+    timestamp: string;
+    source: "mdbmcp";
+    properties: T & {
+        component: string;
+        duration_ms: number;
+        result: TelemetryResult;
+        category: string;
+    };
+};
+export type BaseEvent = TelemetryEvent<unknown>;
+/**
+ * Interface for tool events
+ */
+export type ToolEventProperties = {
+    command: string;
+    error_code?: string;
+    error_type?: string;
+    project_id?: string;
+    org_id?: string;
+    cluster_name?: string;
+    is_atlas?: boolean;
+};
+export type ToolEvent = TelemetryEvent<ToolEventProperties>;
+/**
+ * Interface for server events
+ */
+export type ServerEventProperties = {
+    command: ServerCommand;
+    reason?: string;
+    startup_time_ms?: number;
+    runtime_duration_ms?: number;
+    read_only_mode?: boolean;
+    disabled_tools?: string[];
+};
+export type ServerEvent = TelemetryEvent<ServerEventProperties>;
+/**
+ * Interface for static properties, they can be fetched once and reused.
+ */
+export type CommonStaticProperties = {
+    mcp_server_version: string;
+    mcp_server_name: string;
+    platform: string;
+    arch: string;
+    os_type: string;
+    os_version?: string;
+};
+/**
+ * Common properties for all events that might change.
+ */
+export type CommonProperties = {
+    device_id?: string;
+    is_container_env?: boolean;
+    mcp_client_version?: string;
+    mcp_client_name?: string;
+    transport?: "stdio" | "http";
+    config_atlas_auth?: TelemetryBoolSet;
+    config_connection_string?: TelemetryBoolSet;
+    session_id?: string;
+} & CommonStaticProperties;
+//# sourceMappingURL=types.d.ts.map

package/dist/esm/telemetry/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/telemetry/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,SAAS,GAAG,SAAS,CAAC;AACpD,MAAM,MAAM,aAAa,GAAG,OAAO,GAAG,MAAM,CAAC;AAC7C,MAAM,MAAM,gBAAgB,GAAG,MAAM,GAAG,OAAO,CAAC;AAEhD;;GAEG;AACH,MAAM,MAAM,cAAc,CAAC,CAAC,IAAI;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,QAAQ,CAAC;IACjB,UAAU,EAAE,CAAC,GAAG;QACZ,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,MAAM,EAAE,eAAe,CAAC;QACxB,QAAQ,EAAE,MAAM,CAAC;KACpB,CAAC;CACL,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;AAEhD;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,OAAO,CAAC;CACtB,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG,cAAc,CAAC,mBAAmB,CAAC,CAAC;AAC5D;;GAEG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAChC,OAAO,EAAE,aAAa,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC7B,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG,cAAc,CAAC,qBAAqB,CAAC,CAAC;AAEhE;;GAEG;AACH,MAAM,MAAM,sBAAsB,GAAG;IACjC,kBAAkB,EAAE,MAAM,CAAC;IAC3B,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,SAAS,CAAC,EAAE,OAAO,GAAG,MAAM,CAAC;IAC7B,iBAAiB,CAAC,EAAE,gBAAgB,CAAC;IACrC,wBAAwB,CAAC,EAAE,gBAAgB,CAAC;IAC5C,UAAU,CAAC,EAAE,MAAM,CAAC;CACvB,GAAG,sBAAsB,CAAC"}

package/dist/esm/telemetry/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/telemetry/types.ts"],"names":[],"mappings":""}

package/dist/esm/tools/atlas/atlasTool.d.ts

@@ -0,0 +1,18 @@
+import type { ToolCategory, TelemetryToolMetadata, ToolArgs } from "../tool.js";
+import { ToolBase } from "../tool.js";
+import type { ToolCallback } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+export declare abstract class AtlasToolBase extends ToolBase {
+    category: ToolCategory;
+    protected verifyAllowed(): boolean;
+    protected handleError(error: unknown, args: ToolArgs<typeof this.argsShape>): Promise<CallToolResult> | CallToolResult;
+    /**
+     *
+     * Resolves the tool metadata from the arguments passed to the tool
+     *
+     * @param args - The arguments passed to the tool
+     * @returns The tool metadata
+     */
+    protected resolveTelemetryMetadata(...args: Parameters<ToolCallback<typeof this.argsShape>>): TelemetryToolMetadata;
+}
+//# sourceMappingURL=atlasTool.d.ts.map

package/dist/esm/tools/atlas/atlasTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"atlasTool.d.ts","sourceRoot":"","sources":["../../../../src/tools/atlas/atlasTool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,qBAAqB,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAChF,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAKzE,8BAAsB,aAAc,SAAQ,QAAQ;IACzC,QAAQ,EAAE,YAAY,CAAW;IAExC,SAAS,CAAC,aAAa,IAAI,OAAO;IAOlC,SAAS,CAAC,WAAW,CACjB,KAAK,EAAE,OAAO,EACd,IAAI,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,GACtC,OAAO,CAAC,cAAc,CAAC,GAAG,cAAc;IAyC3C;;;;;;OAMG;IACH,SAAS,CAAC,wBAAwB,CAC9B,GAAG,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,GACzD,qBAAqB;CAgC3B"}

package/dist/{tools → esm/tools}/atlas/atlasTool.js

@@ -1,5 +1,5 @@
 import { ToolBase } from "../tool.js";
-import logger, { LogId } from "../../logger.js";
+import { LogId } from "../../common/logger.js";
 import { z } from "zod";
 import { ApiClientError } from "../../common/atlas/apiClientError.js";
 export class AtlasToolBase extends ToolBase {
@@ -23,7 +23,7 @@ export class AtlasToolBase extends ToolBase {
                             type: "text",
                             text: `Unable to authenticate with MongoDB Atlas, API error: ${error.message}
 
-Hint: Your API credentials may be invalid, expired or lack permissions. 
+Hint: Your API credentials may be invalid, expired or lack permissions.
 Please check your Atlas API credentials and ensure they have the appropriate permissions.
 For more information on setting up API keys, visit: https://www.mongodb.com/docs/atlas/configure-api-access/`,
                         },
@@ -37,7 +37,7 @@ For more information on setting up API keys, visit: https://www.mongodb.com/docs
                         {
                             type: "text",
                             text: `Received a Forbidden API Error: ${error.message}
-                            
+
 You don't have sufficient permissions to perform this action in MongoDB Atlas
 Please ensure your API key has the necessary roles assigned.
 For more information on Atlas API access roles, visit: https://www.mongodb.com/docs/atlas/api/service-accounts-overview/`,
@@ -66,7 +66,11 @@ For more information on Atlas API access roles, visit: https://www.mongodb.com/d
         const argsShape = z.object(this.argsShape);
         const parsedResult = argsShape.safeParse(args[0]);
         if (!parsedResult.success) {
-            logger.debug(LogId.telemetryMetadataError, "tool", `Error parsing tool arguments: ${parsedResult.error.message}`);
+            this.session.logger.debug({
+                id: LogId.telemetryMetadataError,
+                context: "tool",
+                message: `Error parsing tool arguments: ${parsedResult.error.message}`,
+            });
             return toolMetadata;
         }
         const data = parsedResult.data;

package/dist/esm/tools/atlas/atlasTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"atlasTool.js","sourceRoot":"","sources":["../../../../src/tools/atlas/atlasTool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAGtC,OAAO,EAAE,KAAK,EAAE,MAAM,wBAAwB,CAAC;AAC/C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,sCAAsC,CAAC;AAEtE,MAAM,OAAgB,aAAc,SAAQ,QAAQ;IAApD;;QACW,aAAQ,GAAiB,OAAO,CAAC;IA8F5C,CAAC;IA5Fa,aAAa;QACnB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;YAC3D,OAAO,KAAK,CAAC;QACjB,CAAC;QACD,OAAO,KAAK,CAAC,aAAa,EAAE,CAAC;IACjC,CAAC;IAES,WAAW,CACjB,KAAc,EACd,IAAqC;QAErC,IAAI,KAAK,YAAY,cAAc,EAAE,CAAC;YAClC,MAAM,UAAU,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YAEzC,IAAI,UAAU,KAAK,GAAG,EAAE,CAAC;gBACrB,OAAO;oBACH,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,yDAAyD,KAAK,CAAC,OAAO;;;;6GAIK;yBACpF;qBACJ;oBACD,OAAO,EAAE,IAAI;iBAChB,CAAC;YACN,CAAC;YAED,IAAI,UAAU,KAAK,GAAG,EAAE,CAAC;gBACrB,OAAO;oBACH,OAAO,EAAE;wBACL;4BACI,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,mCAAmC,KAAK,CAAC,OAAO;;;;yHAIuC;yBAChG;qBACJ;oBACD,OAAO,EAAE,IAAI;iBAChB,CAAC;YACN,CAAC;QACL,CAAC;QAED,gFAAgF;QAChF,OAAO,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED;;;;;;OAMG;IACO,wBAAwB,CAC9B,GAAG,IAAqD;QAExD,MAAM,YAAY,GAA0B,EAAE,CAAC;QAC/C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACf,OAAO,YAAY,CAAC;QACxB,CAAC;QAED,sDAAsD;QACtD,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC3C,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAElD,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;gBACtB,EAAE,EAAE,KAAK,CAAC,sBAAsB;gBAChC,OAAO,EAAE,MAAM;gBACf,OAAO,EAAE,iCAAiC,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE;aACzE,CAAC,CAAC;YACH,OAAO,YAAY,CAAC;QACxB,CAAC;QAED,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC;QAE/B,qCAAqC;QACrC,IAAI,WAAW,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAC5F,YAAY,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAC5C,CAAC;QAED,iCAAiC;QACjC,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAChF,YAAY,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACpC,CAAC;QACD,OAAO,YAAY,CAAC;IACxB,CAAC;CACJ"}

package/dist/esm/tools/atlas/connect/connectCluster.d.ts

@@ -0,0 +1,18 @@
+import { z } from "zod";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { AtlasToolBase } from "../atlasTool.js";
+import type { ToolArgs, OperationType } from "../../tool.js";
+export declare class ConnectClusterTool extends AtlasToolBase {
+    name: string;
+    protected description: string;
+    operationType: OperationType;
+    protected argsShape: {
+        projectId: z.ZodString;
+        clusterName: z.ZodString;
+    };
+    private queryConnection;
+    private prepareClusterConnection;
+    private connectToCluster;
+    protected execute({ projectId, clusterName }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult>;
+}
+//# sourceMappingURL=connectCluster.d.ts.map

package/dist/esm/tools/atlas/connect/connectCluster.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"connectCluster.d.ts","sourceRoot":"","sources":["../../../../../src/tools/atlas/connect/connectCluster.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAc7D,qBAAa,kBAAmB,SAAQ,aAAa;IAC1C,IAAI,SAA2B;IACtC,SAAS,CAAC,WAAW,SAAsC;IACpD,aAAa,EAAE,aAAa,CAAa;IAChD,SAAS,CAAC,SAAS;;;MAGjB;IAEF,OAAO,CAAC,eAAe;YAmCT,wBAAwB;YAsDxB,gBAAgB;cA4Ed,OAAO,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;CA0DhH"}

package/dist/esm/tools/atlas/connect/connectCluster.js

@@ -0,0 +1,211 @@
+import { z } from "zod";
+import { AtlasToolBase } from "../atlasTool.js";
+import { generateSecurePassword } from "../../../helpers/generatePassword.js";
+import { LogId } from "../../../common/logger.js";
+import { inspectCluster } from "../../../common/atlas/cluster.js";
+import { ensureCurrentIpInAccessList } from "../../../common/atlas/accessListUtils.js";
+import { getDefaultRoleFromConfig } from "../../../common/atlas/roles.js";
+const EXPIRY_MS = 1000 * 60 * 60 * 12; // 12 hours
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+export class ConnectClusterTool extends AtlasToolBase {
+    constructor() {
+        super(...arguments);
+        this.name = "atlas-connect-cluster";
+        this.description = "Connect to MongoDB Atlas cluster";
+        this.operationType = "connect";
+        this.argsShape = {
+            projectId: z.string().describe("Atlas project ID"),
+            clusterName: z.string().describe("Atlas cluster name"),
+        };
+    }
+    queryConnection(projectId, clusterName) {
+        if (!this.session.connectedAtlasCluster) {
+            if (this.session.isConnectedToMongoDB) {
+                return "connected-to-other-cluster";
+            }
+            return "disconnected";
+        }
+        const currentConectionState = this.session.connectionManager.currentConnectionState;
+        if (this.session.connectedAtlasCluster.projectId !== projectId ||
+            this.session.connectedAtlasCluster.clusterName !== clusterName) {
+            return "connected-to-other-cluster";
+        }
+        switch (currentConectionState.tag) {
+            case "connecting":
+            case "disconnected": // we might still be calling Atlas APIs and not attempted yet to connect to MongoDB, but we are still "connecting"
+                return "connecting";
+            case "connected":
+                return "connected";
+            case "errored":
+                this.session.logger.debug({
+                    id: LogId.atlasConnectFailure,
+                    context: "atlas-connect-cluster",
+                    message: `error querying cluster: ${currentConectionState.errorReason}`,
+                });
+                return "unknown";
+        }
+    }
+    async prepareClusterConnection(projectId, clusterName) {
+        const cluster = await inspectCluster(this.session.apiClient, projectId, clusterName);
+        if (!cluster.connectionString) {
+            throw new Error("Connection string not available");
+        }
+        const username = `mcpUser${Math.floor(Math.random() * 100000)}`;
+        const password = await generateSecurePassword();
+        const expiryDate = new Date(Date.now() + EXPIRY_MS);
+        const role = getDefaultRoleFromConfig(this.config);
+        await this.session.apiClient.createDatabaseUser({
+            params: {
+                path: {
+                    groupId: projectId,
+                },
+            },
+            body: {
+                databaseName: "admin",
+                groupId: projectId,
+                roles: [role],
+                scopes: [{ type: "CLUSTER", name: clusterName }],
+                username,
+                password,
+                awsIAMType: "NONE",
+                ldapAuthType: "NONE",
+                oidcAuthType: "NONE",
+                x509Type: "NONE",
+                deleteAfterDate: expiryDate.toISOString(),
+                description: "MDB MCP Temporary user, see https://dochub.mongodb.org/core/mongodb-mcp-server-tools-considerations",
+            },
+        });
+        const connectedAtlasCluster = {
+            username,
+            projectId,
+            clusterName,
+            expiryDate,
+        };
+        const cn = new URL(cluster.connectionString);
+        cn.username = username;
+        cn.password = password;
+        cn.searchParams.set("authSource", "admin");
+        return { connectionString: cn.toString(), atlas: connectedAtlasCluster };
+    }
+    async connectToCluster(connectionString, atlas) {
+        let lastError = undefined;
+        this.session.logger.debug({
+            id: LogId.atlasConnectAttempt,
+            context: "atlas-connect-cluster",
+            message: `attempting to connect to cluster: ${this.session.connectedAtlasCluster?.clusterName}`,
+            noRedaction: true,
+        });
+        // try to connect for about 5 minutes
+        for (let i = 0; i < 600; i++) {
+            try {
+                lastError = undefined;
+                await this.session.connectToMongoDB({ connectionString, atlas });
+                break;
+            }
+            catch (err) {
+                const error = err instanceof Error ? err : new Error(String(err));
+                lastError = error;
+                this.session.logger.debug({
+                    id: LogId.atlasConnectFailure,
+                    context: "atlas-connect-cluster",
+                    message: `error connecting to cluster: ${error.message}`,
+                });
+                await sleep(500); // wait for 500ms before retrying
+            }
+            if (!this.session.connectedAtlasCluster ||
+                this.session.connectedAtlasCluster.projectId !== atlas.projectId ||
+                this.session.connectedAtlasCluster.clusterName !== atlas.clusterName) {
+                throw new Error("Cluster connection aborted");
+            }
+        }
+        if (lastError) {
+            if (this.session.connectedAtlasCluster?.projectId === atlas.projectId &&
+                this.session.connectedAtlasCluster?.clusterName === atlas.clusterName &&
+                this.session.connectedAtlasCluster?.username) {
+                void this.session.apiClient
+                    .deleteDatabaseUser({
+                    params: {
+                        path: {
+                            groupId: this.session.connectedAtlasCluster.projectId,
+                            username: this.session.connectedAtlasCluster.username,
+                            databaseName: "admin",
+                        },
+                    },
+                })
+                    .catch((err) => {
+                    const error = err instanceof Error ? err : new Error(String(err));
+                    this.session.logger.debug({
+                        id: LogId.atlasConnectFailure,
+                        context: "atlas-connect-cluster",
+                        message: `error deleting database user: ${error.message}`,
+                    });
+                });
+            }
+            throw lastError;
+        }
+        this.session.logger.debug({
+            id: LogId.atlasConnectSucceeded,
+            context: "atlas-connect-cluster",
+            message: `connected to cluster: ${this.session.connectedAtlasCluster?.clusterName}`,
+            noRedaction: true,
+        });
+    }
+    async execute({ projectId, clusterName }) {
+        await ensureCurrentIpInAccessList(this.session.apiClient, projectId);
+        for (let i = 0; i < 60; i++) {
+            const state = this.queryConnection(projectId, clusterName);
+            switch (state) {
+                case "connected": {
+                    return {
+                        content: [
+                            {
+                                type: "text",
+                                text: `Connected to cluster "${clusterName}".`,
+                            },
+                        ],
+                    };
+                }
+                case "connecting":
+                case "unknown": {
+                    break;
+                }
+                case "connected-to-other-cluster":
+                case "disconnected":
+                default: {
+                    await this.session.disconnect();
+                    const { connectionString, atlas } = await this.prepareClusterConnection(projectId, clusterName);
+                    // try to connect for about 5 minutes asynchronously
+                    void this.connectToCluster(connectionString, atlas).catch((err) => {
+                        const error = err instanceof Error ? err : new Error(String(err));
+                        this.session.logger.error({
+                            id: LogId.atlasConnectFailure,
+                            context: "atlas-connect-cluster",
+                            message: `error connecting to cluster: ${error.message}`,
+                        });
+                    });
+                    break;
+                }
+            }
+            await sleep(500);
+        }
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `Attempting to connect to cluster "${clusterName}"...`,
+                },
+                {
+                    type: "text",
+                    text: `Warning: Provisioning a user and connecting to the cluster may take more time, please check again in a few seconds.`,
+                },
+                {
+                    type: "text",
+                    text: `Warning: Make sure your IP address was enabled in the allow list setting of the Atlas cluster.`,
+                },
+            ],
+        };
+    }
+}
+//# sourceMappingURL=connectCluster.js.map

package/dist/esm/tools/atlas/connect/connectCluster.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connectCluster.js","sourceRoot":"","sources":["../../../../../src/tools/atlas/connect/connectCluster.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAAE,sBAAsB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,KAAK,EAAE,MAAM,2BAA2B,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AAEvF,OAAO,EAAE,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAE1E,MAAM,SAAS,GAAG,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,WAAW;AAElD,SAAS,KAAK,CAAC,EAAU;IACrB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC7D,CAAC;AAED,MAAM,OAAO,kBAAmB,SAAQ,aAAa;IAArD;;QACW,SAAI,GAAG,uBAAuB,CAAC;QAC5B,gBAAW,GAAG,kCAAkC,CAAC;QACpD,kBAAa,GAAkB,SAAS,CAAC;QACtC,cAAS,GAAG;YAClB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;YAClD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;SACzD,CAAC;IAiON,CAAC;IA/NW,eAAe,CACnB,SAAiB,EACjB,WAAmB;QAEnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;YACtC,IAAI,IAAI,CAAC,OAAO,CAAC,oBAAoB,EAAE,CAAC;gBACpC,OAAO,4BAA4B,CAAC;YACxC,CAAC;YACD,OAAO,cAAc,CAAC;QAC1B,CAAC;QAED,MAAM,qBAAqB,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,sBAAsB,CAAC;QACpF,IACI,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,SAAS,KAAK,SAAS;YAC1D,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,WAAW,KAAK,WAAW,EAChE,CAAC;YACC,OAAO,4BAA4B,CAAC;QACxC,CAAC;QAED,QAAQ,qBAAqB,CAAC,GAAG,EAAE,CAAC;YAChC,KAAK,YAAY,CAAC;YAClB,KAAK,cAAc,EAAE,kHAAkH;gBACnI,OAAO,YAAY,CAAC;YACxB,KAAK,WAAW;gBACZ,OAAO,WAAW,CAAC;YACvB,KAAK,SAAS;gBACV,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;oBACtB,EAAE,EAAE,KAAK,CAAC,mBAAmB;oBAC7B,OAAO,EAAE,uBAAuB;oBAChC,OAAO,EAAE,2BAA2B,qBAAqB,CAAC,WAAW,EAAE;iBAC1E,CAAC,CAAC;gBACH,OAAO,SAAS,CAAC;QACzB,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,wBAAwB,CAClC,SAAiB,EACjB,WAAmB;QAEnB,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC;QAErF,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,QAAQ,GAAG,UAAU,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QAChE,MAAM,QAAQ,GAAG,MAAM,sBAAsB,EAAE,CAAC;QAEhD,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;QACpD,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEnD,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,kBAAkB,CAAC;YAC5C,MAAM,EAAE;gBACJ,IAAI,EAAE;oBACF,OAAO,EAAE,SAAS;iBACrB;aACJ;YACD,IAAI,EAAE;gBACF,YAAY,EAAE,OAAO;gBACrB,OAAO,EAAE,SAAS;gBAClB,KAAK,EAAE,CAAC,IAAI,CAAC;gBACb,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;gBAChD,QAAQ;gBACR,QAAQ;gBACR,UAAU,EAAE,MAAM;gBAClB,YAAY,EAAE,MAAM;gBACpB,YAAY,EAAE,MAAM;gBACpB,QAAQ,EAAE,MAAM;gBAChB,eAAe,EAAE,UAAU,CAAC,WAAW,EAAE;gBACzC,WAAW,EACP,qGAAqG;aAC5G;SACJ,CAAC,CAAC;QAEH,MAAM,qBAAqB,GAAG;YAC1B,QAAQ;YACR,SAAS;YACT,WAAW;YACX,UAAU;SACb,CAAC;QAEF,MAAM,EAAE,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAC7C,EAAE,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACvB,EAAE,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACvB,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAE3C,OAAO,EAAE,gBAAgB,EAAE,EAAE,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE,CAAC;IAC7E,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,gBAAwB,EAAE,KAAiC;QACtF,IAAI,SAAS,GAAsB,SAAS,CAAC;QAE7C,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YACtB,EAAE,EAAE,KAAK,CAAC,mBAAmB;YAC7B,OAAO,EAAE,uBAAuB;YAChC,OAAO,EAAE,qCAAqC,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,WAAW,EAAE;YAC/F,WAAW,EAAE,IAAI;SACpB,CAAC,CAAC;QAEH,qCAAqC;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACD,SAAS,GAAG,SAAS,CAAC;gBAEtB,MAAM,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;gBACjE,MAAM;YACV,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAElE,SAAS,GAAG,KAAK,CAAC;gBAElB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;oBACtB,EAAE,EAAE,KAAK,CAAC,mBAAmB;oBAC7B,OAAO,EAAE,uBAAuB;oBAChC,OAAO,EAAE,gCAAgC,KAAK,CAAC,OAAO,EAAE;iBAC3D,CAAC,CAAC;gBAEH,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,iCAAiC;YACvD,CAAC;YAED,IACI,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB;gBACnC,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,SAAS,KAAK,KAAK,CAAC,SAAS;gBAChE,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,WAAW,KAAK,KAAK,CAAC,WAAW,EACtE,CAAC;gBACC,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAClD,CAAC;QACL,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACZ,IACI,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,SAAS,KAAK,KAAK,CAAC,SAAS;gBACjE,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,WAAW,KAAK,KAAK,CAAC,WAAW;gBACrE,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,QAAQ,EAC9C,CAAC;gBACC,KAAK,IAAI,CAAC,OAAO,CAAC,SAAS;qBACtB,kBAAkB,CAAC;oBAChB,MAAM,EAAE;wBACJ,IAAI,EAAE;4BACF,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,SAAS;4BACrD,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,QAAQ;4BACrD,YAAY,EAAE,OAAO;yBACxB;qBACJ;iBACJ,CAAC;qBACD,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;oBACpB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;oBAClE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;wBACtB,EAAE,EAAE,KAAK,CAAC,mBAAmB;wBAC7B,OAAO,EAAE,uBAAuB;wBAChC,OAAO,EAAE,iCAAiC,KAAK,CAAC,OAAO,EAAE;qBAC5D,CAAC,CAAC;gBACP,CAAC,CAAC,CAAC;YACX,CAAC;YACD,MAAM,SAAS,CAAC;QACpB,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;YACtB,EAAE,EAAE,KAAK,CAAC,qBAAqB;YAC/B,OAAO,EAAE,uBAAuB;YAChC,OAAO,EAAE,yBAAyB,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,WAAW,EAAE;YACnF,WAAW,EAAE,IAAI;SACpB,CAAC,CAAC;IACP,CAAC;IAES,KAAK,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,WAAW,EAAmC;QAC/E,MAAM,2BAA2B,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACrE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YAC3D,QAAQ,KAAK,EAAE,CAAC;gBACZ,KAAK,WAAW,CAAC,CAAC,CAAC;oBACf,OAAO;wBACH,OAAO,EAAE;4BACL;gCACI,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,yBAAyB,WAAW,IAAI;6BACjD;yBACJ;qBACJ,CAAC;gBACN,CAAC;gBACD,KAAK,YAAY,CAAC;gBAClB,KAAK,SAAS,CAAC,CAAC,CAAC;oBACb,MAAM;gBACV,CAAC;gBACD,KAAK,4BAA4B,CAAC;gBAClC,KAAK,cAAc,CAAC;gBACpB,OAAO,CAAC,CAAC,CAAC;oBACN,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;oBAChC,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,wBAAwB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;oBAEhG,oDAAoD;oBACpD,KAAK,IAAI,CAAC,gBAAgB,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;wBACvE,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;wBAClE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;4BACtB,EAAE,EAAE,KAAK,CAAC,mBAAmB;4BAC7B,OAAO,EAAE,uBAAuB;4BAChC,OAAO,EAAE,gCAAgC,KAAK,CAAC,OAAO,EAAE;yBAC3D,CAAC,CAAC;oBACP,CAAC,CAAC,CAAC;oBACH,MAAM;gBACV,CAAC;YACL,CAAC;YAED,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;QAED,OAAO;YACH,OAAO,EAAE;gBACL;oBACI,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,qCAAqC,WAAW,MAAM;iBAC/D;gBACD;oBACI,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,qHAAqH;iBAC9H;gBACD;oBACI,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,gGAAgG;iBACzG;aACJ;SACJ,CAAC;IACN,CAAC;CACJ"}

package/dist/esm/tools/atlas/create/createAccessList.d.ts

@@ -0,0 +1,18 @@
+import { z } from "zod";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { AtlasToolBase } from "../atlasTool.js";
+import type { ToolArgs, OperationType } from "../../tool.js";
+export declare class CreateAccessListTool extends AtlasToolBase {
+    name: string;
+    protected description: string;
+    operationType: OperationType;
+    protected argsShape: {
+        projectId: z.ZodString;
+        ipAddresses: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+        cidrBlocks: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+        currentIpAddress: z.ZodDefault<z.ZodBoolean>;
+        comment: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+    };
+    protected execute({ projectId, ipAddresses, cidrBlocks, comment, currentIpAddress, }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult>;
+}
+//# sourceMappingURL=createAccessList.d.ts.map

package/dist/esm/tools/atlas/create/createAccessList.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createAccessList.d.ts","sourceRoot":"","sources":["../../../../../src/tools/atlas/create/createAccessList.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAG7D,qBAAa,oBAAqB,SAAQ,aAAa;IAC5C,IAAI,SAA8B;IACzC,SAAS,CAAC,WAAW,SAAiE;IAC/E,aAAa,EAAE,aAAa,CAAY;IAC/C,SAAS,CAAC,SAAS;;;;;;MAajB;cAEc,OAAO,CAAC,EACpB,SAAS,EACT,WAAW,EACX,UAAU,EACV,OAAO,EACP,gBAAgB,GACnB,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;CA8C/D"}

package/dist/{tools → esm/tools}/atlas/create/createAccessList.js

@@ -1,6 +1,6 @@
 import { z } from "zod";
 import { AtlasToolBase } from "../atlasTool.js";
-const DEFAULT_COMMENT = "Added by Atlas MCP";
+import { makeCurrentIpAccessListEntry, DEFAULT_ACCESS_LIST_COMMENT } from "../../../common/atlas/accessListUtils.js";
 export class CreateAccessListTool extends AtlasToolBase {
     constructor() {
         super(...arguments);
@@ -15,7 +15,11 @@ export class CreateAccessListTool extends AtlasToolBase {
                 .optional(),
             cidrBlocks: z.array(z.string().cidr()).describe("CIDR blocks to allow access from").optional(),
             currentIpAddress: z.boolean().describe("Add the current IP address").default(false),
-            comment: z.string().describe("Comment for the access list entries").default(DEFAULT_COMMENT).optional(),
+            comment: z
+                .string()
+                .describe("Comment for the access list entries")
+                .default(DEFAULT_ACCESS_LIST_COMMENT)
+                .optional(),
         };
     }
     async execute({ projectId, ipAddresses, cidrBlocks, comment, currentIpAddress, }) {
@@ -25,21 +29,16 @@ export class CreateAccessListTool extends AtlasToolBase {
         const ipInputs = (ipAddresses || []).map((ipAddress) => ({
             groupId: projectId,
             ipAddress,
-            comment: comment || DEFAULT_COMMENT,
+            comment: comment || DEFAULT_ACCESS_LIST_COMMENT,
         }));
         if (currentIpAddress) {
-            const currentIp = await this.session.apiClient.getIpInfo();
-            const input = {
-                groupId: projectId,
-                ipAddress: currentIp.currentIpv4Address,
-                comment: comment || DEFAULT_COMMENT,
-            };
+            const input = await makeCurrentIpAccessListEntry(this.session.apiClient, projectId, comment || DEFAULT_ACCESS_LIST_COMMENT);
             ipInputs.push(input);
         }
         const cidrInputs = (cidrBlocks || []).map((cidrBlock) => ({
             groupId: projectId,
             cidrBlock,
-            comment: comment || DEFAULT_COMMENT,
+            comment: comment || DEFAULT_ACCESS_LIST_COMMENT,
         }));
         const inputs = [...ipInputs, ...cidrInputs];
         await this.session.apiClient.createProjectIpAccessList({

package/dist/esm/tools/atlas/create/createAccessList.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createAccessList.js","sourceRoot":"","sources":["../../../../../src/tools/atlas/create/createAccessList.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAAE,4BAA4B,EAAE,2BAA2B,EAAE,MAAM,0CAA0C,CAAC;AAErH,MAAM,OAAO,oBAAqB,SAAQ,aAAa;IAAvD;;QACW,SAAI,GAAG,0BAA0B,CAAC;QAC/B,gBAAW,GAAG,6DAA6D,CAAC;QAC/E,kBAAa,GAAkB,QAAQ,CAAC;QACrC,cAAS,GAAG;YAClB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;YAClD,WAAW,EAAE,CAAC;iBACT,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;iBACvC,QAAQ,CAAC,mCAAmC,CAAC;iBAC7C,QAAQ,EAAE;YACf,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,kCAAkC,CAAC,CAAC,QAAQ,EAAE;YAC9F,gBAAgB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;YACnF,OAAO,EAAE,CAAC;iBACL,MAAM,EAAE;iBACR,QAAQ,CAAC,qCAAqC,CAAC;iBAC/C,OAAO,CAAC,2BAA2B,CAAC;iBACpC,QAAQ,EAAE;SAClB,CAAC;IAsDN,CAAC;IApDa,KAAK,CAAC,OAAO,CAAC,EACpB,SAAS,EACT,WAAW,EACX,UAAU,EACV,OAAO,EACP,gBAAgB,GACc;QAC9B,IAAI,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACnE,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;QAC3F,CAAC;QAED,MAAM,QAAQ,GAAG,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACrD,OAAO,EAAE,SAAS;YAClB,SAAS;YACT,OAAO,EAAE,OAAO,IAAI,2BAA2B;SAClD,CAAC,CAAC,CAAC;QAEJ,IAAI,gBAAgB,EAAE,CAAC;YACnB,MAAM,KAAK,GAAG,MAAM,4BAA4B,CAC5C,IAAI,CAAC,OAAO,CAAC,SAAS,EACtB,SAAS,EACT,OAAO,IAAI,2BAA2B,CACzC,CAAC;YACF,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YACtD,OAAO,EAAE,SAAS;YAClB,SAAS;YACT,OAAO,EAAE,OAAO,IAAI,2BAA2B;SAClD,CAAC,CAAC,CAAC;QAEJ,MAAM,MAAM,GAAG,CAAC,GAAG,QAAQ,EAAE,GAAG,UAAU,CAAC,CAAC;QAE5C,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,yBAAyB,CAAC;YACnD,MAAM,EAAE;gBACJ,IAAI,EAAE;oBACF,OAAO,EAAE,SAAS;iBACrB;aACJ;YACD,IAAI,EAAE,MAAM;SACf,CAAC,CAAC;QAEH,OAAO;YACH,OAAO,EAAE;gBACL;oBACI,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,mDAAmD,SAAS,GAAG;iBACxE;aACJ;SACJ,CAAC;IACN,CAAC;CACJ"}

package/dist/esm/tools/atlas/create/createDBUser.d.ts

@@ -0,0 +1,30 @@
+import { z } from "zod";
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { AtlasToolBase } from "../atlasTool.js";
+import type { ToolArgs, OperationType } from "../../tool.js";
+export declare class CreateDBUserTool extends AtlasToolBase {
+    name: string;
+    protected description: string;
+    operationType: OperationType;
+    protected argsShape: {
+        projectId: z.ZodString;
+        username: z.ZodString;
+        password: z.ZodNullable<z.ZodOptional<z.ZodString>>;
+        roles: z.ZodArray<z.ZodObject<{
+            roleName: z.ZodString;
+            databaseName: z.ZodDefault<z.ZodString>;
+            collectionName: z.ZodOptional<z.ZodString>;
+        }, "strip", z.ZodTypeAny, {
+            roleName: string;
+            databaseName: string;
+            collectionName?: string | undefined;
+        }, {
+            roleName: string;
+            databaseName?: string | undefined;
+            collectionName?: string | undefined;
+        }>, "many">;
+        clusters: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    };
+    protected execute({ projectId, username, password, roles, clusters, }: ToolArgs<typeof this.argsShape>): Promise<CallToolResult>;
+}
+//# sourceMappingURL=createDBUser.d.ts.map