javascript 树结构过滤保留原始结构

需求：有一个树结构的JSON，被过滤后，希望结构保留树结构。

var tree = [{text: "Parent 1",nodes: [{text: "Child 1",type: "Child",nodes: [{text: "Grandchild 1"type: "Grandchild"},{text: "Grandchild 2"type: "Grandchild"}]},{text: "Child 2",type: "Child"}]},{text: "Parent 2",type: "Parent"},{text: "Parent 3",type: "Parent"}
];

例子 :

1)如果搜索查询是Parent 1

预期结果 :

 [{text: "Parent 1",nodes: [{text: "Child 1",type: "Child",nodes: [{text: "Grandchild 1"type: "Grandchild"},{text: "Grandchild 2"type: "Grandchild"}]},{text: "Child 2",type: "Child"}]}
]

2)如果搜索查询是Child 1

预期结果 :

[{text: "Parent 1",nodes: [{text: "Child 1",type: "Child",nodes: [{text: "Grandchild 1"type: "Grandchild"},{text: "Grandchild 2"type: "Grandchild"}]}]}
]

如果搜索查询是 Grandchild 2

预期结果 :

[{text: "Parent 1",nodes: [{text: "Child 1",type: "Child",nodes: [{text: "Grandchild 2"type: "Grandchild"}]}]}
]

最佳答案

function filter(array, text) {const getNodes = (result, object) => {if (object.text === text) {result.push(object);return result;}if (Array.isArray(object.nodes)) {const nodes = object.nodes.reduce(getNodes, []);if (nodes.length) result.push({ ...object, nodes });}return result;};return array.reduce(getNodes, []);
}var tree = [{ text: "Parent 1", nodes: [{ text: "Child 1", type: "Child", nodes: [{ text: "Grandchild 1", type: "Grandchild" }, { text: "Grandchild 2", type: "Grandchild" }] }, { text: "Child 2", type: "Child" }] }, { text: "Parent 2", type: "Parent" }, { text: "Parent 3", type: "Parent" }];console.log(filter(tree, 'Parent 1'));
console.log(filter(tree, 'Child 1'));
console.log(filter(tree, 'Grandchild 2'));