mongodb-mcp-server 0.2.0 → 0.3.1-prerelease.1

package/dist/cjs/common/logger.js

@@ -0,0 +1,275 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NullLogger = exports.CompositeLogger = exports.McpLogger = exports.DiskLogger = exports.ConsoleLogger = exports.LoggerBase = exports.LogId = void 0;
+const promises_1 = __importDefault(require("fs/promises"));
+const mongodb_log_writer_1 = require("mongodb-log-writer");
+const mongodb_redact_1 = require("mongodb-redact");
+const events_1 = require("events");
+exports.LogId = {
+    serverStartFailure: (0, mongodb_log_writer_1.mongoLogId)(1000001),
+    serverInitialized: (0, mongodb_log_writer_1.mongoLogId)(1000002),
+    serverCloseRequested: (0, mongodb_log_writer_1.mongoLogId)(1000003),
+    serverClosed: (0, mongodb_log_writer_1.mongoLogId)(1000004),
+    serverCloseFailure: (0, mongodb_log_writer_1.mongoLogId)(1000005),
+    serverDuplicateLoggers: (0, mongodb_log_writer_1.mongoLogId)(1000006),
+    serverMcpClientSet: (0, mongodb_log_writer_1.mongoLogId)(1000007),
+    atlasCheckCredentials: (0, mongodb_log_writer_1.mongoLogId)(1001001),
+    atlasDeleteDatabaseUserFailure: (0, mongodb_log_writer_1.mongoLogId)(1001002),
+    atlasConnectFailure: (0, mongodb_log_writer_1.mongoLogId)(1001003),
+    atlasInspectFailure: (0, mongodb_log_writer_1.mongoLogId)(1001004),
+    atlasConnectAttempt: (0, mongodb_log_writer_1.mongoLogId)(1001005),
+    atlasConnectSucceeded: (0, mongodb_log_writer_1.mongoLogId)(1001006),
+    atlasApiRevokeFailure: (0, mongodb_log_writer_1.mongoLogId)(1001007),
+    atlasIpAccessListAdded: (0, mongodb_log_writer_1.mongoLogId)(1001008),
+    atlasIpAccessListAddFailure: (0, mongodb_log_writer_1.mongoLogId)(1001009),
+    telemetryDisabled: (0, mongodb_log_writer_1.mongoLogId)(1002001),
+    telemetryEmitFailure: (0, mongodb_log_writer_1.mongoLogId)(1002002),
+    telemetryEmitStart: (0, mongodb_log_writer_1.mongoLogId)(1002003),
+    telemetryEmitSuccess: (0, mongodb_log_writer_1.mongoLogId)(1002004),
+    telemetryMetadataError: (0, mongodb_log_writer_1.mongoLogId)(1002005),
+    deviceIdResolutionError: (0, mongodb_log_writer_1.mongoLogId)(1002006),
+    deviceIdTimeout: (0, mongodb_log_writer_1.mongoLogId)(1002007),
+    telemetryClose: (0, mongodb_log_writer_1.mongoLogId)(1002008),
+    toolExecute: (0, mongodb_log_writer_1.mongoLogId)(1003001),
+    toolExecuteFailure: (0, mongodb_log_writer_1.mongoLogId)(1003002),
+    toolDisabled: (0, mongodb_log_writer_1.mongoLogId)(1003003),
+    mongodbConnectFailure: (0, mongodb_log_writer_1.mongoLogId)(1004001),
+    mongodbDisconnectFailure: (0, mongodb_log_writer_1.mongoLogId)(1004002),
+    mongodbConnectTry: (0, mongodb_log_writer_1.mongoLogId)(1004003),
+    toolUpdateFailure: (0, mongodb_log_writer_1.mongoLogId)(1005001),
+    resourceUpdateFailure: (0, mongodb_log_writer_1.mongoLogId)(1005002),
+    streamableHttpTransportStarted: (0, mongodb_log_writer_1.mongoLogId)(1006001),
+    streamableHttpTransportSessionCloseFailure: (0, mongodb_log_writer_1.mongoLogId)(1006002),
+    streamableHttpTransportSessionCloseNotification: (0, mongodb_log_writer_1.mongoLogId)(1006003),
+    streamableHttpTransportSessionCloseNotificationFailure: (0, mongodb_log_writer_1.mongoLogId)(1006004),
+    streamableHttpTransportRequestFailure: (0, mongodb_log_writer_1.mongoLogId)(1006005),
+    streamableHttpTransportCloseFailure: (0, mongodb_log_writer_1.mongoLogId)(1006006),
+    streamableHttpTransportKeepAliveFailure: (0, mongodb_log_writer_1.mongoLogId)(1006007),
+    streamableHttpTransportKeepAlive: (0, mongodb_log_writer_1.mongoLogId)(1006008),
+    exportCleanupError: (0, mongodb_log_writer_1.mongoLogId)(1007001),
+    exportCreationError: (0, mongodb_log_writer_1.mongoLogId)(1007002),
+    exportCreationCleanupError: (0, mongodb_log_writer_1.mongoLogId)(1007003),
+    exportReadError: (0, mongodb_log_writer_1.mongoLogId)(1007004),
+    exportCloseError: (0, mongodb_log_writer_1.mongoLogId)(1007005),
+    exportedDataListError: (0, mongodb_log_writer_1.mongoLogId)(1007006),
+    exportedDataAutoCompleteError: (0, mongodb_log_writer_1.mongoLogId)(1007007),
+    exportLockError: (0, mongodb_log_writer_1.mongoLogId)(1007008),
+    oidcFlow: (0, mongodb_log_writer_1.mongoLogId)(1008001),
+};
+class LoggerBase extends events_1.EventEmitter {
+    constructor(keychain) {
+        super();
+        this.keychain = keychain;
+        this.defaultUnredactedLogger = "mcp";
+    }
+    log(level, payload) {
+        // If no explicit value is supplied for unredacted loggers, default to "mcp"
+        const noRedaction = payload.noRedaction !== undefined ? payload.noRedaction : this.defaultUnredactedLogger;
+        this.logCore(level, {
+            ...payload,
+            message: this.redactIfNecessary(payload.message, noRedaction),
+        });
+    }
+    redactIfNecessary(message, noRedaction) {
+        if (typeof noRedaction === "boolean" && noRedaction) {
+            // If the consumer has supplied noRedaction: true, we don't redact the log message
+            // regardless of the logger type
+            return message;
+        }
+        if (typeof noRedaction === "string" && noRedaction === this.type) {
+            // If the consumer has supplied noRedaction: logger-type, we skip redacting if
+            // our logger type is the same as what the consumer requested
+            return message;
+        }
+        if (typeof noRedaction === "object" &&
+            Array.isArray(noRedaction) &&
+            this.type &&
+            noRedaction.indexOf(this.type) !== -1) {
+            // If the consumer has supplied noRedaction: array, we skip redacting if our logger
+            // type is included in that array
+            return message;
+        }
+        return (0, mongodb_redact_1.redact)(message, this.keychain?.allSecrets ?? []);
+    }
+    info(payload) {
+        this.log("info", payload);
+    }
+    error(payload) {
+        this.log("error", payload);
+    }
+    debug(payload) {
+        this.log("debug", payload);
+    }
+    notice(payload) {
+        this.log("notice", payload);
+    }
+    warning(payload) {
+        this.log("warning", payload);
+    }
+    critical(payload) {
+        this.log("critical", payload);
+    }
+    alert(payload) {
+        this.log("alert", payload);
+    }
+    emergency(payload) {
+        this.log("emergency", payload);
+    }
+    mapToMongoDBLogLevel(level) {
+        switch (level) {
+            case "info":
+                return "info";
+            case "warning":
+                return "warn";
+            case "error":
+                return "error";
+            case "notice":
+            case "debug":
+                return "debug";
+            case "critical":
+            case "alert":
+            case "emergency":
+                return "fatal";
+            default:
+                return "info";
+        }
+    }
+}
+exports.LoggerBase = LoggerBase;
+class ConsoleLogger extends LoggerBase {
+    constructor(keychain) {
+        super(keychain);
+        this.type = "console";
+    }
+    logCore(level, payload) {
+        const { id, context, message } = payload;
+        console.error(`[${level.toUpperCase()}] ${id.__value} - ${context}: ${message} (${process.pid}${this.serializeAttributes(payload.attributes)})`);
+    }
+    serializeAttributes(attributes) {
+        if (!attributes || Object.keys(attributes).length === 0) {
+            return "";
+        }
+        return `, ${Object.entries(attributes)
+            .map(([key, value]) => `${key}=${value}`)
+            .join(", ")}`;
+    }
+}
+exports.ConsoleLogger = ConsoleLogger;
+class DiskLogger extends LoggerBase {
+    constructor(logPath, onError, keychain) {
+        super(keychain);
+        this.bufferedMessages = [];
+        this.type = "disk";
+        void this.initialize(logPath, onError);
+    }
+    async initialize(logPath, onError) {
+        try {
+            await promises_1.default.mkdir(logPath, { recursive: true });
+            const manager = new mongodb_log_writer_1.MongoLogManager({
+                directory: logPath,
+                retentionDays: 30,
+                onwarn: console.warn,
+                onerror: console.error,
+                gzip: false,
+                retentionGB: 1,
+            });
+            await manager.cleanupOldLogFiles();
+            this.logWriter = await manager.createLogWriter();
+            for (const message of this.bufferedMessages) {
+                this.logCore(message.level, message.payload);
+            }
+            this.bufferedMessages = [];
+            this.emit("initialized");
+        }
+        catch (error) {
+            onError(error);
+        }
+    }
+    logCore(level, payload) {
+        if (!this.logWriter) {
+            // If the log writer is not initialized, buffer the message
+            this.bufferedMessages.push({ level, payload });
+            return;
+        }
+        const { id, context, message } = payload;
+        const mongoDBLevel = this.mapToMongoDBLogLevel(level);
+        this.logWriter[mongoDBLevel]("MONGODB-MCP", id, context, message, payload.attributes);
+    }
+}
+exports.DiskLogger = DiskLogger;
+class McpLogger extends LoggerBase {
+    constructor(server, keychain) {
+        super(keychain);
+        this.server = server;
+        this.type = "mcp";
+    }
+    logCore(level, payload) {
+        // Only log if the server is connected
+        if (!this.server.mcpServer.isConnected()) {
+            return;
+        }
+        const minimumLevel = McpLogger.LOG_LEVELS.indexOf(this.server.mcpLogLevel);
+        const currentLevel = McpLogger.LOG_LEVELS.indexOf(level);
+        if (minimumLevel > currentLevel) {
+            // Don't log if the requested level is lower than the minimum level
+            return;
+        }
+        void this.server.mcpServer.server.sendLoggingMessage({
+            level,
+            data: `[${payload.context}]: ${payload.message}`,
+        });
+    }
+}
+exports.McpLogger = McpLogger;
+McpLogger.LOG_LEVELS = [
+    "debug",
+    "info",
+    "notice",
+    "warning",
+    "error",
+    "critical",
+    "alert",
+    "emergency",
+];
+class CompositeLogger extends LoggerBase {
+    constructor(...loggers) {
+        // composite logger does not redact, only the actual delegates do the work
+        // so we don't need the Keychain here
+        super(undefined);
+        this.loggers = [];
+        this.attributes = {};
+        this.loggers = loggers;
+    }
+    addLogger(logger) {
+        this.loggers.push(logger);
+    }
+    log(level, payload) {
+        // Override the public method to avoid the base logger redacting the message payload
+        for (const logger of this.loggers) {
+            const attributes = Object.keys(this.attributes).length > 0 || payload.attributes
+                ? { ...this.attributes, ...payload.attributes }
+                : undefined;
+            logger.log(level, { ...payload, attributes });
+        }
+    }
+    logCore() {
+        throw new Error("logCore should never be invoked on CompositeLogger");
+    }
+    setAttribute(key, value) {
+        this.attributes[key] = value;
+    }
+}
+exports.CompositeLogger = CompositeLogger;
+class NullLogger extends LoggerBase {
+    constructor() {
+        super(undefined);
+    }
+    logCore() {
+        // No-op logger, does not log anything
+    }
+}
+exports.NullLogger = NullLogger;
+//# sourceMappingURL=logger.js.map

