UNPKG

@osmos/arraytotree

Version:

transform flat array to tree structure

74 lines (73 loc) 2.29 kB
"use strict"; 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