feat: pop the section stack based on heading difference

Author: pngwn

dist/action.js

@@ -19436,7 +19436,9 @@ function linkify_headings() {
 					prev_section[prev_section.length - 1].sections || []
 				);
 			} else if (node.depth < data.prev_level) {
-				data.section_stack.pop();
+				for (let i = 0; i < data.prev_level - node.depth; i++) {
+					data.section_stack.pop();
+				}
 			}
 
 			data.section_stack[data.section_stack.length - 1].push({

dist/cli.js

@@ -18199,7 +18199,9 @@ function linkify_headings() {
 					prev_section[prev_section.length - 1].sections || []
 				);
 			} else if (node.depth < data.prev_level) {
-				data.section_stack.pop();
+				for (let i = 0; i < data.prev_level - node.depth; i++) {
+					data.section_stack.pop();
+				}
 			}
 
 			data.section_stack[data.section_stack.length - 1].push({

src/format/headings.test.ts

@@ -221,7 +221,9 @@ _headings(
 
 #### one
 
-#### two
+##### two
+
+### style
 `,
 			data: {
 				dir: "docs",
@@ -242,8 +244,31 @@ _headings(
 			output.contents,
 			`<h3><span id="component-format-script" class="offset-anchor"></span><a href="docs#component-format-script" class="anchor" aria-hidden></a>script</h3>
 <h4><span id="component-format-script-one" class="offset-anchor"></span><a href="docs#component-format-script-one" class="anchor" aria-hidden></a>one</h4>
-<h4><span id="component-format-script-two" class="offset-anchor"></span><a href="docs#component-format-script-two" class="anchor" aria-hidden></a>two</h4>`
+<h5><span id="component-format-script-one-two" class="offset-anchor" data-scrollignore></span><a href="docs#component-format-script-one-two" class="anchor" aria-hidden></a>two</h5>
+<h3><span id="component-format-style" class="offset-anchor"></span><a href="docs#component-format-style" class="anchor" aria-hidden></a>style</h3>`
 		);
+
+		//@ts-ignore
+		assert.equal(output.data.sections, [
+			{
+				slug: "component-format-script",
+				title: "script",
+				sections: [
+					{
+						slug: "component-format-script-one",
+						title: "one",
+						sections: [
+							{
+								slug: "component-format-script-one-two",
+								title: "two",
+								sections: [],
+							},
+						],
+					},
+				],
+			},
+			{ slug: "component-format-style", title: "style", sections: [] },
+		]);
 	}
 );
 

src/format/headings.ts

@@ -97,7 +97,6 @@ export function linkify_headings(): Transformer {
 			}
 
 			data.slugs.push(slug);
-			console.log(data.slugs);
 
 			// We keep a 'section_stack' to keep track of the section structure
 			if (node.depth > data.prev_level) {
@@ -108,7 +107,9 @@ export function linkify_headings(): Transformer {
 					prev_section[prev_section.length - 1].sections || []
 				);
 			} else if (node.depth < data.prev_level) {
-				data.section_stack.pop();
+				for (let i = 0; i < data.prev_level - node.depth; i++) {
+					data.section_stack.pop();
+				}
 			}
 
 			data.section_stack[data.section_stack.length - 1].push({