fix: cannot close notification manually (langgenius#11490)

taowang1993 · Dec 9, 2024 · 2866383 · 2866383
1 parent 00ac7ed
commit 2866383
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/web/app/components/base/toast/index.tsx b/web/app/components/base/toast/index.tsx
@@ -123,11 +123,25 @@ Toast.notify = ({
     const holder = document.createElement('div')
     const root = createRoot(holder)
 
-    root.render(<Toast type={type} size={size} message={message} duration={duration} className={className} />)
+    root.render(
+      <ToastContext.Provider value={{
+        notify: () => {},
+        close: () => {
+          if (holder) {
+            root.unmount()
+            holder.remove()
+          }
+        },
+      }}>
+        <Toast type={type} size={size} message={message} duration={duration} className={className} />
+      </ToastContext.Provider>,
+    )
     document.body.appendChild(holder)
     setTimeout(() => {
-      if (holder)
+      if (holder) {
+        root.unmount()
         holder.remove()
+      }
     }, duration || defaultDuring)
   }
 }