You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							434 lines
						
					
					
						
							10 KiB
						
					
					
				
			
		
		
	
	
							434 lines
						
					
					
						
							10 KiB
						
					
					
				| 'use strict';
 | |
| 
 | |
| /*!
 | |
|  * Module requirements.
 | |
|  */
 | |
| 
 | |
| const MongooseError = require('../error/index');
 | |
| const SchemaNumberOptions = require('../options/SchemaNumberOptions');
 | |
| const SchemaType = require('../schematype');
 | |
| const castNumber = require('../cast/number');
 | |
| const handleBitwiseOperator = require('./operators/bitwise');
 | |
| const utils = require('../utils');
 | |
| 
 | |
| const CastError = SchemaType.CastError;
 | |
| 
 | |
| /**
 | |
|  * Number SchemaType constructor.
 | |
|  *
 | |
|  * @param {String} key
 | |
|  * @param {Object} options
 | |
|  * @inherits SchemaType
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| function SchemaNumber(key, options) {
 | |
|   SchemaType.call(this, key, options, 'Number');
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Attaches a getter for all Number instances.
 | |
|  *
 | |
|  * ####Example:
 | |
|  *
 | |
|  *     // Make all numbers round down
 | |
|  *     mongoose.Number.get(function(v) { return Math.floor(v); });
 | |
|  *
 | |
|  *     const Model = mongoose.model('Test', new Schema({ test: Number }));
 | |
|  *     new Model({ test: 3.14 }).test; // 3
 | |
|  *
 | |
|  * @param {Function} getter
 | |
|  * @return {this}
 | |
|  * @function get
 | |
|  * @static
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.get = SchemaType.get;
 | |
| 
 | |
| /**
 | |
|  * Sets a default option for all Number instances.
 | |
|  *
 | |
|  * ####Example:
 | |
|  *
 | |
|  *     // Make all numbers have option `min` equal to 0.
 | |
|  *     mongoose.Schema.Number.set('min', 0);
 | |
|  *
 | |
|  *     const Order = mongoose.model('Order', new Schema({ amount: Number }));
 | |
|  *     new Order({ amount: -10 }).validateSync().errors.amount.message; // Path `amount` must be larger than 0.
 | |
|  *
 | |
|  * @param {String} option - The option you'd like to set the value for
 | |
|  * @param {*} value - value for option
 | |
|  * @return {undefined}
 | |
|  * @function set
 | |
|  * @static
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.set = SchemaType.set;
 | |
| 
 | |
| /*!
 | |
|  * ignore
 | |
|  */
 | |
| 
 | |
| SchemaNumber._cast = castNumber;
 | |
| 
 | |
| /**
 | |
|  * Get/set the function used to cast arbitrary values to numbers.
 | |
|  *
 | |
|  * ####Example:
 | |
|  *
 | |
|  *     // Make Mongoose cast empty strings '' to 0 for paths declared as numbers
 | |
|  *     const original = mongoose.Number.cast();
 | |
|  *     mongoose.Number.cast(v => {
 | |
|  *       if (v === '') { return 0; }
 | |
|  *       return original(v);
 | |
|  *     });
 | |
|  *
 | |
|  *     // Or disable casting entirely
 | |
|  *     mongoose.Number.cast(false);
 | |
|  *
 | |
|  * @param {Function} caster
 | |
|  * @return {Function}
 | |
|  * @function get
 | |
|  * @static
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.cast = function cast(caster) {
 | |
|   if (arguments.length === 0) {
 | |
|     return this._cast;
 | |
|   }
 | |
|   if (caster === false) {
 | |
|     caster = this._defaultCaster;
 | |
|   }
 | |
|   this._cast = caster;
 | |
| 
 | |
|   return this._cast;
 | |
| };
 | |
| 
 | |
| /*!
 | |
|  * ignore
 | |
|  */
 | |
| 
 | |
