fix: filter real markdown headings

Signed-off-by: Innei <i@innei.in>
Innei · Jan 30, 2024 · 3baf262 · 3baf262
1 parent 5915e47
commit 3baf262
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/src/components/modules/toc/TocAside.tsx b/src/components/modules/toc/TocAside.tsx
@@ -65,9 +65,13 @@ export const TocAside = forwardRef<
       if (!$article) {
         return []
       }
-      return [
-        ...$article.querySelectorAll('h1,h2,h3,h4,h5,h6'),
-      ] as HTMLHeadingElement[]
+      return [...$article.querySelectorAll('h1,h2,h3,h4,h5,h6')].filter(
+        ($heading) => {
+          if (($heading as HTMLElement).dataset['markdownHeading'] === 'true')
+            return true
+          return false
+        },
+      ) as HTMLHeadingElement[]
     }, [$article, updated])
 
     useEffect(() => {

diff --git a/src/components/modules/toc/TocFAB.tsx b/src/components/modules/toc/TocFAB.tsx
@@ -22,7 +22,10 @@ export const TocFAB = () => {
       ...$mainMarkdownRender.querySelectorAll('h1,h2,h3,h4,h5,h6'),
     ] as HTMLHeadingElement[]
 
-    return $headings
+    return $headings.filter(($heading) => {
+      if ($heading.dataset['markdownHeading'] === 'true') return true
+      return false
+    })
   }, [])
   const presentToc = useCallback(() => {
     const dispose = present({

diff --git a/src/components/ui/markdown/renderers/heading.tsx b/src/components/ui/markdown/renderers/heading.tsx
@@ -21,6 +21,8 @@ export const MHeader = (props: HeadingProps) => {
     {
       id: nextId,
       className: 'group flex items-center',
+
+      'data-markdown-heading': true,
     } as any,
     null,
     <>