diff options
author | 2017-08-07 11:39:39 +0800 | |
---|---|---|
committer | 2017-08-07 11:39:39 +0800 | |
commit | 21d72c4a80fe2937d0c4ddd20624b27adbcd989b (patch) | |
tree | e5013ee12f74f8452e01cbff16e7b0158bc456cb /common/src/main/webapp/usageguide/appserver/node_modules/mongoose/lib/model.js | |
parent | f533e73e2ae32e010b16abdcf7985abaf31ab843 (diff) |
Upload the ESR GUI seed code
Issue-ID: AAI-68
Change-Id: Ia50ce0570c2fabecd77199d4e8454f56fe587c4e
Signed-off-by: lizi00164331 <li.zi30@zte.com.cn>
Diffstat (limited to 'common/src/main/webapp/usageguide/appserver/node_modules/mongoose/lib/model.js')
-rw-r--r-- | common/src/main/webapp/usageguide/appserver/node_modules/mongoose/lib/model.js | 3414 |
1 files changed, 3414 insertions, 0 deletions
diff --git a/common/src/main/webapp/usageguide/appserver/node_modules/mongoose/lib/model.js b/common/src/main/webapp/usageguide/appserver/node_modules/mongoose/lib/model.js new file mode 100644 index 0000000..4a7cbc4 --- /dev/null +++ b/common/src/main/webapp/usageguide/appserver/node_modules/mongoose/lib/model.js @@ -0,0 +1,3414 @@ +/*! + * Module dependencies. + */ + +var Document = require('./document'); +var MongooseError = require('./error'); +var VersionError = MongooseError.VersionError; +var DivergentArrayError = MongooseError.DivergentArrayError; +var Query = require('./query'); +var Aggregate = require('./aggregate'); +var Schema = require('./schema'); +var utils = require('./utils'); +var hasOwnProperty = utils.object.hasOwnProperty; +var isMongooseObject = utils.isMongooseObject; +var EventEmitter = require('events').EventEmitter; +var MongooseArray = require('./types/array'); +var util = require('util'); +var tick = utils.tick; + +var parallel = require('async/parallel'); +var PromiseProvider = require('./promise_provider'); + +var VERSION_WHERE = 1, + VERSION_INC = 2, + VERSION_ALL = VERSION_WHERE | VERSION_INC; + +var POJO_TO_OBJECT_OPTIONS = { + depopulate: true, + transform: false, + _skipDepopulateTopLevel: true +}; + +/** + * Model constructor + * + * Provides the interface to MongoDB collections as well as creates document instances. + * + * @param {Object} doc values with which to create the document + * @inherits Document http://mongoosejs.com/docs/api.html#document-js + * @event `error`: If listening to this event, it is emitted when a document was saved without passing a callback and an `error` occurred. If not listening, the event bubbles to the connection used to create this Model. + * @event `index`: Emitted after `Model#ensureIndexes` completes. If an error occurred it is passed with the event. + * @event `index-single-start`: Emitted when an individual index starts within `Model#ensureIndexes`. The fields and options being used to build the index are also passed with the event. + * @event `index-single-done`: Emitted when an individual index finishes within `Model#ensureIndexes`. If an error occurred it is passed with the event. The fields, options, and index name are also passed. + * @api public + */ + +function Model(doc, fields, skipId) { + Document.call(this, doc, fields, skipId); +} + +/*! + * Inherits from Document. + * + * All Model.prototype features are available on + * top level (non-sub) documents. + */ + +Model.prototype.__proto__ = Document.prototype; + +/** + * Connection the model uses. + * + * @api public + * @property db + */ + +Model.prototype.db; + +/** + * Collection the model uses. + * + * @api public + * @property collection + */ + +Model.prototype.collection; + +/** + * The name of the model + * + * @api public + * @property modelName + */ + +Model.prototype.modelName; + +/** + * If this is a discriminator model, `baseModelName` is the name of + * the base model. + * + * @api public + * @property baseModelName + */ + +Model.prototype.baseModelName; + +Model.prototype.$__handleSave = function(options, callback) { + var _this = this; + if (!options.safe && this.schema.options.safe) { + options.safe = this.schema.options.safe; + } + if (typeof options.safe === 'boolean') { + options.safe = null; + } + + if (this.isNew) { + // send entire doc + var toObjectOptions = {}; + + toObjectOptions.retainKeyOrder = this.schema.options.retainKeyOrder; + toObjectOptions.depopulate = 1; + toObjectOptions._skipDepopulateTopLevel = true; + toObjectOptions.transform = false; + + var obj = this.toObject(toObjectOptions); + + if (!utils.object.hasOwnProperty(obj || {}, '_id')) { + // documents must have an _id else mongoose won't know + // what to update later if more changes are made. the user + // wouldn't know what _id was generated by mongodb either + // nor would the ObjectId generated my mongodb necessarily + // match the schema definition. + setTimeout(function() { + callback(new Error('document must have an _id before saving')); + }, 0); + return; + } + + this.$__version(true, obj); + this.collection.insert(obj, options.safe, function(err, ret) { + if (err) { + _this.isNew = true; + _this.emit('isNew', true); + + callback(err); + return; + } + + callback(null, ret); + }); + this.$__reset(); + this.isNew = false; + this.emit('isNew', false); + // Make it possible to retry the insert + this.$__.inserting = true; + } else { + // Make sure we don't treat it as a new object on error, + // since it already exists + this.$__.inserting = false; + + var delta = this.$__delta(); + + if (delta) { + if (delta instanceof Error) { + callback(delta); + return; + } + + var where = this.$__where(delta[0]); + + if (where instanceof Error) { + callback(where); + return; + } + + this.collection.update(where, delta[1], options.safe, function(err, ret) { + if (err) { + callback(err); + return; + } + callback(null, ret); + }); + } else { + this.$__reset(); + callback(); + return; + } + + this.emit('isNew', false); + } +}; + +/*! + * ignore + */ + +Model.prototype.$__save = function(options, callback) { + var _this = this; + + _this.$__handleSave(options, function(error, result) { + if (error) { + return _this.schema.s.hooks.execPost('save:error', _this, [_this], { error: error }, function(error) { + callback(error); + }); + } + + _this.$__reset(); + _this.$__storeShard(); + + var numAffected = 0; + if (result) { + if (Array.isArray(result)) { + numAffected = result.length; + } else if (result.result && result.result.n !== undefined) { + numAffected = result.result.n; + } else if (result.result && result.result.nModified !== undefined) { + numAffected = result.result.nModified; + } else { + numAffected = result; + } + } + + // was this an update that required a version bump? + if (_this.$__.version && !_this.$__.inserting) { + var doIncrement = VERSION_INC === (VERSION_INC & _this.$__.version); + _this.$__.version = undefined; + + if (numAffected <= 0) { + // the update failed. pass an error back + var err = new VersionError(_this); + return callback(err); + } + + // increment version if was successful + if (doIncrement) { + var key = _this.schema.options.versionKey; + var version = _this.getValue(key) | 0; + _this.setValue(key, version + 1); + } + } + + _this.emit('save', _this, numAffected); + callback(null, _this, numAffected); + }); +}; + +/** + * Saves this document. + * + * ####Example: + * + * product.sold = Date.now(); + * product.save(function (err, product, numAffected) { + * if (err) .. + * }) + * + * The callback will receive three parameters + * + * 1. `err` if an error occurred + * 2. `product` which is the saved `product` + * 3. `numAffected` will be 1 when the document was successfully persisted to MongoDB, otherwise 0. Unless you tweak mongoose's internals, you don't need to worry about checking this parameter for errors - checking `err` is sufficient to make sure your document was properly saved. + * + * As an extra measure of flow control, save will return a Promise. + * ####Example: + * product.save().then(function(product) { + * ... + * }); + * + * For legacy reasons, mongoose stores object keys in reverse order on initial + * save. That is, `{ a: 1, b: 2 }` will be saved as `{ b: 2, a: 1 }` in + * MongoDB. To override this behavior, set + * [the `toObject.retainKeyOrder` option](http://mongoosejs.com/docs/api.html#document_Document-toObject) + * to true on your schema. + * + * @param {Object} [options] options optional options + * @param {Object} [options.safe] overrides [schema's safe option](http://mongoosejs.com//docs/guide.html#safe) + * @param {Boolean} [options.validateBeforeSave] set to false to save without validating. + * @param {Function} [fn] optional callback + * @return {Promise} Promise + * @api public + * @see middleware http://mongoosejs.com/docs/middleware.html + */ + +Model.prototype.save = function(options, fn) { + if (typeof options === 'function') { + fn = options; + options = undefined; + } + + if (!options) { + options = {}; + } + + if (fn) { + fn = this.constructor.$wrapCallback(fn); + } + + return this.$__save(options, fn); +}; + +/*! + * Determines whether versioning should be skipped for the given path + * + * @param {Document} self + * @param {String} path + * @return {Boolean} true if versioning should be skipped for the given path + */ +function shouldSkipVersioning(self, path) { + var skipVersioning = self.schema.options.skipVersioning; + if (!skipVersioning) return false; + + // Remove any array indexes from the path + path = path.replace(/\.\d+\./, '.'); + + return skipVersioning[path]; +} + +/*! + * Apply the operation to the delta (update) clause as + * well as track versioning for our where clause. + * + * @param {Document} self + * @param {Object} where + * @param {Object} delta + * @param {Object} data + * @param {Mixed} val + * @param {String} [operation] + */ + +function operand(self, where, delta, data, val, op) { + // delta + op || (op = '$set'); + if (!delta[op]) delta[op] = {}; + delta[op][data.path] = val; + + // disabled versioning? + if (self.schema.options.versionKey === false) return; + + // path excluded from versioning? + if (shouldSkipVersioning(self, data.path)) return; + + // already marked for versioning? + if (VERSION_ALL === (VERSION_ALL & self.$__.version)) return; + + switch (op) { + case '$set': + case '$unset': + case '$pop': + case '$pull': + case '$pullAll': + case '$push': + case '$pushAll': + case '$addToSet': + break; + default: + // nothing to do + return; + } + + // ensure updates sent with positional notation are + // editing the correct array element. + // only increment the version if an array position changes. + // modifying elements of an array is ok if position does not change. + + if (op === '$push' || op === '$pushAll' || op === '$addToSet') { + self.$__.version = VERSION_INC; + } else if (/^\$p/.test(op)) { + // potentially changing array positions + self.increment(); + } else if (Array.isArray(val)) { + // $set an array + self.increment(); + } else if (/\.\d+\.|\.\d+$/.test(data.path)) { + // now handling $set, $unset + // subpath of array + self.$__.version = VERSION_WHERE; + } +} + +/*! + * Compiles an update and where clause for a `val` with _atomics. + * + * @param {Document} self + * @param {Object} where + * @param {Object} delta + * @param {Object} data + * @param {Array} value + */ + +function handleAtomics(self, where, delta, data, value) { + if (delta.$set && delta.$set[data.path]) { + // $set has precedence over other atomics + return; + } + + if (typeof value.$__getAtomics === 'function') { + value.$__getAtomics().forEach(function(atomic) { + var op = atomic[0]; + var val = atomic[1]; + if (self.schema.options.usePushEach && op === '$pushAll') { + op = '$push'; + val = { $each: val }; + } + operand(self, where, delta, data, val, op); + }); + return; + } + + // legacy support for plugins + + var atomics = value._atomics, + ops = Object.keys(atomics), + i = ops.length, + val, + op; + + if (i === 0) { + // $set + + if (isMongooseObject(value)) { + value = value.toObject({depopulate: 1, _isNested: true}); + } else if (value.valueOf) { + value = value.valueOf(); + } + + return operand(self, where, delta, data, value); + } + + function iter(mem) { + return isMongooseObject(mem) + ? mem.toObject({depopulate: 1, _isNested: true}) + : mem; + } + + while (i--) { + op = ops[i]; + val = atomics[op]; + + if (isMongooseObject(val)) { + val = val.toObject({depopulate: true, transform: false, _isNested: true}); + } else if (Array.isArray(val)) { + val = val.map(iter); + } else if (val.valueOf) { + val = val.valueOf(); + } + + if (op === '$addToSet') { + val = {$each: val}; + } + + operand(self, where, delta, data, val, op); + } +} + +/** + * Produces a special query document of the modified properties used in updates. + * + * @api private + * @method $__delta + * @memberOf Model + */ + +Model.prototype.$__delta = function() { + var dirty = this.$__dirty(); + if (!dirty.length && VERSION_ALL !== this.$__.version) return; + + var where = {}, + delta = {}, + len = dirty.length, + divergent = [], + d = 0; + + where._id = this._doc._id; + if (where._id.toObject) { + where._id = where._id.toObject({ transform: false, depopulate: true }); + } + + for (; d < len; ++d) { + var data = dirty[d]; + var value = data.value; + + var match = checkDivergentArray(this, data.path, value); + if (match) { + divergent.push(match); + continue; + } + + var pop = this.populated(data.path, true); + if (!pop && this.$__.selected) { + // If any array was selected using an $elemMatch projection, we alter the path and where clause + // NOTE: MongoDB only supports projected $elemMatch on top level array. + var pathSplit = data.path.split('.'); + var top = pathSplit[0]; + if (this.$__.selected[top] && this.$__.selected[top].$elemMatch) { + // If the selected array entry was modified + if (pathSplit.length > 1 && pathSplit[1] == 0 && typeof where[top] === 'undefined') { + where[top] = this.$__.selected[top]; + pathSplit[1] = '$'; + data.path = pathSplit.join('.'); + } + // if the selected array was modified in any other way throw an error + else { + divergent.push(data.path); + continue; + } + } + } + + if (divergent.length) continue; + + if (undefined === value) { + operand(this, where, delta, data, 1, '$unset'); + } else if (value === null) { + operand(this, where, delta, data, null); + } else if (value._path && value._atomics) { + // arrays and other custom types (support plugins etc) + handleAtomics(this, where, delta, data, value); + } else if (value._path && Buffer.isBuffer(value)) { + // MongooseBuffer + value = value.toObject(); + operand(this, where, delta, data, value); + } else { + value = utils.clone(value, {depopulate: 1, _isNested: true}); + operand(this, where, delta, data, value); + } + } + + if (divergent.length) { + return new DivergentArrayError(divergent); + } + + if (this.$__.version) { + this.$__version(where, delta); + } + + return [where, delta]; +}; + +/*! + * Determine if array was populated with some form of filter and is now + * being updated in a manner which could overwrite data unintentionally. + * + * @see https://github.com/Automattic/mongoose/issues/1334 + * @param {Document} doc + * @param {String} path + * @return {String|undefined} + */ + +function checkDivergentArray(doc, path, array) { + // see if we populated this path + var pop = doc.populated(path, true); + + if (!pop && doc.$__.selected) { + // If any array was selected using an $elemMatch projection, we deny the update. + // NOTE: MongoDB only supports projected $elemMatch on top level array. + var top = path.split('.')[0]; + if (doc.$__.selected[top + '.$']) { + return top; + } + } + + if (!(pop && array && array.isMongooseArray)) return; + + // If the array was populated using options that prevented all + // documents from being returned (match, skip, limit) or they + // deselected the _id field, $pop and $set of the array are + // not safe operations. If _id was deselected, we do not know + // how to remove elements. $pop will pop off the _id from the end + // of the array in the db which is not guaranteed to be the + // same as the last element we have here. $set of the entire array + // would be similarily destructive as we never received all + // elements of the array and potentially would overwrite data. + var check = pop.options.match || + pop.options.options && hasOwnProperty(pop.options.options, 'limit') || // 0 is not permitted + pop.options.options && pop.options.options.skip || // 0 is permitted + pop.options.select && // deselected _id? + (pop.options.select._id === 0 || + /\s?-_id\s?/.test(pop.options.select)); + + if (check) { + var atomics = array._atomics; + if (Object.keys(atomics).length === 0 || atomics.$set || atomics.$pop) { + return path; + } + } +} + +/** + * Appends versioning to the where and update clauses. + * + * @api private + * @method $__version + * @memberOf Model + */ + +Model.prototype.$__version = function(where, delta) { + var key = this.schema.options.versionKey; + + if (where === true) { + // this is an insert + if (key) this.setValue(key, delta[key] = 0); + return; + } + + // updates + + // only apply versioning if our versionKey was selected. else + // there is no way to select the correct version. we could fail + // fast here and force them to include the versionKey but + // thats a bit intrusive. can we do this automatically? + if (!this.isSelected(key)) { + return; + } + + // $push $addToSet don't need the where clause set + if (VERSION_WHERE === (VERSION_WHERE & this.$__.version)) { + where[key] = this.getValue(key); + } + + if (VERSION_INC === (VERSION_INC & this.$__.version)) { + if (!delta.$set || typeof delta.$set[key] === 'undefined') { + delta.$inc || (delta.$inc = {}); + delta.$inc[key] = 1; + } + } +}; + +/** + * Signal that we desire an increment of this documents version. + * + * ####Example: + * + * Model.findById(id, function (err, doc) { + * doc.increment(); + * doc.save(function (err) { .. }) + * }) + * + * @see versionKeys http://mongoosejs.com/docs/guide.html#versionKey + * @api public + */ + +Model.prototype.increment = function increment() { + this.$__.version = VERSION_ALL; + return this; +}; + +/** + * Returns a query object which applies shardkeys if they exist. + * + * @api private + * @method $__where + * @memberOf Model + */ + +Model.prototype.$__where = function _where(where) { + where || (where = {}); + + var paths, + len; + + if (!where._id) { + where._id = this._doc._id; + } + + if (this.$__.shardval) { + paths = Object.keys(this.$__.shardval); + len = paths.length; + + for (var i = 0; i < len; ++i) { + where[paths[i]] = this.$__.shardval[paths[i]]; + } + } + + if (this._doc._id == null) { + return new Error('No _id found on document!'); + } + + return where; +}; + +/** + * Removes this document from the db. + * + * ####Example: + * product.remove(function (err, product) { + * if (err) return handleError(err); + * Product.findById(product._id, function (err, product) { + * console.log(product) // null + * }) + * }) + * + * + * As an extra measure of flow control, remove will return a Promise (bound to `fn` if passed) so it could be chained, or hooked to recive errors + * + * ####Example: + * product.remove().then(function (product) { + * ... + * }).onRejected(function (err) { + * assert.ok(err) + * }) + * + * @param {function(err,product)} [fn] optional callback + * @return {Promise} Promise + * @api public + */ + +Model.prototype.remove = function remove(options, fn) { + if (typeof options === 'function') { + fn = options; + options = undefined; + } + + if (!options) { + options = {}; + } + + if (this.$__.removing) { + if (fn) { + this.$__.removing.then( + function(res) { fn(null, res); }, + function(err) { fn(err); }); + } + return this; + } + + var _this = this; + var Promise = PromiseProvider.get(); + + if (fn) { + fn = this.constructor.$wrapCallback(fn); + } + + this.$__.removing = new Promise.ES6(function(resolve, reject) { + var where = _this.$__where(); + if (where instanceof Error) { + reject(where); + fn && fn(where); + return; + } + + if (!options.safe && _this.schema.options.safe) { + options.safe = _this.schema.options.safe; + } + + _this.collection.remove(where, options, function(err) { + if (!err) { + _this.emit('remove', _this); + resolve(_this); + fn && fn(null, _this); + return; + } + reject(err); + fn && fn(err); + }); + }); + return this.$__.removing; +}; + +/** + * Returns another Model instance. + * + * ####Example: + * + * var doc = new Tank; + * doc.model('User').findById(id, callback); + * + * @param {String} name model name + * @api public + */ + +Model.prototype.model = function model(name) { + return this.db.model(name); +}; + +/** + * Adds a discriminator type. + * + * ####Example: + * + * function BaseSchema() { + * Schema.apply(this, arguments); + * + * this.add({ + * name: String, + * createdAt: Date + * }); + * } + * util.inherits(BaseSchema, Schema); + * + * var PersonSchema = new BaseSchema(); + * var BossSchema = new BaseSchema({ department: String }); + * + * var Person = mongoose.model('Person', PersonSchema); + * var Boss = Person.discriminator('Boss', BossSchema); + * + * @param {String} name discriminator model name + * @param {Schema} schema discriminator model schema + * @api public + */ + +Model.discriminator = function discriminator(name, schema) { + var CUSTOMIZABLE_DISCRIMINATOR_OPTIONS = { + toJSON: true, + toObject: true, + _id: true, + id: true + }; + + if (!(schema && schema.instanceOfSchema)) { + throw new Error('You must pass a valid discriminator Schema'); + } + + if (this.schema.discriminatorMapping && !this.schema.discriminatorMapping.isRoot) { + throw new Error('Discriminator "' + name + + '" can only be a discriminator of the root model'); + } + + var key = this.schema.options.discriminatorKey; + if (schema.path(key)) { + throw new Error('Discriminator "' + name + + '" cannot have field with name "' + key + '"'); + } + + function merge(schema, baseSchema) { + utils.merge(schema, baseSchema); + + var obj = {}; + obj[key] = { + default: name, + set: function(newName) { + if (newName === name) { + return name; + } + throw new Error('Can\'t set discriminator key "' + key + '"'); + } + }; + obj[key][schema.options.typeKey] = String; + schema.add(obj); + schema.discriminatorMapping = {key: key, value: name, isRoot: false}; + + if (baseSchema.options.collection) { + schema.options.collection = baseSchema.options.collection; + } + + var toJSON = schema.options.toJSON; + var toObject = schema.options.toObject; + var _id = schema.options._id; + var id = schema.options.id; + + var keys = Object.keys(schema.options); + + for (var i = 0; i < keys.length; ++i) { + var _key = keys[i]; + if (!CUSTOMIZABLE_DISCRIMINATOR_OPTIONS[_key]) { + if (!utils.deepEqual(schema.options[_key], baseSchema.options[_key])) { + throw new Error('Can\'t customize discriminator option ' + _key + + ' (can only modify ' + + Object.keys(CUSTOMIZABLE_DISCRIMINATOR_OPTIONS).join(', ') + + ')'); + } + } + } + + schema.options = utils.clone(baseSchema.options); + if (toJSON) schema.options.toJSON = toJSON; + if (toObject) schema.options.toObject = toObject; + if (typeof _id !== 'undefined') { + schema.options._id = _id; + } + schema.options.id = id; + + schema.callQueue = baseSchema.callQueue.concat(schema.callQueue.slice(schema._defaultMiddleware.length)); + schema._requiredpaths = undefined; // reset just in case Schema#requiredPaths() was called on either schema + } + + // merges base schema into new discriminator schema and sets new type field. + merge(schema, this.schema); + + if (!this.discriminators) { + this.discriminators = {}; + } + + if (!this.schema.discriminatorMapping) { + this.schema.discriminatorMapping = {key: key, value: null, isRoot: true}; + } + + if (this.discriminators[name]) { + throw new Error('Discriminator with name "' + name + '" already exists'); + } + if (this.db.models[name]) { + throw new MongooseError.OverwriteModelError(name); + } + + this.discriminators[name] = this.db.model(name, schema, this.collection.name); + this.discriminators[name].prototype.__proto__ = this.prototype; + Object.defineProperty(this.discriminators[name], 'baseModelName', { + value: this.modelName, + configurable: true, + writable: false + }); + + // apply methods and statics + applyMethods(this.discriminators[name], schema); + applyStatics(this.discriminators[name], schema); + + return this.discriminators[name]; +}; + +// Model (class) features + +/*! + * Give the constructor the ability to emit events. + */ + +for (var i in EventEmitter.prototype) { + Model[i] = EventEmitter.prototype[i]; +} + +/** + * Called when the model compiles. + * + * @api private + */ + +Model.init = function init() { + if ((this.schema.options.autoIndex) || + (this.schema.options.autoIndex === null && this.db.config.autoIndex)) { + this.ensureIndexes({ __noPromise: true, _automatic: true }); + } + + this.schema.emit('init', this); +}; + +/** + * Sends `ensureIndex` commands to mongo for each index declared in the schema. + * + * ####Example: + * + * Event.ensureIndexes(function (err) { + * if (err) return handleError(err); + * }); + * + * After completion, an `index` event is emitted on this `Model` passing an error if one occurred. + * + * ####Example: + * + * var eventSchema = new Schema({ thing: { type: 'string', unique: true }}) + * var Event = mongoose.model('Event', eventSchema); + * + * Event.on('index', function (err) { + * if (err) console.error(err); // error occurred during index creation + * }) + * + * _NOTE: It is not recommended that you run this in production. Index creation may impact database performance depending on your load. Use with caution._ + * + * The `ensureIndex` commands are not sent in parallel. This is to avoid the `MongoError: cannot add index with a background operation in progress` error. See [this ticket](https://github.com/Automattic/mongoose/issues/1365) for more information. + * + * @param {Object} [options] internal options + * @param {Function} [cb] optional callback + * @return {Promise} + * @api public + */ + +Model.ensureIndexes = function ensureIndexes(options, callback) { + if (typeof options === 'function') { + callback = options; + options = null; + } + + if (options && options.__noPromise) { + _ensureIndexes(this, options, callback); + return; + } + + if (callback) { + callback = this.$wrapCallback(callback); + } + + var _this = this; + var Promise = PromiseProvider.get(); + return new Promise.ES6(function(resolve, reject) { + _ensureIndexes(_this, options || {}, function(error) { + if (error) { + callback && callback(error); + reject(error); + } + callback && callback(); + resolve(); + }); + }); +}; + +function _ensureIndexes(model, options, callback) { + var indexes = model.schema.indexes(); + if (!indexes.length) { + setImmediate(function() { + callback && callback(); + }); + return; + } + // Indexes are created one-by-one to support how MongoDB < 2.4 deals + // with background indexes. + + var done = function(err) { + if (err && model.schema.options.emitIndexErrors) { + model.emit('error', err); + } + model.emit('index', err); + callback && callback(err); + }; + + var indexSingleDone = function(err, fields, options, name) { + model.emit('index-single-done', err, fields, options, name); + }; + var indexSingleStart = function(fields, options) { + model.emit('index-single-start', fields, options); + }; + + var create = function() { + var index = indexes.shift(); + if (!index) return done(); + + var indexFields = index[0]; + var options = index[1]; + _handleSafe(options); + + indexSingleStart(indexFields, options); + + model.collection.ensureIndex(indexFields, options, tick(function(err, name) { + indexSingleDone(err, indexFields, options, name); + if (err) { + return done(err); + } + create(); + })); + }; + + setImmediate(function() { + // If buffering is off, do this manually. + if (options._automatic && !model.collection.collection) { + model.collection.addQueue(create, []); + } else { + create(); + } + }); +} + +function _handleSafe(options) { + if (options.safe) { + if (typeof options.safe === 'boolean') { + options.w = options.safe; + delete options.safe; + } + if (typeof options.safe === 'object') { + options.w = options.safe.w; + options.j = options.safe.j; + options.wtimeout = options.safe.wtimeout; + delete options.safe; + } + } +} + +/** + * Schema the model uses. + * + * @property schema + * @receiver Model + * @api public + */ + +Model.schema; + +/*! + * Connection instance the model uses. + * + * @property db + * @receiver Model + * @api public + */ + +Model.db; + +/*! + * Collection the model uses. + * + * @property collection + * @receiver Model + * @api public + */ + +Model.collection; + +/** + * Base Mongoose instance the model uses. + * + * @property base + * @receiver Model + * @api public + */ + +Model.base; + +/** + * Registered discriminators for this model. + * + * @property discriminators + * @receiver Model + * @api public + */ + +Model.discriminators; + +/** + * Removes documents from the collection. + * + * ####Example: + * + * Comment.remove({ title: 'baby born from alien father' }, function (err) { + * + * }); + * + * ####Note: + * + * To remove documents without waiting for a response from MongoDB, do not pass a `callback`, then call `exec` on the returned [Query](#query-js): + * + * var query = Comment.remove({ _id: id }); + * query.exec(); + * + * ####Note: + * + * This method sends a remove command directly to MongoDB, no Mongoose documents are involved. Because no Mongoose documents are involved, _no middleware (hooks) are executed_. + * + * @param {Object} conditions + * @param {Function} [callback] + * @return {Query} + * @api public + */ + +Model.remove = function remove(conditions, callback) { + if (typeof conditions === 'function') { + callback = conditions; + conditions = {}; + } + + // get the mongodb collection object + var mq = new this.Query(conditions, {}, this, this.collection); + + if (callback) { + callback = this.$wrapCallback(callback); + } + + return mq.remove(callback); +}; + +/** + * Finds documents + * + * The `conditions` are cast to their respective SchemaTypes before the command is sent. + * + * ####Examples: + * + * // named john and at least 18 + * MyModel.find({ name: 'john', age: { $gte: 18 }}); + * + * // executes immediately, passing results to callback + * MyModel.find({ name: 'john', age: { $gte: 18 }}, function (err, docs) {}); + * + * // name LIKE john and only selecting the "name" and "friends" fields, executing immediately + * MyModel.find({ name: /john/i }, 'name friends', function (err, docs) { }) + * + * // passing options + * MyModel.find({ name: /john/i }, null, { skip: 10 }) + * + * // passing options and executing immediately + * MyModel.find({ name: /john/i }, null, { skip: 10 }, function (err, docs) {}); + * + * // executing a query explicitly + * var query = MyModel.find({ name: /john/i }, null, { skip: 10 }) + * query.exec(function (err, docs) {}); + * + * // using the promise returned from executing a query + * var query = MyModel.find({ name: /john/i }, null, { skip: 10 }); + * var promise = query.exec(); + * promise.addBack(function (err, docs) {}); + * + * @param {Object} conditions + * @param {Object} [projection] optional fields to return (http://bit.ly/1HotzBo) + * @param {Object} [options] optional + * @param {Function} [callback] + * @return {Query} + * @see field selection #query_Query-select + * @see promise #promise-js + * @api public + */ + +Model.find = function find(conditions, projection, options, callback) { + if (typeof conditions === 'function') { + callback = conditions; + conditions = {}; + projection = null; + options = null; + } else if (typeof projection === 'function') { + callback = projection; + projection = null; + options = null; + } else if (typeof options === 'function') { + callback = options; + options = null; + } + + var mq = new this.Query({}, {}, this, this.collection); + mq.select(projection); + mq.setOptions(options); + if (this.schema.discriminatorMapping && mq.selectedInclusively()) { + mq.select(this.schema.options.discriminatorKey); + } + + if (callback) { + callback = this.$wrapCallback(callback); + } + + return mq.find(conditions, callback); +}; + +/** + * Finds a single document by its _id field. `findById(id)` is almost* + * equivalent to `findOne({ _id: id })`. If you want to query by a document's + * `_id`, use `findById()` instead of `findOne()`. + * + * The `id` is cast based on the Schema before sending the command. + * + * Note: `findById()` triggers `findOne` hooks. + * + * * Except for how it treats `undefined`. If you use `findOne()`, you'll see + * that `findOne(undefined)` and `findOne({ _id: undefined })` are equivalent + * to `findOne({})` and return arbitrary documents. However, mongoose + * translates `findById(undefined)` into `findOne({ _id: null })`. + * + * ####Example: + * + * // find adventure by id and execute immediately + * Adventure.findById(id, function (err, adventure) {}); + * + * // same as above + * Adventure.findById(id).exec(callback); + * + * // select only the adventures name and length + * Adventure.findById(id, 'name length', function (err, adventure) {}); + * + * // same as above + * Adventure.findById(id, 'name length').exec(callback); + * + * // include all properties except for `length` + * Adventure.findById(id, '-length').exec(function (err, adventure) {}); + * + * // passing options (in this case return the raw js objects, not mongoose documents by passing `lean` + * Adventure.findById(id, 'name', { lean: true }, function (err, doc) {}); + * + * // same as above + * Adventure.findById(id, 'name').lean().exec(function (err, doc) {}); + * + * @param {Object|String|Number} id value of `_id` to query by + * @param {Object} [projection] optional fields to return (http://bit.ly/1HotzBo) + * @param {Object} [options] optional + * @param {Function} [callback] + * @return {Query} + * @see field selection #query_Query-select + * @see lean queries #query_Query-lean + * @api public + */ + +Model.findById = function findById(id, projection, options, callback) { + if (typeof id === 'undefined') { + id = null; + } + + if (callback) { + callback = this.$wrapCallback(callback); + } + + return this.findOne({_id: id}, projection, options, callback); +}; + +/** + * Finds one document. + * + * The `conditions` are cast to their respective SchemaTypes before the command is sent. + * + * *Note:* `conditions` is optional, and if `conditions` is null or undefined, + * mongoose will send an empty `findOne` command to MongoDB, which will return + * an arbitrary document. If you're querying by `_id`, use `findById()` instead. + * + * ####Example: + * + * // find one iphone adventures - iphone adventures?? + * Adventure.findOne({ type: 'iphone' }, function (err, adventure) {}); + * + * // same as above + * Adventure.findOne({ type: 'iphone' }).exec(function (err, adventure) {}); + * + * // select only the adventures name + * Adventure.findOne({ type: 'iphone' }, 'name', function (err, adventure) {}); + * + * // same as above + * Adventure.findOne({ type: 'iphone' }, 'name').exec(function (err, adventure) {}); + * + * // specify options, in this case lean + * Adventure.findOne({ type: 'iphone' }, 'name', { lean: true }, callback); + * + * // same as above + * Adventure.findOne({ type: 'iphone' }, 'name', { lean: true }).exec(callback); + * + * // chaining findOne queries (same as above) + * Adventure.findOne({ type: 'iphone' }).select('name').lean().exec(callback); + * + * @param {Object} [conditions] + * @param {Object} [projection] optional fields to return (http://bit.ly/1HotzBo) + * @param {Object} [options] optional + * @param {Function} [callback] + * @return {Query} + * @see field selection #query_Query-select + * @see lean queries #query_Query-lean + * @api public + */ + +Model.findOne = function findOne(conditions, projection, options, callback) { + if (typeof options === 'function') { + callback = options; + options = null; + } else if (typeof projection === 'function') { + callback = projection; + projection = null; + options = null; + } else if (typeof conditions === 'function') { + callback = conditions; + conditions = {}; + projection = null; + options = null; + } + + // get the mongodb collection object + var mq = new this.Query({}, {}, this, this.collection); + mq.select(projection); + mq.setOptions(options); + if (this.schema.discriminatorMapping && mq.selectedInclusively()) { + mq.select(this.schema.options.discriminatorKey); + } + + if (callback) { + callback = this.$wrapCallback(callback); + } + + return mq.findOne(conditions, callback); +}; + +/** + * Counts number of matching documents in a database collection. + * + * ####Example: + * + * Adventure.count({ type: 'jungle' }, function (err, count) { + * if (err) .. + * console.log('there are %d jungle adventures', count); + * }); + * + * @param {Object} conditions + * @param {Function} [callback] + * @return {Query} + * @api public + */ + +Model.count = function count(conditions, callback) { + if (typeof conditions === 'function') { + callback = conditions; + conditions = {}; + } + + // get the mongodb collection object + var mq = new this.Query({}, {}, this, this.collection); + + if (callback) { + callback = this.$wrapCallback(callback); + } + + return mq.count(conditions, callback); +}; + +/** + * Creates a Query for a `distinct` operation. + * + * Passing a `callback` immediately executes the query. + * + * ####Example + * + * Link.distinct('url', { clicks: {$gt: 100}}, function (err, result) { + * if (err) return handleError(err); + * + * assert(Array.isArray(result)); + * console.log('unique urls with more than 100 clicks', result); + * }) + * + * var query = Link.distinct('url'); + * query.exec(callback); + * + * @param {String} field + * @param {Object} [conditions] optional + * @param {Function} [callback] + * @return {Query} + * @api public + */ + +Model.distinct = function distinct(field, conditions, callback) { + // get the mongodb collection object + var mq = new this.Query({}, {}, this, this.collection); + + if (typeof conditions === 'function') { + callback = conditions; + conditions = {}; + } + if (callback) { + callback = this.$wrapCallback(callback); + } + + return mq.distinct(field, conditions, callback); +}; + +/** + * Creates a Query, applies the passed conditions, and returns the Query. + * + * For example, instead of writing: + * + * User.find({age: {$gte: 21, $lte: 65}}, callback); + * + * we can instead write: + * + * User.where('age').gte(21).lte(65).exec(callback); + * + * Since the Query class also supports `where` you can continue chaining + * + * User + * .where('age').gte(21).lte(65) + * .where('name', /^b/i) + * ... etc + * + * @param {String} path + * @param {Object} [val] optional value + * @return {Query} + * @api public + */ + +Model.where = function where(path, val) { + void val; // eslint + // get the mongodb collection object + var mq = new this.Query({}, {}, this, this.collection).find({}); + return mq.where.apply(mq, arguments); +}; + +/** + * Creates a `Query` and specifies a `$where` condition. + * + * Sometimes you need to query for things in mongodb using a JavaScript expression. You can do so via `find({ $where: javascript })`, or you can use the mongoose shortcut method $where via a Query chain or from your mongoose Model. + * + * Blog.$where('this.username.indexOf("val") !== -1').exec(function (err, docs) {}); + * + * @param {String|Function} argument is a javascript string or anonymous function + * @method $where + * @memberOf Model + * @return {Query} + * @see Query.$where #query_Query-%24where + * @api public + */ + +Model.$where = function $where() { + var mq = new this.Query({}, {}, this, this.collection).find({}); + return mq.$where.apply(mq, arguments); +}; + +/** + * Issues a mongodb findAndModify update command. + * + * Finds a matching document, updates it according to the `update` arg, passing any `options`, and returns the found document (if any) to the callback. The query executes immediately if `callback` is passed else a Query object is returned. + * + * ####Options: + * + * - `new`: bool - if true, return the modified document rather than the original. defaults to false (changed in 4.0) + * - `upsert`: bool - creates the object if it doesn't exist. defaults to false. + * - `fields`: {Object|String} - Field selection. Equivalent to `.select(fields).findOneAndUpdate()` + * - `maxTimeMS`: puts a time limit on the query - requires mongodb >= 2.6.0 + * - `sort`: if multiple docs are found by the conditions, sets the sort order to choose which doc to update + * - `runValidators`: if true, runs [update validators](/docs/validation.html#update-validators) on this command. Update validators validate the update operation against the model's schema. + * - `setDefaultsOnInsert`: if this and `upsert` are true, mongoose will apply the [defaults](http://mongoosejs.com/docs/defaults.html) specified in the model's schema if a new document is created. This option only works on MongoDB >= 2.4 because it relies on [MongoDB's `$setOnInsert` operator](https://docs.mongodb.org/v2.4/reference/operator/update/setOnInsert/). + * - `passRawResult`: if true, passes the [raw result from the MongoDB driver as the third callback parameter](http://mongodb.github.io/node-mongodb-native/2.0/api/Collection.html#findAndModify) + * + * + * ####Examples: + * + * A.findOneAndUpdate(conditions, update, options, callback) // executes + * A.findOneAndUpdate(conditions, update, options) // returns Query + * A.findOneAndUpdate(conditions, update, callback) // executes + * A.findOneAndUpdate(conditions, update) // returns Query + * A.findOneAndUpdate() // returns Query + * + * ####Note: + * + * All top level update keys which are not `atomic` operation names are treated as set operations: + * + * ####Example: + * + * var query = { name: 'borne' }; + * Model.findOneAndUpdate(query, { name: 'jason borne' }, options, callback) + * + * // is sent as + * Model.findOneAndUpdate(query, { $set: { name: 'jason borne' }}, options, callback) + * + * This helps prevent accidentally overwriting your document with `{ name: 'jason borne' }`. + * + * ####Note: + * + * Values are cast to their appropriate types when using the findAndModify helpers. + * However, the below are never executed. + * + * - defaults + * - setters + * + * `findAndModify` helpers support limited defaults and validation. You can + * enable these by setting the `setDefaultsOnInsert` and `runValidators` options, + * respectively. + * + * If you need full-fledged validation, use the traditional approach of first + * retrieving the document. + * + * Model.findById(id, function (err, doc) { + * if (err) .. + * doc.name = 'jason borne'; + * doc.save(callback); + * }); + * + * @param {Object} [conditions] + * @param {Object} [update] + * @param {Object} [options] + * @param {Function} [callback] + * @return {Query} + * @see mongodb http://www.mongodb.org/display/DOCS/findAndModify+Command + * @api public + */ + +Model.findOneAndUpdate = function(conditions, update, options, callback) { + if (typeof options === 'function') { + callback = options; + options = null; + } else if (arguments.length === 1) { + if (typeof conditions === 'function') { + var msg = 'Model.findOneAndUpdate(): First argument must not be a function.\n\n' + + ' ' + this.modelName + '.findOneAndUpdate(conditions, update, options, callback)\n' + + ' ' + this.modelName + '.findOneAndUpdate(conditions, update, options)\n' + + ' ' + this.modelName + '.findOneAndUpdate(conditions, update)\n' + + ' ' + this.modelName + '.findOneAndUpdate(update)\n' + + ' ' + this.modelName + '.findOneAndUpdate()\n'; + throw new TypeError(msg); + } + update = conditions; + conditions = undefined; + } + if (callback) { + callback = this.$wrapCallback(callback); + } + + var fields; + if (options && options.fields) { + fields = options.fields; + } + + update = utils.clone(update, {depopulate: 1, _isNested: true}); + if (this.schema.options.versionKey && options && options.upsert) { + if (!update.$setOnInsert) { + update.$setOnInsert = {}; + } + update.$setOnInsert[this.schema.options.versionKey] = 0; + } + + var mq = new this.Query({}, {}, this, this.collection); + mq.select(fields); + + return mq.findOneAndUpdate(conditions, update, options, callback); +}; + +/** + * Issues a mongodb findAndModify update command by a document's _id field. + * `findByIdAndUpdate(id, ...)` is equivalent to `findOneAndUpdate({ _id: id }, ...)`. + * + * Finds a matching document, updates it according to the `update` arg, + * passing any `options`, and returns the found document (if any) to the + * callback. The query executes immediately if `callback` is passed else a + * Query object is returned. + * + * This function triggers `findOneAndUpdate` middleware. + * + * ####Options: + * + * - `new`: bool - true to return the modified document rather than the original. defaults to false + * - `upsert`: bool - creates the object if it doesn't exist. defaults to false. + * - `runValidators`: if true, runs [update validators](/docs/validation.html#update-validators) on this command. Update validators validate the update operation against the model's schema. + * - `setDefaultsOnInsert`: if this and `upsert` are true, mongoose will apply the [defaults](http://mongoosejs.com/docs/defaults.html) specified in the model's schema if a new document is created. This option only works on MongoDB >= 2.4 because it relies on [MongoDB's `$setOnInsert` operator](https://docs.mongodb.org/v2.4/reference/operator/update/setOnInsert/). + * - `sort`: if multiple docs are found by the conditions, sets the sort order to choose which doc to update + * - `select`: sets the document fields to return + * + * ####Examples: + * + * A.findByIdAndUpdate(id, update, options, callback) // executes + * A.findByIdAndUpdate(id, update, options) // returns Query + * A.findByIdAndUpdate(id, update, callback) // executes + * A.findByIdAndUpdate(id, update) // returns Query + * A.findByIdAndUpdate() // returns Query + * + * ####Note: + * + * All top level update keys which are not `atomic` operation names are treated as set operations: + * + * ####Example: + * + * Model.findByIdAndUpdate(id, { name: 'jason borne' }, options, callback) + * + * // is sent as + * Model.findByIdAndUpdate(id, { $set: { name: 'jason borne' }}, options, callback) + * + * This helps prevent accidentally overwriting your document with `{ name: 'jason borne' }`. + * + * ####Note: + * + * Values are cast to their appropriate types when using the findAndModify helpers. + * However, the below are never executed. + * + * - defaults + * - setters + * + * `findAndModify` helpers support limited defaults and validation. You can + * enable these by setting the `setDefaultsOnInsert` and `runValidators` options, + * respectively. + * + * If you need full-fledged validation, use the traditional approach of first + * retrieving the document. + * + * Model.findById(id, function (err, doc) { + * if (err) .. + * doc.name = 'jason borne'; + * doc.save(callback); + * }); + * + * @param {Object|Number|String} id value of `_id` to query by + * @param {Object} [update] + * @param {Object} [options] + * @param {Function} [callback] + * @return {Query} + * @see Model.findOneAndUpdate #model_Model.findOneAndUpdate + * @see mongodb http://www.mongodb.org/display/DOCS/findAndModify+Command + * @api public + */ + +Model.findByIdAndUpdate = function(id, update, options, callback) { + if (callback) { + callback = this.$wrapCallback(callback); + } + if (arguments.length === 1) { + if (typeof id === 'function') { + var msg = 'Model.findByIdAndUpdate(): First argument must not be a function.\n\n' + + ' ' + this.modelName + '.findByIdAndUpdate(id, callback)\n' + + ' ' + this.modelName + '.findByIdAndUpdate(id)\n' + + ' ' + this.modelName + '.findByIdAndUpdate()\n'; + throw new TypeError(msg); + } + return this.findOneAndUpdate({_id: id}, undefined); + } + + // if a model is passed in instead of an id + if (id instanceof Document) { + id = id._id; + } + + return this.findOneAndUpdate.call(this, {_id: id}, update, options, callback); +}; + +/** + * Issue a mongodb findAndModify remove command. + * + * Finds a matching document, removes it, passing the found document (if any) to the callback. + * + * Executes immediately if `callback` is passed else a Query object is returned. + * + * ####Options: + * + * - `sort`: if multiple docs are found by the conditions, sets the sort order to choose which doc to update + * - `maxTimeMS`: puts a time limit on the query - requires mongodb >= 2.6.0 + * - `select`: sets the document fields to return + * + * ####Examples: + * + * A.findOneAndRemove(conditions, options, callback) // executes + * A.findOneAndRemove(conditions, options) // return Query + * A.findOneAndRemove(conditions, callback) // executes + * A.findOneAndRemove(conditions) // returns Query + * A.findOneAndRemove() // returns Query + * + * Values are cast to their appropriate types when using the findAndModify helpers. + * However, the below are never executed. + * + * - defaults + * - setters + * + * `findAndModify` helpers support limited defaults and validation. You can + * enable these by setting the `setDefaultsOnInsert` and `runValidators` options, + * respectively. + * + * If you need full-fledged validation, use the traditional approach of first + * retrieving the document. + * + * Model.findById(id, function (err, doc) { + * if (err) .. + * doc.name = 'jason borne'; + * doc.save(callback); + * }); + * + * @param {Object} conditions + * @param {Object} [options] + * @param {Function} [callback] + * @return {Query} + * @see mongodb http://www.mongodb.org/display/DOCS/findAndModify+Command + * @api public + */ + +Model.findOneAndRemove = function(conditions, options, callback) { + if (arguments.length === 1 && typeof conditions === 'function') { + var msg = 'Model.findOneAndRemove(): First argument must not be a function.\n\n' + + ' ' + this.modelName + '.findOneAndRemove(conditions, callback)\n' + + ' ' + this.modelName + '.findOneAndRemove(conditions)\n' + + ' ' + this.modelName + '.findOneAndRemove()\n'; + throw new TypeError(msg); + } + + if (typeof options === 'function') { + callback = options; + options = undefined; + } + if (callback) { + callback = this.$wrapCallback(callback); + } + + var fields; + if (options) { + fields = options.select; + options.select = undefined; + } + + var mq = new this.Query({}, {}, this, this.collection); + mq.select(fields); + + return mq.findOneAndRemove(conditions, options, callback); +}; + +/** + * Issue a mongodb findAndModify remove command by a document's _id field. `findByIdAndRemove(id, ...)` is equivalent to `findOneAndRemove({ _id: id }, ...)`. + * + * Finds a matching document, removes it, passing the found document (if any) to the callback. + * + * Executes immediately if `callback` is passed, else a `Query` object is returned. + * + * ####Options: + * + * - `sort`: if multiple docs are found by the conditions, sets the sort order to choose which doc to update + * - `select`: sets the document fields to return + * + * ####Examples: + * + * A.findByIdAndRemove(id, options, callback) // executes + * A.findByIdAndRemove(id, options) // return Query + * A.findByIdAndRemove(id, callback) // executes + * A.findByIdAndRemove(id) // returns Query + * A.findByIdAndRemove() // returns Query + * + * @param {Object|Number|String} id value of `_id` to query by + * @param {Object} [options] + * @param {Function} [callback] + * @return {Query} + * @see Model.findOneAndRemove #model_Model.findOneAndRemove + * @see mongodb http://www.mongodb.org/display/DOCS/findAndModify+Command + */ + +Model.findByIdAndRemove = function(id, options, callback) { + if (arguments.length === 1 && typeof id === 'function') { + var msg = 'Model.findByIdAndRemove(): First argument must not be a function.\n\n' + + ' ' + this.modelName + '.findByIdAndRemove(id, callback)\n' + + ' ' + this.modelName + '.findByIdAndRemove(id)\n' + + ' ' + this.modelName + '.findByIdAndRemove()\n'; + throw new TypeError(msg); + } + if (callback) { + callback = this.$wrapCallback(callback); + } + + return this.findOneAndRemove({_id: id}, options, callback); +}; + +/** + * Shortcut for saving one or more documents to the database. + * `MyModel.create(docs)` does `new MyModel(doc).save()` for every doc in + * docs. + * + * Hooks Triggered: + * - `save()` + * + * ####Example: + * + * // pass individual docs + * Candy.create({ type: 'jelly bean' }, { type: 'snickers' }, function (err, jellybean, snickers) { + * if (err) // ... + * }); + * + * // pass an array + * var array = [{ type: 'jelly bean' }, { type: 'snickers' }]; + * Candy.create(array, function (err, candies) { + * if (err) // ... + * + * var jellybean = candies[0]; + * var snickers = candies[1]; + * // ... + * }); + * + * // callback is optional; use the returned promise if you like: + * var promise = Candy.create({ type: 'jawbreaker' }); + * promise.then(function (jawbreaker) { + * // ... + * }) + * + * @param {Array|Object|*} doc(s) + * @param {Function} [callback] callback + * @return {Promise} + * @api public + */ + +Model.create = function create(doc, callback) { + var args; + var cb; + + if (Array.isArray(doc)) { + args = doc; + cb = callback; + } else { + var last = arguments[arguments.length - 1]; + if (typeof last === 'function') { + cb = last; + args = utils.args(arguments, 0, arguments.length - 1); + } else { + args = utils.args(arguments); + } + } + + var Promise = PromiseProvider.get(); + var _this = this; + if (cb) { + cb = this.$wrapCallback(cb); + } + + var promise = new Promise.ES6(function(resolve, reject) { + if (args.length === 0) { + setImmediate(function() { + cb && cb(null); + resolve(null); + }); + return; + } + + var toExecute = []; + args.forEach(function(doc) { + toExecute.push(function(callback) { + var toSave = doc instanceof _this ? doc : new _this(doc); + var callbackWrapper = function(error, doc) { + if (error) { + return callback(error); + } + callback(null, doc); + }; + + // Hack to avoid getting a promise because of + // $__registerHooksFromSchema + if (toSave.$__original_save) { + toSave.$__original_save({ __noPromise: true }, callbackWrapper); + } else { + toSave.save({ __noPromise: true }, callbackWrapper); + } + }); + }); + + parallel(toExecute, function(error, savedDocs) { + if (error) { + if (cb) { + cb(error); + } else { + reject(error); + } + return; + } + + if (doc instanceof Array) { + resolve(savedDocs); + cb && cb.call(_this, null, savedDocs); + } else { + resolve.apply(promise, savedDocs); + if (cb) { + savedDocs.unshift(null); + cb.apply(_this, savedDocs); + } + } + }); + }); + + return promise; +}; + +/** + * Shortcut for validating an array of documents and inserting them into + * MongoDB if they're all valid. This function is faster than `.create()` + * because it only sends one operation to the server, rather than one for each + * document. + * + * This function does **not** trigger save middleware. + * + * ####Example: + * + * var arr = [{ name: 'Star Wars' }, { name: 'The Empire Strikes Back' }]; + * Movies.insertMany(arr, function(error, docs) {}); + * + * @param {Array|Object|*} doc(s) + * @param {Function} [callback] callback + * @return {Promise} + * @api public + */ + +Model.insertMany = function(arr, callback) { + var _this = this; + if (callback) { + callback = this.$wrapCallback(callback); + } + + var toExecute = []; + arr.forEach(function(doc) { + toExecute.push(function(callback) { + doc = new _this(doc); + doc.validate({ __noPromise: true }, function(error) { + if (error) { + return callback(error); + } + callback(null, doc); + }); + }); + }); + + parallel(toExecute, function(error, docs) { + if (error) { + callback && callback(error); + return; + } + var docObjects = docs.map(function(doc) { + if (doc.schema.options.versionKey) { + doc[doc.schema.options.versionKey] = 0; + } + if (doc.initializeTimestamps) { + return doc.initializeTimestamps().toObject(POJO_TO_OBJECT_OPTIONS); + } + return doc.toObject(POJO_TO_OBJECT_OPTIONS); + }); + _this.collection.insertMany(docObjects, function(error) { + if (error) { + callback && callback(error); + return; + } + for (var i = 0; i < docs.length; ++i) { + docs[i].isNew = false; + docs[i].emit('isNew', false); + } + callback && callback(null, docs); + }); + }); +}; + +/** + * Shortcut for creating a new Document from existing raw data, pre-saved in the DB. + * The document returned has no paths marked as modified initially. + * + * ####Example: + * + * // hydrate previous data into a Mongoose document + * var mongooseCandy = Candy.hydrate({ _id: '54108337212ffb6d459f854c', type: 'jelly bean' }); + * + * @param {Object} obj + * @return {Document} + * @api public + */ + +Model.hydrate = function(obj) { + var model = require('./queryhelpers').createModel(this, obj); + model.init(obj); + return model; +}; + +/** + * Updates documents in the database without returning them. + * + * ####Examples: + * + * MyModel.update({ age: { $gt: 18 } }, { oldEnough: true }, fn); + * MyModel.update({ name: 'Tobi' }, { ferret: true }, { multi: true }, function (err, raw) { + * if (err) return handleError(err); + * console.log('The raw response from Mongo was ', raw); + * }); + * + * ####Valid options: + * + * - `safe` (boolean) safe mode (defaults to value set in schema (true)) + * - `upsert` (boolean) whether to create the doc if it doesn't match (false) + * - `multi` (boolean) whether multiple documents should be updated (false) + * - `runValidators`: if true, runs [update validators](/docs/validation.html#update-validators) on this command. Update validators validate the update operation against the model's schema. + * - `setDefaultsOnInsert`: if this and `upsert` are true, mongoose will apply the [defaults](http://mongoosejs.com/docs/defaults.html) specified in the model's schema if a new document is created. This option only works on MongoDB >= 2.4 because it relies on [MongoDB's `$setOnInsert` operator](https://docs.mongodb.org/v2.4/reference/operator/update/setOnInsert/). + * - `strict` (boolean) overrides the `strict` option for this update + * - `overwrite` (boolean) disables update-only mode, allowing you to overwrite the doc (false) + * + * All `update` values are cast to their appropriate SchemaTypes before being sent. + * + * The `callback` function receives `(err, rawResponse)`. + * + * - `err` is the error if any occurred + * - `rawResponse` is the full response from Mongo + * + * ####Note: + * + * All top level keys which are not `atomic` operation names are treated as set operations: + * + * ####Example: + * + * var query = { name: 'borne' }; + * Model.update(query, { name: 'jason borne' }, options, callback) + * + * // is sent as + * Model.update(query, { $set: { name: 'jason borne' }}, options, callback) + * // if overwrite option is false. If overwrite is true, sent without the $set wrapper. + * + * This helps prevent accidentally overwriting all documents in your collection with `{ name: 'jason borne' }`. + * + * ####Note: + * + * Be careful to not use an existing model instance for the update clause (this won't work and can cause weird behavior like infinite loops). Also, ensure that the update clause does not have an _id property, which causes Mongo to return a "Mod on _id not allowed" error. + * + * ####Note: + * + * To update documents without waiting for a response from MongoDB, do not pass a `callback`, then call `exec` on the returned [Query](#query-js): + * + * Comment.update({ _id: id }, { $set: { text: 'changed' }}).exec(); + * + * ####Note: + * + * Although values are casted to their appropriate types when using update, the following are *not* applied: + * + * - defaults + * - setters + * - validators + * - middleware + * + * If you need those features, use the traditional approach of first retrieving the document. + * + * Model.findOne({ name: 'borne' }, function (err, doc) { + * if (err) .. + * doc.name = 'jason borne'; + * doc.save(callback); + * }) + * + * @see strict http://mongoosejs.com/docs/guide.html#strict + * @see response http://docs.mongodb.org/v2.6/reference/command/update/#output + * @param {Object} conditions + * @param {Object} doc + * @param {Object} [options] + * @param {Function} [callback] + * @return {Query} + * @api public + */ + +Model.update = function update(conditions, doc, options, callback) { + var mq = new this.Query({}, {}, this, this.collection); + if (callback) { + callback = this.$wrapCallback(callback); + } + // gh-2406 + // make local deep copy of conditions + if (conditions instanceof Document) { + conditions = conditions.toObject(); + } else { + conditions = utils.clone(conditions, {retainKeyOrder: true}); + } + options = typeof options === 'function' ? options : utils.clone(options); + + if (this.schema.options.versionKey && options && options.upsert) { + if (!doc.$setOnInsert) { + doc.$setOnInsert = {}; + } + doc.$setOnInsert[this.schema.options.versionKey] = 0; + } + + return mq.update(conditions, doc, options, callback); +}; + +/** + * Executes a mapReduce command. + * + * `o` is an object specifying all mapReduce options as well as the map and reduce functions. All options are delegated to the driver implementation. See [node-mongodb-native mapReduce() documentation](http://mongodb.github.io/node-mongodb-native/api-generated/collection.html#mapreduce) for more detail about options. + * + * ####Example: + * + * var o = {}; + * o.map = function () { emit(this.name, 1) } + * o.reduce = function (k, vals) { return vals.length } + * User.mapReduce(o, function (err, results) { + * console.log(results) + * }) + * + * ####Other options: + * + * - `query` {Object} query filter object. + * - `sort` {Object} sort input objects using this key + * - `limit` {Number} max number of documents + * - `keeptemp` {Boolean, default:false} keep temporary data + * - `finalize` {Function} finalize function + * - `scope` {Object} scope variables exposed to map/reduce/finalize during execution + * - `jsMode` {Boolean, default:false} it is possible to make the execution stay in JS. Provided in MongoDB > 2.0.X + * - `verbose` {Boolean, default:false} provide statistics on job execution time. + * - `readPreference` {String} + * - `out*` {Object, default: {inline:1}} sets the output target for the map reduce job. + * + * ####* out options: + * + * - `{inline:1}` the results are returned in an array + * - `{replace: 'collectionName'}` add the results to collectionName: the results replace the collection + * - `{reduce: 'collectionName'}` add the results to collectionName: if dups are detected, uses the reducer / finalize functions + * - `{merge: 'collectionName'}` add the results to collectionName: if dups exist the new docs overwrite the old + * + * If `options.out` is set to `replace`, `merge`, or `reduce`, a Model instance is returned that can be used for further querying. Queries run against this model are all executed with the `lean` option; meaning only the js object is returned and no Mongoose magic is applied (getters, setters, etc). + * + * ####Example: + * + * var o = {}; + * o.map = function () { emit(this.name, 1) } + * o.reduce = function (k, vals) { return vals.length } + * o.out = { replace: 'createdCollectionNameForResults' } + * o.verbose = true; + * + * User.mapReduce(o, function (err, model, stats) { + * console.log('map reduce took %d ms', stats.processtime) + * model.find().where('value').gt(10).exec(function (err, docs) { + * console.log(docs); + * }); + * }) + * + * // a promise is returned so you may instead write + * var promise = User.mapReduce(o); + * promise.then(function (model, stats) { + * console.log('map reduce took %d ms', stats.processtime) + * return model.find().where('value').gt(10).exec(); + * }).then(function (docs) { + * console.log(docs); + * }).then(null, handleError).end() + * + * @param {Object} o an object specifying map-reduce options + * @param {Function} [callback] optional callback + * @see http://www.mongodb.org/display/DOCS/MapReduce + * @return {Promise} + * @api public + */ + +Model.mapReduce = function mapReduce(o, callback) { + var _this = this; + if (callback) { + callback = this.$wrapCallback(callback); + } + var Promise = PromiseProvider.get(); + return new Promise.ES6(function(resolve, reject) { + if (!Model.mapReduce.schema) { + var opts = {noId: true, noVirtualId: true, strict: false}; + Model.mapReduce.schema = new Schema({}, opts); + } + + if (!o.out) o.out = {inline: 1}; + if (o.verbose !== false) o.verbose = true; + + o.map = String(o.map); + o.reduce = String(o.reduce); + + if (o.query) { + var q = new _this.Query(o.query); + q.cast(_this); + o.query = q._conditions; + q = undefined; + } + + _this.collection.mapReduce(null, null, o, function(err, ret, stats) { + if (err) { + callback && callback(err); + reject(err); + return; + } + + if (ret.findOne && ret.mapReduce) { + // returned a collection, convert to Model + var model = Model.compile( + '_mapreduce_' + ret.collectionName + , Model.mapReduce.schema + , ret.collectionName + , _this.db + , _this.base); + + model._mapreduce = true; + + callback && callback(null, model, stats); + return resolve(model, stats); + } + + callback && callback(null, ret, stats); + resolve(ret, stats); + }); + }); +}; + +/** + * geoNear support for Mongoose + * + * ####Options: + * - `lean` {Boolean} return the raw object + * - All options supported by the driver are also supported + * + * ####Example: + * + * // Legacy point + * Model.geoNear([1,3], { maxDistance : 5, spherical : true }, function(err, results, stats) { + * console.log(results); + * }); + * + * // geoJson + * var point = { type : "Point", coordinates : [9,9] }; + * Model.geoNear(point, { maxDistance : 5, spherical : true }, function(err, results, stats) { + * console.log(results); + * }); + * + * @param {Object|Array} GeoJSON point or legacy coordinate pair [x,y] to search near + * @param {Object} options for the qurery + * @param {Function} [callback] optional callback for the query + * @return {Promise} + * @see http://docs.mongodb.org/manual/core/2dsphere/ + * @see http://mongodb.github.io/node-mongodb-native/api-generated/collection.html?highlight=geonear#geoNear + * @api public + */ + +Model.geoNear = function(near, options, callback) { + if (typeof options === 'function') { + callback = options; + options = {}; + } + + if (callback) { + callback = this.$wrapCallback(callback); + } + + var _this = this; + var Promise = PromiseProvider.get(); + if (!near) { + return new Promise.ES6(function(resolve, reject) { + var error = new Error('Must pass a near option to geoNear'); + reject(error); + callback && callback(error); + }); + } + + var x, y; + + return new Promise.ES6(function(resolve, reject) { + var handler = function(err, res) { + if (err) { + reject(err); + callback && callback(err); + return; + } + if (options.lean) { + resolve(res.results, res.stats); + callback && callback(null, res.results, res.stats); + return; + } + + var count = res.results.length; + // if there are no results, fulfill the promise now + if (count === 0) { + resolve(res.results, res.stats); + callback && callback(null, res.results, res.stats); + return; + } + + var errSeen = false; + + function init(err) { + if (err && !errSeen) { + errSeen = true; + reject(err); + callback && callback(err); + return; + } + if (--count <= 0) { + resolve(res.results, res.stats); + callback && callback(null, res.results, res.stats); + } + } + + for (var i = 0; i < res.results.length; i++) { + var temp = res.results[i].obj; + res.results[i].obj = new _this(); + res.results[i].obj.init(temp, init); + } + }; + + if (Array.isArray(near)) { + if (near.length !== 2) { + var error = new Error('If using legacy coordinates, must be an array ' + + 'of size 2 for geoNear'); + reject(error); + callback && callback(error); + return; + } + x = near[0]; + y = near[1]; + _this.collection.geoNear(x, y, options, handler); + } else { + if (near.type !== 'Point' || !Array.isArray(near.coordinates)) { + error = new Error('Must pass either a legacy coordinate array or ' + + 'GeoJSON Point to geoNear'); + reject(error); + callback && callback(error); + return; + } + + _this.collection.geoNear(near, options, handler); + } + }); +}; + +/** + * Performs [aggregations](http://docs.mongodb.org/manual/applications/aggregation/) on the models collection. + * + * If a `callback` is passed, the `aggregate` is executed and a `Promise` is returned. If a callback is not passed, the `aggregate` itself is returned. + * + * ####Example: + * + * // Find the max balance of all accounts + * Users.aggregate( + * { $group: { _id: null, maxBalance: { $max: '$balance' }}} + * , { $project: { _id: 0, maxBalance: 1 }} + * , function (err, res) { + * if (err) return handleError(err); + * console.log(res); // [ { maxBalance: 98000 } ] + * }); + * + * // Or use the aggregation pipeline builder. + * Users.aggregate() + * .group({ _id: null, maxBalance: { $max: '$balance' } }) + * .select('-id maxBalance') + * .exec(function (err, res) { + * if (err) return handleError(err); + * console.log(res); // [ { maxBalance: 98 } ] + * }); + * + * ####NOTE: + * + * - Arguments are not cast to the model's schema because `$project` operators allow redefining the "shape" of the documents at any stage of the pipeline, which may leave documents in an incompatible format. + * - The documents returned are plain javascript objects, not mongoose documents (since any shape of document can be returned). + * - Requires MongoDB >= 2.1 + * + * @see Aggregate #aggregate_Aggregate + * @see MongoDB http://docs.mongodb.org/manual/applications/aggregation/ + * @param {Object|Array} [...] aggregation pipeline operator(s) or operator array + * @param {Function} [callback] + * @return {Aggregate|Promise} + * @api public + */ + +Model.aggregate = function aggregate() { + var args = [].slice.call(arguments), + aggregate, + callback; + + if (typeof args[args.length - 1] === 'function') { + callback = args.pop(); + } + + if (args.length === 1 && util.isArray(args[0])) { + aggregate = new Aggregate(args[0]); + } else { + aggregate = new Aggregate(args); + } + + aggregate.model(this); + + if (typeof callback === 'undefined') { + return aggregate; + } + + if (callback) { + callback = this.$wrapCallback(callback); + } + + aggregate.exec(callback); +}; + +/** + * Implements `$geoSearch` functionality for Mongoose + * + * ####Example: + * + * var options = { near: [10, 10], maxDistance: 5 }; + * Locations.geoSearch({ type : "house" }, options, function(err, res) { + * console.log(res); + * }); + * + * ####Options: + * - `near` {Array} x,y point to search for + * - `maxDistance` {Number} the maximum distance from the point near that a result can be + * - `limit` {Number} The maximum number of results to return + * - `lean` {Boolean} return the raw object instead of the Mongoose Model + * + * @param {Object} conditions an object that specifies the match condition (required) + * @param {Object} options for the geoSearch, some (near, maxDistance) are required + * @param {Function} [callback] optional callback + * @return {Promise} + * @see http://docs.mongodb.org/manual/reference/command/geoSearch/ + * @see http://docs.mongodb.org/manual/core/geohaystack/ + * @api public + */ + +Model.geoSearch = function(conditions, options, callback) { + if (typeof options === 'function') { + callback = options; + options = {}; + } + if (callback) { + callback = this.$wrapCallback(callback); + } + + var _this = this; + var Promise = PromiseProvider.get(); + return new Promise.ES6(function(resolve, reject) { + var error; + if (conditions === undefined || !utils.isObject(conditions)) { + error = new Error('Must pass conditions to geoSearch'); + } else if (!options.near) { + error = new Error('Must specify the near option in geoSearch'); + } else if (!Array.isArray(options.near)) { + error = new Error('near option must be an array [x, y]'); + } + + if (error) { + callback && callback(error); + reject(error); + return; + } + + // send the conditions in the options object + options.search = conditions; + + _this.collection.geoHaystackSearch(options.near[0], options.near[1], options, function(err, res) { + // have to deal with driver problem. Should be fixed in a soon-ish release + // (7/8/2013) + if (err) { + callback && callback(err); + reject(err); + return; + } + + var count = res.results.length; + if (options.lean || count === 0) { + callback && callback(null, res.results, res.stats); + resolve(res.results, res.stats); + return; + } + + var errSeen = false; + + function init(err) { + if (err && !errSeen) { + callback && callback(err); + reject(err); + return; + } + + if (!--count && !errSeen) { + callback && callback(null, res.results, res.stats); + resolve(res.results, res.stats); + } + } + + for (var i = 0; i < res.results.length; i++) { + var temp = res.results[i]; + res.results[i] = new _this(); + res.results[i].init(temp, {}, init); + } + }); + }); +}; + +/** + * Populates document references. + * + * ####Available options: + * + * - path: space delimited path(s) to populate + * - select: optional fields to select + * - match: optional query conditions to match + * - model: optional name of the model to use for population + * - options: optional query options like sort, limit, etc + * + * ####Examples: + * + * // populates a single object + * User.findById(id, function (err, user) { + * var opts = [ + * { path: 'company', match: { x: 1 }, select: 'name' } + * , { path: 'notes', options: { limit: 10 }, model: 'override' } + * ] + * + * User.populate(user, opts, function (err, user) { + * console.log(user); + * }) + * }) + * + * // populates an array of objects + * User.find(match, function (err, users) { + * var opts = [{ path: 'company', match: { x: 1 }, select: 'name' }] + * + * var promise = User.populate(users, opts); + * promise.then(console.log).end(); + * }) + * + * // imagine a Weapon model exists with two saved documents: + * // { _id: 389, name: 'whip' } + * // { _id: 8921, name: 'boomerang' } + * + * var user = { name: 'Indiana Jones', weapon: 389 } + * Weapon.populate(user, { path: 'weapon', model: 'Weapon' }, function (err, user) { + * console.log(user.weapon.name) // whip + * }) + * + * // populate many plain objects + * var users = [{ name: 'Indiana Jones', weapon: 389 }] + * users.push({ name: 'Batman', weapon: 8921 }) + * Weapon.populate(users, { path: 'weapon' }, function (err, users) { + * users.forEach(function (user) { + * console.log('%s uses a %s', users.name, user.weapon.name) + * // Indiana Jones uses a whip + * // Batman uses a boomerang + * }) + * }) + * // Note that we didn't need to specify the Weapon model because + * // we were already using it's populate() method. + * + * @param {Document|Array} docs Either a single document or array of documents to populate. + * @param {Object} options A hash of key/val (path, options) used for population. + * @param {Function} [callback(err,doc)] Optional callback, executed upon completion. Receives `err` and the `doc(s)`. + * @return {Promise} + * @api public + */ + +Model.populate = function(docs, paths, callback) { + var _this = this; + if (callback) { + callback = this.$wrapCallback(callback); + } + + // normalized paths + var noPromise = paths && !!paths.__noPromise; + paths = utils.populate(paths); + + // data that should persist across subPopulate calls + var cache = {}; + + if (noPromise) { + _populate(this, docs, paths, cache, callback); + } else { + var Promise = PromiseProvider.get(); + return new Promise.ES6(function(resolve, reject) { + _populate(_this, docs, paths, cache, function(error, docs) { + if (error) { + callback && callback(error); + reject(error); + } else { + callback && callback(null, docs); + resolve(docs); + } + }); + }); + } +}; + +/*! + * Populate helper + * + * @param {Model} model the model to use + * @param {Document|Array} docs Either a single document or array of documents to populate. + * @param {Object} paths + * @param {Function} [cb(err,doc)] Optional callback, executed upon completion. Receives `err` and the `doc(s)`. + * @return {Function} + * @api private + */ + +function _populate(model, docs, paths, cache, callback) { + var pending = paths.length; + + if (pending === 0) { + return callback(null, docs); + } + + // each path has its own query options and must be executed separately + var i = pending; + var path; + while (i--) { + path = paths[i]; + populate(model, docs, path, next); + } + + function next(err) { + if (err) { + return callback(err); + } + if (--pending) { + return; + } + callback(null, docs); + } +} + +/*! + * Populates `docs` + */ +var excludeIdReg = /\s?-_id\s?/, + excludeIdRegGlobal = /\s?-_id\s?/g; + +function populate(model, docs, options, callback) { + var modelsMap; + + // normalize single / multiple docs passed + if (!Array.isArray(docs)) { + docs = [docs]; + } + + if (docs.length === 0 || docs.every(utils.isNullOrUndefined)) { + return callback(); + } + + modelsMap = getModelsMapForPopulate(model, docs, options); + + var i, len = modelsMap.length, + mod, match, select, vals = []; + + function flatten(item) { + // no need to include undefined values in our query + return undefined !== item; + } + + var _remaining = len; + var hasOne = false; + for (i = 0; i < len; i++) { + mod = modelsMap[i]; + select = mod.options.select; + + if (mod.options.match) { + match = utils.object.shallowCopy(mod.options.match); + } else { + match = {}; + } + + var ids = utils.array.flatten(mod.ids, flatten); + ids = utils.array.unique(ids); + + if (ids.length === 0 || ids.every(utils.isNullOrUndefined)) { + --_remaining; + continue; + } + + hasOne = true; + if (mod.foreignField !== '_id' || !match['_id']) { + match[mod.foreignField] = { $in: ids }; + } + + var assignmentOpts = {}; + assignmentOpts.sort = mod.options.options && mod.options.options.sort || undefined; + assignmentOpts.excludeId = excludeIdReg.test(select) || (select && select._id === 0); + + if (assignmentOpts.excludeId) { + // override the exclusion from the query so we can use the _id + // for document matching during assignment. we'll delete the + // _id back off before returning the result. + if (typeof select === 'string') { + select = select.replace(excludeIdRegGlobal, ' '); + } else { + // preserve original select conditions by copying + select = utils.object.shallowCopy(select); + delete select._id; + } + } + + if (mod.options.options && mod.options.options.limit) { + assignmentOpts.originalLimit = mod.options.options.limit; + mod.options.options.limit = mod.options.options.limit * ids.length; + } + + var subPopulate = mod.options.populate; + var query = mod.Model.find(match, select, mod.options.options); + if (subPopulate) { + query.populate(subPopulate); + } + query.exec(next.bind(this, mod, assignmentOpts)); + } + + if (!hasOne) { + return callback(); + } + + function next(options, assignmentOpts, err, valsFromDb) { + if (err) return callback(err); + vals = vals.concat(valsFromDb); + _assign(null, vals, options, assignmentOpts); + if (--_remaining === 0) { + callback(); + } + } + + function _assign(err, vals, mod, assignmentOpts) { + if (err) return callback(err); + + var options = mod.options; + var _val; + var lean = options.options && options.options.lean, + len = vals.length, + rawOrder = {}, rawDocs = {}, key, val; + + // optimization: + // record the document positions as returned by + // the query result. + for (var i = 0; i < len; i++) { + val = vals[i]; + if (val) { + _val = utils.getValue(mod.foreignField, val); + if (Array.isArray(_val)) { + var _valLength = _val.length; + for (var j = 0; j < _valLength; ++j) { + if (_val[j] instanceof Document) { + _val[j] = _val[j]._id; + } + key = String(_val[j]); + if (rawDocs[key]) { + if (Array.isArray(rawDocs[key])) { + rawDocs[key].push(val); + rawOrder[key].push(i); + } else { + rawDocs[key] = [rawDocs[key], val]; + rawOrder[key] = [rawOrder[key], i]; + } + } else { + rawDocs[key] = val; + rawOrder[key] = i; + } + } + } else { + if (_val instanceof Document) { + _val = _val._id; + } + key = String(_val); + if (rawDocs[key]) { + if (Array.isArray(rawDocs[key])) { + rawDocs[key].push(val); + rawOrder[key].push(i); + } else { + rawDocs[key] = [rawDocs[key], val]; + rawOrder[key] = [rawOrder[key], i]; + } + } else { + rawDocs[key] = val; + rawOrder[key] = i; + } + } + // flag each as result of population + if (!lean) { + val.$__.wasPopulated = true; + } + } + } + + assignVals({ + originalModel: model, + rawIds: mod.ids, + localField: mod.localField, + foreignField: mod.foreignField, + rawDocs: rawDocs, + rawOrder: rawOrder, + docs: mod.docs, + path: options.path, + options: assignmentOpts, + justOne: mod.justOne, + isVirtual: mod.isVirtual + }); + } +} + +/*! + * Assigns documents returned from a population query back + * to the original document path. + */ + +function assignVals(o) { + // replace the original ids in our intermediate _ids structure + // with the documents found by query + assignRawDocsToIdStructure(o.rawIds, o.rawDocs, o.rawOrder, o.options, + o.localField, o.foreignField); + + // now update the original documents being populated using the + // result structure that contains real documents. + + var docs = o.docs; + var rawIds = o.rawIds; + var options = o.options; + + function setValue(val) { + return valueFilter(val, options); + } + + for (var i = 0; i < docs.length; ++i) { + if (utils.getValue(o.path, docs[i]) == null && + !o.originalModel.schema._getVirtual(o.path)) { + continue; + } + + if (o.isVirtual && !o.justOne && !Array.isArray(rawIds[i])) { + rawIds[i] = [rawIds[i]]; + } + utils.setValue(o.path, rawIds[i], docs[i], setValue); + } +} + +/*! + * Assign `vals` returned by mongo query to the `rawIds` + * structure returned from utils.getVals() honoring + * query sort order if specified by user. + * + * This can be optimized. + * + * Rules: + * + * if the value of the path is not an array, use findOne rules, else find. + * for findOne the results are assigned directly to doc path (including null results). + * for find, if user specified sort order, results are assigned directly + * else documents are put back in original order of array if found in results + * + * @param {Array} rawIds + * @param {Array} vals + * @param {Boolean} sort + * @api private + */ + +function assignRawDocsToIdStructure(rawIds, resultDocs, resultOrder, options, localFields, foreignFields, recursed) { + // honor user specified sort order + var newOrder = []; + var sorting = options.sort && rawIds.length > 1; + var doc; + var sid; + var id; + + for (var i = 0; i < rawIds.length; ++i) { + id = rawIds[i]; + + if (Array.isArray(id)) { + // handle [ [id0, id2], [id3] ] + assignRawDocsToIdStructure(id, resultDocs, resultOrder, options, localFields, foreignFields, true); + newOrder.push(id); + continue; + } + + if (id === null && !sorting) { + // keep nulls for findOne unless sorting, which always + // removes them (backward compat) + newOrder.push(id); + continue; + } + + sid = String(id); + + if (recursed) { + // apply find behavior + + // assign matching documents in original order unless sorting + doc = resultDocs[sid]; + if (doc) { + if (sorting) { + newOrder[resultOrder[sid]] = doc; + } else { + newOrder.push(doc); + } + } else { + newOrder.push(id); + } + } else { + // apply findOne behavior - if document in results, assign, else assign null + newOrder[i] = doc = resultDocs[sid] || null; + } + } + + rawIds.length = 0; + if (newOrder.length) { + // reassign the documents based on corrected order + + // forEach skips over sparse entries in arrays so we + // can safely use this to our advantage dealing with sorted + // result sets too. + newOrder.forEach(function(doc, i) { + if (!doc) { + return; + } + rawIds[i] = doc; + }); + } +} + +function getModelsMapForPopulate(model, docs, options) { + var i, doc, len = docs.length, + available = {}, + map = [], + modelNameFromQuery = options.model && options.model.modelName || options.model, + schema, refPath, Model, currentOptions, modelNames, modelName, discriminatorKey, modelForFindSchema; + + var originalOptions = utils.clone(options); + var isVirtual = false; + + schema = model._getSchema(options.path); + + if (schema && schema.caster) { + schema = schema.caster; + } + + if (!schema && model.discriminators) { + discriminatorKey = model.schema.discriminatorMapping.key; + } + + refPath = schema && schema.options && schema.options.refPath; + + for (i = 0; i < len; i++) { + doc = docs[i]; + + if (refPath) { + modelNames = utils.getValue(refPath, doc); + if (Array.isArray(modelNames)) { + modelNames = modelNames.filter(function(v) { + return v != null; + }); + } + } else { + if (!modelNameFromQuery) { + var modelForCurrentDoc = model; + var schemaForCurrentDoc; + + if (!schema && discriminatorKey) { + modelForFindSchema = utils.getValue(discriminatorKey, doc); + + if (modelForFindSchema) { + modelForCurrentDoc = model.db.model(modelForFindSchema); + schemaForCurrentDoc = modelForCurrentDoc._getSchema(options.path); + + if (schemaForCurrentDoc && schemaForCurrentDoc.caster) { + schemaForCurrentDoc = schemaForCurrentDoc.caster; + } + } + } else { + schemaForCurrentDoc = schema; + } + var virtual = modelForCurrentDoc.schema._getVirtual(options.path); + + if (schemaForCurrentDoc && schemaForCurrentDoc.options && schemaForCurrentDoc.options.ref) { + modelNames = [schemaForCurrentDoc.options.ref]; + } else if (virtual && virtual.options && virtual.options.ref) { + modelNames = [virtual && virtual.options && virtual.options.ref]; + isVirtual = true; + } else { + modelNames = [model.modelName]; + } + } else { + modelNames = [modelNameFromQuery]; // query options + } + } + + if (!modelNames) { + continue; + } + + if (!Array.isArray(modelNames)) { + modelNames = [modelNames]; + } + + virtual = model.schema._getVirtual(options.path); + var localField = virtual && virtual.options ? + (virtual.$nestedSchemaPath ? virtual.$nestedSchemaPath + '.' : '') + virtual.options.localField : + options.path; + var foreignField = virtual && virtual.options ? + virtual.options.foreignField : + '_id'; + var justOne = virtual && virtual.options && virtual.options.justOne; + if (virtual && virtual.options && virtual.options.ref) { + isVirtual = true; + } + + if (virtual && (!localField || !foreignField)) { + throw new Error('If you are populating a virtual, you must set the ' + + 'localField and foreignField options'); + } + + options.isVirtual = isVirtual; + var ret = convertTo_id(utils.getValue(localField, doc)); + var id = String(utils.getValue(foreignField, doc)); + options._docs[id] = Array.isArray(ret) ? ret.slice() : ret; + if (doc.$__) { + doc.populated(options.path, options._docs[id], options); + } + + var k = modelNames.length; + while (k--) { + modelName = modelNames[k]; + Model = originalOptions.model && originalOptions.model.modelName ? + originalOptions.model : + model.db.model(modelName); + + if (!available[modelName]) { + currentOptions = { + model: Model + }; + + if (isVirtual && virtual.options && virtual.options.options) { + currentOptions.options = utils.clone(virtual.options.options, { + retainKeyOrder: true + }); + } + utils.merge(currentOptions, options); + if (schema && !discriminatorKey) { + currentOptions.model = Model; + } + options.model = Model; + + available[modelName] = { + Model: Model, + options: currentOptions, + docs: [doc], + ids: [ret], + // Assume only 1 localField + foreignField + localField: localField, + foreignField: foreignField, + justOne: justOne, + isVirtual: isVirtual + }; + map.push(available[modelName]); + } else { + available[modelName].docs.push(doc); + available[modelName].ids.push(ret); + } + } + } + + return map; +} + +/*! + * Retrieve the _id of `val` if a Document or Array of Documents. + * + * @param {Array|Document|Any} val + * @return {Array|Document|Any} + */ + +function convertTo_id(val) { + if (val instanceof Model) return val._id; + + if (Array.isArray(val)) { + for (var i = 0; i < val.length; ++i) { + if (val[i] instanceof Model) { + val[i] = val[i]._id; + } + } + if (val.isMongooseArray) { + return new MongooseArray(val, val._path, val._parent); + } + + return [].concat(val); + } + + return val; +} + +/*! + * 1) Apply backwards compatible find/findOne behavior to sub documents + * + * find logic: + * a) filter out non-documents + * b) remove _id from sub docs when user specified + * + * findOne + * a) if no doc found, set to null + * b) remove _id from sub docs when user specified + * + * 2) Remove _ids when specified by users query. + * + * background: + * _ids are left in the query even when user excludes them so + * that population mapping can occur. + */ + +function valueFilter(val, assignmentOpts) { + if (Array.isArray(val)) { + // find logic + var ret = []; + var numValues = val.length; + for (var i = 0; i < numValues; ++i) { + var subdoc = val[i]; + if (!isDoc(subdoc)) continue; + maybeRemoveId(subdoc, assignmentOpts); + ret.push(subdoc); + if (assignmentOpts.originalLimit && + ret.length >= assignmentOpts.originalLimit) { + break; + } + } + + // Since we don't want to have to create a new mongoosearray, make sure to + // modify the array in place + while (val.length > ret.length) { + Array.prototype.pop.apply(val, []); + } + for (i = 0; i < ret.length; ++i) { + val[i] = ret[i]; + } + return val; + } + + // findOne + if (isDoc(val)) { + maybeRemoveId(val, assignmentOpts); + return val; + } + + return null; +} + +/*! + * Remove _id from `subdoc` if user specified "lean" query option + */ + +function maybeRemoveId(subdoc, assignmentOpts) { + if (assignmentOpts.excludeId) { + if (typeof subdoc.setValue === 'function') { + delete subdoc._doc._id; + } else { + delete subdoc._id; + } + } +} + +/*! + * Determine if `doc` is a document returned + * by a populate query. + */ + +function isDoc(doc) { + if (doc == null) { + return false; + } + + var type = typeof doc; + if (type === 'string') { + return false; + } + + if (type === 'number') { + return false; + } + + if (Buffer.isBuffer(doc)) { + return false; + } + + if (doc.constructor.name === 'ObjectID') { + return false; + } + + // only docs + return true; +} + +/** + * Finds the schema for `path`. This is different than + * calling `schema.path` as it also resolves paths with + * positional selectors (something.$.another.$.path). + * + * @param {String} path + * @return {Schema} + * @api private + */ + +Model._getSchema = function _getSchema(path) { + return this.schema._getSchema(path); +}; + +/*! + * Compiler utility. + * + * @param {String|Function} name model name or class extending Model + * @param {Schema} schema + * @param {String} collectionName + * @param {Connection} connection + * @param {Mongoose} base mongoose instance + */ + +Model.compile = function compile(name, schema, collectionName, connection, base) { + var versioningEnabled = schema.options.versionKey !== false; + + if (versioningEnabled && !schema.paths[schema.options.versionKey]) { + // add versioning to top level documents only + var o = {}; + o[schema.options.versionKey] = Number; + schema.add(o); + } + + var model; + if (typeof name === 'function' && name.prototype instanceof Model) { + model = name; + name = model.name; + schema.loadClass(model, true); + } else { + // generate new class + model = function model(doc, fields, skipId) { + if (!(this instanceof model)) { + return new model(doc, fields, skipId); + } + Model.call(this, doc, fields, skipId); + }; + } + + model.hooks = schema.s.hooks.clone(); + model.base = base; + model.modelName = name; + if (!(model.prototype instanceof Model)) { + model.__proto__ = Model; + model.prototype.__proto__ = Model.prototype; + } + model.model = Model.prototype.model; + model.db = model.prototype.db = connection; + model.discriminators = model.prototype.discriminators = undefined; + + model.prototype.$__setSchema(schema); + + var collectionOptions = { + bufferCommands: schema.options.bufferCommands, + capped: schema.options.capped + }; + + model.prototype.collection = connection.collection( + collectionName + , collectionOptions + ); + + // apply methods and statics + applyMethods(model, schema); + applyStatics(model, schema); + + model.schema = model.prototype.schema; + model.collection = model.prototype.collection; + + // Create custom query constructor + model.Query = function() { + Query.apply(this, arguments); + this.options.retainKeyOrder = model.schema.options.retainKeyOrder; + }; + model.Query.prototype = Object.create(Query.prototype); + model.Query.base = Query.base; + applyQueryMethods(model, schema.query); + + var kareemOptions = { useErrorHandlers: true }; + model.$__insertMany = model.hooks.createWrapper('insertMany', + model.insertMany, model, kareemOptions); + model.insertMany = function(arr, callback) { + var Promise = PromiseProvider.get(); + return new Promise.ES6(function(resolve, reject) { + model.$__insertMany(arr, function(error, result) { + if (error) { + callback && callback(error); + return reject(error); + } + callback && callback(null, result); + resolve(result); + }); + }); + }; + + return model; +}; + +/*! + * Register methods for this model + * + * @param {Model} model + * @param {Schema} schema + */ +var applyMethods = function(model, schema) { + function apply(method, schema) { + Object.defineProperty(model.prototype, method, { + get: function() { + var h = {}; + for (var k in schema.methods[method]) { + h[k] = schema.methods[method][k].bind(this); + } + return h; + }, + configurable: true + }); + } + for (var method in schema.methods) { + if (schema.tree.hasOwnProperty(method)) { + throw new Error('You have a method and a property in your schema both ' + + 'named "' + method + '"'); + } + if (typeof schema.methods[method] === 'function') { + model.prototype[method] = schema.methods[method]; + } else { + apply(method, schema); + } + } +}; + +/*! + * Register statics for this model + * @param {Model} model + * @param {Schema} schema + */ +var applyStatics = function(model, schema) { + for (var i in schema.statics) { + model[i] = schema.statics[i]; + } +}; + +/*! + * Register custom query methods for this model + * + * @param {Model} model + * @param {Schema} schema + */ + +function applyQueryMethods(model, methods) { + for (var i in methods) { + model.Query.prototype[i] = methods[i]; + } +} + +/*! + * Subclass this model with `conn`, `schema`, and `collection` settings. + * + * @param {Connection} conn + * @param {Schema} [schema] + * @param {String} [collection] + * @return {Model} + */ + +Model.__subclass = function subclass(conn, schema, collection) { + // subclass model using this connection and collection name + var _this = this; + + var Model = function Model(doc, fields, skipId) { + if (!(this instanceof Model)) { + return new Model(doc, fields, skipId); + } + _this.call(this, doc, fields, skipId); + }; + + Model.__proto__ = _this; + Model.prototype.__proto__ = _this.prototype; + Model.db = Model.prototype.db = conn; + + var s = schema && typeof schema !== 'string' + ? schema + : _this.prototype.schema; + + var options = s.options || {}; + + if (!collection) { + collection = _this.prototype.schema.get('collection') + || utils.toCollectionName(_this.modelName, options); + } + + var collectionOptions = { + bufferCommands: s ? options.bufferCommands : true, + capped: s && options.capped + }; + + Model.prototype.collection = conn.collection(collection, collectionOptions); + Model.collection = Model.prototype.collection; + Model.init(); + return Model; +}; + +Model.$wrapCallback = function(callback) { + var _this = this; + return function() { + try { + callback.apply(null, arguments); + } catch (error) { + _this.emit('error', error); + } + }; +}; + +/*! + * Module exports. + */ + +module.exports = exports = Model; |