package/dist/cjs/common/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../../src/common/logger.ts"],"names":[],"mappings":";;;;;;AAAA,2DAA6B;AAE7B,2DAAiE;AACjE,mDAAwC;AAExC,mCAAsC;AAMzB,QAAA,KAAK,GAAG;IACjB,kBAAkB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACzC,iBAAiB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACxC,oBAAoB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC3C,YAAY,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACnC,kBAAkB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACzC,sBAAsB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC7C,kBAAkB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAEzC,qBAAqB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC5C,8BAA8B,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACrD,mBAAmB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC1C,mBAAmB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC1C,mBAAmB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC1C,qBAAqB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC5C,qBAAqB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC5C,sBAAsB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC7C,2BAA2B,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAElD,iBAAiB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACxC,oBAAoB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC3C,kBAAkB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACzC,oBAAoB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC3C,sBAAsB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC7C,uBAAuB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC9C,eAAe,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACtC,cAAc,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAErC,WAAW,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAClC,kBAAkB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACzC,YAAY,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAEnC,qBAAqB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC5C,wBAAwB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC/C,iBAAiB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAExC,iBAAiB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACxC,qBAAqB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAE5C,8BAA8B,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACrD,0CAA0C,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACjE,+CAA+C,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACtE,sDAAsD,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC7E,qCAAqC,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC5D,mCAAmC,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC1D,uCAAuC,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC9D,gCAAgC,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAEvD,kBAAkB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACzC,mBAAmB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC1C,0BAA0B,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACjD,eAAe,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACtC,gBAAgB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACvC,qBAAqB,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAC5C,6BAA6B,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IACpD,eAAe,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;IAEtC,QAAQ,EAAE,IAAA,+BAAU,EAAC,OAAS,CAAC;CACzB,CAAC;AAgBX,MAAsB,UAAoD,SAAQ,qBAAe;IAG7F,YAA6B,QAA8B;QACvD,KAAK,EAAE,CAAC;QADiB,aAAQ,GAAR,QAAQ,CAAsB;QAF1C,4BAAuB,GAAe,KAAK,CAAC;IAI7D,CAAC;IAEM,GAAG,CAAC,KAAe,EAAE,OAAmB;QAC3C,4EAA4E;QAC5E,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC;QAE3G,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE;YAChB,GAAG,OAAO;YACV,OAAO,EAAE,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,CAAC;SAChE,CAAC,CAAC;IACP,CAAC;IAMO,iBAAiB,CAAC,OAAe,EAAE,WAAsC;QAC7E,IAAI,OAAO,WAAW,KAAK,SAAS,IAAI,WAAW,EAAE,CAAC;YAClD,kFAAkF;YAClF,gCAAgC;YAChC,OAAO,OAAO,CAAC;QACnB,CAAC;QAED,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,WAAW,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;YAC/D,8EAA8E;YAC9E,6DAA6D;YAC7D,OAAO,OAAO,CAAC;QACnB,CAAC;QAED,IACI,OAAO,WAAW,KAAK,QAAQ;YAC/B,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC;YAC1B,IAAI,CAAC,IAAI;YACT,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EACvC,CAAC;YACC,mFAAmF;YACnF,iCAAiC;YACjC,OAAO,OAAO,CAAC;QACnB,CAAC;QAED,OAAO,IAAA,uBAAM,EAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;IAEM,IAAI,CAAC,OAAmB;QAC3B,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC9B,CAAC;IAEM,KAAK,CAAC,OAAmB;QAC5B,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/B,CAAC;IACM,KAAK,CAAC,OAAmB;QAC5B,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/B,CAAC;IAEM,MAAM,CAAC,OAAmB;QAC7B,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChC,CAAC;IAEM,OAAO,CAAC,OAAmB;QAC9B,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACjC,CAAC;IAEM,QAAQ,CAAC,OAAmB;QAC/B,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAClC,CAAC;IAEM,KAAK,CAAC,OAAmB;QAC5B,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/B,CAAC;IAEM,SAAS,CAAC,OAAmB;QAChC,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAES,oBAAoB,CAAC,KAAe;QAC1C,QAAQ,KAAK,EAAE,CAAC;YACZ,KAAK,MAAM;gBACP,OAAO,MAAM,CAAC;YAClB,KAAK,SAAS;gBACV,OAAO,MAAM,CAAC;YAClB,KAAK,OAAO;gBACR,OAAO,OAAO,CAAC;YACnB,KAAK,QAAQ,CAAC;YACd,KAAK,OAAO;gBACR,OAAO,OAAO,CAAC;YACnB,KAAK,UAAU,CAAC;YAChB,KAAK,OAAO,CAAC;YACb,KAAK,WAAW;gBACZ,OAAO,OAAO,CAAC;YACnB;gBACI,OAAO,MAAM,CAAC;QACtB,CAAC;IACL,CAAC;CACJ;AAlGD,gCAkGC;AAED,MAAa,aAAc,SAAQ,UAAU;IAGzC,YAAmB,QAAkB;QACjC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAHD,SAAI,GAAe,SAAS,CAAC;IAIhD,CAAC;IAES,OAAO,CAAC,KAAe,EAAE,OAAmB;QAClD,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QACzC,OAAO,CAAC,KAAK,CACT,IAAI,KAAK,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC,OAAO,MAAM,OAAO,KAAK,OAAO,KAAK,OAAO,CAAC,GAAG,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CACpI,CAAC;IACN,CAAC;IAEO,mBAAmB,CAAC,UAAmC;QAC3D,IAAI,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,OAAO,EAAE,CAAC;QACd,CAAC;QACD,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;aACjC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,KAAK,EAAE,CAAC;aACxC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACtB,CAAC;CACJ;AAtBD,sCAsBC;AAED,MAAa,UAAW,SAAQ,UAA+B;IAI3D,YAAmB,OAAe,EAAE,OAA+B,EAAE,QAAkB;QACnF,KAAK,CAAC,QAAQ,CAAC,CAAC;QAJZ,qBAAgB,GAA+C,EAAE,CAAC;QAoChE,SAAI,GAAe,MAAM,CAAC;QA9BhC,KAAK,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,OAAe,EAAE,OAA+B;QACrE,IAAI,CAAC;YACD,MAAM,kBAAE,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE7C,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC;gBAChC,SAAS,EAAE,OAAO;gBAClB,aAAa,EAAE,EAAE;gBACjB,MAAM,EAAE,OAAO,CAAC,IAAI;gBACpB,OAAO,EAAE,OAAO,CAAC,KAAK;gBACtB,IAAI,EAAE,KAAK;gBACX,WAAW,EAAE,CAAC;aACjB,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;YAEnC,IAAI,CAAC,SAAS,GAAG,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC;YAEjD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAC1C,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;YACjD,CAAC;YACD,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACtB,OAAO,CAAC,KAAc,CAAC,CAAC;QAC5B,CAAC;IACL,CAAC;IAIS,OAAO,CAAC,KAAe,EAAE,OAAmB;QAClD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YAClB,2DAA2D;YAC3D,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YAC/C,OAAO;QACX,CAAC;QAED,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;QACzC,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAEtD,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,aAAa,EAAE,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAC1F,CAAC;CACJ;AAnDD,gCAmDC;AAED,MAAa,SAAU,SAAQ,UAAU;IAYrC,YACqB,MAAc,EAC/B,QAAkB;QAElB,KAAK,CAAC,QAAQ,CAAC,CAAC;QAHC,WAAM,GAAN,MAAM,CAAQ;QAMhB,SAAI,GAAe,KAAK,CAAC;IAF5C,CAAC;IAIS,OAAO,CAAC,KAAe,EAAE,OAAmB;QAClD,sCAAsC;QACtC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC;YACvC,OAAO;QACX,CAAC;QAED,MAAM,YAAY,GAAG,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QAC3E,MAAM,YAAY,GAAG,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzD,IAAI,YAAY,GAAG,YAAY,EAAE,CAAC;YAC9B,mEAAmE;YACnE,OAAO;QACX,CAAC;QAED,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,kBAAkB,CAAC;YACjD,KAAK;YACL,IAAI,EAAE,IAAI,OAAO,CAAC,OAAO,MAAM,OAAO,CAAC,OAAO,EAAE;SACnD,CAAC,CAAC;IACP,CAAC;;AAtCL,8BAuCC;AAtC0B,oBAAU,GAAe;IAC5C,OAAO;IACP,MAAM;IACN,QAAQ;IACR,SAAS;IACT,OAAO;IACP,UAAU;IACV,OAAO;IACP,WAAW;CACL,AATuB,CAStB;AA+Bf,MAAa,eAAgB,SAAQ,UAAU;IAM3C,YAAY,GAAG,OAAqB;QAChC,0EAA0E;QAC1E,qCAAqC;QACrC,KAAK,CAAC,SAAS,CAAC,CAAC;QANJ,YAAO,GAAiB,EAAE,CAAC;QAC3B,eAAU,GAA2B,EAAE,CAAC;QAOrD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IAC3B,CAAC;IAEM,SAAS,CAAC,MAAkB;QAC/B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAEM,GAAG,CAAC,KAAe,EAAE,OAAmB;QAC3C,oFAAoF;QACpF,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAChC,MAAM,UAAU,GACZ,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,UAAU;gBACzD,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE;gBAC/C,CAAC,CAAC,SAAS,CAAC;YACpB,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QAClD,CAAC;IACL,CAAC;IAES,OAAO;QACb,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;IAC1E,CAAC;IAEM,YAAY,CAAC,GAAW,EAAE,KAAa;QAC1C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IACjC,CAAC;CACJ;AApCD,0CAoCC;AAED,MAAa,UAAW,SAAQ,UAAU;IAGtC;QACI,KAAK,CAAC,SAAS,CAAC,CAAC;IACrB,CAAC;IAES,OAAO;QACb,sCAAsC;IAC1C,CAAC;CACJ;AAVD,gCAUC"}

