123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402 |
- 'use strict';
- const MongooseError = require('../error/index');
- const SchemaDateOptions = require('../options/SchemaDateOptions');
- const SchemaType = require('../schematype');
- const castDate = require('../cast/date');
- const utils = require('../utils');
- const CastError = SchemaType.CastError;
- function SchemaDate(key, options) {
- SchemaType.call(this, key, options, 'Date');
- }
- SchemaDate.schemaName = 'Date';
- SchemaDate.defaultOptions = {};
- SchemaDate.prototype = Object.create(SchemaType.prototype);
- SchemaDate.prototype.constructor = SchemaDate;
- SchemaDate.prototype.OptionsConstructor = SchemaDateOptions;
- SchemaDate._cast = castDate;
- SchemaDate.set = SchemaType.set;
- SchemaDate.cast = function cast(caster) {
- if (arguments.length === 0) {
- return this._cast;
- }
- if (caster === false) {
- caster = this._defaultCaster;
- }
- this._cast = caster;
- return this._cast;
- };
- SchemaDate._defaultCaster = v => {
- if (v != null && !(v instanceof Date)) {
- throw new Error();
- }
- return v;
- };
- SchemaDate.prototype.expires = function(when) {
- if (!this._index || this._index.constructor.name !== 'Object') {
- this._index = {};
- }
- this._index.expires = when;
- utils.expires(this._index);
- return this;
- };
- SchemaDate._checkRequired = v => v instanceof Date;
- SchemaDate.checkRequired = SchemaType.checkRequired;
- SchemaDate.prototype.checkRequired = function(value, doc) {
- if (SchemaType._isRef(this, value, doc, true)) {
- return !!value;
- }
-
-
- const _checkRequired = typeof this.constructor.checkRequired == 'function' ?
- this.constructor.checkRequired() :
- SchemaDate.checkRequired();
- return _checkRequired(value);
- };
- SchemaDate.prototype.min = function(value, message) {
- if (this.minValidator) {
- this.validators = this.validators.filter(function(v) {
- return v.validator !== this.minValidator;
- }, this);
- }
- if (value) {
- let msg = message || MongooseError.messages.Date.min;
- if (typeof msg === 'string') {
- msg = msg.replace(/{MIN}/, (value === Date.now ? 'Date.now()' : value.toString()));
- }
- const _this = this;
- this.validators.push({
- validator: this.minValidator = function(val) {
- let _value = value;
- if (typeof value === 'function' && value !== Date.now) {
- _value = _value.call(this);
- }
- const min = (_value === Date.now ? _value() : _this.cast(_value));
- return val === null || val.valueOf() >= min.valueOf();
- },
- message: msg,
- type: 'min',
- min: value
- });
- }
- return this;
- };
- SchemaDate.prototype.max = function(value, message) {
- if (this.maxValidator) {
- this.validators = this.validators.filter(function(v) {
- return v.validator !== this.maxValidator;
- }, this);
- }
- if (value) {
- let msg = message || MongooseError.messages.Date.max;
- if (typeof msg === 'string') {
- msg = msg.replace(/{MAX}/, (value === Date.now ? 'Date.now()' : value.toString()));
- }
- const _this = this;
- this.validators.push({
- validator: this.maxValidator = function(val) {
- let _value = value;
- if (typeof _value === 'function' && _value !== Date.now) {
- _value = _value.call(this);
- }
- const max = (_value === Date.now ? _value() : _this.cast(_value));
- return val === null || val.valueOf() <= max.valueOf();
- },
- message: msg,
- type: 'max',
- max: value
- });
- }
- return this;
- };
- SchemaDate.prototype.cast = function(value) {
- let castDate;
- if (typeof this._castFunction === 'function') {
- castDate = this._castFunction;
- } else if (typeof this.constructor.cast === 'function') {
- castDate = this.constructor.cast();
- } else {
- castDate = SchemaDate.cast();
- }
- try {
- return castDate(value);
- } catch (error) {
- throw new CastError('date', value, this.path, error, this);
- }
- };
- function handleSingle(val) {
- return this.cast(val);
- }
- SchemaDate.prototype.$conditionalHandlers =
- utils.options(SchemaType.prototype.$conditionalHandlers, {
- $gt: handleSingle,
- $gte: handleSingle,
- $lt: handleSingle,
- $lte: handleSingle
- });
- SchemaDate.prototype.castForQuery = function($conditional, val) {
- if (arguments.length !== 2) {
- return this._castForQuery($conditional);
- }
- const handler = this.$conditionalHandlers[$conditional];
- if (!handler) {
- throw new Error('Can\'t use ' + $conditional + ' with Date.');
- }
- return handler.call(this, val);
- };
- module.exports = SchemaDate;
|