refactor: uploadrundialog

1 files changed, 4 insertions, 0 deletions

diff --git a/frontend/src/hooks/UseMessage.tsx b/frontend/src/hooks/UseMessage.tsx
index 249a3bf..ebc4276 100644
--- a/frontend/src/hooks/UseMessage.tsx
+++ b/frontend/src/hooks/UseMessage.tsx
@@ -3,6 +3,7 @@ import MessageDialog from "../components/MessageDialog";
 
 const useMessage = () => {
     const [isOpen, setIsOpen] = useState(false);
+        const [resolvePromise, setResolvePromise] = useState<((value: boolean) => void) | null>(null);
 
     const [title, setTitle] = useState<string>("");
     const [subtitle, setSubtitle] = useState<string>("");
@@ -11,6 +12,9 @@ const useMessage = () => {
         setIsOpen(true);
         setTitle(title);
         setSubtitle(subtitle);
+                return new Promise((resolve) => {
+                        setResolvePromise(() => resolve);
+                });
     };
 
     const handleClose = () => {