package/dist/cjs/common/managedTimeout.d.ts

@@ -0,0 +1,6 @@
+export interface ManagedTimeout {
+    cancel: () => void;
+    restart: () => void;
+}
+export declare function setManagedTimeout(callback: () => Promise<void> | void, timeoutMS: number): ManagedTimeout;
+//# sourceMappingURL=managedTimeout.d.ts.map

package/dist/cjs/common/managedTimeout.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"managedTimeout.d.ts","sourceRoot":"","sources":["../../../src/common/managedTimeout.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,cAAc;IAC3B,MAAM,EAAE,MAAM,IAAI,CAAC;IACnB,OAAO,EAAE,MAAM,IAAI,CAAC;CACvB;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,SAAS,EAAE,MAAM,GAAG,cAAc,CAqBzG"}

package/dist/cjs/common/managedTimeout.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setManagedTimeout = setManagedTimeout;
+function setManagedTimeout(callback, timeoutMS) {
+    let timeoutId = setTimeout(() => {
+        void callback();
+    }, timeoutMS);
+    function cancel() {
+        clearTimeout(timeoutId);
+        timeoutId = undefined;
+    }
+    function restart() {
+        cancel();
+        timeoutId = setTimeout(() => {
+            void callback();
+        }, timeoutMS);
+    }
+    return {
+        cancel,
+        restart,
+    };
+}
+//# sourceMappingURL=managedTimeout.js.map

