Skip to content

Commit 3baf262

Browse files
committed
fix: filter real markdown headings
Signed-off-by: Innei <[email protected]>
1 parent 5915e47 commit 3baf262

File tree

3 files changed

+13
-4
lines changed

3 files changed

+13
-4
lines changed

src/components/modules/toc/TocAside.tsx

+7-3
Original file line numberDiff line numberDiff line change
@@ -65,9 +65,13 @@ export const TocAside = forwardRef<
6565
if (!$article) {
6666
return []
6767
}
68-
return [
69-
...$article.querySelectorAll('h1,h2,h3,h4,h5,h6'),
70-
] as HTMLHeadingElement[]
68+
return [...$article.querySelectorAll('h1,h2,h3,h4,h5,h6')].filter(
69+
($heading) => {
70+
if (($heading as HTMLElement).dataset['markdownHeading'] === 'true')
71+
return true
72+
return false
73+
},
74+
) as HTMLHeadingElement[]
7175
}, [$article, updated])
7276

7377
useEffect(() => {

src/components/modules/toc/TocFAB.tsx

+4-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,10 @@ export const TocFAB = () => {
2222
...$mainMarkdownRender.querySelectorAll('h1,h2,h3,h4,h5,h6'),
2323
] as HTMLHeadingElement[]
2424

25-
return $headings
25+
return $headings.filter(($heading) => {
26+
if ($heading.dataset['markdownHeading'] === 'true') return true
27+
return false
28+
})
2629
}, [])
2730
const presentToc = useCallback(() => {
2831
const dispose = present({

src/components/ui/markdown/renderers/heading.tsx

+2
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,8 @@ export const MHeader = (props: HeadingProps) => {
2121
{
2222
id: nextId,
2323
className: 'group flex items-center',
24+
25+
'data-markdown-heading': true,
2426
} as any,
2527
null,
2628
<>

0 commit comments

Comments
 (0)