@librechat/data-schemas 0.0.43 → 0.0.44

package/dist/methods/key.cjs

@@ -0,0 +1,136 @@
+'use strict';
+
+var librechatDataProvider = require('librechat-data-provider');
+var index = require('../crypto/index.cjs');
+var winston = require('../config/winston.cjs');
+
+/** Factory function that takes mongoose instance and returns the key methods */
+function createKeyMethods(mongoose) {
+    /**
+     * Retrieves and decrypts the key value for a given user identified by userId and identifier name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @returns The decrypted key value
+     * @throws Error if the key is not found or if there is a problem during key retrieval
+     * @description This function searches for a user's key in the database using their userId and name.
+     *              If found, it decrypts the value of the key and returns it. If no key is found, it throws
+     *              an error indicating that there is no user key available.
+     */
+    async function getUserKey(params) {
+        const { userId, name } = params;
+        const Key = mongoose.models.Key;
+        const keyValue = (await Key.findOne({ userId, name }).lean());
+        if (!keyValue) {
+            throw new Error(JSON.stringify({
+                type: librechatDataProvider.ErrorTypes.NO_USER_KEY,
+            }));
+        }
+        return await index.decrypt(keyValue.value);
+    }
+    /**
+     * Retrieves, decrypts, and parses the key values for a given user identified by userId and name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @returns The decrypted and parsed key values
+     * @throws Error if the key is invalid or if there is a problem during key value parsing
+     * @description This function retrieves a user's encrypted key using their userId and name, decrypts it,
+     *              and then attempts to parse the decrypted string into a JSON object. If the parsing fails,
+     *              it throws an error indicating that the user key is invalid.
+     */
+    async function getUserKeyValues(params) {
+        const { userId, name } = params;
+        const userValues = await getUserKey({ userId, name });
+        try {
+            return JSON.parse(userValues);
+        }
+        catch (e) {
+            winston.error('[getUserKeyValues]', e);
+            throw new Error(JSON.stringify({
+                type: librechatDataProvider.ErrorTypes.INVALID_USER_KEY,
+            }));
+        }
+    }
+    /**
+     * Retrieves the expiry information of a user's key identified by userId and name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @returns The expiry date of the key or null if the key doesn't exist
+     * @description This function fetches a user's key from the database using their userId and name and
+     *              returns its expiry date. If the key is not found, it returns null for the expiry date.
+     */
+    async function getUserKeyExpiry(params) {
+        const { userId, name } = params;
+        const Key = mongoose.models.Key;
+        const keyValue = (await Key.findOne({ userId, name }).lean());
+        if (!keyValue) {
+            return { expiresAt: null };
+        }
+        return { expiresAt: keyValue.expiresAt || 'never' };
+    }
+    /**
+     * Updates or inserts a new key for a given user identified by userId and name, with a specified value and expiry date.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @param params.value - The value to be encrypted and stored as the key's value
+     * @param params.expiresAt - The expiry date for the key [optional]
+     * @returns The updated or newly inserted key document
+     * @description This function either updates an existing user key or inserts a new one into the database,
+     *              after encrypting the provided value. It sets the provided expiry date for the key (or unsets for no expiry).
+     */
+    async function updateUserKey(params) {
+        const { userId, name, value, expiresAt = null } = params;
+        const Key = mongoose.models.Key;
+        const encryptedValue = await index.encrypt(value);
+        const updateObject = {
+            userId,
+            name,
+            value: encryptedValue,
+        };
+        const updateQuery = {
+            $set: updateObject,
+        };
+        if (expiresAt) {
+            updateObject.expiresAt = new Date(expiresAt);
+        }
+        else {
+            updateQuery.$unset = { expiresAt: '' };
+        }
+        return await Key.findOneAndUpdate({ userId, name }, updateQuery, {
+            upsert: true,
+            new: true,
+        }).lean();
+    }
+    /**
+     * Deletes a key or all keys for a given user identified by userId, optionally based on a specified name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key to delete. If not provided and all is true, deletes all keys
+     * @param params.all - Whether to delete all keys for the user
+     * @returns The result of the deletion operation
+     * @description This function deletes a specific key or all keys for a user from the database.
+     *              If a name is provided and all is false, it deletes only the key with that name.
+     *              If all is true, it ignores the name and deletes all keys for the user.
+     */
+    async function deleteUserKey(params) {
+        const { userId, name, all = false } = params;
+        const Key = mongoose.models.Key;
+        if (all) {
+            return await Key.deleteMany({ userId });
+        }
+        return await Key.findOneAndDelete({ userId, name }).lean();
+    }
+    return {
+        getUserKey,
+        updateUserKey,
+        deleteUserKey,
+        getUserKeyValues,
+        getUserKeyExpiry,
+    };
+}
+
+exports.createKeyMethods = createKeyMethods;
+//# sourceMappingURL=key.cjs.map

