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.
		
		
		
		
		
			
		
			
				
					70 lines
				
				1.8 KiB
			
		
		
			
		
	
	
					70 lines
				
				1.8 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								'use strict'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								const Ajv = require('ajv')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function ValidatorSelector () {
							 | 
						||
| 
								 | 
							
								  const validatorPool = new Map()
							 | 
						||
| 
								 | 
							
								  const cache = new Map()
							 | 
						||
| 
								 | 
							
								  cache.put = cache.set
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return function buildCompilerFromPool (externalSchemas, options) {
							 | 
						||
| 
								 | 
							
								    const externals = JSON.stringify(externalSchemas)
							 | 
						||
| 
								 | 
							
								    const ajvConfig = JSON.stringify(options.customOptions)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    const uniqueAjvKey = `${externals}${ajvConfig}`
							 | 
						||
| 
								 | 
							
								    if (validatorPool.has(uniqueAjvKey)) {
							 | 
						||
| 
								 | 
							
								      return validatorPool.get(uniqueAjvKey)
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    const compiler = ValidatorCompiler(externalSchemas, options, cache)
							 | 
						||
| 
								 | 
							
								    validatorPool.set(uniqueAjvKey, compiler)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    return compiler
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function ValidatorCompiler (externalSchemas, options, cache) {
							 | 
						||
| 
								 | 
							
								  // This instance of Ajv is private
							 | 
						||
| 
								 | 
							
								  // it should not be customized or used
							 | 
						||
| 
								 | 
							
								  const ajv = new Ajv(Object.assign({
							 | 
						||
| 
								 | 
							
								    coerceTypes: true,
							 | 
						||
| 
								 | 
							
								    useDefaults: true,
							 | 
						||
| 
								 | 
							
								    removeAdditional: true,
							 | 
						||
| 
								 | 
							
								    // Explicitly set allErrors to `false`.
							 | 
						||
| 
								 | 
							
								    // When set to `true`, a DoS attack is possible.
							 | 
						||
| 
								 | 
							
								    allErrors: false,
							 | 
						||
| 
								 | 
							
								    nullable: true
							 | 
						||
| 
								 | 
							
								  }, options.customOptions, { cache }))
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (options.plugins && options.plugins.length > 0) {
							 | 
						||
| 
								 | 
							
								    for (const plugin of options.plugins) {
							 | 
						||
| 
								 | 
							
								      if (Array.isArray(plugin)) {
							 | 
						||
| 
								 | 
							
								        plugin[0](ajv, plugin[1])
							 | 
						||
| 
								 | 
							
								      } else {
							 | 
						||
| 
								 | 
							
								        plugin(ajv)
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const sourceSchemas = Object.values(externalSchemas)
							 | 
						||
| 
								 | 
							
								  for (const extSchema of sourceSchemas) {
							 | 
						||
| 
								 | 
							
								    ajv.addSchema(extSchema)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return function ({ schema/*, method, url, httpPart */ }) {
							 | 
						||
| 
								 | 
							
								    // Ajv does not support compiling two schemas with the same
							 | 
						||
| 
								 | 
							
								    // id inside the same instance. Therefore if we have already
							 | 
						||
| 
								 | 
							
								    // compiled the schema with the given id, we just return it.
							 | 
						||
| 
								 | 
							
								    if (schema.$id) {
							 | 
						||
| 
								 | 
							
								      const stored = ajv.getSchema(schema.$id)
							 | 
						||
| 
								 | 
							
								      if (stored) {
							 | 
						||
| 
								 | 
							
								        return stored
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    return ajv.compile(schema)
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								module.exports = ValidatorSelector
							 |