From 347e10e21572d12f3703b804d5b4bdf986c4534c Mon Sep 17 00:00:00 2001
From: DIYgod <i@diygod.me>
Date: Mon, 24 Apr 2023 12:59:06 +0100
Subject: [PATCH] feat: enable more HTML restrictions in comments

---
 src/components/common/CommentItem.tsx | 1 +
 src/components/common/PageContent.tsx | 6 ++++--
 src/markdown/index.ts                 | 3 ++-
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/components/common/CommentItem.tsx b/src/components/common/CommentItem.tsx
index 2d1bcb28..08d7210e 100644
--- a/src/components/common/CommentItem.tsx
+++ b/src/components/common/CommentItem.tsx
@@ -95,6 +95,7 @@ export const CommentItem: React.FC<{
           </div>
           <PageContent
             content={comment.metadata?.content?.content}
+            isComment={true}
           ></PageContent>
           <div className="mt-1 flex items-center">
             <Reactions
diff --git a/src/components/common/PageContent.tsx b/src/components/common/PageContent.tsx
index 048d6f50..ceba87d0 100644
--- a/src/components/common/PageContent.tsx
+++ b/src/components/common/PageContent.tsx
@@ -14,6 +14,7 @@ export const PageContent: React.FC<{
   onScroll?: (scrollTop: number) => void
   onMouseEnter?: () => void
   parsedContent?: ReturnType<typeof renderPageContent>
+  isComment?: boolean
 }> = ({
   className,
   content,
@@ -22,6 +23,7 @@ export const PageContent: React.FC<{
   onScroll,
   onMouseEnter,
   parsedContent,
+  isComment,
 }) => {
   useCodeCopy()
 
@@ -29,12 +31,12 @@ export const PageContent: React.FC<{
     if (parsedContent) {
       return parsedContent
     } else if (content) {
-      const result = renderPageContent(content)
+      const result = renderPageContent(content, false, isComment)
       return result
     } else {
       return null
     }
-  }, [content, parsedContent])
+  }, [content, isComment, parsedContent])
 
   useEffect(() => {
     const hashChangeHandler = () => {
diff --git a/src/markdown/index.ts b/src/markdown/index.ts
index 166c05e5..6f528caf 100644
--- a/src/markdown/index.ts
+++ b/src/markdown/index.ts
@@ -75,6 +75,7 @@ const rehypePrism = rehypePrismGenerator(refractor)
 export const renderPageContent = (
   content: string,
   html?: boolean,
+  simple?: boolean,
 ): Rendered => {
   const env: MarkdownEnv = {
     excerpt: "",
@@ -162,7 +163,7 @@ export const renderPageContent = (
           ]
         },
       })
-      .use(rehypeSanitize, sanitizeScheme)
+      .use(rehypeSanitize, simple ? undefined : sanitizeScheme)
       .use(rehypeTable)
       .use(rehypeExternalLink)
       .use(rehypeWrapCode)