package/dist/methods/key.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"key.cjs","sources":["../../src/methods/key.ts"],"sourcesContent":[null],"names":["ErrorTypes","decrypt","logger","encrypt"],"mappings":";;;;;;AAKA;AACM,SAAU,gBAAgB,CAAC,QAAmC,EAAA;AAClE;;;;;;;;;;AAUG;IACH,eAAe,UAAU,CAAC,MAAwC,EAAA;AAChE,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM;AAC/B,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;AAC/B,QAAA,MAAM,QAAQ,IAAI,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAEpD;QACR,IAAI,CAAC,QAAQ,EAAE;AACb,YAAA,MAAM,IAAI,KAAK,CACb,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAEA,gCAAU,CAAC,WAAW;AAC7B,aAAA,CAAC,CACH;QACH;AACA,QAAA,OAAO,MAAMC,aAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;IACtC;AAEA;;;;;;;;;;AAUG;IACH,eAAe,gBAAgB,CAAC,MAG/B,EAAA;AACC,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM;QAC/B,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AACrD,QAAA,IAAI;AACF,YAAA,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAA2B;QACzD;QAAE,OAAO,CAAC,EAAE;AACV,YAAAC,OAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE,CAAC,CAAC;AACrC,YAAA,MAAM,IAAI,KAAK,CACb,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAEF,gCAAU,CAAC,gBAAgB;AAClC,aAAA,CAAC,CACH;QACH;IACF;AAEA;;;;;;;;AAQG;IACH,eAAe,gBAAgB,CAAC,MAG/B,EAAA;AACC,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM;AAC/B,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;AAC/B,QAAA,MAAM,QAAQ,IAAI,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAEpD;QACR,IAAI,CAAC,QAAQ,EAAE;AACb,YAAA,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE;QAC5B;QACA,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,OAAO,EAAE;IACrD;AAEA;;;;;;;;;;AAUG;IACH,eAAe,aAAa,CAAC,MAK5B,EAAA;AACC,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,GAAG,IAAI,EAAE,GAAG,MAAM;AACxD,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;AAC/B,QAAA,MAAM,cAAc,GAAG,MAAMG,aAAO,CAAC,KAAK,CAAC;AAC3C,QAAA,MAAM,YAAY,GAAsE;YACtF,MAAM;YACN,IAAI;AACJ,YAAA,KAAK,EAAE,cAAc;SACtB;AACD,QAAA,MAAM,WAAW,GAAkE;AACjF,YAAA,IAAI,EAAE,YAAY;SACnB;QACD,IAAI,SAAS,EAAE;YACb,YAAY,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC;QAC9C;aAAO;YACL,WAAW,CAAC,MAAM,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE;QACxC;AACA,QAAA,OAAO,MAAM,GAAG,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,WAAW,EAAE;AAC/D,YAAA,MAAM,EAAE,IAAI;AACZ,YAAA,GAAG,EAAE,IAAI;SACV,CAAC,CAAC,IAAI,EAAE;IACX;AAEA;;;;;;;;;;AAUG;IACH,eAAe,aAAa,CAAC,MAI5B,EAAA;QACC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,KAAK,EAAE,GAAG,MAAM;AAC5C,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;QAC/B,IAAI,GAAG,EAAE;YACP,OAAO,MAAM,GAAG,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACzC;AACA,QAAA,OAAO,MAAM,GAAG,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;IAC5D;IAEA,OAAO;QACL,UAAU;QACV,aAAa;QACb,aAAa;QACb,gBAAgB;QAChB,gBAAgB;KACjB;AACH;;;;"}

package/dist/methods/key.es.js

