feat: add rehypeRemoveH1 and rehypeMention, remove rehypeRewrite

package.json

@@ -130,7 +130,6 @@
     "rehype-katex": "7.0.0",
     "rehype-prism-plus": "2.0.0",
     "rehype-raw": "7.0.0",
-    "rehype-rewrite": "4.0.2",
     "rehype-sanitize": "6.0.0",
     "rehype-slug": "6.0.0",
     "remark-breaks": "4.0.0",

src/markdown/index.ts

@@ -17,7 +17,6 @@ import rehypeInferDescriptionMeta from "rehype-infer-description-meta"
 import rehypeKatex from "rehype-katex"
 import rehypePrismGenerator from "rehype-prism-plus/generator"
 import rehypeRaw, { Options as RehypeRawOptions } from "rehype-raw"
-import rehypeRewrite from "rehype-rewrite"
 import rehypeSanitize from "rehype-sanitize"
 import rehypeSlug from "rehype-slug"
 import remarkBreaks from "remark-breaks"
@@ -48,6 +47,8 @@ import {
 } from "./rehype-custom-wrapper"
 import { rehypeEmbed } from "./rehype-embed"
 import { rehypeIpfs } from "./rehype-ipfs"
+import { rehypeMention } from "./rehype-mention"
+import { rehypeRemoveH1 } from "./rehype-remove-h1"
 import { rehypeTable } from "./rehype-table"
 import { rehypeWrapCode } from "./rehype-wrap-code"
 import { rehypeExternalLink } from "./rehyper-external-link"
@@ -129,54 +130,15 @@ export const renderPageContent = (content: string, strictMode?: boolean) => {
       .use(rehypeEmbed, {
         transformers,
       })
+      .use(rehypeRemoveH1)
       .use(rehypePrism, {
         ignoreMissing: true,
         showLineNumbers: true,
       })
       .use(rehypeKatex, {
         strict: false,
       })
-      .use(rehypeRewrite, {
-        selector: "p, li, h1",
-        rewrite: (node: any) => {
-          if (node.tagName === "h1") {
-            node.tagName = "h2"
-            return
-          }
-          if (node.children) {
-            node.children = node.children.flatMap((child: any) => {
-              if (child.type === "text") {
-                const mentionRegex = /(@[\w-]+)/g
-                if (mentionRegex.test(child.value)) {
-                  const parts = child.value.split(mentionRegex)
-                  return parts.map((part: string) => {
-                    if (part.startsWith("@")) {
-                      return {
-                        type: "element",
-                        tagName: "mention",
-                        children: [{ type: "text", value: part }],
-                      }
-                    } else {
-                      return {
-                        type: "text",
-                        value: part,
-                      }
-                    }
-                  })
-                } else {
-                  return child
-                }
-              } else {
-                return child
-              }
-            })
-          }
-        },
-      })
-      // TODO
-      .use(rehypeRaw, {
-        passThrough: allowedCustomWrappers,
-      } as RehypeRawOptions)
+      .use(rehypeMention)
 
     // markdown abstract syntax tree
     mdastTree = pipeline.parse(file)

src/markdown/rehype-mention.ts

@@ -0,0 +1,42 @@
+import type { Root } from "hast"
+import { Plugin } from "unified"
+import { visit } from "unist-util-visit"
+
+export const rehypeMention: Plugin<Array<void>, Root> = () => {
+  return (tree: Root) => {
+    visit(tree, (node, i, parent) => {
+      if (node.type === "element") {
+        if (node.tagName === "p" || node.tagName === "li") {
+          if (node.children) {
+            node.children = node.children.flatMap((child: any) => {
+              if (child.type === "text") {
+                const mentionRegex = /(@[\w-]+)/g
+                if (mentionRegex.test(child.value)) {
+                  const parts = child.value.split(mentionRegex)
+                  return parts.map((part: string) => {
+                    if (part.startsWith("@")) {
+                      return {
+                        type: "element",
+                        tagName: "mention",
+                        children: [{ type: "text", value: part }],
+                      }
+                    } else {
+                      return {
+                        type: "text",
+                        value: part,
+                      }
+                    }
+                  })
+                } else {
+                  return child
+                }
+              } else {
+                return child
+              }
+            })
+          }
+        }
+      }
+    })
+  }
+}

src/markdown/rehype-remove-h1.ts

@@ -0,0 +1,15 @@
+import type { Root } from "hast"
+import { Plugin } from "unified"
+import { visit } from "unist-util-visit"
+
+export const rehypeRemoveH1: Plugin<Array<void>, Root> = () => {
+  return (tree: Root) => {
+    visit(tree, (node, i, parent) => {
+      if (node.type === "element") {
+        if (node.tagName === "h1") {
+          node.tagName = "h2"
+        }
+      }
+    })
+  }
+}