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.
306 lines
7.1 KiB
306 lines
7.1 KiB
3 months ago
|
/*!
|
||
|
* Module dependencies.
|
||
|
*/
|
||
|
|
||
|
'use strict';
|
||
|
|
||
|
const MongooseBuffer = require('../types/buffer');
|
||
|
const SchemaType = require('../schemaType');
|
||
|
const CastError = SchemaType.CastError;
|
||
|
const castUUID = require('../cast/uuid');
|
||
|
const createJSONSchemaTypeDefinition = require('../helpers/createJSONSchemaTypeDefinition');
|
||
|
const utils = require('../utils');
|
||
|
const handleBitwiseOperator = require('./operators/bitwise');
|
||
|
|
||
|
const UUID_FORMAT = castUUID.UUID_FORMAT;
|
||
|
const Binary = MongooseBuffer.Binary;
|
||
|
|
||
|
/**
|
||
|
* Convert binary to a uuid string
|
||
|
* @param {Buffer|Binary|String} uuidBin The value to process
|
||
|
* @returns {String} The completed uuid-string
|
||
|
* @api private
|
||
|
*/
|
||
|
function binaryToString(uuidBin) {
|
||
|
// i(hasezoey) dont quite know why, but "uuidBin" may sometimes also be the already processed string
|
||
|
let hex;
|
||
|
if (typeof uuidBin !== 'string' && uuidBin != null) {
|
||
|
hex = uuidBin.toString('hex');
|
||
|
const uuidStr = hex.substring(0, 8) + '-' + hex.substring(8, 8 + 4) + '-' + hex.substring(12, 12 + 4) + '-' + hex.substring(16, 16 + 4) + '-' + hex.substring(20, 20 + 12);
|
||
|
return uuidStr;
|
||
|
}
|
||
|
return uuidBin;
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* UUIDv1 SchemaType constructor.
|
||
|
*
|
||
|
* @param {String} key
|
||
|
* @param {Object} options
|
||
|
* @inherits SchemaType
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
function SchemaUUID(key, options) {
|
||
|
SchemaType.call(this, key, options, 'UUID');
|
||
|
this.getters.push(function(value) {
|
||
|
// For populated
|
||
|
if (value != null && value.$__ != null) {
|
||
|
return value;
|
||
|
}
|
||
|
if (Buffer.isBuffer(value)) {
|
||
|
return binaryToString(value);
|
||
|
} else if (value instanceof Binary) {
|
||
|
return binaryToString(value.buffer);
|
||
|
} else if (utils.isPOJO(value) && value.type === 'Buffer' && Array.isArray(value.data)) {
|
||
|
// Cloned buffers look like `{ type: 'Buffer', data: [5, 224, ...] }`
|
||
|
return binaryToString(Buffer.from(value.data));
|
||
|
}
|
||
|
return value;
|
||
|
});
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* This schema type's name, to defend against minifiers that mangle
|
||
|
* function names.
|
||
|
*
|
||
|
* @api public
|
||
|
*/
|
||
|
SchemaUUID.schemaName = 'UUID';
|
||
|
|
||
|
SchemaUUID.defaultOptions = {};
|
||
|
|
||
|
/*!
|
||
|
* Inherits from SchemaType.
|
||
|
*/
|
||
|
SchemaUUID.prototype = Object.create(SchemaType.prototype);
|
||
|
SchemaUUID.prototype.constructor = SchemaUUID;
|
||
|
|
||
|
/*!
|
||
|
* ignore
|
||
|
*/
|
||
|
|
||
|
SchemaUUID._cast = castUUID;
|
||
|
|
||
|
/**
|
||
|
* Attaches a getter for all UUID instances.
|
||
|
*
|
||
|
* #### Example:
|
||
|
*
|
||
|
* // Note that `v` is a string by default
|
||
|
* mongoose.Schema.UUID.get(v => v.toUpperCase());
|
||
|
*
|
||
|
* const Model = mongoose.model('Test', new Schema({ test: 'UUID' }));
|
||
|
* new Model({ test: uuid.v4() }).test; // UUID with all uppercase
|
||
|
*
|
||
|
* @param {Function} getter
|
||
|
* @return {this}
|
||
|
* @function get
|
||
|
* @static
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.get = SchemaType.get;
|
||
|
|
||
|
/**
|
||
|
* Sets a default option for all UUID instances.
|
||
|
*
|
||
|
* #### Example:
|
||
|
*
|
||
|
* // Make all UUIDs have `required` of true by default.
|
||
|
* mongoose.Schema.UUID.set('required', true);
|
||
|
*
|
||
|
* const User = mongoose.model('User', new Schema({ test: mongoose.UUID }));
|
||
|
* new User({ }).validateSync().errors.test.message; // Path `test` is required.
|
||
|
*
|
||
|
* @param {String} option The option you'd like to set the value for
|
||
|
* @param {Any} value value for option
|
||
|
* @return {undefined}
|
||
|
* @function set
|
||
|
* @static
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.set = SchemaType.set;
|
||
|
|
||
|
SchemaUUID.setters = [];
|
||
|
|
||
|
/**
|
||
|
* Get/set the function used to cast arbitrary values to UUIDs.
|
||
|
*
|
||
|
* #### Example:
|
||
|
*
|
||
|
* // Make Mongoose refuse to cast UUIDs with 0 length
|
||
|
* const original = mongoose.Schema.Types.UUID.cast();
|
||
|
* mongoose.UUID.cast(v => {
|
||
|
* assert.ok(typeof v === "string" && v.length > 0);
|
||
|
* return original(v);
|
||
|
* });
|
||
|
*
|
||
|
* // Or disable casting entirely
|
||
|
* mongoose.UUID.cast(false);
|
||
|
*
|
||
|
* @param {Function} [caster]
|
||
|
* @return {Function}
|
||
|
* @function get
|
||
|
* @static
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.cast = function cast(caster) {
|
||
|
if (arguments.length === 0) {
|
||
|
return this._cast;
|
||
|
}
|
||
|
if (caster === false) {
|
||
|
caster = this._defaultCaster;
|
||
|
}
|
||
|
this._cast = caster;
|
||
|
|
||
|
return this._cast;
|
||
|
};
|
||
|
|
||
|
/*!
|
||
|
* ignore
|
||
|
*/
|
||
|
|
||
|
SchemaUUID._checkRequired = v => v != null;
|
||
|
|
||
|
/**
|
||
|
* 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
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.checkRequired = SchemaType.checkRequired;
|
||
|
|
||
|
/**
|
||
|
* Check if the given value satisfies a required validator.
|
||
|
*
|
||
|
* @param {Any} value
|
||
|
* @return {Boolean}
|
||
|
* @api public
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.prototype.checkRequired = function checkRequired(value) {
|
||
|
if (Buffer.isBuffer(value)) {
|
||
|
value = binaryToString(value);
|
||
|
}
|
||
|
return value != null && UUID_FORMAT.test(value);
|
||
|
};
|
||
|
|
||
|
/**
|
||
|
* Casts to UUID
|
||
|
*
|
||
|
* @param {Object} value
|
||
|
* @param {Object} doc
|
||
|
* @param {Boolean} init whether this is an initialization cast
|
||
|
* @api private
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.prototype.cast = function(value, doc, init, prev, options) {
|
||
|
if (utils.isNonBuiltinObject(value) &&
|
||
|
SchemaType._isRef(this, value, doc, init)) {
|
||
|
return this._castRef(value, doc, init, options);
|
||
|
}
|
||
|
|
||
|
let castFn;
|
||
|
if (typeof this._castFunction === 'function') {
|
||
|
castFn = this._castFunction;
|
||
|
} else if (typeof this.constructor.cast === 'function') {
|
||
|
castFn = this.constructor.cast();
|
||
|
} else {
|
||
|
castFn = SchemaUUID.cast();
|
||
|
}
|
||
|
|
||
|
try {
|
||
|
return castFn(value);
|
||
|
} catch (error) {
|
||
|
throw new CastError(SchemaUUID.schemaName, value, this.path, error, this);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
/*!
|
||
|
* ignore
|
||
|
*/
|
||
|
|
||
|
function handleSingle(val) {
|
||
|
return this.cast(val);
|
||
|
}
|
||
|
|
||
|
/*!
|
||
|
* ignore
|
||
|
*/
|
||
|
|
||
|
function handleArray(val) {
|
||
|
return val.map((m) => {
|
||
|
return this.cast(m);
|
||
|
});
|
||
|
}
|
||
|
|
||
|
SchemaUUID.prototype.$conditionalHandlers = {
|
||
|
...SchemaType.prototype.$conditionalHandlers,
|
||
|
$bitsAllClear: handleBitwiseOperator,
|
||
|
$bitsAnyClear: handleBitwiseOperator,
|
||
|
$bitsAllSet: handleBitwiseOperator,
|
||
|
$bitsAnySet: handleBitwiseOperator,
|
||
|
$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
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.prototype.castForQuery = function($conditional, val, context) {
|
||
|
let handler;
|
||
|
if ($conditional != null) {
|
||
|
handler = this.$conditionalHandlers[$conditional];
|
||
|
if (!handler)
|
||
|
throw new Error('Can\'t use ' + $conditional + ' with UUID.');
|
||
|
return handler.call(this, val, context);
|
||
|
}
|
||
|
|
||
|
try {
|
||
|
return this.applySetters(val, context);
|
||
|
} catch (err) {
|
||
|
if (err instanceof CastError && err.path === this.path && this.$fullPath != null) {
|
||
|
err.path = this.$fullPath;
|
||
|
}
|
||
|
throw err;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
/**
|
||
|
* Returns this schema type's representation in a JSON schema.
|
||
|
*
|
||
|
* @param [options]
|
||
|
* @param [options.useBsonType=false] If true, return a representation with `bsonType` for use with MongoDB's `$jsonSchema`.
|
||
|
* @returns {Object} JSON schema properties
|
||
|
*/
|
||
|
|
||
|
SchemaUUID.prototype.toJSONSchema = function toJSONSchema(options) {
|
||
|
const isRequired = this.options.required && typeof this.options.required !== 'function';
|
||
|
return createJSONSchemaTypeDefinition('string', 'binData', options?.useBsonType, isRequired);
|
||
|
};
|
||
|
|
||
|
/*!
|
||
|
* Module exports.
|
||
|
*/
|
||
|
|
||
|
module.exports = SchemaUUID;
|