@uiw/react-native
Version:
UIW for React Native
28 lines (26 loc) • 655 B
text/typescript
function arrayTreeFilter<T>(
data: T[],
filterFn: (item: T, level: number) => boolean,
options?: {
childrenKeyName?: string;
},
) {
options = options || {};
options.childrenKeyName = options.childrenKeyName || 'children';
var children = data || [];
var result: T[] = [];
var level = 0;
do {
var foundItem: T = children.filter(function (item) {
return filterFn(item, level);
})[0];
if (!foundItem) {
break;
}
result.push(foundItem);
children = (foundItem as any)[options.childrenKeyName] || [];
level += 1;
} while (children.length > 0);
return result;
}
export default arrayTreeFilter;