@@ -0,0 +1,134 @@
+import { ErrorTypes } from 'librechat-data-provider';
+import { encrypt, decrypt } from '../crypto/index.es.js';
+import logger from '../config/winston.es.js';
+
+/** Factory function that takes mongoose instance and returns the key methods */
+function createKeyMethods(mongoose) {
+    /**
+     * Retrieves and decrypts the key value for a given user identified by userId and identifier name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @returns The decrypted key value
+     * @throws Error if the key is not found or if there is a problem during key retrieval
+     * @description This function searches for a user's key in the database using their userId and name.
+     *              If found, it decrypts the value of the key and returns it. If no key is found, it throws
+     *              an error indicating that there is no user key available.
+     */
+    async function getUserKey(params) {
+        const { userId, name } = params;
+        const Key = mongoose.models.Key;
+        const keyValue = (await Key.findOne({ userId, name }).lean());
+        if (!keyValue) {
+            throw new Error(JSON.stringify({
+                type: ErrorTypes.NO_USER_KEY,
+            }));
+        }
+        return await decrypt(keyValue.value);
+    }
+    /**
+     * Retrieves, decrypts, and parses the key values for a given user identified by userId and name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @returns The decrypted and parsed key values
+     * @throws Error if the key is invalid or if there is a problem during key value parsing
+     * @description This function retrieves a user's encrypted key using their userId and name, decrypts it,
+     *              and then attempts to parse the decrypted string into a JSON object. If the parsing fails,
+     *              it throws an error indicating that the user key is invalid.
+     */
+    async function getUserKeyValues(params) {
+        const { userId, name } = params;
+        const userValues = await getUserKey({ userId, name });
+        try {
+            return JSON.parse(userValues);
+        }
+        catch (e) {
+            logger.error('[getUserKeyValues]', e);
+            throw new Error(JSON.stringify({
+                type: ErrorTypes.INVALID_USER_KEY,
+            }));
+        }
+    }
+    /**
+     * Retrieves the expiry information of a user's key identified by userId and name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @returns The expiry date of the key or null if the key doesn't exist
+     * @description This function fetches a user's key from the database using their userId and name and
+     *              returns its expiry date. If the key is not found, it returns null for the expiry date.
+     */
+    async function getUserKeyExpiry(params) {
+        const { userId, name } = params;
+        const Key = mongoose.models.Key;
+        const keyValue = (await Key.findOne({ userId, name }).lean());
+        if (!keyValue) {
+            return { expiresAt: null };
+        }
+        return { expiresAt: keyValue.expiresAt || 'never' };
+    }
+    /**
+     * Updates or inserts a new key for a given user identified by userId and name, with a specified value and expiry date.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key
+     * @param params.value - The value to be encrypted and stored as the key's value
+     * @param params.expiresAt - The expiry date for the key [optional]
+     * @returns The updated or newly inserted key document
+     * @description This function either updates an existing user key or inserts a new one into the database,
+     *              after encrypting the provided value. It sets the provided expiry date for the key (or unsets for no expiry).
+     */
+    async function updateUserKey(params) {
+        const { userId, name, value, expiresAt = null } = params;
+        const Key = mongoose.models.Key;
+        const encryptedValue = await encrypt(value);
+        const updateObject = {
+            userId,
+            name,
+            value: encryptedValue,
+        };
+        const updateQuery = {
+            $set: updateObject,
+        };
+        if (expiresAt) {
+            updateObject.expiresAt = new Date(expiresAt);
+        }
+        else {
+            updateQuery.$unset = { expiresAt: '' };
+        }
+        return await Key.findOneAndUpdate({ userId, name }, updateQuery, {
+            upsert: true,
+            new: true,
+        }).lean();
+    }
+    /**
+     * Deletes a key or all keys for a given user identified by userId, optionally based on a specified name.
+     * @param params - The parameters object
+     * @param params.userId - The unique identifier for the user
+     * @param params.name - The name associated with the key to delete. If not provided and all is true, deletes all keys
+     * @param params.all - Whether to delete all keys for the user
+     * @returns The result of the deletion operation
+     * @description This function deletes a specific key or all keys for a user from the database.
+     *              If a name is provided and all is false, it deletes only the key with that name.
+     *              If all is true, it ignores the name and deletes all keys for the user.
+     */
+    async function deleteUserKey(params) {
+        const { userId, name, all = false } = params;
+        const Key = mongoose.models.Key;
+        if (all) {
+            return await Key.deleteMany({ userId });
+        }
+        return await Key.findOneAndDelete({ userId, name }).lean();
+    }
+    return {
+        getUserKey,
+        updateUserKey,
+        deleteUserKey,
+        getUserKeyValues,
+        getUserKeyExpiry,
+    };
+}
+
+export { createKeyMethods };
+//# sourceMappingURL=key.es.js.map

