1 |
3a515b92
|
cagy
|
var baseSetData = require('./_baseSetData'),
|
2 |
|
|
createBind = require('./_createBind'),
|
3 |
|
|
createCurry = require('./_createCurry'),
|
4 |
|
|
createHybrid = require('./_createHybrid'),
|
5 |
|
|
createPartial = require('./_createPartial'),
|
6 |
|
|
getData = require('./_getData'),
|
7 |
|
|
mergeData = require('./_mergeData'),
|
8 |
|
|
setData = require('./_setData'),
|
9 |
|
|
setWrapToString = require('./_setWrapToString'),
|
10 |
|
|
toInteger = require('./toInteger');
|
11 |
|
|
|
12 |
|
|
/** Error message constants. */
|
13 |
|
|
var FUNC_ERROR_TEXT = 'Expected a function';
|
14 |
|
|
|
15 |
|
|
/** Used to compose bitmasks for function metadata. */
|
16 |
|
|
var WRAP_BIND_FLAG = 1,
|
17 |
|
|
WRAP_BIND_KEY_FLAG = 2,
|
18 |
|
|
WRAP_CURRY_FLAG = 8,
|
19 |
|
|
WRAP_CURRY_RIGHT_FLAG = 16,
|
20 |
|
|
WRAP_PARTIAL_FLAG = 32,
|
21 |
|
|
WRAP_PARTIAL_RIGHT_FLAG = 64;
|
22 |
|
|
|
23 |
|
|
/* Built-in method references for those with the same name as other `lodash` methods. */
|
24 |
|
|
var nativeMax = Math.max;
|
25 |
|
|
|
26 |
|
|
/**
|
27 |
|
|
* Creates a function that either curries or invokes `func` with optional
|
28 |
|
|
* `this` binding and partially applied arguments.
|
29 |
|
|
*
|
30 |
|
|
* @private
|
31 |
|
|
* @param {Function|string} func The function or method name to wrap.
|
32 |
|
|
* @param {number} bitmask The bitmask flags.
|
33 |
|
|
* 1 - `_.bind`
|
34 |
|
|
* 2 - `_.bindKey`
|
35 |
|
|
* 4 - `_.curry` or `_.curryRight` of a bound function
|
36 |
|
|
* 8 - `_.curry`
|
37 |
|
|
* 16 - `_.curryRight`
|
38 |
|
|
* 32 - `_.partial`
|
39 |
|
|
* 64 - `_.partialRight`
|
40 |
|
|
* 128 - `_.rearg`
|
41 |
|
|
* 256 - `_.ary`
|
42 |
|
|
* 512 - `_.flip`
|
43 |
|
|
* @param {*} [thisArg] The `this` binding of `func`.
|
44 |
|
|
* @param {Array} [partials] The arguments to be partially applied.
|
45 |
|
|
* @param {Array} [holders] The `partials` placeholder indexes.
|
46 |
|
|
* @param {Array} [argPos] The argument positions of the new function.
|
47 |
|
|
* @param {number} [ary] The arity cap of `func`.
|
48 |
|
|
* @param {number} [arity] The arity of `func`.
|
49 |
|
|
* @returns {Function} Returns the new wrapped function.
|
50 |
|
|
*/
|
51 |
|
|
function createWrap(func, bitmask, thisArg, partials, holders, argPos, ary, arity) {
|
52 |
|
|
var isBindKey = bitmask & WRAP_BIND_KEY_FLAG;
|
53 |
|
|
if (!isBindKey && typeof func != 'function') {
|
54 |
|
|
throw new TypeError(FUNC_ERROR_TEXT);
|
55 |
|
|
}
|
56 |
|
|
var length = partials ? partials.length : 0;
|
57 |
|
|
if (!length) {
|
58 |
|
|
bitmask &= ~(WRAP_PARTIAL_FLAG | WRAP_PARTIAL_RIGHT_FLAG);
|
59 |
|
|
partials = holders = undefined;
|
60 |
|
|
}
|
61 |
|
|
ary = ary === undefined ? ary : nativeMax(toInteger(ary), 0);
|
62 |
|
|
arity = arity === undefined ? arity : toInteger(arity);
|
63 |
|
|
length -= holders ? holders.length : 0;
|
64 |
|
|
|
65 |
|
|
if (bitmask & WRAP_PARTIAL_RIGHT_FLAG) {
|
66 |
|
|
var partialsRight = partials,
|
67 |
|
|
holdersRight = holders;
|
68 |
|
|
|
69 |
|
|
partials = holders = undefined;
|
70 |
|
|
}
|
71 |
|
|
var data = isBindKey ? undefined : getData(func);
|
72 |
|
|
|
73 |
|
|
var newData = [
|
74 |
|
|
func, bitmask, thisArg, partials, holders, partialsRight, holdersRight,
|
75 |
|
|
argPos, ary, arity
|
76 |
|
|
];
|
77 |
|
|
|
78 |
|
|
if (data) {
|
79 |
|
|
mergeData(newData, data);
|
80 |
|
|
}
|
81 |
|
|
func = newData[0];
|
82 |
|
|
bitmask = newData[1];
|
83 |
|
|
thisArg = newData[2];
|
84 |
|
|
partials = newData[3];
|
85 |
|
|
holders = newData[4];
|
86 |
|
|
arity = newData[9] = newData[9] === undefined
|
87 |
|
|
? (isBindKey ? 0 : func.length)
|
88 |
|
|
: nativeMax(newData[9] - length, 0);
|
89 |
|
|
|
90 |
|
|
if (!arity && bitmask & (WRAP_CURRY_FLAG | WRAP_CURRY_RIGHT_FLAG)) {
|
91 |
|
|
bitmask &= ~(WRAP_CURRY_FLAG | WRAP_CURRY_RIGHT_FLAG);
|
92 |
|
|
}
|
93 |
|
|
if (!bitmask || bitmask == WRAP_BIND_FLAG) {
|
94 |
|
|
var result = createBind(func, bitmask, thisArg);
|
95 |
|
|
} else if (bitmask == WRAP_CURRY_FLAG || bitmask == WRAP_CURRY_RIGHT_FLAG) {
|
96 |
|
|
result = createCurry(func, bitmask, arity);
|
97 |
|
|
} else if ((bitmask == WRAP_PARTIAL_FLAG || bitmask == (WRAP_BIND_FLAG | WRAP_PARTIAL_FLAG)) && !holders.length) {
|
98 |
|
|
result = createPartial(func, bitmask, thisArg, partials);
|
99 |
|
|
} else {
|
100 |
|
|
result = createHybrid.apply(undefined, newData);
|
101 |
|
|
}
|
102 |
|
|
var setter = data ? baseSetData : setData;
|
103 |
|
|
return setWrapToString(setter(result, newData), func, bitmask);
|
104 |
|
|
}
|
105 |
|
|
|
106 |
|
|
module.exports = createWrap;
|