swimlane/ngx-ui

/* eslint-disable security/detect-object-injection */
import { getType } from '../../utils/get-type/get-type.util';

import { TreeNode } from '../../components/tree/tree-node.model';

export const jsonToTree = (value: any, label?: string): TreeNode => {
  const type = getType(value);

  let children: any[];
  let expandable = false;

  if (typeof label === 'undefined') {
    label = label || type;
  }

  switch (type) {
    case 'object':
      children = Object.keys(value).map(key => jsonToTree(value[key], key));
      expandable = children.length > 0;

      return {
        label,
        expandable,
        expanded: true,
        model: { type, value },
        children
      };
    case 'array':
      children = value.map(jsonToTree);
      expandable = children.length > 0;

      return {
        label,
        expandable,
        expanded: true,
        model: { type, value },
        children
      };
    default:
      return {
        label,
        expandable,
        model: { type, value }
      };
  }
};