123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207 |
- /* eslint no-empty: 1 */
- /*!
- * Module dependencies.
- */
- var SchemaType = require('../schematype'),
- CastError = SchemaType.CastError,
- oid = require('../types/objectid'),
- utils = require('../utils'),
- Document;
- /**
- * ObjectId SchemaType constructor.
- *
- * @param {String} key
- * @param {Object} options
- * @inherits SchemaType
- * @api public
- */
- function ObjectId(key, options) {
- SchemaType.call(this, key, options, 'ObjectID');
- }
- /**
- * This schema type's name, to defend against minifiers that mangle
- * function names.
- *
- * @api public
- */
- ObjectId.schemaName = 'ObjectId';
- /*!
- * Inherits from SchemaType.
- */
- ObjectId.prototype = Object.create(SchemaType.prototype);
- ObjectId.prototype.constructor = ObjectId;
- /**
- * Adds an auto-generated ObjectId default if turnOn is true.
- * @param {Boolean} turnOn auto generated ObjectId defaults
- * @api public
- * @return {SchemaType} this
- */
- ObjectId.prototype.auto = function(turnOn) {
- if (turnOn) {
- this.default(defaultId);
- this.set(resetId);
- }
- return this;
- };
- /**
- * Check if the given value satisfies a required validator.
- *
- * @param {Any} value
- * @param {Document} doc
- * @return {Boolean}
- * @api public
- */
- ObjectId.prototype.checkRequired = function checkRequired(value, doc) {
- if (SchemaType._isRef(this, value, doc, true)) {
- return !!value;
- }
- return value instanceof oid;
- };
- /**
- * Casts to ObjectId
- *
- * @param {Object} value
- * @param {Object} doc
- * @param {Boolean} init whether this is an initialization cast
- * @api private
- */
- ObjectId.prototype.cast = function(value, doc, init) {
- if (SchemaType._isRef(this, value, doc, init)) {
- // wait! we may need to cast this to a document
- if (value === null || value === undefined) {
- return value;
- }
- // lazy load
- Document || (Document = require('./../document'));
- if (value instanceof Document) {
- value.$__.wasPopulated = true;
- return value;
- }
- // setting a populated path
- if (value instanceof oid) {
- return value;
- } else if (Buffer.isBuffer(value) || !utils.isObject(value)) {
- throw new CastError('ObjectId', value, this.path);
- }
- // Handle the case where user directly sets a populated
- // path to a plain object; cast to the Model used in
- // the population query.
- var path = doc.$__fullPath(this.path);
- var owner = doc.ownerDocument ? doc.ownerDocument() : doc;
- var pop = owner.populated(path, true);
- var ret = value;
- if (!doc.$__.populated ||
- !doc.$__.populated[path] ||
- !doc.$__.populated[path].options ||
- !doc.$__.populated[path].options.options ||
- !doc.$__.populated[path].options.options.lean) {
- ret = new pop.options.model(value);
- ret.$__.wasPopulated = true;
- }
- return ret;
- }
- if (value === null || value === undefined) {
- return value;
- }
- if (value instanceof oid) {
- return value;
- }
- if (value._id) {
- if (value._id instanceof oid) {
- return value._id;
- }
- if (value._id.toString instanceof Function) {
- try {
- return new oid(value._id.toString());
- } catch (e) {
- }
- }
- }
- if (value.toString instanceof Function) {
- try {
- return new oid(value.toString());
- } catch (err) {
- throw new CastError('ObjectId', value, this.path);
- }
- }
- throw new CastError('ObjectId', value, this.path);
- };
- /*!
- * ignore
- */
- function handleSingle(val) {
- return this.cast(val);
- }
- ObjectId.prototype.$conditionalHandlers =
- utils.options(SchemaType.prototype.$conditionalHandlers, {
- $gt: handleSingle,
- $gte: handleSingle,
- $lt: handleSingle,
- $lte: handleSingle
- });
- /**
- * Casts contents for queries.
- *
- * @param {String} $conditional
- * @param {any} [val]
- * @api private
- */
- ObjectId.prototype.castForQuery = function($conditional, val) {
- var handler;
- if (arguments.length === 2) {
- handler = this.$conditionalHandlers[$conditional];
- if (!handler) {
- throw new Error('Can\'t use ' + $conditional + ' with ObjectId.');
- }
- return handler.call(this, val);
- }
- return this.cast($conditional);
- };
- /*!
- * ignore
- */
- function defaultId() {
- return new oid();
- }
- function resetId(v) {
- this.$__._id = null;
- return v;
- }
- /*!
- * Module exports.
- */
- module.exports = ObjectId;
|