Как сформировать содержание из массива заголовков?

Question

[Сергей]

Есть объёмная статья. Задача - собрать все её заголовки и сформировать из них содержание, сохранив при этом иерархическую структуру. Например, у меня будет следующий массив заголовков:
var arr = [h2, h3, h3, h2, h2, h2, h3, h4];
Соответственно должен получиться такой список:

<ul>
  <li>h2
    <ul>
      <li>h3</li>
      <li>h3</li>
    </ul>
  </li>
  <li>h2</li>
  <li>h2</li>
  <li>h2
    <ul>
      <li>h3
        <ul>
          <li>h4</li>
        </ul>
      </li>
    </ul>
  </li>
</ul>

Буду рад любым подсказкам и идеям. Спасибо.

Answer 1

[h88p]

var arr2 = ['h2', 'h3', 'h3', 'h2', 'h2', 'h2', 'h3', 'h4'];

function setList(arr) {
	if (arr.length === 0) return '<ul>\n</ul>';
	let fullList = '<ul>\n';
	let ends = [];
	arr.forEach((w,e) => {
		if((arr[e+1]) && (parseInt(w[1]) < parseInt(arr[e+1][1]))){
			fullList += `<li>${w}\n<ul>\n`;
			ends.push('</ul>\n</li>');
		} else if ((arr[e+1]) && (parseInt(w[1]) === parseInt(arr[e+1][1]))) {
			fullList += `<li>${w}</li>\n`;
		} else {
			fullList +=`<li>${w}</li>\n` +  ends.join('\n') + '\n';
			ends = [];
		}
	}) 
	fullList = fullList + '</ul>';
	return fullList;
}

Answer 2

[0xD34F]

function createList(data) {
  const list = document.createElement('ul');
  const stack = [];
  let [ ul, level ] = [ list, +/\d+/.exec(data[0]) ];

  for (const n of data) {
    const currLevel = +/\d+/.exec(n);
    if (currLevel > level) {
      stack.push([ ul, level ]);
      [ ul, level ] = [ document.createElement('ul'), currLevel ];
      stack[stack.length - 1][0].lastChild.append(ul);
    } else {
      for (; currLevel !== level; [ ul, level ] = stack.pop()) ;
    }

    ul.append(document.createElement('li'));
    ul.lastChild.textContent = n;
  }

  return list;
}