123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986 |
- 'use strict';
-
- const applyWriteConcern = require('../utils').applyWriteConcern;
- const Code = require('mongodb-core').BSON.Code;
- const resolveReadPreference = require('../utils').resolveReadPreference;
- const crypto = require('crypto');
- const debugOptions = require('../utils').debugOptions;
- const handleCallback = require('../utils').handleCallback;
- const MongoError = require('mongodb-core').MongoError;
- const parseIndexOptions = require('../utils').parseIndexOptions;
- const ReadPreference = require('mongodb-core').ReadPreference;
- const toError = require('../utils').toError;
- const CONSTANTS = require('../constants');
-
- const count = require('./collection_ops').count;
- const findOne = require('./collection_ops').findOne;
- const remove = require('./collection_ops').remove;
- const updateOne = require('./collection_ops').updateOne;
-
- const debugFields = [
- 'authSource',
- 'w',
- 'wtimeout',
- 'j',
- 'native_parser',
- 'forceServerObjectId',
- 'serializeFunctions',
- 'raw',
- 'promoteLongs',
- 'promoteValues',
- 'promoteBuffers',
- 'bufferMaxEntries',
- 'numberOfRetries',
- 'retryMiliSeconds',
- 'readPreference',
- 'pkFactory',
- 'parentDb',
- 'promiseLibrary',
- 'noListener'
- ];
-
- // Filter out any write concern options
- const illegalCommandFields = [
- 'w',
- 'wtimeout',
- 'j',
- 'fsync',
- 'autoIndexId',
- 'strict',
- 'serializeFunctions',
- 'pkFactory',
- 'raw',
- 'readPreference',
- 'session'
- ];
-
- /**
- * Add a user to the database.
- * @method
- * @param {Db} db The Db instance on which to add a user.
- * @param {string} username The username.
- * @param {string} password The password.
- * @param {object} [options] Optional settings. See Db.prototype.addUser for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function addUser(db, username, password, options, callback) {
- const Db = require('../db');
-
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed())
- return callback(new MongoError('topology was destroyed'));
- // Attempt to execute auth command
- executeAuthCreateUserCommand(db, username, password, options, (err, r) => {
- // We need to perform the backward compatible insert operation
- if (err && err.code === -5000) {
- const finalOptions = applyWriteConcern(Object.assign({}, options), { db }, options);
-
- // Use node md5 generator
- const md5 = crypto.createHash('md5');
- // Generate keys used for authentication
- md5.update(username + ':mongo:' + password);
- const userPassword = md5.digest('hex');
-
- // If we have another db set
- const db = options.dbName ? new Db(options.dbName, db.s.topology, db.s.options) : db;
-
- // Fetch a user collection
- const collection = db.collection(CONSTANTS.SYSTEM_USER_COLLECTION);
-
- // Check if we are inserting the first user
- count(collection, {}, finalOptions, (err, count) => {
- // We got an error (f.ex not authorized)
- if (err != null) return handleCallback(callback, err, null);
- // Check if the user exists and update i
- const findOptions = Object.assign({ projection: { dbName: 1 } }, finalOptions);
- collection.find({ user: username }, findOptions).toArray(err => {
- // We got an error (f.ex not authorized)
- if (err != null) return handleCallback(callback, err, null);
- // Add command keys
- finalOptions.upsert = true;
-
- // We have a user, let's update the password or upsert if not
- updateOne(
- collection,
- { user: username },
- { $set: { user: username, pwd: userPassword } },
- finalOptions,
- err => {
- if (count === 0 && err)
- return handleCallback(callback, null, [{ user: username, pwd: userPassword }]);
- if (err) return handleCallback(callback, err, null);
- handleCallback(callback, null, [{ user: username, pwd: userPassword }]);
- }
- );
- });
- });
-
- return;
- }
-
- if (err) return handleCallback(callback, err);
- handleCallback(callback, err, r);
- });
- }
-
- /**
- * Fetch all collections for the current db.
- *
- * @method
- * @param {Db} db The Db instance on which to fetch collections.
- * @param {object} [options] Optional settings. See Db.prototype.collections for a list of options.
- * @param {Db~collectionsResultCallback} [callback] The results callback
- */
- function collections(db, options, callback) {
- const Collection = require('../collection');
-
- options = Object.assign({}, options, { nameOnly: true });
- // Let's get the collection names
- db.listCollections({}, options).toArray((err, documents) => {
- if (err != null) return handleCallback(callback, err, null);
- // Filter collections removing any illegal ones
- documents = documents.filter(doc => {
- return doc.name.indexOf('$') === -1;
- });
-
- // Return the collection objects
- handleCallback(
- callback,
- null,
- documents.map(d => {
- return new Collection(
- db,
- db.s.topology,
- db.s.databaseName,
- d.name,
- db.s.pkFactory,
- db.s.options
- );
- })
- );
- });
- }
-
- /**
- * Create a new collection on a server with the specified options. Use this to create capped collections.
- * More information about command options available at https://docs.mongodb.com/manual/reference/command/create/
- *
- * @method
- * @param {Db} db The Db instance on which to create the collection.
- * @param {string} name The collection name to create.
- * @param {object} [options] Optional settings. See Db.prototype.createCollection for a list of options.
- * @param {Db~collectionResultCallback} [callback] The results callback
- */
- function createCollection(db, name, options, callback) {
- const Collection = require('../collection');
-
- // Get the write concern options
- const finalOptions = applyWriteConcern(Object.assign({}, options), { db }, options);
-
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed()) {
- return callback(new MongoError('topology was destroyed'));
- }
-
- const listCollectionOptions = Object.assign({}, finalOptions, { nameOnly: true });
-
- // Check if we have the name
- db
- .listCollections({ name }, listCollectionOptions)
- .setReadPreference(ReadPreference.PRIMARY)
- .toArray((err, collections) => {
- if (err != null) return handleCallback(callback, err, null);
- if (collections.length > 0 && finalOptions.strict) {
- return handleCallback(
- callback,
- MongoError.create({
- message: `Collection ${name} already exists. Currently in strict mode.`,
- driver: true
- }),
- null
- );
- } else if (collections.length > 0) {
- try {
- return handleCallback(
- callback,
- null,
- new Collection(db, db.s.topology, db.s.databaseName, name, db.s.pkFactory, options)
- );
- } catch (err) {
- return handleCallback(callback, err);
- }
- }
-
- // Create collection command
- const cmd = { create: name };
-
- // Decorate command with writeConcern if supported
- applyWriteConcern(cmd, { db }, options);
-
- // Add all optional parameters
- for (let n in options) {
- if (
- options[n] != null &&
- typeof options[n] !== 'function' &&
- illegalCommandFields.indexOf(n) === -1
- ) {
- cmd[n] = options[n];
- }
- }
-
- // Force a primary read Preference
- finalOptions.readPreference = ReadPreference.PRIMARY;
- // Execute command
- executeCommand(db, cmd, finalOptions, err => {
- if (err) return handleCallback(callback, err);
- handleCallback(
- callback,
- null,
- new Collection(db, db.s.topology, db.s.databaseName, name, db.s.pkFactory, options)
- );
- });
- });
- }
-
- /**
- * Creates an index on the db and collection.
- * @method
- * @param {Db} db The Db instance on which to create an index.
- * @param {string} name Name of the collection to create the index on.
- * @param {(string|object)} fieldOrSpec Defines the index.
- * @param {object} [options] Optional settings. See Db.prototype.createIndex for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function createIndex(db, name, fieldOrSpec, options, callback) {
- // Get the write concern options
- let finalOptions = Object.assign({}, { readPreference: ReadPreference.PRIMARY }, options);
- finalOptions = applyWriteConcern(finalOptions, { db }, options);
-
- // Ensure we have a callback
- if (finalOptions.writeConcern && typeof callback !== 'function') {
- throw MongoError.create({
- message: 'Cannot use a writeConcern without a provided callback',
- driver: true
- });
- }
-
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed())
- return callback(new MongoError('topology was destroyed'));
-
- // Attempt to run using createIndexes command
- createIndexUsingCreateIndexes(db, name, fieldOrSpec, finalOptions, (err, result) => {
- if (err == null) return handleCallback(callback, err, result);
-
- /**
- * The following errors mean that the server recognized `createIndex` as a command so we don't need to fallback to an insert:
- * 67 = 'CannotCreateIndex' (malformed index options)
- * 85 = 'IndexOptionsConflict' (index already exists with different options)
- * 86 = 'IndexKeySpecsConflict' (index already exists with the same name)
- * 11000 = 'DuplicateKey' (couldn't build unique index because of dupes)
- * 11600 = 'InterruptedAtShutdown' (interrupted at shutdown)
- * 197 = 'InvalidIndexSpecificationOption' (`_id` with `background: true`)
- */
- if (
- err.code === 67 ||
- err.code === 11000 ||
- err.code === 85 ||
- err.code === 86 ||
- err.code === 11600 ||
- err.code === 197
- ) {
- return handleCallback(callback, err, result);
- }
-
- // Create command
- const doc = createCreateIndexCommand(db, name, fieldOrSpec, options);
- // Set no key checking
- finalOptions.checkKeys = false;
- // Insert document
- db.s.topology.insert(
- `${db.s.databaseName}.${CONSTANTS.SYSTEM_INDEX_COLLECTION}`,
- doc,
- finalOptions,
- (err, result) => {
- if (callback == null) return;
- if (err) return handleCallback(callback, err);
- if (result == null) return handleCallback(callback, null, null);
- if (result.result.writeErrors)
- return handleCallback(callback, MongoError.create(result.result.writeErrors[0]), null);
- handleCallback(callback, null, doc.name);
- }
- );
- });
- }
-
- // Add listeners to topology
- function createListener(db, e, object) {
- function listener(err) {
- if (object.listeners(e).length > 0) {
- object.emit(e, err, db);
-
- // Emit on all associated db's if available
- for (let i = 0; i < db.s.children.length; i++) {
- db.s.children[i].emit(e, err, db.s.children[i]);
- }
- }
- }
- return listener;
- }
-
- /**
- * Drop a collection from the database, removing it permanently. New accesses will create a new collection.
- *
- * @method
- * @param {Db} db The Db instance on which to drop the collection.
- * @param {string} name Name of collection to drop
- * @param {Object} [options] Optional settings. See Db.prototype.dropCollection for a list of options.
- * @param {Db~resultCallback} [callback] The results callback
- */
- function dropCollection(db, name, options, callback) {
- executeCommand(db, name, options, (err, result) => {
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed()) {
- return callback(new MongoError('topology was destroyed'));
- }
-
- if (err) return handleCallback(callback, err);
- if (result.ok) return handleCallback(callback, null, true);
- handleCallback(callback, null, false);
- });
- }
-
- /**
- * Drop a database, removing it permanently from the server.
- *
- * @method
- * @param {Db} db The Db instance to drop.
- * @param {Object} cmd The command document.
- * @param {Object} [options] Optional settings. See Db.prototype.dropDatabase for a list of options.
- * @param {Db~resultCallback} [callback] The results callback
- */
- function dropDatabase(db, cmd, options, callback) {
- executeCommand(db, cmd, options, (err, result) => {
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed()) {
- return callback(new MongoError('topology was destroyed'));
- }
-
- if (callback == null) return;
- if (err) return handleCallback(callback, err, null);
- handleCallback(callback, null, result.ok ? true : false);
- });
- }
-
- /**
- * Ensures that an index exists. If it does not, creates it.
- *
- * @method
- * @param {Db} db The Db instance on which to ensure the index.
- * @param {string} name The index name
- * @param {(string|object)} fieldOrSpec Defines the index.
- * @param {object} [options] Optional settings. See Db.prototype.ensureIndex for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function ensureIndex(db, name, fieldOrSpec, options, callback) {
- // Get the write concern options
- const finalOptions = applyWriteConcern({}, { db }, options);
- // Create command
- const selector = createCreateIndexCommand(db, name, fieldOrSpec, options);
- const index_name = selector.name;
-
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed())
- return callback(new MongoError('topology was destroyed'));
-
- // Merge primary readPreference
- finalOptions.readPreference = ReadPreference.PRIMARY;
-
- // Check if the index already exists
- indexInformation(db, name, finalOptions, (err, indexInformation) => {
- if (err != null && err.code !== 26) return handleCallback(callback, err, null);
- // If the index does not exist, create it
- if (indexInformation == null || !indexInformation[index_name]) {
- createIndex(db, name, fieldOrSpec, options, callback);
- } else {
- if (typeof callback === 'function') return handleCallback(callback, null, index_name);
- }
- });
- }
-
- /**
- * Evaluate JavaScript on the server
- *
- * @method
- * @param {Db} db The Db instance.
- * @param {Code} code JavaScript to execute on server.
- * @param {(object|array)} parameters The parameters for the call.
- * @param {object} [options] Optional settings. See Db.prototype.eval for a list of options.
- * @param {Db~resultCallback} [callback] The results callback
- * @deprecated Eval is deprecated on MongoDB 3.2 and forward
- */
- function evaluate(db, code, parameters, options, callback) {
- let finalCode = code;
- let finalParameters = [];
-
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed())
- return callback(new MongoError('topology was destroyed'));
-
- // If not a code object translate to one
- if (!(finalCode && finalCode._bsontype === 'Code')) finalCode = new Code(finalCode);
- // Ensure the parameters are correct
- if (parameters != null && !Array.isArray(parameters) && typeof parameters !== 'function') {
- finalParameters = [parameters];
- } else if (parameters != null && Array.isArray(parameters) && typeof parameters !== 'function') {
- finalParameters = parameters;
- }
-
- // Create execution selector
- let cmd = { $eval: finalCode, args: finalParameters };
- // Check if the nolock parameter is passed in
- if (options['nolock']) {
- cmd['nolock'] = options['nolock'];
- }
-
- // Set primary read preference
- options.readPreference = new ReadPreference(ReadPreference.PRIMARY);
-
- // Execute the command
- executeCommand(db, cmd, options, (err, result) => {
- if (err) return handleCallback(callback, err, null);
- if (result && result.ok === 1) return handleCallback(callback, null, result.retval);
- if (result)
- return handleCallback(
- callback,
- MongoError.create({ message: `eval failed: ${result.errmsg}`, driver: true }),
- null
- );
- handleCallback(callback, err, result);
- });
- }
-
- /**
- * Execute a command
- *
- * @method
- * @param {Db} db The Db instance on which to execute the command.
- * @param {object} command The command hash
- * @param {object} [options] Optional settings. See Db.prototype.command for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function executeCommand(db, command, options, callback) {
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed())
- return callback(new MongoError('topology was destroyed'));
- // Get the db name we are executing against
- const dbName = options.dbName || options.authdb || db.s.databaseName;
-
- // Convert the readPreference if its not a write
- options.readPreference = resolveReadPreference(options, { db, default: ReadPreference.primary });
-
- // Debug information
- if (db.s.logger.isDebug())
- db.s.logger.debug(
- `executing command ${JSON.stringify(
- command
- )} against ${dbName}.$cmd with options [${JSON.stringify(
- debugOptions(debugFields, options)
- )}]`
- );
-
- // Execute command
- db.s.topology.command(`${dbName}.$cmd`, command, options, (err, result) => {
- if (err) return handleCallback(callback, err);
- if (options.full) return handleCallback(callback, null, result);
- handleCallback(callback, null, result.result);
- });
- }
-
- /**
- * Runs a command on the database as admin.
- *
- * @method
- * @param {Db} db The Db instance on which to execute the command.
- * @param {object} command The command hash
- * @param {object} [options] Optional settings. See Db.prototype.executeDbAdminCommand for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function executeDbAdminCommand(db, command, options, callback) {
- db.s.topology.command('admin.$cmd', command, options, (err, result) => {
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed()) {
- return callback(new MongoError('topology was destroyed'));
- }
-
- if (err) return handleCallback(callback, err);
- handleCallback(callback, null, result.result);
- });
- }
-
- /**
- * Retrieves this collections index info.
- *
- * @method
- * @param {Db} db The Db instance on which to retrieve the index info.
- * @param {string} name The name of the collection.
- * @param {object} [options] Optional settings. See Db.prototype.indexInformation for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function indexInformation(db, name, options, callback) {
- // If we specified full information
- const full = options['full'] == null ? false : options['full'];
-
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed())
- return callback(new MongoError('topology was destroyed'));
- // Process all the results from the index command and collection
- function processResults(indexes) {
- // Contains all the information
- let info = {};
- // Process all the indexes
- for (let i = 0; i < indexes.length; i++) {
- const index = indexes[i];
- // Let's unpack the object
- info[index.name] = [];
- for (let name in index.key) {
- info[index.name].push([name, index.key[name]]);
- }
- }
-
- return info;
- }
-
- // Get the list of indexes of the specified collection
- db
- .collection(name)
- .listIndexes(options)
- .toArray((err, indexes) => {
- if (err) return callback(toError(err));
- if (!Array.isArray(indexes)) return handleCallback(callback, null, []);
- if (full) return handleCallback(callback, null, indexes);
- handleCallback(callback, null, processResults(indexes));
- });
- }
-
- // Transformation methods for cursor results
- function listCollectionsTransforms(databaseName) {
- const matching = `${databaseName}.`;
-
- return {
- doc: doc => {
- const index = doc.name.indexOf(matching);
- // Remove database name if available
- if (doc.name && index === 0) {
- doc.name = doc.name.substr(index + matching.length);
- }
-
- return doc;
- }
- };
- }
-
- /**
- * Retrive the current profiling information for MongoDB
- *
- * @method
- * @param {Db} db The Db instance on which to retrieve the profiling info.
- * @param {Object} [options] Optional settings. See Db.protoype.profilingInfo for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback.
- * @deprecated Query the system.profile collection directly.
- */
- function profilingInfo(db, options, callback) {
- try {
- db
- .collection('system.profile')
- .find({}, options)
- .toArray(callback);
- } catch (err) {
- return callback(err, null);
- }
- }
-
- /**
- * Retrieve the current profiling level for MongoDB
- *
- * @method
- * @param {Db} db The Db instance on which to retrieve the profiling level.
- * @param {Object} [options] Optional settings. See Db.prototype.profilingLevel for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function profilingLevel(db, options, callback) {
- executeCommand(db, { profile: -1 }, options, (err, doc) => {
- if (err == null && doc.ok === 1) {
- const was = doc.was;
- if (was === 0) return callback(null, 'off');
- if (was === 1) return callback(null, 'slow_only');
- if (was === 2) return callback(null, 'all');
- return callback(new Error('Error: illegal profiling level value ' + was), null);
- } else {
- err != null ? callback(err, null) : callback(new Error('Error with profile command'), null);
- }
- });
- }
-
- /**
- * Remove a user from a database
- *
- * @method
- * @param {Db} db The Db instance on which to remove the user.
- * @param {string} username The username.
- * @param {object} [options] Optional settings. See Db.prototype.removeUser for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function removeUser(db, username, options, callback) {
- const Db = require('../db');
-
- // Attempt to execute command
- executeAuthRemoveUserCommand(db, username, options, (err, result) => {
- if (err && err.code === -5000) {
- const finalOptions = applyWriteConcern(Object.assign({}, options), { db }, options);
- // If we have another db set
- const db = options.dbName ? new Db(options.dbName, db.s.topology, db.s.options) : db;
-
- // Fetch a user collection
- const collection = db.collection(CONSTANTS.SYSTEM_USER_COLLECTION);
-
- // Locate the user
- findOne(collection, { user: username }, finalOptions, (err, user) => {
- if (user == null) return handleCallback(callback, err, false);
- remove(collection, { user: username }, finalOptions, err => {
- handleCallback(callback, err, true);
- });
- });
-
- return;
- }
-
- if (err) return handleCallback(callback, err);
- handleCallback(callback, err, result);
- });
- }
-
- /**
- * Set the current profiling level of MongoDB
- *
- * @method
- * @param {Db} db The Db instance on which to execute the command.
- * @param {string} level The new profiling level (off, slow_only, all).
- * @param {Object} [options] Optional settings. See Db.prototype.setProfilingLevel for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback.
- */
- function setProfilingLevel(db, level, options, callback) {
- const command = {};
- let profile = 0;
-
- if (level === 'off') {
- profile = 0;
- } else if (level === 'slow_only') {
- profile = 1;
- } else if (level === 'all') {
- profile = 2;
- } else {
- return callback(new Error('Error: illegal profiling level value ' + level));
- }
-
- // Set up the profile number
- command['profile'] = profile;
-
- executeCommand(db, command, options, (err, doc) => {
- if (err == null && doc.ok === 1) return callback(null, level);
- return err != null
- ? callback(err, null)
- : callback(new Error('Error with profile command'), null);
- });
- }
-
- // Validate the database name
- function validateDatabaseName(databaseName) {
- if (typeof databaseName !== 'string')
- throw MongoError.create({ message: 'database name must be a string', driver: true });
- if (databaseName.length === 0)
- throw MongoError.create({ message: 'database name cannot be the empty string', driver: true });
- if (databaseName === '$external') return;
-
- const invalidChars = [' ', '.', '$', '/', '\\'];
- for (let i = 0; i < invalidChars.length; i++) {
- if (databaseName.indexOf(invalidChars[i]) !== -1)
- throw MongoError.create({
- message: "database names cannot contain the character '" + invalidChars[i] + "'",
- driver: true
- });
- }
- }
-
- /**
- * Create the command object for Db.prototype.createIndex.
- *
- * @param {Db} db The Db instance on which to create the command.
- * @param {string} name Name of the collection to create the index on.
- * @param {(string|object)} fieldOrSpec Defines the index.
- * @param {Object} [options] Optional settings. See Db.prototype.createIndex for a list of options.
- * @return {Object} The insert command object.
- */
- function createCreateIndexCommand(db, name, fieldOrSpec, options) {
- const indexParameters = parseIndexOptions(fieldOrSpec);
- const fieldHash = indexParameters.fieldHash;
-
- // Generate the index name
- const indexName = typeof options.name === 'string' ? options.name : indexParameters.name;
- const selector = {
- ns: db.databaseName + '.' + name,
- key: fieldHash,
- name: indexName
- };
-
- // Ensure we have a correct finalUnique
- const finalUnique = options == null || 'object' === typeof options ? false : options;
- // Set up options
- options = options == null || typeof options === 'boolean' ? {} : options;
-
- // Add all the options
- const keysToOmit = Object.keys(selector);
- for (let optionName in options) {
- if (keysToOmit.indexOf(optionName) === -1) {
- selector[optionName] = options[optionName];
- }
- }
-
- if (selector['unique'] == null) selector['unique'] = finalUnique;
-
- // Remove any write concern operations
- const removeKeys = ['w', 'wtimeout', 'j', 'fsync', 'readPreference', 'session'];
- for (let i = 0; i < removeKeys.length; i++) {
- delete selector[removeKeys[i]];
- }
-
- // Return the command creation selector
- return selector;
- }
-
- /**
- * Create index using the createIndexes command.
- *
- * @param {Db} db The Db instance on which to execute the command.
- * @param {string} name Name of the collection to create the index on.
- * @param {(string|object)} fieldOrSpec Defines the index.
- * @param {Object} [options] Optional settings. See Db.prototype.createIndex for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback.
- */
- function createIndexUsingCreateIndexes(db, name, fieldOrSpec, options, callback) {
- // Build the index
- const indexParameters = parseIndexOptions(fieldOrSpec);
- // Generate the index name
- const indexName = typeof options.name === 'string' ? options.name : indexParameters.name;
- // Set up the index
- const indexes = [{ name: indexName, key: indexParameters.fieldHash }];
- // merge all the options
- const keysToOmit = Object.keys(indexes[0]).concat([
- 'writeConcern',
- 'w',
- 'wtimeout',
- 'j',
- 'fsync',
- 'readPreference',
- 'session'
- ]);
-
- for (let optionName in options) {
- if (keysToOmit.indexOf(optionName) === -1) {
- indexes[0][optionName] = options[optionName];
- }
- }
-
- // Get capabilities
- const capabilities = db.s.topology.capabilities();
-
- // Did the user pass in a collation, check if our write server supports it
- if (indexes[0].collation && capabilities && !capabilities.commandsTakeCollation) {
- // Create a new error
- const error = new MongoError('server/primary/mongos does not support collation');
- error.code = 67;
- // Return the error
- return callback(error);
- }
-
- // Create command, apply write concern to command
- const cmd = applyWriteConcern({ createIndexes: name, indexes }, { db }, options);
-
- // ReadPreference primary
- options.readPreference = ReadPreference.PRIMARY;
-
- // Build the command
- executeCommand(db, cmd, options, (err, result) => {
- if (err) return handleCallback(callback, err, null);
- if (result.ok === 0) return handleCallback(callback, toError(result), null);
- // Return the indexName for backward compatibility
- handleCallback(callback, null, indexName);
- });
- }
-
- /**
- * Run the createUser command.
- *
- * @param {Db} db The Db instance on which to execute the command.
- * @param {string} username The username of the user to add.
- * @param {string} password The password of the user to add.
- * @param {object} [options] Optional settings. See Db.prototype.addUser for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function executeAuthCreateUserCommand(db, username, password, options, callback) {
- // Special case where there is no password ($external users)
- if (typeof username === 'string' && password != null && typeof password === 'object') {
- options = password;
- password = null;
- }
-
- // Unpack all options
- if (typeof options === 'function') {
- callback = options;
- options = {};
- }
-
- // Error out if we digestPassword set
- if (options.digestPassword != null) {
- return callback(
- toError(
- "The digestPassword option is not supported via add_user. Please use db.command('createUser', ...) instead for this option."
- )
- );
- }
-
- // Get additional values
- const customData = options.customData != null ? options.customData : {};
- let roles = Array.isArray(options.roles) ? options.roles : [];
- const maxTimeMS = typeof options.maxTimeMS === 'number' ? options.maxTimeMS : null;
-
- // If not roles defined print deprecated message
- if (roles.length === 0) {
- console.log('Creating a user without roles is deprecated in MongoDB >= 2.6');
- }
-
- // Get the error options
- const commandOptions = { writeCommand: true };
- if (options['dbName']) commandOptions.dbName = options['dbName'];
-
- // Add maxTimeMS to options if set
- if (maxTimeMS != null) commandOptions.maxTimeMS = maxTimeMS;
-
- // Check the db name and add roles if needed
- if (
- (db.databaseName.toLowerCase() === 'admin' || options.dbName === 'admin') &&
- !Array.isArray(options.roles)
- ) {
- roles = ['root'];
- } else if (!Array.isArray(options.roles)) {
- roles = ['dbOwner'];
- }
-
- const digestPassword = db.s.topology.lastIsMaster().maxWireVersion >= 7;
-
- // Build the command to execute
- let command = {
- createUser: username,
- customData: customData,
- roles: roles,
- digestPassword
- };
-
- // Apply write concern to command
- command = applyWriteConcern(command, { db }, options);
-
- let userPassword = password;
-
- if (!digestPassword) {
- // Use node md5 generator
- const md5 = crypto.createHash('md5');
- // Generate keys used for authentication
- md5.update(username + ':mongo:' + password);
- userPassword = md5.digest('hex');
- }
-
- // No password
- if (typeof password === 'string') {
- command.pwd = userPassword;
- }
-
- // Force write using primary
- commandOptions.readPreference = ReadPreference.primary;
-
- // Execute the command
- executeCommand(db, command, commandOptions, (err, result) => {
- if (err && err.ok === 0 && err.code === undefined)
- return handleCallback(callback, { code: -5000 }, null);
- if (err) return handleCallback(callback, err, null);
- handleCallback(
- callback,
- !result.ok ? toError(result) : null,
- result.ok ? [{ user: username, pwd: '' }] : null
- );
- });
- }
-
- /**
- * Run the dropUser command.
- *
- * @param {Db} db The Db instance on which to execute the command.
- * @param {string} username The username of the user to remove.
- * @param {object} [options] Optional settings. See Db.prototype.removeUser for a list of options.
- * @param {Db~resultCallback} [callback] The command result callback
- */
- function executeAuthRemoveUserCommand(db, username, options, callback) {
- if (typeof options === 'function') (callback = options), (options = {});
- options = options || {};
-
- // Did the user destroy the topology
- if (db.serverConfig && db.serverConfig.isDestroyed())
- return callback(new MongoError('topology was destroyed'));
- // Get the error options
- const commandOptions = { writeCommand: true };
- if (options['dbName']) commandOptions.dbName = options['dbName'];
-
- // Get additional values
- const maxTimeMS = typeof options.maxTimeMS === 'number' ? options.maxTimeMS : null;
-
- // Add maxTimeMS to options if set
- if (maxTimeMS != null) commandOptions.maxTimeMS = maxTimeMS;
-
- // Build the command to execute
- let command = {
- dropUser: username
- };
-
- // Apply write concern to command
- command = applyWriteConcern(command, { db }, options);
-
- // Force write using primary
- commandOptions.readPreference = ReadPreference.primary;
-
- // Execute the command
- executeCommand(db, command, commandOptions, (err, result) => {
- if (err && !err.ok && err.code === undefined) return handleCallback(callback, { code: -5000 });
- if (err) return handleCallback(callback, err, null);
- handleCallback(callback, null, result.ok ? true : false);
- });
- }
-
- module.exports = {
- addUser,
- collections,
- createCollection,
- createListener,
- createIndex,
- dropCollection,
- dropDatabase,
- ensureIndex,
- evaluate,
- executeCommand,
- executeDbAdminCommand,
- listCollectionsTransforms,
- indexInformation,
- profilingInfo,
- profilingLevel,
- removeUser,
- setProfilingLevel,
- validateDatabaseName
- };
|