package/dist/methods/key.es.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"key.es.js","sources":["../../src/methods/key.ts"],"sourcesContent":[null],"names":[],"mappings":";;;;AAKA;AACM,SAAU,gBAAgB,CAAC,QAAmC,EAAA;AAClE;;;;;;;;;;AAUG;IACH,eAAe,UAAU,CAAC,MAAwC,EAAA;AAChE,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM;AAC/B,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;AAC/B,QAAA,MAAM,QAAQ,IAAI,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAEpD;QACR,IAAI,CAAC,QAAQ,EAAE;AACb,YAAA,MAAM,IAAI,KAAK,CACb,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,UAAU,CAAC,WAAW;AAC7B,aAAA,CAAC,CACH;QACH;AACA,QAAA,OAAO,MAAM,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;IACtC;AAEA;;;;;;;;;;AAUG;IACH,eAAe,gBAAgB,CAAC,MAG/B,EAAA;AACC,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM;QAC/B,MAAM,UAAU,GAAG,MAAM,UAAU,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AACrD,QAAA,IAAI;AACF,YAAA,OAAO,IAAI,CAAC,KAAK,CAAC,UAAU,CAA2B;QACzD;QAAE,OAAO,CAAC,EAAE;AACV,YAAA,MAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE,CAAC,CAAC;AACrC,YAAA,MAAM,IAAI,KAAK,CACb,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,UAAU,CAAC,gBAAgB;AAClC,aAAA,CAAC,CACH;QACH;IACF;AAEA;;;;;;;;AAQG;IACH,eAAe,gBAAgB,CAAC,MAG/B,EAAA;AACC,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM;AAC/B,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;AAC/B,QAAA,MAAM,QAAQ,IAAI,MAAM,GAAG,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAEpD;QACR,IAAI,CAAC,QAAQ,EAAE;AACb,YAAA,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE;QAC5B;QACA,OAAO,EAAE,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,OAAO,EAAE;IACrD;AAEA;;;;;;;;;;AAUG;IACH,eAAe,aAAa,CAAC,MAK5B,EAAA;AACC,QAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,GAAG,IAAI,EAAE,GAAG,MAAM;AACxD,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;AAC/B,QAAA,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC;AAC3C,QAAA,MAAM,YAAY,GAAsE;YACtF,MAAM;YACN,IAAI;AACJ,YAAA,KAAK,EAAE,cAAc;SACtB;AACD,QAAA,MAAM,WAAW,GAAkE;AACjF,YAAA,IAAI,EAAE,YAAY;SACnB;QACD,IAAI,SAAS,EAAE;YACb,YAAY,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC;QAC9C;aAAO;YACL,WAAW,CAAC,MAAM,GAAG,EAAE,SAAS,EAAE,EAAE,EAAE;QACxC;AACA,QAAA,OAAO,MAAM,GAAG,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,WAAW,EAAE;AAC/D,YAAA,MAAM,EAAE,IAAI;AACZ,YAAA,GAAG,EAAE,IAAI;SACV,CAAC,CAAC,IAAI,EAAE;IACX;AAEA;;;;;;;;;;AAUG;IACH,eAAe,aAAa,CAAC,MAI5B,EAAA;QACC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,KAAK,EAAE,GAAG,MAAM;AAC5C,QAAA,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG;QAC/B,IAAI,GAAG,EAAE;YACP,OAAO,MAAM,GAAG,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACzC;AACA,QAAA,OAAO,MAAM,GAAG,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;IAC5D;IAEA,OAAO;QACL,UAAU;QACV,aAAa;QACb,aAAa;QACb,gBAAgB;QAChB,gBAAgB;KACjB;AACH;;;;"}

package/dist/methods/mcpServer.cjs

