1 | var LazyWrapper = require('./_LazyWrapper'),
|
---|
2 | LodashWrapper = require('./_LodashWrapper'),
|
---|
3 | baseAt = require('./_baseAt'),
|
---|
4 | flatRest = require('./_flatRest'),
|
---|
5 | isIndex = require('./_isIndex'),
|
---|
6 | thru = require('./thru');
|
---|
7 |
|
---|
8 | /**
|
---|
9 | * This method is the wrapper version of `_.at`.
|
---|
10 | *
|
---|
11 | * @name at
|
---|
12 | * @memberOf _
|
---|
13 | * @since 1.0.0
|
---|
14 | * @category Seq
|
---|
15 | * @param {...(string|string[])} [paths] The property paths to pick.
|
---|
16 | * @returns {Object} Returns the new `lodash` wrapper instance.
|
---|
17 | * @example
|
---|
18 | *
|
---|
19 | * var object = { 'a': [{ 'b': { 'c': 3 } }, 4] };
|
---|
20 | *
|
---|
21 | * _(object).at(['a[0].b.c', 'a[1]']).value();
|
---|
22 | * // => [3, 4]
|
---|
23 | */
|
---|
24 | var wrapperAt = flatRest(function(paths) {
|
---|
25 | var length = paths.length,
|
---|
26 | start = length ? paths[0] : 0,
|
---|
27 | value = this.__wrapped__,
|
---|
28 | interceptor = function(object) { return baseAt(object, paths); };
|
---|
29 |
|
---|
30 | if (length > 1 || this.__actions__.length ||
|
---|
31 | !(value instanceof LazyWrapper) || !isIndex(start)) {
|
---|
32 | return this.thru(interceptor);
|
---|
33 | }
|
---|
34 | value = value.slice(start, +start + (length ? 1 : 0));
|
---|
35 | value.__actions__.push({
|
---|
36 | 'func': thru,
|
---|
37 | 'args': [interceptor],
|
---|
38 | 'thisArg': undefined
|
---|
39 | });
|
---|
40 | return new LodashWrapper(value, this.__chain__).thru(function(array) {
|
---|
41 | if (length && !array.length) {
|
---|
42 | array.push(undefined);
|
---|
43 | }
|
---|
44 | return array;
|
---|
45 | });
|
---|
46 | });
|
---|
47 |
|
---|
48 | module.exports = wrapperAt;
|
---|