Fix TOC generation

JS-DevTools · Jan 25, 2021 · 6646eed · 6646eed
1 parent 04aef46
commit 6646eed
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/rehype-toc.ts b/src/rehype-toc.ts
@@ -23,18 +23,19 @@ export function toc(this: Processor, opts?: Options): Transformer {
 
     let mainNode: Element | undefined;
     let mainParent: Parent | undefined;
+    let placeholderNode: Element | undefined;
+    let placeholderParent: Parent | undefined;
 
     if (options.placeholder) {
       // Find the <target> element if option is given
-      [mainNode, mainParent] = findPlaceholderNode(root, options.placeholder);
+      [placeholderNode, placeholderParent] = findPlaceholderNode(root, options.placeholder);
 
-      if (!mainNode || !mainParent) { return root; }
-    }
-    else {
-      // Find the <main> or <body> element
-      [mainNode, mainParent] = findMainNode(root);
+      if (!placeholderNode || !placeholderParent) { return root; }
     }
 
+    // Find the <main> or <body> element
+    [mainNode, mainParent] = findMainNode(root);
+
     // Find all heading elements
     let headings = findHeadings(mainNode, options);
 
@@ -46,7 +47,7 @@ export function toc(this: Processor, opts?: Options): Transformer {
 
     if (node) {
       // Add the table of contents to the <main> element
-      insertTOC(node, mainNode, mainParent, { ...options, replace: !!options.placeholder });
+      insertTOC(node, placeholderNode || mainNode, placeholderParent || mainParent, { ...options, replace: !!options.placeholder });
     }
 
     return root;