| SchemaNumber._defaultCaster = v => {
 | |
|   if (typeof v !== 'number') {
 | |
|     throw new Error();
 | |
|   }
 | |
|   return v;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * This schema type's name, to defend against minifiers that mangle
 | |
|  * function names.
 | |
|  *
 | |
|  * @api public
 | |
|  */
 | |
| SchemaNumber.schemaName = 'Number';
 | |
| 
 | |
| SchemaNumber.defaultOptions = {};
 | |
| 
 | |
| /*!
 | |
|  * Inherits from SchemaType.
 | |
|  */
 | |
| SchemaNumber.prototype = Object.create(SchemaType.prototype);
 | |
| SchemaNumber.prototype.constructor = SchemaNumber;
 | |
| SchemaNumber.prototype.OptionsConstructor = SchemaNumberOptions;
 | |
| 
 | |
| /*!
 | |
|  * ignore
 | |
|  */
 | |
| 
 | |
| SchemaNumber._checkRequired = v => typeof v === 'number' || v instanceof Number;
 | |
| 
 | |
| /**
 | |
|  * Override the function the required validator uses to check whether a string
 | |
|  * passes the `required` check.
 | |
|  *
 | |
|  * @param {Function} fn
 | |
|  * @return {Function}
 | |
|  * @function checkRequired
 | |
|  * @static
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.checkRequired = SchemaType.checkRequired;
 | |
| 
 | |
| /**
 | |
|  * Check if the given value satisfies a required validator.
 | |
|  *
 | |
|  * @param {Any} value
 | |
|  * @param {Document} doc
 | |
|  * @return {Boolean}
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.prototype.checkRequired = function checkRequired(value, doc) {
 | |
|   if (SchemaType._isRef(this, value, doc, true)) {
 | |
|     return !!value;
 | |
|   }
 | |
| 
 | |
|   // `require('util').inherits()` does **not** copy static properties, and
 | |
|   // plugins like mongoose-float use `inherits()` for pre-ES6.
 | |
|   const _checkRequired = typeof this.constructor.checkRequired == 'function' ?
 | |
|     this.constructor.checkRequired() :
 | |
|     SchemaNumber.checkRequired();
 | |
| 
 | |
|   return _checkRequired(value);
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Sets a minimum number validator.
 | |
|  *
 | |
|  * ####Example:
 | |
|  *
 | |
|  *     const s = new Schema({ n: { type: Number, min: 10 })
 | |
|  *     const M = db.model('M', s)
 | |
|  *     const m = new M({ n: 9 })
 | |
|  *     m.save(function (err) {
 | |
|  *       console.error(err) // validator error
 | |
|  *       m.n = 10;
 | |
|  *       m.save() // success
 | |
|  *     })
 | |
|  *
 | |
|  *     // custom error messages
 | |
|  *     // We can also use the special {MIN} token which will be replaced with the invalid value
 | |
|  *     const min = [10, 'The value of path `{PATH}` ({VALUE}) is beneath the limit ({MIN}).'];
 | |
|  *     const schema = new Schema({ n: { type: Number, min: min })
 | |
|  *     const M = mongoose.model('Measurement', schema);
 | |
|  *     const s= new M({ n: 4 });
 | |
|  *     s.validate(function (err) {
 | |
|  *       console.log(String(err)) // ValidationError: The value of path `n` (4) is beneath the limit (10).
 | |
|  *     })
 | |
|  *
 | |
|  * @param {Number} value minimum number
 | |
|  * @param {String} [message] optional custom error message
 | |
|  * @return {SchemaType} this
 | |
|  * @see Customized Error Messages #error_messages_MongooseError-messages
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.prototype.min = function(value, message) {
 | |
|   if (this.minValidator) {
 | |
|     this.validators = this.validators.filter(function(v) {
 | |
|       return v.validator !== this.minValidator;
 | |
|     }, this);
 | |
|   }
 | |
| 
 | |
|   if (value !== null && value !== undefined) {
 | |
|     let msg = message || MongooseError.messages.Number.min;
 | |
|     msg = msg.replace(/{MIN}/, value);
 | |
|     this.validators.push({
 | |
|       validator: this.minValidator = function(v) {
 | |
|         return v == null || v >= value;
 | |
|       },
 | |
|       message: msg,
 | |
|       type: 'min',
 | |
|       min: value
 | |
|     });
 | |
|   }
 | |
| 
 | |
|   return this;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Sets a maximum number validator.
 | |
|  *
 | |
|  * ####Example:
 | |
|  *
 | |
|  *     const s = new Schema({ n: { type: Number, max: 10 })
 | |
|  *     const M = db.model('M', s)
 | |
|  *     const m = new M({ n: 11 })
 | |
|  *     m.save(function (err) {
 | |
|  *       console.error(err) // validator error
 | |
|  *       m.n = 10;
 | |
|  *       m.save() // success
 | |
|  *     })
 | |
|  *
 | |
|  *     // custom error messages
 | |
|  *     // We can also use the special {MAX} token which will be replaced with the invalid value
 | |
|  *     const max = [10, 'The value of path `{PATH}` ({VALUE}) exceeds the limit ({MAX}).'];
 | |
|  *     const schema = new Schema({ n: { type: Number, max: max })
 | |
|  *     const M = mongoose.model('Measurement', schema);
 | |
|  *     const s= new M({ n: 4 });
 | |
|  *     s.validate(function (err) {
 | |
|  *       console.log(String(err)) // ValidationError: The value of path `n` (4) exceeds the limit (10).
 | |
|  *     })
 | |
|  *
 | |
|  * @param {Number} maximum number
 | |
|  * @param {String} [message] optional custom error message
 | |
|  * @return {SchemaType} this
 | |
|  * @see Customized Error Messages #error_messages_MongooseError-messages
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.prototype.max = function(value, message) {
 | |
|   if (this.maxValidator) {
 | |
|     this.validators = this.validators.filter(function(v) {
 | |
|       return v.validator !== this.maxValidator;
 | |
|     }, this);
 | |
|   }
 | |
| 
 | |
|   if (value !== null && value !== undefined) {
 | |
|     let msg = message || MongooseError.messages.Number.max;
 | |
|     msg = msg.replace(/{MAX}/, value);
 | |
|     this.validators.push({
 | |
|       validator: this.maxValidator = function(v) {
 | |
|         return v == null || v <= value;
 | |
|       },
 | |
|       message: msg,
 | |
|       type: 'max',
 | |
|       max: value
 | |
|     });
 | |
|   }
 | |
| 
 | |
|   return this;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Sets a enum validator
 | |
|  *
 | |
|  * ####Example:
 | |
|  *
 | |
|  *     const s = new Schema({ n: { type: Number, enum: [1, 2, 3] });
 | |
|  *     const M = db.model('M', s);
 | |
|  *
 | |
|  *     const m = new M({ n: 4 });
 | |
|  *     await m.save(); // throws validation error
 | |
|  *
 | |
|  *     m.n = 3;
 | |
|  *     await m.save(); // succeeds
 | |
|  *
 | |
|  * @param {Array} values allowed values
 | |
|  * @param {String} [message] optional custom error message
 | |
|  * @return {SchemaType} this
 | |
|  * @see Customized Error Messages #error_messages_MongooseError-messages
 | |
|  * @api public
 | |
|  */
 | |
| 
 | |
| SchemaNumber.prototype.enum = function(values, message) {
 | |
|   if (this.enumValidator) {
 | |
|     this.validators = this.validators.filter(function(v) {
 | |
|       return v.validator !== this.enumValidator;
 | |
|     }, this);
 | |
|   }
 | |
| 
 | |
| 
 | |
|   if (!Array.isArray(values)) {
 | |
|     if (utils.isObject(values)) {
 | |
|       values = utils.object.vals(values);
 | |
|     } else {
 | |
|       values = Array.prototype.slice.call(arguments);
 | |
|     }
 | |
|     message = MongooseError.messages.Number.enum;
 | |
|   }
 | |
| 
 | |
|   message = message == null ? MongooseError.messages.Number.enum : message;
 | |
| 
 | |
|   this.enumValidator = v => v == null || values.indexOf(v) !== -1;
 | |
|   this.validators.push({
 | |
|     validator: this.enumValidator,
 | |
|     message: message,
 | |
|     type: 'enum',
 | |
|     enumValues: values
 | |
|   });
 | |
| 
 | |
|   return this;
 | |
| };
 | |
| 
 | |
| /**
 | |
|  * Casts to number
 | |
|  *
 | |
|  * @param {Object} value value to cast
 | |
|  * @param {Document} doc document that triggers the casting
 | |
|  * @param {Boolean} init
 | |
|  * @api private
 | |
|  */
 | |
| 
 | |
| SchemaNumber.prototype.cast = function(value, doc, init) {
 | |
|   if (SchemaType._isRef(this, value, doc, init)) {
 | |
|     if (typeof value === 'number') {
 | |
|       return value;
 | |
|     }
 | |
| 
 | |
|     return this._castRef(value, doc, init);
 | |
|   }
 | |
| 
 | |
|   const val = value && typeof value._id !== 'undefined' ?
 | |
|     value._id : // documents
 | |
|     value;
 | |
| 
 | |
|   let castNumber;
 | |
|   if (typeof this._castFunction === 'function') {
 | |
|     castNumber = this._castFunction;
 | |
|   } else if (typeof this.constructor.cast === 'function') {
 | |
|     castNumber = this.constructor.cast();
 | |
|   } else {
 | |
|     castNumber = SchemaNumber.cast();
 | |
|   }
 | |
| 
 | |
|   try {
 | |
|     return castNumber(val);
 | |
|   } catch (err) {
 | |
|     throw new CastError('Number', val, this.path, err, this);
 | |
|   }
 | |
| };
 | |
| 
 | |
| /*!
 | |
|  * ignore
 | |
|  */
 | |
| 
 | |
| function handleSingle(val) {
 | |
|   return this.cast(val);
 | |
| }
 | |
| 
 | |
| function handleArray(val) {
 | |
|   const _this = this;
 | |
|   if (!Array.isArray(val)) {
 | |
|     return [this.cast(val)];
 | |
|   }
 | |
|   return val.map(function(m) {
 | |
|     return _this.cast(m);
 | |
|   });
 | |
| }
 | |
| 
 | |
| SchemaNumber.prototype.$conditionalHandlers =
 | |
|     utils.options(SchemaType.prototype.$conditionalHandlers, {
 | |
|       $bitsAllClear: handleBitwiseOperator,
 | |
|       $bitsAnyClear: handleBitwiseOperator,
 | |
|       $bitsAllSet: handleBitwiseOperator,
 | |
|       $bitsAnySet: handleBitwiseOperator,
 | |
|       $gt: handleSingle,
 | |
|       $gte: handleSingle,
 | |
|       $lt: handleSingle,
 | |
|       $lte: handleSingle,
 | |
|       $mod: handleArray
 | |
|     });
 | |
| 
 | |
| /**
 | |
|  * Casts contents for queries.
 | |
|  *
 | |
|  * @param {String} $conditional
 | |
|  * @param {any} [value]
 | |
|  * @api private
 | |
|  */
 | |
| 
 | |
| SchemaNumber.prototype.castForQuery = function($conditional, val) {
 | |
|   let handler;
 | |
|   if (arguments.length === 2) {
 | |
|     handler = this.$conditionalHandlers[$conditional];
 | |
|     if (!handler) {
 | |
|       throw new CastError('number', val, this.path, null, this);
 | |
|     }
 | |
|     return handler.call(this, val);
 | |
|   }
 | |
|   val = this._castForQuery($conditional);
 | |
|   return val;
 | |
| };
 | |
| 
 | |
| /*!
 | |
|  * Module exports.
 | |
|  */
 | |
| 
 | |
| module.exports = SchemaNumber;
 |