@@ -0,0 +1,259 @@
+'use strict';
+
+var winston = require('../config/winston.cjs');
+var nanoid = require('nanoid');
+
+const NORMALIZED_LIMIT_DEFAULT = 20;
+const MAX_CREATE_RETRIES = 5;
+const RETRY_BASE_DELAY_MS = 25;
+/**
+ * Helper to check if an error is a MongoDB duplicate key error.
+ * Since serverName is the only unique index on MCPServer, any E11000 error
+ * during creation is necessarily a serverName collision.
+ */
+function isDuplicateKeyError(error) {
+    if (error && typeof error === 'object' && 'code' in error) {
+        const mongoError = error;
+        return mongoError.code === 11000;
+    }
+    return false;
+}
+/**
+ * Escapes special regex characters in a string so they are treated literally.
+ */
+function escapeRegex(str) {
+    return str.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
+}
+/**
+ * Generates a URL-friendly server name from a title.
+ * Converts to lowercase, replaces spaces with hyphens, removes special characters.
+ */
+function generateServerNameFromTitle(title) {
+    const slug = title
+        .toLowerCase()
+        .trim()
+        .replace(/[^a-z0-9\s-]/g, '') // Remove special chars except spaces and hyphens
+        .replace(/\s+/g, '-') // Replace spaces with hyphens
+        .replace(/-+/g, '-') // Remove consecutive hyphens
+        .replace(/^-|-$/g, ''); // Trim leading/trailing hyphens
+    return slug || 'mcp-server'; // Fallback if empty
+}
+function createMCPServerMethods(mongoose) {
+    /**
+     * Finds the next available server name by checking for duplicates.
+     * If baseName exists, returns baseName-2, baseName-3, etc.
+     */
+    async function findNextAvailableServerName(baseName) {
+        const MCPServer = mongoose.models.MCPServer;
+        // Find all servers with matching base name pattern (baseName or baseName-N)
+        const escapedBaseName = escapeRegex(baseName);
+        const existing = await MCPServer.find({
+            serverName: { $regex: `^${escapedBaseName}(-\\d+)?$` },
+        })
+            .select('serverName')
+            .lean();
+        if (existing.length === 0) {
+            return baseName;
+        }
+        // Extract numbers from existing names
+        const numbers = existing.map((s) => {
+            const match = s.serverName.match(/-(\d+)$/);
+            return match ? parseInt(match[1], 10) : 1;
+        });
+        const maxNumber = Math.max(...numbers);
+        return `${baseName}-${maxNumber + 1}`;
+    }
+    /**
+     * Create a new MCP server with retry logic for handling race conditions.
+     * When multiple requests try to create servers with the same title simultaneously,
+     * they may get the same serverName from findNextAvailableServerName() before any
+     * creates the record (TOCTOU race condition). This is handled by retrying with
+     * exponential backoff when a duplicate key error occurs.
+     * @param data - Object containing config (with title, description, url, etc.) and author
+     * @returns The created MCP server document
+     */
+    async function createMCPServer(data) {
+        const MCPServer = mongoose.models.MCPServer;
+        let lastError;
+        for (let attempt = 0; attempt < MAX_CREATE_RETRIES; attempt++) {
+            try {
+                // Generate serverName from title, with fallback to nanoid if no title
+                // Important: regenerate on each attempt to get fresh available name
+                let serverName;
+                if (data.config.title) {
+                    const baseSlug = generateServerNameFromTitle(data.config.title);
+                    serverName = await findNextAvailableServerName(baseSlug);
+                }
+                else {
+                    serverName = `mcp-${nanoid.nanoid(16)}`;
+                }
+                const newServer = await MCPServer.create({
+                    serverName,
+                    config: data.config,
+                    author: data.author,
+                });
+                return newServer.toObject();
+            }
+            catch (error) {
+                lastError = error;
+                // Only retry on duplicate key errors (serverName collision)
+                if (isDuplicateKeyError(error) && attempt < MAX_CREATE_RETRIES - 1) {
+                    // Exponential backoff: 10ms, 20ms, 40ms
+                    const delay = RETRY_BASE_DELAY_MS * Math.pow(2, attempt);
+                    winston.debug(`[createMCPServer] Duplicate serverName detected, retrying (attempt ${attempt + 2}/${MAX_CREATE_RETRIES}) after ${delay}ms`);
+                    await new Promise((resolve) => setTimeout(resolve, delay));
+                    continue;
+                }
+                // Not a duplicate key error or out of retries - throw immediately
+                throw error;
+            }
+        }
+        // Should not reach here, but TypeScript requires a return
+        throw lastError;
+    }
+    /**
+     * Find an MCP server by serverName
+     * @param serverName - The unique server name identifier
+     * @returns The MCP server document or null
+     */
+    async function findMCPServerByServerName(serverName) {
+        const MCPServer = mongoose.models.MCPServer;
+        return await MCPServer.findOne({ serverName }).lean();
+    }
+    /**
+     * Find an MCP server by MongoDB ObjectId
+     * @param _id - The MongoDB ObjectId
+     * @returns The MCP server document or null
+     */
+    async function findMCPServerByObjectId(_id) {
+        const MCPServer = mongoose.models.MCPServer;
+        return await MCPServer.findById(_id).lean();
+    }
+    /**
+     * Find MCP servers by author
+     * @param authorId - The author's ObjectId or string
+     * @returns Array of MCP server documents
+     */
+    async function findMCPServersByAuthor(authorId) {
+        const MCPServer = mongoose.models.MCPServer;
+        return await MCPServer.find({ author: authorId }).sort({ updatedAt: -1 }).lean();
+    }
+    /**
+     * Get a paginated list of MCP servers by IDs with filtering and search
+     * @param ids - Array of ObjectIds to include
+     * @param otherParams - Additional filter parameters (e.g., search)
+     * @param limit - Page size limit (null for no pagination)
+     * @param after - Cursor for pagination
+     * @returns Paginated list of MCP servers
+     */
+    async function getListMCPServersByIds({ ids = [], otherParams = {}, limit = null, after = null, }) {
+        const MCPServer = mongoose.models.MCPServer;
+        const isPaginated = limit !== null && limit !== undefined;
+        const normalizedLimit = isPaginated
+            ? Math.min(Math.max(1, parseInt(String(limit)) || NORMALIZED_LIMIT_DEFAULT), 100)
+            : null;
+        // Build base query combining accessible servers with other filters
+        const baseQuery = { ...otherParams, _id: { $in: ids } };
+        // Add cursor condition
+        if (after) {
+            try {
+                const cursor = JSON.parse(Buffer.from(after, 'base64').toString('utf8'));
+                const { updatedAt, _id } = cursor;
+                const cursorCondition = {
+                    $or: [
+                        { updatedAt: { $lt: new Date(updatedAt) } },
+                        { updatedAt: new Date(updatedAt), _id: { $gt: new mongoose.Types.ObjectId(_id) } },
+                    ],
+                };
+                // Merge cursor condition with base query
+                if (Object.keys(baseQuery).length > 0) {
+                    baseQuery.$and = [{ ...baseQuery }, cursorCondition];
+                    // Remove the original conditions from baseQuery to avoid duplication
+                    Object.keys(baseQuery).forEach((key) => {
+                        if (key !== '$and') {
+                            delete baseQuery[key];
+                        }
+                    });
+                }
+            }
+            catch (error) {
+                // Invalid cursor, ignore
+                winston.warn('[getListMCPServersByIds] Invalid cursor provided', error);
+            }
+        }
+        if (normalizedLimit === null) {
+            // No pagination - return all matching servers
+            const servers = await MCPServer.find(baseQuery).sort({ updatedAt: -1, _id: 1 }).lean();
+            return {
+                data: servers,
+                has_more: false,
+                after: null,
+            };
+        }
+        // Paginated query - assign to const to help TypeScript
+        const servers = await MCPServer.find(baseQuery)
+            .sort({ updatedAt: -1, _id: 1 })
+            .limit(normalizedLimit + 1)
+            .lean();
+        const hasMore = servers.length > normalizedLimit;
+        const data = hasMore ? servers.slice(0, normalizedLimit) : servers;
+        let nextCursor = null;
+        if (hasMore && data.length > 0) {
+            const lastItem = data[data.length - 1];
+            nextCursor = Buffer.from(JSON.stringify({
+                updatedAt: lastItem.updatedAt,
+                _id: lastItem._id,
+            })).toString('base64');
+        }
+        return {
+            data,
+            has_more: hasMore,
+            after: nextCursor,
+        };
+    }
+    /**
+     * Update an MCP server
+     * @param serverName - The MCP server ID
+     * @param updateData - Object containing config to update
+     * @returns The updated MCP server document or null
+     */
+    async function updateMCPServer(serverName, updateData) {
+        const MCPServer = mongoose.models.MCPServer;
+        return await MCPServer.findOneAndUpdate({ serverName }, { $set: updateData }, { new: true, runValidators: true }).lean();
+    }
+    /**
+     * Delete an MCP server
+     * @param serverName - The MCP server ID
+     * @returns The deleted MCP server document or null
+     */
+    async function deleteMCPServer(serverName) {
+        const MCPServer = mongoose.models.MCPServer;
+        return await MCPServer.findOneAndDelete({ serverName }).lean();
+    }
+    /**
+     * Get MCP servers by their serverName strings
+     * @param names - Array of serverName strings to fetch
+     * @returns Object containing array of MCP server documents
+     */
+    async function getListMCPServersByNames({ names = [] }) {
+        if (names.length === 0) {
+            return { data: [] };
+        }
+        const MCPServer = mongoose.models.MCPServer;
+        const servers = await MCPServer.find({ serverName: { $in: names } }).lean();
+        return { data: servers };
+    }
+    return {
+        createMCPServer,
+        findMCPServerByServerName,
+        findMCPServerByObjectId,
+        findMCPServersByAuthor,
+        getListMCPServersByIds,
+        getListMCPServersByNames,
+        updateMCPServer,
+        deleteMCPServer,
+    };
+}
+
+exports.createMCPServerMethods = createMCPServerMethods;
+//# sourceMappingURL=mcpServer.cjs.map