package/dist/cjs/common/managedTimeout.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"managedTimeout.js","sourceRoot":"","sources":["../../../src/common/managedTimeout.ts"],"names":[],"mappings":";;AAKA,8CAqBC;AArBD,SAAgB,iBAAiB,CAAC,QAAoC,EAAE,SAAiB;IACrF,IAAI,SAAS,GAA+B,UAAU,CAAC,GAAG,EAAE;QACxD,KAAK,QAAQ,EAAE,CAAC;IACpB,CAAC,EAAE,SAAS,CAAC,CAAC;IAEd,SAAS,MAAM;QACX,YAAY,CAAC,SAAS,CAAC,CAAC;QACxB,SAAS,GAAG,SAAS,CAAC;IAC1B,CAAC;IAED,SAAS,OAAO;QACZ,MAAM,EAAE,CAAC;QACT,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YACxB,KAAK,QAAQ,EAAE,CAAC;QACpB,CAAC,EAAE,SAAS,CAAC,CAAC;IAClB,CAAC;IAED,OAAO;QACH,MAAM;QACN,OAAO;KACV,CAAC;AACN,CAAC"}

package/dist/cjs/common/packageInfo.d.ts

@@ -0,0 +1,5 @@
+export declare const packageInfo: {
+    version: string;
+    mcpServerName: string;
+};
+//# sourceMappingURL=packageInfo.d.ts.map

