jcubic / 10xDevs / 18569861134

import { parse, type MarkdownNode, type HeaderNode } from './index';

export function extractHeaders(content: string): Header[] {

  if (!content) return [];

  try {

    const nodes: MarkdownNode[] = parse(content);

    const headers: Header[] = [];

    nodes.forEach((node, index) => {

      if (node.type === 'header') {

        const headerNode = node as HeaderNode;

        headers.push({

          id: `header-${index}-${headerNode.level}`,

          text: headerNode.content.replace(/^#+\s*/, '').trim(),

          level: headerNode.level as 1 | 2 | 3 | 4 | 5 | 6,

          line: headerNode.loc.start.line

        });

      }

    });

    return headers;

  } catch (error) {

    console.error('Failed to parse markdown content:', error);

    return [];

  }

}

export function buildHeaderTree(headers: Header[]): Header[] {

  if (headers.length === 0) return [];

  const result: Header[] = [];

  const stack: Header[] = [];

  headers.forEach((header) => {

    while (stack.length > 0 && stack[stack.length - 1].level >= header.level) {

      stack.pop();

    }

    const headerCopy: Header = {

      id: header.id,

      text: header.text,

      level: header.level,

      line: header.line

    };

    if (stack.length === 0) {

      result.push(headerCopy);

    } else {

      const parent = stack[stack.length - 1];

      if (!parent.children) {

        parent.children = [];

      }

      parent.children.push(headerCopy);

    }

    stack.push(headerCopy);

  });

  return result;

}