47 lines
1.6 KiB
TypeScript
47 lines
1.6 KiB
TypeScript
import deepmerge from 'deepmerge';
|
|
|
|
export const flatToTree = (flatArray, options) => {
|
|
options = {
|
|
id: "id",
|
|
parentId: "parentId",
|
|
children: "children",
|
|
...options
|
|
};
|
|
const dictionary = {}; // a hash table mapping to the specific array objects with their ids as key
|
|
const tree = [];
|
|
const children = options.children;
|
|
flatArray.forEach(node => {
|
|
const nodeId = node[options.id];
|
|
const nodeParentId = node[options.parentId];
|
|
// set up current node data in dictionary
|
|
dictionary[nodeId] = {
|
|
[children]: [], // init a children property
|
|
...node, // add other propertys
|
|
...dictionary[nodeId] // children will be replaced if this node already has children property which was set below
|
|
};
|
|
dictionary[nodeParentId] = dictionary[nodeParentId] || { [children]: [] }; // if it's not exist in dictionary, init an object with children property
|
|
dictionary[nodeParentId][children].push(dictionary[nodeId]); // add reference to current node object in parent node object
|
|
});
|
|
// find root nodes
|
|
Object.values(dictionary).forEach(obj => {
|
|
if (typeof obj[options.id] === "undefined") {
|
|
tree.push(...obj[children]);
|
|
}
|
|
});
|
|
return treeData(tree);
|
|
};
|
|
|
|
function treeData(pages: Array<any>) {
|
|
return pages.map(data => {
|
|
return { ...data, children: data.children && data.children.length ? treeData(data.children) : undefined }
|
|
});
|
|
}
|
|
|
|
const overwriteMerge = (destinationArray, sourceArray, options) => sourceArray
|
|
|
|
export function merge(obj1: any, obj2: any) {
|
|
return deepmerge(obj1, obj2, {
|
|
arrayMerge: overwriteMerge,
|
|
});
|
|
}
|