package/dist/cjs/common/packageInfo.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"packageInfo.d.ts","sourceRoot":"","sources":["../../../src/common/packageInfo.ts"],"names":[],"mappings":"AACA,eAAO,MAAM,WAAW;;;CAGvB,CAAC"}

package/dist/cjs/common/packageInfo.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.packageInfo = void 0;
+// This file was generated by scripts/updatePackageVersion.ts - Do not edit it manually.
+exports.packageInfo = {
+    version: "0.3.1-prerelease.1",
+    mcpServerName: "MongoDB MCP Server",
+};
+//# sourceMappingURL=packageInfo.js.map

package/dist/cjs/common/packageInfo.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"packageInfo.js","sourceRoot":"","sources":["../../../src/common/packageInfo.ts"],"names":[],"mappings":";;;AAAA,wFAAwF;AAC3E,QAAA,WAAW,GAAG;IACvB,OAAO,EAAE,oBAAoB;IAC7B,aAAa,EAAE,oBAAoB;CACtC,CAAC"}

package/dist/cjs/common/session.d.ts

@@ -0,0 +1,45 @@
+import { ApiClient } from "./atlas/apiClient.js";
+import type { Implementation } from "@modelcontextprotocol/sdk/types.js";
+import type { CompositeLogger } from "./logger.js";
+import EventEmitter from "events";
+import type { AtlasClusterConnectionInfo, ConnectionManager, ConnectionSettings, ConnectionStateErrored } from "./connectionManager.js";
+import type { NodeDriverServiceProvider } from "@mongosh/service-provider-node-driver";
+import type { ExportsManager } from "./exportsManager.js";
+import type { Keychain } from "./keychain.js";
+export interface SessionOptions {
+    apiBaseUrl: string;
+    apiClientId?: string;
+    apiClientSecret?: string;
+    logger: CompositeLogger;
+    exportsManager: ExportsManager;
+    connectionManager: ConnectionManager;
+    keychain: Keychain;
+}
+export type SessionEvents = {
+    connect: [];
+    close: [];
+    disconnect: [];
+    "connection-error": [ConnectionStateErrored];
+};
+export declare class Session extends EventEmitter<SessionEvents> {
+    readonly sessionId: string;
+    readonly exportsManager: ExportsManager;
+    readonly connectionManager: ConnectionManager;
+    readonly apiClient: ApiClient;
+    readonly keychain: Keychain;
+    mcpClient?: {
+        name?: string;
+        version?: string;
+        title?: string;
+    };
+    logger: CompositeLogger;
+    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, }: SessionOptions);
+    setMcpClient(mcpClient: Implementation | undefined): void;
+    disconnect(): Promise<void>;
+    close(): Promise<void>;
+    connectToMongoDB(settings: ConnectionSettings): Promise<void>;
+    get isConnectedToMongoDB(): boolean;
+    get serviceProvider(): NodeDriverServiceProvider;
+    get connectedAtlasCluster(): AtlasClusterConnectionInfo | undefined;
+}
+//# sourceMappingURL=session.d.ts.map

package/dist/cjs/common/session.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEnD,OAAO,YAAY,MAAM,QAAQ,CAAC;AAClC,OAAO,KAAK,EACR,0BAA0B,EAC1B,iBAAiB,EACjB,kBAAkB,EAElB,sBAAsB,EACzB,MAAM,wBAAwB,CAAC;AAChC,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AAEvF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAE9C,MAAM,WAAW,cAAc;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,EAAE,eAAe,CAAC;IACxB,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,QAAQ,EAAE,QAAQ,CAAC;CACtB;AAED,MAAM,MAAM,aAAa,GAAG;IACxB,OAAO,EAAE,EAAE,CAAC;IACZ,KAAK,EAAE,EAAE,CAAC;IACV,UAAU,EAAE,EAAE,CAAC;IACf,kBAAkB,EAAE,CAAC,sBAAsB,CAAC,CAAC;CAChD,CAAC;AAEF,qBAAa,OAAQ,SAAQ,YAAY,CAAC,aAAa,CAAC;IACpD,QAAQ,CAAC,SAAS,EAAE,MAAM,CAA6B;IACvD,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;IACxC,QAAQ,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IAC9C,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IAC9B,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAE5B,SAAS,CAAC,EAAE;QACR,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,KAAK,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC;IAEK,MAAM,EAAE,eAAe,CAAC;gBAEnB,EACR,UAAU,EACV,WAAW,EACX,eAAe,EACf,MAAM,EACN,iBAAiB,EACjB,cAAc,EACd,QAAQ,GACX,EAAE,cAAc;IAsBjB,YAAY,CAAC,SAAS,EAAE,cAAc,GAAG,SAAS,GAAG,IAAI;IAoBnD,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAoC3B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAOtB,gBAAgB,CAAC,QAAQ,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IAInE,IAAI,oBAAoB,IAAI,OAAO,CAElC;IAED,IAAI,eAAe,IAAI,yBAAyB,CAO/C;IAED,IAAI,qBAAqB,IAAI,0BAA0B,GAAG,SAAS,CAElE;CACJ"}

