refactor: update rehypeCollectHeadings to accept parameters and improve heading collection

Author: LadyBluenotes

src/utils/markdown/plugins/collectHeadings.ts

@@ -10,8 +10,8 @@ export type MarkdownHeading = {
   level: number
 }
 
-export const rehypeCollectHeadings = (headings: MarkdownHeading[]) => {
-  return (tree: Root, file) => {
+export async function rehypeCollectHeadings (tree, file, headings: MarkdownHeading[]) {
+  return (tree) => {
     visit(tree, 'element', (node) => {
       if (!isHeading(node)) {
         return
@@ -29,6 +29,8 @@ export const rehypeCollectHeadings = (headings: MarkdownHeading[]) => {
       })
     })
 
-    file.data.headings = headings
+    if (file) {
+      file.data.headings = headings
+    }
   }
 }

src/utils/markdown/processor.ts

@@ -69,7 +69,7 @@ export function renderMarkdown(content): MarkdownRenderResult {
         className: ['anchor-heading'],
       },
     })
-    .use(rehypeCollectHeadings(headings))
+    .use((node, file) => rehypeCollectHeadings(node, file, headings))
 
   const file = processor.use(rehypeStringify).processSync(content)