refactor: touch ups

2 files changed, 63 insertions, 0 deletions

diff --git a/frontend/src/hooks/UseConfirm.tsx b/frontend/src/hooks/UseConfirm.tsx
new file mode 100644
index 0000000..6de7b10
--- /dev/null
+++ b/frontend/src/hooks/UseConfirm.tsx
@@ -0,0 +1,36 @@
+import React, { useState } from 'react';
+import ConfirmDialog from '../components/ConfirmDialog';
+
+const useConfirm = ( title: string, subtitle: string ) => {
+    const [isOpen, setIsOpen] = useState(false);
+    const [resolvePromise, setResolvePromise] = useState<((value: boolean) => void) | null>(null);
+
+    const confirm = () => {
+        setIsOpen(true);
+        return new Promise((resolve) => {
+            setResolvePromise(() => resolve);
+        });
+    };
+
+    const handleConfirm = () => {
+        setIsOpen(false);
+        if (resolvePromise) {
+            resolvePromise(true);
+        }
+    }
+
+    const handleCancel = () => {
+        setIsOpen(false);
+        if (resolvePromise) {
+            resolvePromise(false);
+        }
+    }
+
+    const ConfirmDialogComponent = isOpen && (
+        <ConfirmDialog title={title} subtitle={subtitle} onConfirm={handleConfirm} onCancel={handleCancel}></ConfirmDialog>
+    );
+
+    return { confirm, ConfirmDialogComponent };
+}
+
+export default useConfirm;
diff --git a/frontend/src/hooks/UseMessage.tsx b/frontend/src/hooks/UseMessage.tsx
new file mode 100644
index 0000000..249a3bf
--- /dev/null
+++ b/frontend/src/hooks/UseMessage.tsx
@@ -0,0 +1,27 @@
+import React, { useState } from 'react';
+import MessageDialog from "../components/MessageDialog";
+
+const useMessage = () => {
+    const [isOpen, setIsOpen] = useState(false);
+
+    const [title, setTitle] = useState<string>("");
+    const [subtitle, setSubtitle] = useState<string>("");
+
+    const message = (title: string, subtitle: string) => {
+        setIsOpen(true);
+        setTitle(title);
+        setSubtitle(subtitle);
+    };
+
+    const handleClose = () => {
+        setIsOpen(false);
+    };
+
+    const MessageDialogComponent = isOpen && (
+        <MessageDialog title={title} subtitle={subtitle} onClose={handleClose}></MessageDialog>
+    );
+
+    return { message, MessageDialogComponent };
+}
+
+export default useMessage;