sjs/batteries
1
0
Fork 0
mirror of https://github.com/samsonjs/batteries.git synced 2026-03-25 09:15:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make flatten work on objects and arrays

Browse source
This commit is contained in:
Sami Samhuri 2011-11-05 16:02:36 -07:00
parent 92912c0678
commit 59e6566d51
1 changed files with 11 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lib/array.js
View file

@ -35,7 +35,6 @@ function at(a, i) {
return a[a.length + i];
}
// TODO make this work for non-array objects
function compact(a) {
var b = []
, i = a.length
@ -50,10 +49,17 @@ function first(a) { return a[0]; }
// Based on underscore.js's flatten
function flatten(a) {
return a.reduce(function(initial, elem) {
if (elem && elem.flatten) initial = initial.concat(elem.flatten());
else initial.push(elem);
return initial;
return a.reduce(function(flat, val) {
if (val && typeof val.flatten === 'function') {
flat = flat.concat(val.flatten());
}
else if (Array.isArray(val)) {
flat = flat.concat(flatten(val));
}
else {
flat.push(val);
}
return flat;
});
}