package/dist/cjs/common/session.js

@@ -0,0 +1,107 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Session = void 0;
+const bson_1 = require("bson");
+const apiClient_js_1 = require("./atlas/apiClient.js");
+const logger_js_1 = require("./logger.js");
+const events_1 = __importDefault(require("events"));
+const errors_js_1 = require("./errors.js");
+class Session extends events_1.default {
+    constructor({ apiBaseUrl, apiClientId, apiClientSecret, logger, connectionManager, exportsManager, keychain, }) {
+        super();
+        this.sessionId = new bson_1.ObjectId().toString();
+        this.keychain = keychain;
+        this.logger = logger;
+        const credentials = apiClientId && apiClientSecret
+            ? {
+                clientId: apiClientId,
+                clientSecret: apiClientSecret,
+            }
+            : undefined;
+        this.apiClient = new apiClient_js_1.ApiClient({ baseUrl: apiBaseUrl, credentials }, logger);
+        this.exportsManager = exportsManager;
+        this.connectionManager = connectionManager;
+        this.connectionManager.events.on("connection-success", () => this.emit("connect"));
+        this.connectionManager.events.on("connection-time-out", (error) => this.emit("connection-error", error));
+        this.connectionManager.events.on("connection-close", () => this.emit("disconnect"));
+        this.connectionManager.events.on("connection-error", (error) => this.emit("connection-error", error));
+    }
+    setMcpClient(mcpClient) {
+        if (!mcpClient) {
+            this.connectionManager.setClientName("unknown");
+            this.logger.debug({
+                id: logger_js_1.LogId.serverMcpClientSet,
+                context: "session",
+                message: "MCP client info not found",
+            });
+        }
+        this.mcpClient = {
+            name: mcpClient?.name || "unknown",
+            version: mcpClient?.version || "unknown",
+            title: mcpClient?.title || "unknown",
+        };
+        // Set the client name on the connection manager for appName generation
+        this.connectionManager.setClientName(this.mcpClient.name || "unknown");
+    }
+    async disconnect() {
+        const atlasCluster = this.connectedAtlasCluster;
+        try {
+            await this.connectionManager.disconnect();
+        }
+        catch (err) {
+            const error = err instanceof Error ? err : new Error(String(err));
+            this.logger.error({
+                id: logger_js_1.LogId.mongodbDisconnectFailure,
+                context: "session",
+                message: `Error closing service provider: ${error.message}`,
+            });
+        }
+        if (atlasCluster?.username && atlasCluster?.projectId) {
+            void this.apiClient
+                .deleteDatabaseUser({
+                params: {
+                    path: {
+                        groupId: atlasCluster.projectId,
+                        username: atlasCluster.username,
+                        databaseName: "admin",
+                    },
+                },
+            })
+                .catch((err) => {
+                const error = err instanceof Error ? err : new Error(String(err));
+                this.logger.error({
+                    id: logger_js_1.LogId.atlasDeleteDatabaseUserFailure,
+                    context: "session",
+                    message: `Error deleting previous database user: ${error.message}`,
+                });
+            });
+        }
+    }
+    async close() {
+        await this.disconnect();
+        await this.apiClient.close();
+        await this.exportsManager.close();
+        this.emit("close");
+    }
+    async connectToMongoDB(settings) {
+        await this.connectionManager.connect({ ...settings });
+    }
+    get isConnectedToMongoDB() {
+        return this.connectionManager.currentConnectionState.tag === "connected";
+    }
+    get serviceProvider() {
+        if (this.isConnectedToMongoDB) {
+            const state = this.connectionManager.currentConnectionState;
+            return state.serviceProvider;
+        }
+        throw new errors_js_1.MongoDBError(errors_js_1.ErrorCodes.NotConnectedToMongoDB, "Not connected to MongoDB");
+    }
+    get connectedAtlasCluster() {
+        return this.connectionManager.currentConnectionState.connectedAtlasCluster;
+    }
+}
+exports.Session = Session;
+//# sourceMappingURL=session.js.map

