feat: sanitizeHtmlPlugin now can pass options

opensquare-network · Jan 18, 2024 · 652bc87 · 652bc87
1 parent 9662431
commit 652bc87
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/src/HtmlPreviewer.tsx b/src/HtmlPreviewer.tsx
@@ -13,7 +13,7 @@ export function HtmlPreviewer(props: PreviewerProps) {
     maxLines,
   } = props;
 
-  const resolvedPlugins = [...plugins, sanitizeHtmlPlugin(allowedTags)];
+  const resolvedPlugins = [...plugins, sanitizeHtmlPlugin({ allowedTags })];
 
   const ref = useRef<HTMLDivElement>(null);
   const [html, setHtml] = useState(content);

diff --git a/src/plugins/sanitizeHtml.ts b/src/plugins/sanitizeHtml.ts
@@ -1,12 +1,20 @@
-import type { Plugin, PreviewerProps } from "../types";
+import type { Plugin } from "../types";
 import sanitizeHtml from "sanitize-html";
 import { getMentionIdentityUserTargetElementAttrs } from "./renderMentionIdentityUser";
 
-export function sanitizeHtmlPlugin(
-  allowedTags: PreviewerProps["allowedTags"] = sanitizeHtml.defaults.allowedTags.concat(
-    ["img", "iframe", "br", "ins", "del"],
-  ),
-): Plugin {
+export function sanitizeHtmlPlugin(options?: sanitizeHtml.IOptions): Plugin {
+  const {
+    allowedTags = sanitizeHtml.defaults.allowedTags.concat([
+      "img",
+      "iframe",
+      "br",
+      "ins",
+      "del",
+    ]),
+    allowedAttributes = {},
+    ...rest
+  } = options ?? {};
+
   return {
     name: "sanitize-html",
 
@@ -21,7 +29,9 @@ export function sanitizeHtmlPlugin(
           td: ["align"],
           th: ["align"],
           li: ["data-list"],
+          ...allowedAttributes,
         },
+        ...rest,
       });
     },
   };