@osmos/arraytotree
Version:
transform flat array to tree structure
74 lines (73 loc) • 2.29 kB
JavaScript
;
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
for (var name in all)
__defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
if (from && typeof from === "object" || typeof from === "function") {
for (let key of __getOwnPropNames(from))
if (!__hasOwnProp.call(to, key) && key !== except)
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
}
return to;
};
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
// src/index.ts
var src_exports = {};
__export(src_exports, {
arrayToTree: () => arrayToTree
});
module.exports = __toCommonJS(src_exports);
function arrayToTree(nodes, option) {
const { idKey, parentKey, childrenKey, sortKey } = option;
const roots = [];
if (sortKey) {
nodes.sort((a, b) => a[sortKey] - b[sortKey]);
}
for (const node of nodes) {
if (!node[childrenKey]) {
node[childrenKey] = [];
}
}
for (const node of nodes) {
let parentId = parentKey.includes(".") ? parentKey.split(".").reduce((obj, key) => obj && obj[key], node) : node[parentKey];
if (parentId && typeof parentId === "object") {
parentId = parentId[idKey];
}
if (parentId === null || parentId === void 0) {
roots.push(node);
} else {
const parent = nodes.find((item) => item[idKey] === parentId);
if (parent) {
parent[childrenKey].push(node);
} else {
roots.push(node);
}
}
}
function findNodeById(id, nodes2) {
if (nodes2) {
for (const node of nodes2) {
if (node[idKey] === id) {
return node;
}
const foundNode = findNodeById(id, node[childrenKey]);
if (foundNode) {
return foundNode;
}
}
}
return null;
}
roots.findNodeById = (id) => findNodeById(id, roots);
return roots;
}
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
arrayToTree
});
//# sourceMappingURL=index.js.map