62 lines
1.6 KiB
JavaScript
62 lines
1.6 KiB
JavaScript
|
import apply from './_apply.js';
|
||
|
|
import arrayMap from './_arrayMap.js';
|
||
|
|
import baseFlatten from './_baseFlatten.js';
|
||
|
|
import baseIteratee from './_baseIteratee.js';
|
||
|
|
import baseRest from './_baseRest.js';
|
||
|
|
import baseUnary from './_baseUnary.js';
|
||
|
|
import castRest from './_castRest.js';
|
||
|
|
import isArray from './isArray.js';
|
||
|
|
|
||
|
|
/* Built-in method references for those with the same name as other `lodash` methods. */
|
||
|
|
var nativeMin = Math.min;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Creates a function that invokes `func` with its arguments transformed.
|
||
|
|
*
|
||
|
|
* @static
|
||
|
|
* @since 4.0.0
|
||
|
|
* @memberOf _
|
||
|
|
* @category Function
|
||
|
|
* @param {Function} func The function to wrap.
|
||
|
|
* @param {...(Function|Function[])} [transforms=[_.identity]]
|
||
|
|
* The argument transforms.
|
||
|
|
* @returns {Function} Returns the new function.
|
||
|
|
* @example
|
||
|
|
*
|
||
|
|
* function doubled(n) {
|
||
|
|
* return n * 2;
|
||
|
|
* }
|
||
|
|
*
|
||
|
|
* function square(n) {
|
||
|
|
* return n * n;
|
||
|
|
* }
|
||
|
|
*
|
||
|
|
* var func = _.overArgs(function(x, y) {
|
||
|
|
* return [x, y];
|
||
|
|
* }, [square, doubled]);
|
||
|
|
*
|
||
|
|
* func(9, 3);
|
||
|
|
* // => [81, 6]
|
||
|
|
*
|
||
|
|
* func(10, 5);
|
||
|
|
* // => [100, 10]
|
||
|
|
*/
|
||
|
|
var overArgs = castRest(function(func, transforms) {
|
||
|
|
transforms = (transforms.length == 1 && isArray(transforms[0]))
|
||
|
|
? arrayMap(transforms[0], baseUnary(baseIteratee))
|
||
|
|
: arrayMap(baseFlatten(transforms, 1), baseUnary(baseIteratee));
|
||
|
|
|
||
|
|
var funcsLength = transforms.length;
|
||
|
|
return baseRest(function(args) {
|
||
|
|
var index = -1,
|
||
|
|
length = nativeMin(args.length, funcsLength);
|
||
|
|
|
||
|
|
while (++index < length) {
|
||
|
|
args[index] = transforms[index].call(this, args[index]);
|
||
|
|
}
|
||
|
|
return apply(func, this, args);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
|
||
|
|
export default overArgs;
|