123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204 |
- /*!
- * 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 private
- */
- function ObjectId (key, options) {
- SchemaType.call(this, key, options, 'ObjectID');
- }
- /**
- * This schema type's name, to defend against minifiers that mangle
- * function names.
- *
- * @api private
- */
- 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 required
- *
- * @api private
- */
- ObjectId.prototype.checkRequired = function checkRequired (value, doc) {
- if (SchemaType._isRef(this, value, doc, true)) {
- return null != value;
- } else {
- 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 (null == value) {
- 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 = new pop.options.model(value);
- ret.$__.wasPopulated = true;
- return ret;
- }
- // If null or undefined
- if (value == null) 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 oid.createFromHexString(value._id.toString());
- } catch(e) {}
- }
- }
- if (value.toString instanceof Function) {
- try {
- return oid.createFromHexString(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);
- }
- function handleArray (val) {
- var self = this;
- return val.map(function (m) {
- return self.cast(m);
- });
- }
- ObjectId.prototype.$conditionalHandlers =
- utils.options(SchemaType.prototype.$conditionalHandlers, {
- '$all': handleArray,
- '$gt': handleSingle,
- '$gte': handleSingle,
- '$in': handleArray,
- '$lt': handleSingle,
- '$lte': handleSingle,
- '$ne': handleSingle,
- '$nin': handleArray
- });
- /**
- * 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);
- } else {
- return this.cast($conditional);
- }
- };
- /*!
- * ignore
- */
- function defaultId () {
- return new oid();
- };
- function resetId (v) {
- this.$__._id = null;
- return v;
- }
- /*!
- * Module exports.
- */
- module.exports = ObjectId;
|