@himanshusinghs/mongodb-mcp-server 0.3.0

package/dist/cjs/common/exportsManager.js

@@ -0,0 +1,288 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExportsManager = exports.jsonExportFormat = void 0;
+exports.ensureExtension = ensureExtension;
+exports.validateExportName = validateExportName;
+exports.isExportExpired = isExportExpired;
+const zod_1 = __importDefault(require("zod"));
+const path_1 = __importDefault(require("path"));
+const promises_1 = __importDefault(require("fs/promises"));
+const events_1 = __importDefault(require("events"));
+const fs_1 = require("fs");
+const bson_1 = require("bson");
+const stream_1 = require("stream");
+const promises_2 = require("stream/promises");
+const logger_js_1 = require("./logger.js");
+exports.jsonExportFormat = zod_1.default.enum(["relaxed", "canonical"]);
+class ExportsManager extends events_1.default {
+    constructor(exportsDirectoryPath, config, logger) {
+        super();
+        this.exportsDirectoryPath = exportsDirectoryPath;
+        this.config = config;
+        this.logger = logger;
+        this.storedExports = {};
+        this.exportsCleanupInProgress = false;
+        this.shutdownController = new AbortController();
+    }
+    get availableExports() {
+        this.assertIsNotShuttingDown();
+        return Object.values(this.storedExports)
+            .filter((storedExport) => {
+            return (storedExport.exportStatus === "ready" &&
+                !isExportExpired(storedExport.exportCreatedAt, this.config.exportTimeoutMs));
+        })
+            .map(({ exportName, exportTitle, exportURI, exportPath }) => ({
+            exportName,
+            exportTitle,
+            exportURI,
+            exportPath,
+        }));
+    }
+    init() {
+        if (!this.exportsCleanupInterval) {
+            this.exportsCleanupInterval = setInterval(() => void this.cleanupExpiredExports(), this.config.exportCleanupIntervalMs);
+        }
+    }
+    async close() {
+        if (this.shutdownController.signal.aborted) {
+            return;
+        }
+        try {
+            clearInterval(this.exportsCleanupInterval);
+            this.shutdownController.abort();
+            await promises_1.default.rm(this.exportsDirectoryPath, { force: true, recursive: true });
+            this.emit("closed");
+        }
+        catch (error) {
+            this.logger.error({
+                id: logger_js_1.LogId.exportCloseError,
+                context: "Error while closing ExportsManager",
+                message: error instanceof Error ? error.message : String(error),
+            });
+        }
+    }
+    async readExport(exportName) {
+        try {
+            this.assertIsNotShuttingDown();
+            exportName = decodeURIComponent(exportName);
+            const exportHandle = this.storedExports[exportName];
+            if (!exportHandle) {
+                throw new Error("Requested export has either expired or does not exist.");
+            }
+            if (exportHandle.exportStatus === "in-progress") {
+                throw new Error("Requested export is still being generated. Try again later.");
+            }
+            const { exportPath } = exportHandle;
+            return promises_1.default.readFile(exportPath, { encoding: "utf8", signal: this.shutdownController.signal });
+        }
+        catch (error) {
+            this.logger.error({
+                id: logger_js_1.LogId.exportReadError,
+                context: `Error when reading export - ${exportName}`,
+                message: error instanceof Error ? error.message : String(error),
+            });
+            throw error;
+        }
+    }
+    async createJSONExport({ input, exportName, exportTitle, jsonExportFormat, }) {
+        try {
+            this.assertIsNotShuttingDown();
+            const exportNameWithExtension = validateExportName(ensureExtension(exportName, "json"));
+            if (this.storedExports[exportNameWithExtension]) {
+                return Promise.reject(new Error("Export with same name is either already available or being generated."));
+            }
+            const exportURI = `exported-data://${encodeURIComponent(exportNameWithExtension)}`;
+            const exportFilePath = path_1.default.join(this.exportsDirectoryPath, exportNameWithExtension);
+            const inProgressExport = (this.storedExports[exportNameWithExtension] = {
+                exportName: exportNameWithExtension,
+                exportTitle,
+                exportPath: exportFilePath,
+                exportURI: exportURI,
+                exportStatus: "in-progress",
+            });
+            void this.startExport({ input, jsonExportFormat, inProgressExport });
+            return Promise.resolve(inProgressExport);
+        }
+        catch (error) {
+            this.logger.error({
+                id: logger_js_1.LogId.exportCreationError,
+                context: "Error when registering JSON export request",
+                message: error instanceof Error ? error.message : String(error),
+            });
+            throw error;
+        }
+    }
+    async startExport({ input, jsonExportFormat, inProgressExport, }) {
+        try {
+            let pipeSuccessful = false;
+            try {
+                await promises_1.default.mkdir(this.exportsDirectoryPath, { recursive: true });
+                const outputStream = (0, fs_1.createWriteStream)(inProgressExport.exportPath);
+                await (0, promises_2.pipeline)([
+                    input.stream(),
+                    this.docToEJSONStream(this.getEJSONOptionsForFormat(jsonExportFormat)),
+                    outputStream,
+                ], { signal: this.shutdownController.signal });
+                pipeSuccessful = true;
+            }
+            catch (error) {
+                // If the pipeline errors out then we might end up with
+                // partial and incorrect export so we remove it entirely.
+                delete this.storedExports[inProgressExport.exportName];
+                // do not block the user, just delete the file in the background
+                void this.silentlyRemoveExport(inProgressExport.exportPath, logger_js_1.LogId.exportCreationCleanupError, `Error when removing incomplete export ${inProgressExport.exportName}`);
+                throw error;
+            }
+            finally {
+                if (pipeSuccessful) {
+                    this.storedExports[inProgressExport.exportName] = {
+                        ...inProgressExport,
+                        exportCreatedAt: Date.now(),
+                        exportStatus: "ready",
+                    };
+                    this.emit("export-available", inProgressExport.exportURI);
+                }
+                void input.close();
+            }
+        }
+        catch (error) {
+            this.logger.error({
+                id: logger_js_1.LogId.exportCreationError,
+                context: `Error when generating JSON export for ${inProgressExport.exportName}`,
+                message: error instanceof Error ? error.message : String(error),
+            });
+        }
+    }
+    getEJSONOptionsForFormat(format) {
+        switch (format) {
+            case "relaxed":
+                return { relaxed: true };
+            case "canonical":
+                return { relaxed: false };
+            default:
+                return undefined;
+        }
+    }
+    docToEJSONStream(ejsonOptions) {
+        let docsTransformed = 0;
+        return new stream_1.Transform({
+            objectMode: true,
+            transform(chunk, encoding, callback) {
+                try {
+                    const doc = bson_1.EJSON.stringify(chunk, undefined, undefined, ejsonOptions);
+                    if (docsTransformed === 0) {
+                        this.push("[" + doc);
+                    }
+                    else {
+                        this.push(",\n" + doc);
+                    }
+                    docsTransformed++;
+                    callback();
+                }
+                catch (err) {
+                    callback(err);
+                }
+            },
+            flush(callback) {
+                if (docsTransformed === 0) {
+                    this.push("[]");
+                }
+                else {
+                    this.push("]");
+                }
+                callback();
+            },
+        });
+    }
+    async cleanupExpiredExports() {
+        if (this.exportsCleanupInProgress) {
+            return;
+        }
+        this.exportsCleanupInProgress = true;
+        try {
+            // first, unregister all exports that are expired, so they are not considered anymore for reading
+            const exportsForCleanup = [];
+            for (const expiredExport of Object.values(this.storedExports)) {
+                if (expiredExport.exportStatus === "ready" &&
+                    isExportExpired(expiredExport.exportCreatedAt, this.config.exportTimeoutMs)) {
+                    exportsForCleanup.push(expiredExport);
+                    delete this.storedExports[expiredExport.exportName];
+                }
+            }
+            // and then remove them (slow operation potentially) from disk.
+            const allDeletionPromises = [];
+            for (const { exportPath, exportName } of exportsForCleanup) {
+                allDeletionPromises.push(this.silentlyRemoveExport(exportPath, logger_js_1.LogId.exportCleanupError, `Considerable error when removing export ${exportName}`));
+            }
+            await Promise.allSettled(allDeletionPromises);
+        }
+        catch (error) {
+            this.logger.error({
+                id: logger_js_1.LogId.exportCleanupError,
+                context: "Error when cleaning up exports",
+                message: error instanceof Error ? error.message : String(error),
+            });
+        }
+        finally {
+            this.exportsCleanupInProgress = false;
+        }
+    }
+    async silentlyRemoveExport(exportPath, logId, logContext) {
+        try {
+            await promises_1.default.unlink(exportPath);
+        }
+        catch (error) {
+            // If the file does not exist or the containing directory itself
+            // does not exist then we can safely ignore that error anything else
+            // we need to flag.
+            if (error.code !== "ENOENT") {
+                this.logger.error({
+                    id: logId,
+                    context: logContext,
+                    message: error instanceof Error ? error.message : String(error),
+                });
+            }
+        }
+    }
+    assertIsNotShuttingDown() {
+        if (this.shutdownController.signal.aborted) {
+            throw new Error("ExportsManager is shutting down.");
+        }
+    }
+    static init(config, logger, sessionId = new bson_1.ObjectId().toString()) {
+        const exportsDirectoryPath = path_1.default.join(config.exportsPath, sessionId);
+        const exportsManager = new ExportsManager(exportsDirectoryPath, config, logger);
+        exportsManager.init();
+        return exportsManager;
+    }
+}
+exports.ExportsManager = ExportsManager;
+/**
+ * Ensures the path ends with the provided extension */
+function ensureExtension(pathOrName, extension) {
+    const extWithDot = extension.startsWith(".") ? extension : `.${extension}`;
+    if (pathOrName.endsWith(extWithDot)) {
+        return pathOrName;
+    }
+    return `${pathOrName}${extWithDot}`;
+}
+/**
+ * Small utility to decoding and validating provided export name for path
+ * traversal or no extension */
+function validateExportName(nameWithExtension) {
+    const decodedName = decodeURIComponent(nameWithExtension);
+    if (!path_1.default.extname(decodedName)) {
+        throw new Error("Provided export name has no extension");
+    }
+    if (decodedName.includes("..") || decodedName.includes("/") || decodedName.includes("\\")) {
+        throw new Error("Invalid export name: path traversal hinted");
+    }
+    return decodedName;
+}
+function isExportExpired(createdAt, exportTimeoutMs) {
+    return Date.now() - createdAt > exportTimeoutMs;
+}
+//# sourceMappingURL=exportsManager.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"exportsManager.js","sourceRoot":"","sources":["../../../src/common/exportsManager.ts"],"names":[],"mappings":";;;;;;AA+WA,0CAMC;AAKD,gDAWC;AAED,0CAEC;AAzYD,8CAAoB;AACpB,gDAAwB;AACxB,2DAA6B;AAC7B,oDAAkC;AAClC,2BAAuC;AAGvC,+BAAuC;AACvC,mCAAmC;AACnC,8CAA2C;AAK3C,2CAAoC;AAEvB,QAAA,gBAAgB,GAAG,aAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC;AAoDjE,MAAa,cAAe,SAAQ,gBAAkC;IAMlE,YACqB,oBAA4B,EAC5B,MAA4B,EAC5B,MAAkB;QAEnC,KAAK,EAAE,CAAC;QAJS,yBAAoB,GAApB,oBAAoB,CAAQ;QAC5B,WAAM,GAAN,MAAM,CAAsB;QAC5B,WAAM,GAAN,MAAM,CAAY;QAR/B,kBAAa,GAAqD,EAAE,CAAC;QACrE,6BAAwB,GAAY,KAAK,CAAC;QAEjC,uBAAkB,GAAoB,IAAI,eAAe,EAAE,CAAC;IAQ7E,CAAC;IAED,IAAW,gBAAgB;QACvB,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC/B,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC;aACnC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE;YACrB,OAAO,CACH,YAAY,CAAC,YAAY,KAAK,OAAO;gBACrC,CAAC,eAAe,CAAC,YAAY,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAC9E,CAAC;QACN,CAAC,CAAC;aACD,GAAG,CAAC,CAAC,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;YAC1D,UAAU;YACV,WAAW;YACX,SAAS;YACT,UAAU;SACb,CAAC,CAAC,CAAC;IACZ,CAAC;IAES,IAAI;QACV,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAC/B,IAAI,CAAC,sBAAsB,GAAG,WAAW,CACrC,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,qBAAqB,EAAE,EACvC,IAAI,CAAC,MAAM,CAAC,uBAAuB,CACtC,CAAC;QACN,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK;QACd,IAAI,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzC,OAAO;QACX,CAAC;QACD,IAAI,CAAC;YACD,aAAa,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAC3C,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;YAChC,MAAM,kBAAE,CAAC,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACzE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,gBAAgB;gBAC1B,OAAO,EAAE,oCAAoC;gBAC7C,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAClE,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,UAAU,CAAC,UAAkB;QACtC,IAAI,CAAC;YACD,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC/B,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YACpD,IAAI,CAAC,YAAY,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;YAC9E,CAAC;YAED,IAAI,YAAY,CAAC,YAAY,KAAK,aAAa,EAAE,CAAC;gBAC9C,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;YACnF,CAAC;YAED,MAAM,EAAE,UAAU,EAAE,GAAG,YAAY,CAAC;YAEpC,OAAO,kBAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,CAAC;QACjG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,eAAe;gBACzB,OAAO,EAAE,+BAA+B,UAAU,EAAE;gBACpD,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAClE,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,EAC1B,KAAK,EACL,UAAU,EACV,WAAW,EACX,gBAAgB,GAMnB;QACG,IAAI,CAAC;YACD,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAC/B,MAAM,uBAAuB,GAAG,kBAAkB,CAAC,eAAe,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;YACxF,IAAI,IAAI,CAAC,aAAa,CAAC,uBAAuB,CAAC,EAAE,CAAC;gBAC9C,OAAO,OAAO,CAAC,MAAM,CACjB,IAAI,KAAK,CAAC,uEAAuE,CAAC,CACrF,CAAC;YACN,CAAC;YACD,MAAM,SAAS,GAAG,mBAAmB,kBAAkB,CAAC,uBAAuB,CAAC,EAAE,CAAC;YACnF,MAAM,cAAc,GAAG,cAAI,CAAC,IAAI,CAAC,IAAI,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,CAAC;YACrF,MAAM,gBAAgB,GAAqB,CAAC,IAAI,CAAC,aAAa,CAAC,uBAAuB,CAAC,GAAG;gBACtF,UAAU,EAAE,uBAAuB;gBACnC,WAAW;gBACX,UAAU,EAAE,cAAc;gBAC1B,SAAS,EAAE,SAAS;gBACpB,YAAY,EAAE,aAAa;aAC9B,CAAC,CAAC;YAEH,KAAK,IAAI,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACrE,OAAO,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,mBAAmB;gBAC7B,OAAO,EAAE,4CAA4C;gBACrD,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAClE,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,EACtB,KAAK,EACL,gBAAgB,EAChB,gBAAgB,GAKnB;QACG,IAAI,CAAC;YACD,IAAI,cAAc,GAAG,KAAK,CAAC;YAC3B,IAAI,CAAC;gBACD,MAAM,kBAAE,CAAC,KAAK,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC/D,MAAM,YAAY,GAAG,IAAA,sBAAiB,EAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;gBACpE,MAAM,IAAA,mBAAQ,EACV;oBACI,KAAK,CAAC,MAAM,EAAE;oBACd,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,wBAAwB,CAAC,gBAAgB,CAAC,CAAC;oBACtE,YAAY;iBACf,EACD,EAAE,MAAM,EAAE,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAC7C,CAAC;gBACF,cAAc,GAAG,IAAI,CAAC;YAC1B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,uDAAuD;gBACvD,yDAAyD;gBACzD,OAAO,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;gBACvD,gEAAgE;gBAChE,KAAK,IAAI,CAAC,oBAAoB,CAC1B,gBAAgB,CAAC,UAAU,EAC3B,iBAAK,CAAC,0BAA0B,EAChC,yCAAyC,gBAAgB,CAAC,UAAU,EAAE,CACzE,CAAC;gBACF,MAAM,KAAK,CAAC;YAChB,CAAC;oBAAS,CAAC;gBACP,IAAI,cAAc,EAAE,CAAC;oBACjB,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,UAAU,CAAC,GAAG;wBAC9C,GAAG,gBAAgB;wBACnB,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE;wBAC3B,YAAY,EAAE,OAAO;qBACxB,CAAC;oBACF,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC;gBAC9D,CAAC;gBACD,KAAK,KAAK,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,mBAAmB;gBAC7B,OAAO,EAAE,yCAAyC,gBAAgB,CAAC,UAAU,EAAE;gBAC/E,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAClE,CAAC,CAAC;QACP,CAAC;IACL,CAAC;IAEO,wBAAwB,CAAC,MAAwB;QACrD,QAAQ,MAAM,EAAE,CAAC;YACb,KAAK,SAAS;gBACV,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC7B,KAAK,WAAW;gBACZ,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC9B;gBACI,OAAO,SAAS,CAAC;QACzB,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,YAAsC;QAC3D,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,OAAO,IAAI,kBAAS,CAAC;YACjB,UAAU,EAAE,IAAI;YAChB,SAAS,CAAC,KAAc,EAAE,QAAQ,EAAE,QAAQ;gBACxC,IAAI,CAAC;oBACD,MAAM,GAAG,GAAG,YAAK,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;oBACvE,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;wBACxB,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;oBACzB,CAAC;yBAAM,CAAC;wBACJ,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC;oBAC3B,CAAC;oBACD,eAAe,EAAE,CAAC;oBAClB,QAAQ,EAAE,CAAC;gBACf,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACX,QAAQ,CAAC,GAAY,CAAC,CAAC;gBAC3B,CAAC;YACL,CAAC;YACD,KAAK,CAAC,QAAQ;gBACV,IAAI,eAAe,KAAK,CAAC,EAAE,CAAC;oBACxB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACpB,CAAC;qBAAM,CAAC;oBACJ,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACnB,CAAC;gBACD,QAAQ,EAAE,CAAC;YACf,CAAC;SACJ,CAAC,CAAC;IACP,CAAC;IAEO,KAAK,CAAC,qBAAqB;QAC/B,IAAI,IAAI,CAAC,wBAAwB,EAAE,CAAC;YAChC,OAAO;QACX,CAAC;QAED,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC;QACrC,IAAI,CAAC;YACD,iGAAiG;YACjG,MAAM,iBAAiB,GAAkB,EAAE,CAAC;YAC5C,KAAK,MAAM,aAAa,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;gBAC5D,IACI,aAAa,CAAC,YAAY,KAAK,OAAO;oBACtC,eAAe,CAAC,aAAa,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,EAC7E,CAAC;oBACC,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;oBACtC,OAAO,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBACxD,CAAC;YACL,CAAC;YAED,+DAA+D;YAC/D,MAAM,mBAAmB,GAAoB,EAAE,CAAC;YAChD,KAAK,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,iBAAiB,EAAE,CAAC;gBACzD,mBAAmB,CAAC,IAAI,CACpB,IAAI,CAAC,oBAAoB,CACrB,UAAU,EACV,iBAAK,CAAC,kBAAkB,EACxB,2CAA2C,UAAU,EAAE,CAC1D,CACJ,CAAC;YACN,CAAC;YAED,MAAM,OAAO,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;QAClD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACd,EAAE,EAAE,iBAAK,CAAC,kBAAkB;gBAC5B,OAAO,EAAE,gCAAgC;gBACzC,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAClE,CAAC,CAAC;QACP,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,wBAAwB,GAAG,KAAK,CAAC;QAC1C,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAAC,UAAkB,EAAE,KAAiB,EAAE,UAAkB;QACxF,IAAI,CAAC;YACD,MAAM,kBAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,gEAAgE;YAChE,oEAAoE;YACpE,mBAAmB;YACnB,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACrD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;oBACd,EAAE,EAAE,KAAK;oBACT,OAAO,EAAE,UAAU;oBACnB,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAClE,CAAC,CAAC;YACP,CAAC;QACL,CAAC;IACL,CAAC;IAEO,uBAAuB;QAC3B,IAAI,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACxD,CAAC;IACL,CAAC;IAED,MAAM,CAAC,IAAI,CACP,MAA4B,EAC5B,MAAkB,EAClB,SAAS,GAAG,IAAI,eAAQ,EAAE,CAAC,QAAQ,EAAE;QAErC,MAAM,oBAAoB,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACtE,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,oBAAoB,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAChF,cAAc,CAAC,IAAI,EAAE,CAAC;QACtB,OAAO,cAAc,CAAC;IAC1B,CAAC;CACJ;AAvSD,wCAuSC;AAED;uDACuD;AACvD,SAAgB,eAAe,CAAC,UAAkB,EAAE,SAAiB;IACjE,MAAM,UAAU,GAAG,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,EAAE,CAAC;IAC3E,IAAI,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QAClC,OAAO,UAAU,CAAC;IACtB,CAAC;IACD,OAAO,GAAG,UAAU,GAAG,UAAU,EAAE,CAAC;AACxC,CAAC;AAED;;+BAE+B;AAC/B,SAAgB,kBAAkB,CAAC,iBAAyB;IACxD,MAAM,WAAW,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,CAAC;IAC1D,IAAI,CAAC,cAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAC7D,CAAC;IAED,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACxF,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;IAClE,CAAC;IAED,OAAO,WAAW,CAAC;AACvB,CAAC;AAED,SAAgB,eAAe,CAAC,SAAiB,EAAE,eAAuB;IACtE,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,eAAe,CAAC;AACpD,CAAC"}

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