package/dist/cjs/common/session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../src/common/session.ts"],"names":[],"mappings":";;;;;;AAAA,+BAAgC;AAEhC,uDAAiD;AAGjD,2CAAoC;AACpC,oDAAkC;AASlC,2CAAuD;AAqBvD,MAAa,OAAQ,SAAQ,gBAA2B;IAepD,YAAY,EACR,UAAU,EACV,WAAW,EACX,eAAe,EACf,MAAM,EACN,iBAAiB,EACjB,cAAc,EACd,QAAQ,GACK;QACb,KAAK,EAAE,CAAC;QAvBH,cAAS,GAAW,IAAI,eAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;QAyBnD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,WAAW,GACb,WAAW,IAAI,eAAe;YAC1B,CAAC,CAAC;gBACI,QAAQ,EAAE,WAAW;gBACrB,YAAY,EAAE,eAAe;aAChC;YACH,CAAC,CAAC,SAAS,CAAC;QAEpB,IAAI,CAAC,SAAS,GAAG,IAAI,wBAAS,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,MAAM,CAAC,CAAC;QAC7E,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,qBAAqB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC,CAAC;QACzG,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QACpF,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC,CAAC;IAC1G,CAAC;IAED,YAAY,CAAC,SAAqC;QAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;YACb,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,2BAA2B;aACvC,CAAC,CAAC;QACP,CAAC;QAED,IAAI,CAAC,SAAS,GAAG;YACb,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,SAAS;YAClC,OAAO,EAAE,SAAS,EAAE,OAAO,IAAI,SAAS;YACxC,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,SAAS;SACvC,CAAC;QAEF,uEAAuE;QACvE,IAAI,CAAC,iBAAiB,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,UAAU;QACZ,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC;QAEhD,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;QAC9C,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACpB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,wBAAwB;gBAClC,OAAO,EAAE,SAAS;gBAClB,OAAO,EAAE,mCAAmC,KAAK,CAAC,OAAO,EAAE;aAC9D,CAAC,CAAC;QACP,CAAC;QAED,IAAI,YAAY,EAAE,QAAQ,IAAI,YAAY,EAAE,SAAS,EAAE,CAAC;YACpD,KAAK,IAAI,CAAC,SAAS;iBACd,kBAAkB,CAAC;gBAChB,MAAM,EAAE;oBACJ,IAAI,EAAE;wBACF,OAAO,EAAE,YAAY,CAAC,SAAS;wBAC/B,QAAQ,EAAE,YAAY,CAAC,QAAQ;wBAC/B,YAAY,EAAE,OAAO;qBACxB;iBACJ;aACJ,CAAC;iBACD,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;gBACpB,MAAM,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAClE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;oBACd,EAAE,EAAE,iBAAK,CAAC,8BAA8B;oBACxC,OAAO,EAAE,SAAS;oBAClB,OAAO,EAAE,0CAA0C,KAAK,CAAC,OAAO,EAAE;iBACrE,CAAC,CAAC;YACP,CAAC,CAAC,CAAC;QACX,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK;QACP,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QACxB,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;QAClC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,QAA4B;QAC/C,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,GAAG,QAAQ,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,oBAAoB;QACpB,OAAO,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,GAAG,KAAK,WAAW,CAAC;IAC7E,CAAC;IAED,IAAI,eAAe;QACf,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,sBAAkD,CAAC;YACxF,OAAO,KAAK,CAAC,eAAe,CAAC;QACjC,CAAC;QAED,MAAM,IAAI,wBAAY,CAAC,sBAAU,CAAC,qBAAqB,EAAE,0BAA0B,CAAC,CAAC;IACzF,CAAC;IAED,IAAI,qBAAqB;QACrB,OAAO,IAAI,CAAC,iBAAiB,CAAC,sBAAsB,CAAC,qBAAqB,CAAC;IAC/E,CAAC;CACJ;AAhID,0BAgIC"}

package/dist/cjs/common/sessionStore.d.ts

