From 48497c749aa243ab4ab2db18f94a20fa9355c40f Mon Sep 17 00:00:00 2001
From: Siddhant Rai <siddhant.rai.5686@gmail.com>
Date: Mon, 3 Jun 2024 20:36:52 +0530
Subject: [PATCH] fix: dompurify import error

---
 extensions/react-widget/package-lock.json     |  6 ++--
 extensions/react-widget/package.json          |  2 +-
 .../src/components/DocsGPTWidget.tsx          | 33 ++++++++++---------
 3 files changed, 22 insertions(+), 19 deletions(-)

diff --git a/extensions/react-widget/package-lock.json b/extensions/react-widget/package-lock.json
index 5f23125..6b861a0 100644
--- a/extensions/react-widget/package-lock.json
+++ b/extensions/react-widget/package-lock.json
@@ -18,7 +18,7 @@
         "@radix-ui/react-icons": "^1.3.0",
         "class-variance-authority": "^0.7.0",
         "clsx": "^2.1.0",
-        "dompurify": "^3.0.9",
+        "dompurify": "^3.1.5",
         "flow-bin": "^0.229.2",
         "i": "^0.3.7",
         "install": "^0.13.0",
@@ -5744,7 +5744,9 @@
     "node_modules/graceful-fs": {
       "version": "4.2.11",
       "resolved": "https://registry.npmjs.org/graceful-fs/-/graceful-fs-4.2.11.tgz",
-      "integrity": "sha512-RbJ5/jmFcNNCcDV5o9eTnBLJ/HszWV0P73bc+Ff4nS/rJj+YaS6IGyiOL0VoBYX+l1Wrl3k63h/KrH+nhJ0XvQ=="
+      "integrity": "sha512-RbJ5/jmFcNNCcDV5o9eTnBLJ/HszWV0P73bc+Ff4nS/rJj+YaS6IGyiOL0VoBYX+l1Wrl3k63h/KrH+nhJ0XvQ==",
+      "dev": true,
+      "peer": true
     },
     "node_modules/has-flag": {
       "version": "3.0.0",
diff --git a/extensions/react-widget/package.json b/extensions/react-widget/package.json
index 6a6284c..cd8fb6e 100644
--- a/extensions/react-widget/package.json
+++ b/extensions/react-widget/package.json
@@ -47,7 +47,7 @@
     "@radix-ui/react-icons": "^1.3.0",
     "class-variance-authority": "^0.7.0",
     "clsx": "^2.1.0",
-    "dompurify": "^3.0.9",
+    "dompurify": "^3.1.5",
     "flow-bin": "^0.229.2",
     "i": "^0.3.7",
     "install": "^0.13.0",
diff --git a/extensions/react-widget/src/components/DocsGPTWidget.tsx b/extensions/react-widget/src/components/DocsGPTWidget.tsx
index b47e2a4..5b7a070 100644
--- a/extensions/react-widget/src/components/DocsGPTWidget.tsx
+++ b/extensions/react-widget/src/components/DocsGPTWidget.tsx
@@ -1,12 +1,13 @@
 "use client";
-import { Fragment, useEffect, useRef, useState } from 'react'
+import React from 'react'
+import DOMPurify from 'dompurify';
+import snarkdown from '@bpmn-io/snarkdown';
+import styled, { keyframes, createGlobalStyle } from 'styled-components';
 import { PaperPlaneIcon, RocketIcon, ExclamationTriangleIcon, Cross2Icon } from '@radix-ui/react-icons';
-import { MESSAGE_TYPE, Query, Status } from '../types/index';
 import MessageIcon from '../assets/message.svg';
+import { MESSAGE_TYPE, Query, Status } from '../types/index';
 import { fetchAnswerStreaming } from '../requests/streamingApi';
-import styled, { keyframes, createGlobalStyle } from 'styled-components';
-import snarkdown from '@bpmn-io/snarkdown';
-import { sanitize } from 'dompurify';
+
 const GlobalStyles = createGlobalStyle`
 .response pre {
     padding: 8px;
@@ -292,13 +293,13 @@ export const DocsGPTWidget = ({
   heroDescription = 'This chatbot is built with DocsGPT and utilises GenAI, please review important information using sources.'
 }) => {
 
-  const [prompt, setPrompt] = useState('');
-  const [status, setStatus] = useState<Status>('idle');
-  const [queries, setQueries] = useState<Query[]>([])
-  const [conversationId, setConversationId] = useState<string | null>(null)
-  const [open, setOpen] = useState<boolean>(false)
-  const [eventInterrupt, setEventInterrupt] = useState<boolean>(false); //click or scroll by user while autoScrolling
-  const endMessageRef = useRef<HTMLDivElement | null>(null);
+  const [prompt, setPrompt] = React.useState('');
+  const [status, setStatus] = React.useState<Status>('idle');
+  const [queries, setQueries] = React.useState<Query[]>([])
+  const [conversationId, setConversationId] = React.useState<string | null>(null)
+  const [open, setOpen] = React.useState<boolean>(false)
+  const [eventInterrupt, setEventInterrupt] = React.useState<boolean>(false); //click or scroll by user while autoScrolling
+  const endMessageRef = React.useRef<HTMLDivElement | null>(null);
   const handleUserInterrupt = () => {
     (status === 'loading') && setEventInterrupt(true);
   }
@@ -316,7 +317,7 @@ export const DocsGPTWidget = ({
     lastChild && scrollToBottom(lastChild)
   };
 
-  useEffect(() => {
+  React.useEffect(() => {
     !eventInterrupt && scrollToBottom(endMessageRef.current);
   }, [queries.length, queries[queries.length - 1]?.response]);
 
@@ -396,7 +397,7 @@ export const DocsGPTWidget = ({
             {
               queries.length > 0 ? queries?.map((query, index) => {
                 return (
-                  <Fragment key={index}>
+                  <React.Fragment key={index}>
                     {
                       query.prompt && <MessageBubble type='QUESTION'>
                         <Message
@@ -412,7 +413,7 @@ export const DocsGPTWidget = ({
                           type='ANSWER'
                           ref={(index === queries.length - 1) ? endMessageRef : null}
                         >
-                          <div className="response" dangerouslySetInnerHTML={{ __html: sanitize(snarkdown(query.response)) }} />
+                          <div className="response" dangerouslySetInnerHTML={{ __html: DOMPurify.sanitize(snarkdown(query.response)) }} />
                         </Message>
                       </MessageBubble>
                         : <div>
@@ -436,7 +437,7 @@ export const DocsGPTWidget = ({
                           }
                         </div>
                     }
-                  </Fragment>)
+                  </React.Fragment>)
               })
                 : <Hero title={heroTitle} description={heroDescription} />
             }