@@ -0,0 +1,118 @@
+import type { MongoLogId } from "mongodb-log-writer";
+import type { LoggingMessageNotification } from "@modelcontextprotocol/sdk/types.js";
+import { EventEmitter } from "events";
+import type { Server } from "../lib.js";
+export type LogLevel = LoggingMessageNotification["params"]["level"];
+export declare const LogId: {
+    readonly serverStartFailure: MongoLogId;
+    readonly serverInitialized: MongoLogId;
+    readonly serverCloseRequested: MongoLogId;
+    readonly serverClosed: MongoLogId;
+    readonly serverCloseFailure: MongoLogId;
+    readonly serverDuplicateLoggers: MongoLogId;
+    readonly serverMcpClientSet: MongoLogId;
+    readonly atlasCheckCredentials: MongoLogId;
+    readonly atlasDeleteDatabaseUserFailure: MongoLogId;
+    readonly atlasConnectFailure: MongoLogId;
+    readonly atlasInspectFailure: MongoLogId;
+    readonly atlasConnectAttempt: MongoLogId;
+    readonly atlasConnectSucceeded: MongoLogId;
+    readonly atlasApiRevokeFailure: MongoLogId;
+    readonly atlasIpAccessListAdded: MongoLogId;
+    readonly atlasIpAccessListAddFailure: MongoLogId;
+    readonly telemetryDisabled: MongoLogId;
+    readonly telemetryEmitFailure: MongoLogId;
+    readonly telemetryEmitStart: MongoLogId;
+    readonly telemetryEmitSuccess: MongoLogId;
+    readonly telemetryMetadataError: MongoLogId;
+    readonly deviceIdResolutionError: MongoLogId;
+    readonly deviceIdTimeout: MongoLogId;
+    readonly toolExecute: MongoLogId;
+    readonly toolExecuteFailure: MongoLogId;
+    readonly toolDisabled: MongoLogId;
+    readonly mongodbConnectFailure: MongoLogId;
+    readonly mongodbDisconnectFailure: MongoLogId;
+    readonly toolUpdateFailure: MongoLogId;
+    readonly resourceUpdateFailure: MongoLogId;
+    readonly streamableHttpTransportStarted: MongoLogId;
+    readonly streamableHttpTransportSessionCloseFailure: MongoLogId;
+    readonly streamableHttpTransportSessionCloseNotification: MongoLogId;
+    readonly streamableHttpTransportSessionCloseNotificationFailure: MongoLogId;
+    readonly streamableHttpTransportRequestFailure: MongoLogId;
+    readonly streamableHttpTransportCloseFailure: MongoLogId;
+    readonly streamableHttpTransportKeepAliveFailure: MongoLogId;
+    readonly streamableHttpTransportKeepAlive: MongoLogId;
+    readonly exportCleanupError: MongoLogId;
+    readonly exportCreationError: MongoLogId;
+    readonly exportCreationCleanupError: MongoLogId;
+    readonly exportReadError: MongoLogId;
+    readonly exportCloseError: MongoLogId;
+    readonly exportedDataListError: MongoLogId;
+    readonly exportedDataAutoCompleteError: MongoLogId;
+    readonly exportLockError: MongoLogId;
+    readonly oidcFlow: MongoLogId;
+};
+export interface LogPayload {
+    id: MongoLogId;
+    context: string;
+    message: string;
+    noRedaction?: boolean | LoggerType | LoggerType[];
+    attributes?: Record<string, string>;
+}
+export type LoggerType = "console" | "disk" | "mcp";
+type EventMap<T> = Record<keyof T, any[]> | DefaultEventMap;
+type DefaultEventMap = [never];
+export declare abstract class LoggerBase<T extends EventMap<T> = DefaultEventMap> extends EventEmitter<T> {
+    private readonly defaultUnredactedLogger;
+    log(level: LogLevel, payload: LogPayload): void;
+    protected abstract readonly type?: LoggerType;
+    protected abstract logCore(level: LogLevel, payload: LogPayload): void;
+    private redactIfNecessary;
+    info(payload: LogPayload): void;
+    error(payload: LogPayload): void;
+    debug(payload: LogPayload): void;
+    notice(payload: LogPayload): void;
+    warning(payload: LogPayload): void;
+    critical(payload: LogPayload): void;
+    alert(payload: LogPayload): void;
+    emergency(payload: LogPayload): void;
+    protected mapToMongoDBLogLevel(level: LogLevel): "info" | "warn" | "error" | "debug" | "fatal";
+}
+export declare class ConsoleLogger extends LoggerBase {
+    protected readonly type: LoggerType;
+    protected logCore(level: LogLevel, payload: LogPayload): void;
+    private serializeAttributes;
+}
+export declare class DiskLogger extends LoggerBase<{
+    initialized: [];
+}> {
+    private bufferedMessages;
+    private logWriter?;
+    constructor(logPath: string, onError: (error: Error) => void);
+    private initialize;
+    protected type: LoggerType;
+    protected logCore(level: LogLevel, payload: LogPayload): void;
+}
+export declare class McpLogger extends LoggerBase {
+    private readonly server;
+    static readonly LOG_LEVELS: LogLevel[];
+    constructor(server: Server);
+    protected readonly type: LoggerType;
+    protected logCore(level: LogLevel, payload: LogPayload): void;
+}
+export declare class CompositeLogger extends LoggerBase {
+    protected readonly type?: LoggerType;
+    private readonly loggers;
+    private readonly attributes;
+    constructor(...loggers: LoggerBase[]);
+    addLogger(logger: LoggerBase): void;
+    log(level: LogLevel, payload: LogPayload): void;
+    protected logCore(): void;
+    setAttribute(key: string, value: string): void;
+}
+export declare class NullLogger extends LoggerBase {
+    protected type?: LoggerType;
+    protected logCore(): void;
+}
+export {};
+//# sourceMappingURL=logger.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../src/common/logger.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAkB,MAAM,oBAAoB,CAAC;AAGrE,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,oCAAoC,CAAC;AACrF,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAExC,MAAM,MAAM,QAAQ,GAAG,0BAA0B,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC;AAErE,eAAO,MAAM,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAwDR,CAAC;AAEX,MAAM,WAAW,UAAU;IACvB,EAAE,EAAE,UAAU,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,OAAO,GAAG,UAAU,GAAG,UAAU,EAAE,CAAC;IAClD,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACvC;AAED,MAAM,MAAM,UAAU,GAAG,SAAS,GAAG,MAAM,GAAG,KAAK,CAAC;AAGpD,KAAK,QAAQ,CAAC,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,GAAG,eAAe,CAAC;AAC5D,KAAK,eAAe,GAAG,CAAC,KAAK,CAAC,CAAC;AAE/B,8BAAsB,UAAU,CAAC,CAAC,SAAS,QAAQ,CAAC,CAAC,CAAC,GAAG,eAAe,CAAE,SAAQ,YAAY,CAAC,CAAC,CAAC;IAC7F,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAqB;IAEtD,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,GAAG,IAAI;IAUtD,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC;IAE9C,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,GAAG,IAAI;IAEtE,OAAO,CAAC,iBAAiB;IA2BlB,IAAI,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAI/B,KAAK,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAGhC,KAAK,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAIhC,MAAM,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAIjC,OAAO,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAIlC,QAAQ,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAInC,KAAK,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAIhC,SAAS,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI;IAI3C,SAAS,CAAC,oBAAoB,CAAC,KAAK,EAAE,QAAQ,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,OAAO,GAAG,OAAO;CAmBjG;AAED,qBAAa,aAAc,SAAQ,UAAU;IACzC,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAa;IAEhD,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,GAAG,IAAI;IAO7D,OAAO,CAAC,mBAAmB;CAQ9B;AAED,qBAAa,UAAW,SAAQ,UAAU,CAAC;IAAE,WAAW,EAAE,EAAE,CAAA;CAAE,CAAC;IAC3D,OAAO,CAAC,gBAAgB,CAAkD;IAC1E,OAAO,CAAC,SAAS,CAAC,CAAiB;gBAEhB,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI;YAMrD,UAAU;IA2BxB,SAAS,CAAC,IAAI,EAAE,UAAU,CAAU;IAEpC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,GAAG,IAAI;CAYhE;AAED,qBAAa,SAAU,SAAQ,UAAU;IAYlB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAX1C,gBAAuB,UAAU,EAAE,QAAQ,EAAE,CASlC;gBAEyB,MAAM,EAAE,MAAM;IAIlD,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAS;IAE5C,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,GAAG,IAAI;CAkBhE;AAED,qBAAa,eAAgB,SAAQ,UAAU;IAC3C,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC;IAErC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoB;IAC5C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA8B;gBAE7C,GAAG,OAAO,EAAE,UAAU,EAAE;IAM7B,SAAS,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI;IAInC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,GAAG,IAAI;IAWtD,SAAS,CAAC,OAAO,IAAI,IAAI;IAIlB,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;CAGxD;AAED,qBAAa,UAAW,SAAQ,UAAU;IACtC,SAAS,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC;IAE5B,SAAS,CAAC,OAAO,IAAI,IAAI;CAG5B"}