116 lines
2.2 KiB
JavaScript
116 lines
2.2 KiB
JavaScript
/*!
|
|
* group-array <https://github.com/doowb/group-array>
|
|
*
|
|
* Copyright (c) 2015, 2017, Brian Woodward.
|
|
* Released under the MIT License.
|
|
*/
|
|
|
|
'use strict';
|
|
|
|
var split = require('split-string');
|
|
var flatten = require('arr-flatten');
|
|
var union = require('union-value');
|
|
var forOwn = require('for-own');
|
|
var typeOf = require('kind-of');
|
|
var get = require('get-value');
|
|
|
|
function groupFn(arr, props) {
|
|
if (arr == null) {
|
|
return [];
|
|
}
|
|
|
|
if (!Array.isArray(arr)) {
|
|
throw new TypeError('group-array expects an array.');
|
|
}
|
|
|
|
if (arguments.length === 1) {
|
|
return arr;
|
|
}
|
|
|
|
var args = flatten([].slice.call(arguments, 1));
|
|
var groups = groupBy(arr, args[0]);
|
|
|
|
for (var i = 1; i < args.length; i++) {
|
|
toGroup(groups, args[i]);
|
|
}
|
|
return groups;
|
|
}
|
|
|
|
function groupBy(arr, prop, key) {
|
|
var groups = {};
|
|
|
|
for (var i = 0; i < arr.length; i++) {
|
|
var obj = arr[i];
|
|
var val;
|
|
|
|
// allow a function to modify the object
|
|
// and/or return a val to use
|
|
if (typeof prop === 'function') {
|
|
val = prop.call(groups, obj, key);
|
|
} else {
|
|
val = get(obj, prop);
|
|
}
|
|
|
|
switch (typeOf(val)) {
|
|
case 'undefined':
|
|
break;
|
|
case 'string':
|
|
case 'number':
|
|
case 'boolean':
|
|
union(groups, escape(String(val)), obj);
|
|
break;
|
|
case 'object':
|
|
case 'array':
|
|
eachValue(groups, obj, val);
|
|
break;
|
|
case 'function':
|
|
throw new Error('invalid argument type: ' + key);
|
|
}
|
|
}
|
|
return groups;
|
|
}
|
|
|
|
function eachValue(groups, obj, val) {
|
|
if (Array.isArray(val)) {
|
|
val.forEach(function(key) {
|
|
union(groups, escape(key), obj);
|
|
});
|
|
} else {
|
|
forOwn(val, function(v, key) {
|
|
union(groups, escape(key), obj);
|
|
});
|
|
}
|
|
}
|
|
|
|
function toGroup(groups, prop) {
|
|
forOwn(groups, function(val, key) {
|
|
if (!Array.isArray(val)) {
|
|
groups[key] = toGroup(val, prop, key);
|
|
} else {
|
|
groups[key] = groupBy(val, prop, key);
|
|
}
|
|
});
|
|
return groups;
|
|
}
|
|
|
|
function escape(str) {
|
|
var opts = {
|
|
strict: false,
|
|
keepEscaping: true,
|
|
keepDoubleQuotes: true,
|
|
keepSingleQuotes: true
|
|
};
|
|
|
|
try {
|
|
return split(str, opts).join('\\.');
|
|
} catch (err) {
|
|
return str;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Expose `groupArray`
|
|
*/
|
|
|
|
module.exports = groupFn;
|