fix: memory leak lowlight dep

janhq · Nov 26, 2024 · bf9bdde · bf9bdde
1 parent b046ee3
commit bf9bdde
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/web/package.json b/web/package.json
@@ -38,7 +38,7 @@
     "react-icons": "^4.12.0",
     "react-markdown": "^9.0.1",
     "react-toastify": "^9.1.3",
-    "rehype-highlight": "^7.0.1",
+    "rehype-highlight": "^6.0.0",
     "rehype-highlight-code-lines": "^1.0.4",
     "rehype-katex": "^7.0.1",
     "rehype-raw": "^7.0.0",

diff --git a/web/screens/Thread/ThreadCenterPanel/SimpleTextMessage/index.tsx b/web/screens/Thread/ThreadCenterPanel/SimpleTextMessage/index.tsx
@@ -95,6 +95,7 @@ const SimpleTextMessage: React.FC<ThreadMessage> = (props) => {
     // Join the lines with newline characters for proper formatting
     return codeLines.join('\n')
   }
+
   function wrapCodeBlocksWithoutVisit() {
     return (tree: { children: any[] }) => {
       tree.children = tree.children.map((node) => {
@@ -394,7 +395,12 @@ const SimpleTextMessage: React.FC<ThreadMessage> = (props) => {
                 rehypePlugins={[
                   [rehypeKatex, { throwOnError: false }],
                   rehypeRaw,
-                  rehypeHighlight,
+                  [
+                    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+                    // @ts-ignore
+                    rehypeHighlight,
+                    { subset: false, plainText: ['txt', 'text'] },
+                  ],
                   [rehypeHighlightCodeLines, { showLineNumbers: true }],
                   wrapCodeBlocksWithoutVisit,
                 ]}