'use strict'; const get = require('../get'); /*! * Register methods for this model * * @param {Model} model * @param {Schema} schema */ module.exports = function applyMethods(model, schema) { function apply(method, schema) { Object.defineProperty(model.prototype, method, { get: function() { const h = {}; for (const k in schema.methods[method]) { h[k] = schema.methods[method][k].bind(this); } return h; }, configurable: true }); } for (const method of Object.keys(schema.methods)) { const fn = schema.methods[method]; if (schema.tree.hasOwnProperty(method)) { throw new Error('You have a method and a property in your schema both ' + 'named "' + method + '"'); } if (schema.reserved[method] && !get(schema, `methodOptions.${method}.suppressWarning`, false)) { console.warn(`mongoose: the method name "${method}" is used by mongoose ` + 'internally, overwriting it may cause bugs. If you\'re sure you know ' + 'what you\'re doing, you can suppress this error by using ' + `\`schema.method('${method}', fn, { suppressWarning: true })\`.`); } if (typeof fn === 'function') { model.prototype[method] = fn; } else { apply(method, schema); } } // Recursively call `applyMethods()` on child schemas model.$appliedMethods = true; for (let i = 0; i < schema.childSchemas.length; ++i) { if (schema.childSchemas[i].model.$appliedMethods) { continue; } applyMethods(schema.childSchemas[i].model, schema.childSchemas[i].schema); } };