Как правильно собрать вложенный объект без указания родителя?

Question

[dake1231]

Здравствуйте! Есть дерево в виде массива с указанием уровня вложенности, нужно из него сделать массив с вложенными объектами.
На вход имеется вот такой массив:

[
{id:1, title: 'test1', level:0},
{id:2, title: 'test2', level:1},
{id:3, title: 'test3', level:2},
{id:4, title: 'test4', level:1},
{id:5, title: 'test5', level:0},
]

на выходе должны получить вот такой:

[
{
   id:1,
   title: 'test1',
   nodes: [
      {
          id:2,
          title: 'test2',
         nodes: [
             id: 3,
             title: 'test3'
             nodes: []
        ]
      },
      {
          id:4,
          title: 'test4',
          nodes: []
      }
   ],
   {
       id:5,
       title: 'test5',
       nodes: []
   }
}
]

Comments

[V Sh.]

Без указания родителя не получится хотя бы потому что мы не будем знать к кому привязывать объекты, уровнем ниже.

Answer 1

[0xD34F]

function createTree(data, levelKey, childrenKey) {
  const tree = [];
  const rootLevel = data.length && data[0][levelKey];

  for (const n of data) {
    let arr = tree;

    for (let level = rootLevel; n[levelKey] > level++;) {
      arr = arr[arr.length - 1][childrenKey];
    }

    arr.push(Object.assign({ [childrenKey]: [] }, n));
  }

  return tree;
}

const tree = createTree(arr, 'level', 'nodes');

или

function createTree(
  data,
  {
    levelKey = 'level',
    childrenKey = 'children',
  } = {}
) {
  const rootLevel = data.length && data[0][levelKey];

  return data.reduce((acc, {...n}) => (
    acc[n[levelKey] + 1] = n[childrenKey] = [],
    acc[n[levelKey]].push(n),
    acc
  ), { [rootLevel]: [] })[rootLevel];
}

const tree = createTree(arr, { childrenKey: 'nodes' });

Comments

[dake1231]

Сколько минут у вас это заняло?

[0xD34F]

dake1231, четыре или пять.

[dake1231]

0xD34F, пипец) Спасибо

[dake1231]

Сейчас встроил ваш код к себе и тут появился баг, если level опускается до 0 и заного становится один то элемент уходит к вышестоящему элементу например:

[
  {id:1, level:0},
  {id:2, level:1},
  {id:3, level:0},
  {id:4, level:1},
]

то на выходе получается массив

{id:1, level:0, node: [
      {id:2, level:1},
      {id:4, level:1}
   ]},
   {id:3, level:3}

Как можно исправить?

[0xD34F]

А, кажется понял, что вам нужно было. Ответ обновил.

[dake1231]

0xD34F, Возьмите меня в ученики сенсей)