From c3abc069780fab2174f66e8f9edda63589b0ab8a Mon Sep 17 00:00:00 2001
From: Stephen Zhou <38493346+hyoban@users.noreply.github.com>
Date: Sat, 12 Oct 2024 09:53:48 +0800
Subject: [PATCH] fix: do not filter figure, add pre fallback

close #847
---
 apps/renderer/src/lib/parse-html.ts | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/apps/renderer/src/lib/parse-html.ts b/apps/renderer/src/lib/parse-html.ts
index 34cfa6a8d..4b6076a76 100644
--- a/apps/renderer/src/lib/parse-html.ts
+++ b/apps/renderer/src/lib/parse-html.ts
@@ -74,7 +74,7 @@ export const parseHtml = (
       (tag) => tag !== "img" && tag !== "picture",
     )
   } else {
-    rehypeSchema.tagNames = [...rehypeSchema.tagNames!, "video", "style"]
+    rehypeSchema.tagNames = [...rehypeSchema.tagNames!, "video", "style", "figure"]
     rehypeSchema.attributes = {
       ...rehypeSchema.attributes,
       "*": renderInlineStyle
@@ -181,7 +181,7 @@ export const parseHtml = (
                 ? propsChildren.find((i) => i.type === "code")
                 : propsChildren
 
-              if (!children) return null
+              if (!children) return createElement("pre", props, props.children)
 
               if (
                 "type" in children &&
@@ -191,7 +191,7 @@ export const parseHtml = (
                 language = children.props.className.replace("language-", "")
               }
               const code = "props" in children && children.props.children
-              if (!code) return null
+              if (!code) createElement("pre", props, props.children)
 
               try {
                 codeString = extractCodeFromHtml(renderToString(code))
@@ -203,7 +203,7 @@ export const parseHtml = (
               }
             }
 
-            if (!codeString) return null
+            if (!codeString) return createElement("pre", props, props.children)
 
             return createElement(ShikiHighLighter, {
               code: codeString.trimEnd(),