@@ -0,0 +1,16 @@
+import type { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import type { LoggerBase } from "./logger.js";
+export declare class SessionStore {
+    private readonly idleTimeoutMS;
+    private readonly notificationTimeoutMS;
+    private readonly logger;
+    private sessions;
+    constructor(idleTimeoutMS: number, notificationTimeoutMS: number, logger: LoggerBase);
+    getSession(sessionId: string): StreamableHTTPServerTransport | undefined;
+    private resetTimeout;
+    private sendNotification;
+    setSession(sessionId: string, transport: StreamableHTTPServerTransport, logger: LoggerBase): void;
+    closeSession(sessionId: string, closeTransport?: boolean): Promise<void>;
+    closeAllSessions(): Promise<void>;
+}
+//# sourceMappingURL=sessionStore.d.ts.map

package/dist/cjs/common/sessionStore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionStore.d.ts","sourceRoot":"","sources":["../../../src/common/sessionStore.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACxG,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAK9C,qBAAa,YAAY;IAWjB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,qBAAqB;IACtC,OAAO,CAAC,QAAQ,CAAC,MAAM;IAZ3B,OAAO,CAAC,QAAQ,CAOT;gBAGc,aAAa,EAAE,MAAM,EACrB,qBAAqB,EAAE,MAAM,EAC7B,MAAM,EAAE,UAAU;IAavC,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,6BAA6B,GAAG,SAAS;IAKxE,OAAO,CAAC,YAAY;IAWpB,OAAO,CAAC,gBAAgB;IAiBxB,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,6BAA6B,EAAE,MAAM,EAAE,UAAU,GAAG,IAAI;IA4B3F,YAAY,CAAC,SAAS,EAAE,MAAM,EAAE,cAAc,GAAE,OAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB9E,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAG1C"}

package/dist/cjs/common/sessionStore.js

@@ -0,0 +1,99 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SessionStore = void 0;
+const logger_js_1 = require("./logger.js");
+const managedTimeout_js_1 = require("./managedTimeout.js");
+class SessionStore {
+    constructor(idleTimeoutMS, notificationTimeoutMS, logger) {
+        this.idleTimeoutMS = idleTimeoutMS;
+        this.notificationTimeoutMS = notificationTimeoutMS;
+        this.logger = logger;
+        this.sessions = {};
+        if (idleTimeoutMS <= 0) {
+            throw new Error("idleTimeoutMS must be greater than 0");
+        }
+        if (notificationTimeoutMS <= 0) {
+            throw new Error("notificationTimeoutMS must be greater than 0");
+        }
+        if (idleTimeoutMS <= notificationTimeoutMS) {
+            throw new Error("idleTimeoutMS must be greater than notificationTimeoutMS");
+        }
+    }
+    getSession(sessionId) {
+        this.resetTimeout(sessionId);
+        return this.sessions[sessionId]?.transport;
+    }
+    resetTimeout(sessionId) {
+        const session = this.sessions[sessionId];
+        if (!session) {
+            return;
+        }
+        session.abortTimeout.restart();
+        session.notificationTimeout.restart();
+    }
+    sendNotification(sessionId) {
+        const session = this.sessions[sessionId];
+        if (!session) {
+            this.logger.warning({
+                id: logger_js_1.LogId.streamableHttpTransportSessionCloseNotificationFailure,
+                context: "sessionStore",
+                message: `session ${sessionId} not found, no notification delivered`,
+            });
+            return;
+        }
+        session.logger.info({
+            id: logger_js_1.LogId.streamableHttpTransportSessionCloseNotification,
+            context: "sessionStore",
+            message: "Session is about to be closed due to inactivity",
+        });
+    }
+    setSession(sessionId, transport, logger) {
+        const session = this.sessions[sessionId];
+        if (session) {
+            throw new Error(`Session ${sessionId} already exists`);
+        }
+        const abortTimeout = (0, managedTimeout_js_1.setManagedTimeout)(async () => {
+            if (this.sessions[sessionId]) {
+                this.sessions[sessionId].logger.info({
+                    id: logger_js_1.LogId.streamableHttpTransportSessionCloseNotification,
+                    context: "sessionStore",
+                    message: "Session closed due to inactivity",
+                });
+                await this.closeSession(sessionId);
+            }
+        }, this.idleTimeoutMS);
+        const notificationTimeout = (0, managedTimeout_js_1.setManagedTimeout)(() => this.sendNotification(sessionId), this.notificationTimeoutMS);
+        this.sessions[sessionId] = {
+            transport,
+            abortTimeout,
+            notificationTimeout,
+            logger,
+        };
+    }
+    async closeSession(sessionId, closeTransport = true) {
+        const session = this.sessions[sessionId];
+        if (!session) {
+            throw new Error(`Session ${sessionId} not found`);
+        }
+        session.abortTimeout.cancel();
+        session.notificationTimeout.cancel();
+        if (closeTransport) {
+            try {
+                await session.transport.close();
+            }
+            catch (error) {
+                this.logger.error({
+                    id: logger_js_1.LogId.streamableHttpTransportSessionCloseFailure,
+                    context: "streamableHttpTransport",
+                    message: `Error closing transport ${sessionId}: ${error instanceof Error ? error.message : String(error)}`,
+                });
+            }
+        }
+        delete this.sessions[sessionId];
+    }
+    async closeAllSessions() {
+        await Promise.all(Object.keys(this.sessions).map((sessionId) => this.closeSession(sessionId)));
+    }
+}
+exports.SessionStore = SessionStore;
+//# sourceMappingURL=sessionStore.js.map

package/dist/cjs/common/sessionStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionStore.js","sourceRoot":"","sources":["../../../src/common/sessionStore.ts"],"names":[],"mappings":";;;AAEA,2CAAoC;AAEpC,2DAAwD;AAExD,MAAa,YAAY;IAUrB,YACqB,aAAqB,EACrB,qBAA6B,EAC7B,MAAkB;QAFlB,kBAAa,GAAb,aAAa,CAAQ;QACrB,0BAAqB,GAArB,qBAAqB,CAAQ;QAC7B,WAAM,GAAN,MAAM,CAAY;QAZ/B,aAAQ,GAOZ,EAAE,CAAC;QAOH,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,qBAAqB,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,aAAa,IAAI,qBAAqB,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAChF,CAAC;IACL,CAAC;IAED,UAAU,CAAC,SAAiB;QACxB,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC7B,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC;IAC/C,CAAC;IAEO,YAAY,CAAC,SAAiB;QAClC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,OAAO;QACX,CAAC;QAED,OAAO,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;QAE/B,OAAO,CAAC,mBAAmB,CAAC,OAAO,EAAE,CAAC;IAC1C,CAAC;IAEO,gBAAgB,CAAC,SAAiB;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;gBAChB,EAAE,EAAE,iBAAK,CAAC,sDAAsD;gBAChE,OAAO,EAAE,cAAc;gBACvB,OAAO,EAAE,WAAW,SAAS,uCAAuC;aACvE,CAAC,CAAC;YACH,OAAO;QACX,CAAC;QACD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC;YAChB,EAAE,EAAE,iBAAK,CAAC,+CAA+C;YACzD,OAAO,EAAE,cAAc;YACvB,OAAO,EAAE,iDAAiD;SAC7D,CAAC,CAAC;IACP,CAAC;IAED,UAAU,CAAC,SAAiB,EAAE,SAAwC,EAAE,MAAkB;QACtF,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,OAAO,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,WAAW,SAAS,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QACD,MAAM,YAAY,GAAG,IAAA,qCAAiB,EAAC,KAAK,IAAI,EAAE;YAC9C,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC;oBACjC,EAAE,EAAE,iBAAK,CAAC,+CAA+C;oBACzD,OAAO,EAAE,cAAc;oBACvB,OAAO,EAAE,kCAAkC;iBAC9C,CAAC,CAAC;gBAEH,MAAM,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;YACvC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACvB,MAAM,mBAAmB,GAAG,IAAA,qCAAiB,EACzC,GAAG,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,EACtC,IAAI,CAAC,qBAAqB,CAC7B,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG;YACvB,SAAS;YACT,YAAY;YACZ,mBAAmB;YACnB,MAAM;SACT,CAAC;IACN,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAiB,EAAE,iBAA0B,IAAI;QAChE,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,WAAW,SAAS,YAAY,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;QAC9B,OAAO,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC;QACrC,IAAI,cAAc,EAAE,CAAC;YACjB,IAAI,CAAC;gBACD,MAAM,OAAO,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YACpC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;oBACd,EAAE,EAAE,iBAAK,CAAC,0CAA0C;oBACpD,OAAO,EAAE,yBAAyB;oBAClC,OAAO,EAAE,2BAA2B,SAAS,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;iBAC7G,CAAC,CAAC;YACP,CAAC;QACL,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,gBAAgB;QAClB,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACnG,CAAC;CACJ;AA/GD,oCA+GC"}