123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355 |
- 'use strict';
- // Load modules
- const Hoek = require('hoek');
- const Any = require('./any');
- const Cast = require('./cast');
- const Errors = require('./errors');
- const Lazy = require('./lazy');
- const Ref = require('./ref');
- // Declare internals
- const internals = {
- alternatives: require('./alternatives'),
- array: require('./array'),
- boolean: require('./boolean'),
- binary: require('./binary'),
- date: require('./date'),
- number: require('./number'),
- object: require('./object'),
- string: require('./string')
- };
- internals.root = function () {
- const any = new Any();
- const root = any.clone();
- root.any = function () {
- return any;
- };
- root.alternatives = root.alt = function () {
- return arguments.length ? internals.alternatives.try.apply(internals.alternatives, arguments) : internals.alternatives;
- };
- root.array = function () {
- return internals.array;
- };
- root.boolean = root.bool = function () {
- return internals.boolean;
- };
- root.binary = function () {
- return internals.binary;
- };
- root.date = function () {
- return internals.date;
- };
- root.func = function () {
- return internals.object._func();
- };
- root.number = function () {
- return internals.number;
- };
- root.object = function () {
- return arguments.length ? internals.object.keys.apply(internals.object, arguments) : internals.object;
- };
- root.string = function () {
- return internals.string;
- };
- root.ref = function () {
- return Ref.create.apply(null, arguments);
- };
- root.isRef = function (ref) {
- return Ref.isRef(ref);
- };
- root.validate = function (value /*, [schema], [options], callback */) {
- const last = arguments[arguments.length - 1];
- const callback = typeof last === 'function' ? last : null;
- const count = arguments.length - (callback ? 1 : 0);
- if (count === 1) {
- return any.validate(value, callback);
- }
- const options = count === 3 ? arguments[2] : {};
- const schema = root.compile(arguments[1]);
- return schema._validateWithOptions(value, options, callback);
- };
- root.describe = function () {
- const schema = arguments.length ? root.compile(arguments[0]) : any;
- return schema.describe();
- };
- root.compile = function (schema) {
- try {
- return Cast.schema(schema);
- }
- catch (err) {
- if (err.hasOwnProperty('path')) {
- err.message = err.message + '(' + err.path + ')';
- }
- throw err;
- }
- };
- root.assert = function (value, schema, message) {
- root.attempt(value, schema, message);
- };
- root.attempt = function (value, schema, message) {
- const result = root.validate(value, schema);
- const error = result.error;
- if (error) {
- if (!message) {
- error.message = error.annotate();
- throw error;
- }
- if (!(message instanceof Error)) {
- error.message = message + ' ' + error.annotate();
- throw error;
- }
- throw message;
- }
- return result.value;
- };
- root.reach = function (schema, path) {
- Hoek.assert(schema && schema instanceof Any, 'you must provide a joi schema');
- Hoek.assert(typeof path === 'string', 'path must be a string');
- if (path === '') {
- return schema;
- }
- const parts = path.split('.');
- const children = schema._inner.children;
- if (!children) {
- return;
- }
- const key = parts[0];
- for (let i = 0; i < children.length; ++i) {
- const child = children[i];
- if (child.key === key) {
- return this.reach(child.schema, path.substr(key.length + 1));
- }
- }
- };
- root.lazy = function (fn) {
- return Lazy.set(fn);
- };
- root.extend = function () {
- const extensions = Hoek.flatten(Array.prototype.slice.call(arguments));
- Hoek.assert(extensions.length > 0, 'You need to provide at least one extension');
- this.assert(extensions, root.extensionsSchema);
- const joi = Object.create(this);
- for (let i = 0; i < extensions.length; ++i) {
- const extension = extensions[i];
- const base = (extension.base || this.any()).clone(); // Cloning because we're going to override language afterwards
- const ctor = base.constructor;
- const type = class extends ctor { // eslint-disable-line no-loop-func
- constructor() {
- super();
- if (extension.base) {
- Object.assign(this, base);
- }
- this._type = extension.name;
- if (extension.language) {
- this._settings = this._settings || { language: {} };
- this._settings.language = Hoek.applyToDefaults(this._settings.language, {
- [extension.name]: extension.language
- });
- }
- }
- };
- if (extension.coerce) {
- type.prototype._coerce = function (value, state, options) {
- if (ctor.prototype._coerce) {
- const baseRet = ctor.prototype._coerce.call(this, value, state, options);
- if (baseRet.errors) {
- return baseRet;
- }
- value = baseRet.value;
- }
- const ret = extension.coerce.call(this, value, state, options);
- if (ret instanceof Errors.Err) {
- return { value, errors: ret };
- }
- return { value: ret };
- };
- }
- if (extension.pre) {
- type.prototype._base = function (value, state, options) {
- if (ctor.prototype._base) {
- const baseRet = ctor.prototype._base.call(this, value, state, options);
- if (baseRet.errors) {
- return baseRet;
- }
- value = baseRet.value;
- }
- const ret = extension.pre.call(this, value, state, options);
- if (ret instanceof Errors.Err) {
- return { value, errors: ret };
- }
- return { value: ret };
- };
- }
- if (extension.rules) {
- for (let j = 0; j < extension.rules.length; ++j) {
- const rule = extension.rules[j];
- const ruleArgs = rule.params ?
- (rule.params instanceof Any ? rule.params._inner.children.map((k) => k.key) : Object.keys(rule.params)) :
- [];
- const validateArgs = rule.params ? Cast.schema(rule.params) : null;
- type.prototype[rule.name] = function () { // eslint-disable-line no-loop-func
- if (arguments.length > ruleArgs.length) {
- throw new Error('Unexpected number of arguments');
- }
- const args = Array.prototype.slice.call(arguments);
- let hasRef = false;
- const arg = {};
- for (let k = 0; k < ruleArgs.length; ++k) {
- arg[ruleArgs[k]] = args[k];
- if (!hasRef && Ref.isRef(args[k])) {
- hasRef = true;
- }
- }
- if (validateArgs) {
- joi.assert(arg, validateArgs);
- }
- let schema;
- if (rule.validate) {
- const validate = function (value, state, options) {
- return rule.validate.call(this, arg, value, state, options);
- };
- schema = this._test(rule.name, arg, validate, {
- description: rule.description,
- hasRef
- });
- }
- else {
- schema = this.clone();
- }
- if (rule.setup) {
- rule.setup.call(schema, arg);
- }
- return schema;
- };
- }
- }
- if (extension.describe) {
- type.prototype.describe = function () {
- const description = ctor.prototype.describe.call(this);
- return extension.describe.call(this, description);
- };
- }
- const instance = new type();
- joi[extension.name] = function () {
- return instance;
- };
- }
- return joi;
- };
- root.extensionsSchema = internals.array.items(internals.object.keys({
- base: internals.object.type(Any, 'Joi object'),
- name: internals.string.required(),
- coerce: internals.object._func().arity(3),
- pre: internals.object._func().arity(3),
- language: internals.object,
- describe: internals.object._func().arity(1),
- rules: internals.array.items(internals.object.keys({
- name: internals.string.required(),
- setup: internals.object._func().arity(1),
- validate: internals.object._func().arity(4),
- params: [
- internals.object.pattern(/.*/, internals.object.type(Any, 'Joi object')),
- internals.object.type(internals.object.constructor, 'Joi object')
- ],
- description: [internals.string, internals.object._func().arity(1)]
- }).or('setup', 'validate'))
- })).strict();
- return root;
- };
- module.exports = internals.root();
|