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.
305 lines
7.1 KiB
305 lines
7.1 KiB
/*! |
|
* 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;
|
|
|