You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
40 lines
794 B
JavaScript
40 lines
794 B
JavaScript
"use strict";
|
|
|
|
// FIXME: Untested
|
|
|
|
const isIterable = require("is-iterable");
|
|
const forEach = require("../for-each");
|
|
|
|
module.exports = function treeFilterFlat(tree, filter, options = {}) {
|
|
let key = options.key ?? "children";
|
|
let recurseFilteredSubtrees = options.recurseFilteredSubtrees ?? false;
|
|
|
|
let results = [];
|
|
|
|
function step(subtree) {
|
|
let matchesFilter = filter(subtree);
|
|
|
|
if (matchesFilter || recurseFilteredSubtrees) {
|
|
if (isIterable(subtree[key])) {
|
|
for (let item of subtree[key]) {
|
|
step(item);
|
|
}
|
|
} else if (subtree[key] != null && typeof subtree[key] === "object") {
|
|
step(subtree[key]);
|
|
}
|
|
}
|
|
|
|
if (matchesFilter) {
|
|
results.push(subtree);
|
|
}
|
|
}
|
|
|
|
if (isIterable(tree)) {
|
|
forEach(tree, step);
|
|
} else {
|
|
step(tree);
|
|
}
|
|
|
|
return results;
|
|
};
|