From 6f3d5762ca55d30a48d5bfb662459337e5a9938a Mon Sep 17 00:00:00 2001
From: ti777777 <a104221087@gmail.com>
Date: Thu, 12 Sep 2024 08:54:33 +0800
Subject: [PATCH] fix: iframe rendering (#3916)

* fix iframe rendering

* fix eslint check
---
 web/src/components/MemoContent/CodeBlock.tsx | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/web/src/components/MemoContent/CodeBlock.tsx b/web/src/components/MemoContent/CodeBlock.tsx
index 4b733015..25d056cc 100644
--- a/web/src/components/MemoContent/CodeBlock.tsx
+++ b/web/src/components/MemoContent/CodeBlock.tsx
@@ -25,7 +25,17 @@ const CodeBlock: React.FC<Props> = ({ language, content }: Props) => {
   // Users can set Markdown code blocks as `__html` to render HTML directly.
   if (formatedLanguage === SpecialLanguage.HTML) {
     const purify = DOMPurify(window);
-    return <div className="w-full overflow-auto !my-2" dangerouslySetInnerHTML={{ __html: purify.sanitize(content) }} />;
+    return (
+      <div
+        className="w-full overflow-auto !my-2"
+        dangerouslySetInnerHTML={{
+          __html: purify.sanitize(content, {
+            ALLOWED_TAGS: ["iframe"],
+            ADD_ATTR: ["allow", "allowfullscreen", "frameborder", "scrolling"],
+          }),
+        }}
+      />
+    );
   } else if (formatedLanguage === SpecialLanguage.MERMAID) {
     return <MermaidBlock content={content} />;
   }