UNPKG

3.71 kBJavaScriptView Raw
1import baseSetData from './_baseSetData.js';
2import createBind from './_createBind.js';
3import createCurry from './_createCurry.js';
4import createHybrid from './_createHybrid.js';
5import createPartial from './_createPartial.js';
6import getData from './_getData.js';
7import mergeData from './_mergeData.js';
8import setData from './_setData.js';
9import setWrapToString from './_setWrapToString.js';
10import toInteger from './toInteger.js';
11
12/** Error message constants. */
13var FUNC_ERROR_TEXT = 'Expected a function';
14
15/** Used to compose bitmasks for function metadata. */
16var 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. */
24var 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 */
51function 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
106export default createWrap;