You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.4 KiB
JavaScript
83 lines
2.4 KiB
JavaScript
9 years ago
|
var arrayCopy = require('../internal/arrayCopy'),
|
||
|
arrayPush = require('../internal/arrayPush'),
|
||
|
baseFunctions = require('../internal/baseFunctions'),
|
||
|
isFunction = require('../lang/isFunction'),
|
||
|
isObject = require('../lang/isObject'),
|
||
|
keys = require('../object/keys');
|
||
|
|
||
|
/**
|
||
|
* Adds all own enumerable function properties of a source object to the
|
||
|
* destination object. If `object` is a function then methods are added to
|
||
|
* its prototype as well.
|
||
|
*
|
||
|
* **Note:** Use `_.runInContext` to create a pristine `lodash` function to
|
||
|
* avoid conflicts caused by modifying the original.
|
||
|
*
|
||
|
* @static
|
||
|
* @memberOf _
|
||
|
* @category Utility
|
||
|
* @param {Function|Object} [object=lodash] The destination object.
|
||
|
* @param {Object} source The object of functions to add.
|
||
|
* @param {Object} [options] The options object.
|
||
|
* @param {boolean} [options.chain=true] Specify whether the functions added
|
||
|
* are chainable.
|
||
|
* @returns {Function|Object} Returns `object`.
|
||
|
* @example
|
||
|
*
|
||
|
* function vowels(string) {
|
||
|
* return _.filter(string, function(v) {
|
||
|
* return /[aeiou]/i.test(v);
|
||
|
* });
|
||
|
* }
|
||
|
*
|
||
|
* _.mixin({ 'vowels': vowels });
|
||
|
* _.vowels('fred');
|
||
|
* // => ['e']
|
||
|
*
|
||
|
* _('fred').vowels().value();
|
||
|
* // => ['e']
|
||
|
*
|
||
|
* _.mixin({ 'vowels': vowels }, { 'chain': false });
|
||
|
* _('fred').vowels();
|
||
|
* // => ['e']
|
||
|
*/
|
||
|
function mixin(object, source, options) {
|
||
|
var methodNames = baseFunctions(source, keys(source));
|
||
|
|
||
|
var chain = true,
|
||
|
index = -1,
|
||
|
isFunc = isFunction(object),
|
||
|
length = methodNames.length;
|
||
|
|
||
|
if (options === false) {
|
||
|
chain = false;
|
||
|
} else if (isObject(options) && 'chain' in options) {
|
||
|
chain = options.chain;
|
||
|
}
|
||
|
while (++index < length) {
|
||
|
var methodName = methodNames[index],
|
||
|
func = source[methodName];
|
||
|
|
||
|
object[methodName] = func;
|
||
|
if (isFunc) {
|
||
|
object.prototype[methodName] = (function(func) {
|
||
|
return function() {
|
||
|
var chainAll = this.__chain__;
|
||
|
if (chain || chainAll) {
|
||
|
var result = object(this.__wrapped__),
|
||
|
actions = result.__actions__ = arrayCopy(this.__actions__);
|
||
|
|
||
|
actions.push({ 'func': func, 'args': arguments, 'thisArg': object });
|
||
|
result.__chain__ = chainAll;
|
||
|
return result;
|
||
|
}
|
||
|
return func.apply(object, arrayPush([this.value()], arguments));
|
||
|
};
|
||
|
}(func));
|
||
|
}
|
||
|
}
|
||
|
return object;
|
||
|
}
|
||
|
|
||
|
module.exports = mixin;
|