package/dist/methods/mcpServer.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcpServer.cjs","sources":["../../src/methods/mcpServer.ts"],"sourcesContent":[null],"names":["nanoid","logger"],"mappings":";;;;;AAMA,MAAM,wBAAwB,GAAG,EAAE;AACnC,MAAM,kBAAkB,GAAG,CAAC;AAC5B,MAAM,mBAAmB,GAAG,EAAE;AAE9B;;;;AAIG;AACH,SAAS,mBAAmB,CAAC,KAAc,EAAA;IACzC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,IAAI,KAAK,EAAE;QACzD,MAAM,UAAU,GAAG,KAAyB;AAC5C,QAAA,OAAO,UAAU,CAAC,IAAI,KAAK,KAAK;IAClC;AACA,IAAA,OAAO,KAAK;AACd;AAEA;;AAEG;AACH,SAAS,WAAW,CAAC,GAAW,EAAA;IAC9B,OAAO,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC;AACnD;AAEA;;;AAGG;AACH,SAAS,2BAA2B,CAAC,KAAa,EAAA;IAChD,MAAM,IAAI,GAAG;AACV,SAAA,WAAW;AACX,SAAA,IAAI;AACJ,SAAA,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC;AAC5B,SAAA,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;AACpB,SAAA,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;AACnB,SAAA,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;AAEzB,IAAA,OAAO,IAAI,IAAI,YAAY,CAAC;AAC9B;AAEM,SAAU,sBAAsB,CAAC,QAAmC,EAAA;AACxE;;;AAGG;IACH,eAAe,2BAA2B,CAAC,QAAgB,EAAA;AACzD,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;;AAGvE,QAAA,MAAM,eAAe,GAAG,WAAW,CAAC,QAAQ,CAAC;AAC7C,QAAA,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC;AACpC,YAAA,UAAU,EAAE,EAAE,MAAM,EAAE,CAAA,CAAA,EAAI,eAAe,WAAW,EAAE;SACvD;aACE,MAAM,CAAC,YAAY;AACnB,aAAA,IAAI,EAAE;AAET,QAAA,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;AACzB,YAAA,OAAO,QAAQ;QACjB;;QAGA,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,KAAI;YACjC,MAAM,KAAK,GAAG,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC;AAC3C,YAAA,OAAO,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC;AAC3C,QAAA,CAAC,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;AACtC,QAAA,OAAO,GAAG,QAAQ,CAAA,CAAA,EAAI,SAAS,GAAG,CAAC,EAAE;IACvC;AAEA;;;;;;;;AAQG;IACH,eAAe,eAAe,CAAC,IAG9B,EAAA;AACC,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;AACvE,QAAA,IAAI,SAAkB;AAEtB,QAAA,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,kBAAkB,EAAE,OAAO,EAAE,EAAE;AAC7D,YAAA,IAAI;;;AAGF,gBAAA,IAAI,UAAkB;AACtB,gBAAA,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE;oBACrB,MAAM,QAAQ,GAAG,2BAA2B,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;AAC/D,oBAAA,UAAU,GAAG,MAAM,2BAA2B,CAAC,QAAQ,CAAC;gBAC1D;qBAAO;AACL,oBAAA,UAAU,GAAG,CAAA,IAAA,EAAOA,aAAM,CAAC,EAAE,CAAC,EAAE;gBAClC;AAEA,gBAAA,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,MAAM,CAAC;oBACvC,UAAU;oBACV,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,MAAM,EAAE,IAAI,CAAC,MAAM;AACpB,iBAAA,CAAC;AAEF,gBAAA,OAAO,SAAS,CAAC,QAAQ,EAAuB;YAClD;YAAE,OAAO,KAAK,EAAE;gBACd,SAAS,GAAG,KAAK;;gBAGjB,IAAI,mBAAmB,CAAC,KAAK,CAAC,IAAI,OAAO,GAAG,kBAAkB,GAAG,CAAC,EAAE;;AAElE,oBAAA,MAAM,KAAK,GAAG,mBAAmB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC;AACxD,oBAAAC,OAAM,CAAC,KAAK,CACV,CAAA,mEAAA,EAAsE,OAAO,GAAG,CAAC,CAAA,CAAA,EAAI,kBAAkB,CAAA,QAAA,EAAW,KAAK,CAAA,EAAA,CAAI,CAC5H;AACD,oBAAA,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;oBAC1D;gBACF;;AAGA,gBAAA,MAAM,KAAK;YACb;QACF;;AAGA,QAAA,MAAM,SAAS;IACjB;AAEA;;;;AAIG;IACH,eAAe,yBAAyB,CAAC,UAAkB,EAAA;AACzD,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;AACvE,QAAA,OAAO,MAAM,SAAS,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE;IACvD;AAEA;;;;AAIG;IACH,eAAe,uBAAuB,CACpC,GAA4B,EAAA;AAE5B,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;QACvE,OAAO,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE;IAC7C;AAEA;;;;AAIG;IACH,eAAe,sBAAsB,CACnC,QAAiC,EAAA;AAEjC,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;QACvE,OAAO,MAAM,SAAS,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE;IAClF;AAEA;;;;;;;AAOG;AACH,IAAA,eAAe,sBAAsB,CAAC,EACpC,GAAG,GAAG,EAAE,EACR,WAAW,GAAG,EAAE,EAChB,KAAK,GAAG,IAAI,EACZ,KAAK,GAAG,IAAI,GAMb,EAAA;AAKC,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;QACvE,MAAM,WAAW,GAAG,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS;QACzD,MAAM,eAAe,GAAG;cACpB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,wBAAwB,CAAC,EAAE,GAAG;cAC9E,IAAI;;AAGR,QAAA,MAAM,SAAS,GAAuC,EAAE,GAAG,WAAW,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;;QAG3F,IAAI,KAAK,EAAE;AACT,YAAA,IAAI;gBACF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AACxE,gBAAA,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,GAAG,MAAM;AAEjC,gBAAA,MAAM,eAAe,GAAG;AACtB,oBAAA,GAAG,EAAE;wBACH,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE;wBAC3C,EAAE,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE;AACnF,qBAAA;iBACF;;gBAGD,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE;oBACrC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,GAAG,SAAS,EAAE,EAAE,eAAe,CAAC;;oBAEpD,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,KAAI;AACrC,wBAAA,IAAI,GAAG,KAAK,MAAM,EAAE;AAClB,4BAAA,OAAO,SAAS,CAAC,GAAG,CAAC;wBACvB;AACF,oBAAA,CAAC,CAAC;gBACJ;YACF;YAAE,OAAO,KAAK,EAAE;;AAEd,gBAAAA,OAAM,CAAC,IAAI,CAAC,kDAAkD,EAAE,KAAK,CAAC;YACxE;QACF;AAEA,QAAA,IAAI,eAAe,KAAK,IAAI,EAAE;;YAE5B,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE;YAEtF,OAAO;AACL,gBAAA,IAAI,EAAE,OAAO;AACb,gBAAA,QAAQ,EAAE,KAAK;AACf,gBAAA,KAAK,EAAE,IAAI;aACZ;QACH;;QAGA,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,SAAS;aAC3C,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE;AAC9B,aAAA,KAAK,CAAC,eAAe,GAAG,CAAC;AACzB,aAAA,IAAI,EAAE;AAET,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,GAAG,eAAe;AAChD,QAAA,MAAM,IAAI,GAAG,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,eAAe,CAAC,GAAG,OAAO;QAElE,IAAI,UAAU,GAAG,IAAI;QACrB,IAAI,OAAO,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE;YAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;YACtC,UAAU,GAAG,MAAM,CAAC,IAAI,CACtB,IAAI,CAAC,SAAS,CAAC;gBACb,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,GAAG,EAAE,QAAQ,CAAC,GAAG;AAClB,aAAA,CAAC,CACH,CAAC,QAAQ,CAAC,QAAQ,CAAC;QACtB;QAEA,OAAO;YACL,IAAI;AACJ,YAAA,QAAQ,EAAE,OAAO;AACjB,YAAA,KAAK,EAAE,UAAU;SAClB;IACH;AAEA;;;;;AAKG;AACH,IAAA,eAAe,eAAe,CAC5B,UAAkB,EAClB,UAAmC,EAAA;AAEnC,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;AACvE,QAAA,OAAO,MAAM,SAAS,CAAC,gBAAgB,CACrC,EAAE,UAAU,EAAE,EACd,EAAE,IAAI,EAAE,UAAU,EAAE,EACpB,EAAE,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CACnC,CAAC,IAAI,EAAE;IACV;AAEA;;;;AAIG;IACH,eAAe,eAAe,CAAC,UAAkB,EAAA;AAC/C,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;AACvE,QAAA,OAAO,MAAM,SAAS,CAAC,gBAAgB,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,IAAI,EAAE;IAChE;AAEA;;;;AAIG;AACH,IAAA,eAAe,wBAAwB,CAAC,EAAE,KAAK,GAAG,EAAE,EAAuB,EAAA;AAGzE,QAAA,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;AACtB,YAAA,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE;QACrB;AACA,QAAA,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,SAAqC;QACvE,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE;AAC3E,QAAA,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE;IAC1B;IAEA,OAAO;QACL,eAAe;QACf,yBAAyB;QACzB,uBAAuB;QACvB,sBAAsB;QACtB,sBAAsB;QACtB,wBAAwB;QACxB,eAAe;QACf,eAAe;KAChB;AACH;;;;"}