31 lines
703 B
JavaScript
31 lines
703 B
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
var variadic = require('../helpers/variadic');
|
||
|
|
|
||
|
|
module.exports = function only() {
|
||
|
|
var _this = this;
|
||
|
|
|
||
|
|
for (var _len = arguments.length, args = Array(_len), _key = 0; _key < _len; _key++) {
|
||
|
|
args[_key] = arguments[_key];
|
||
|
|
}
|
||
|
|
|
||
|
|
var properties = variadic(args);
|
||
|
|
|
||
|
|
if (Array.isArray(this.items)) {
|
||
|
|
var _collection = this.items.filter(function (item) {
|
||
|
|
return properties.indexOf(item) !== -1;
|
||
|
|
});
|
||
|
|
|
||
|
|
return new this.constructor(_collection);
|
||
|
|
}
|
||
|
|
|
||
|
|
var collection = {};
|
||
|
|
|
||
|
|
Object.keys(this.items).forEach(function (prop) {
|
||
|
|
if (properties.indexOf(prop) !== -1) {
|
||
|
|
collection[prop] = _this.items[prop];
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
return new this.constructor(collection);
|
||
|
|
};
|