Projekt

Obecné

Profil

Stáhnout (3.63 KB) Statistiky
| Větev: | Revize:
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;