From 3b2f070128d17331211da5f2d63b69fca0567fd2 Mon Sep 17 00:00:00 2001
From: yassinedorbozgithub <yassine.dorboz.2022@gmail.com>
Date: Tue, 21 Jan 2025 08:24:54 +0100
Subject: [PATCH] fix: apply feedback updates

---
 frontend/src/hooks/useDialog.tsx | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/frontend/src/hooks/useDialog.tsx b/frontend/src/hooks/useDialog.tsx
index 867dbfb0..20026651 100644
--- a/frontend/src/hooks/useDialog.tsx
+++ b/frontend/src/hooks/useDialog.tsx
@@ -7,7 +7,7 @@
  */
 
 import { DialogProps } from "@mui/material";
-import { useState } from "react";
+import { Dispatch, SetStateAction, useState } from "react";
 
 export type DialogControlProps<T, C = never> = Omit<
   DialogControl<T, C>,
@@ -20,7 +20,7 @@ type TCloseDialog = <E extends React.MouseEvent | Event | Object>(
 type TFnVoid<T> = (data?: T) => void;
 export type DialogControl<T = null, C = never> = DialogProps & {
   data?: T;
-  saveData?: TFnVoid<T>;
+  setData?: Dispatch<SetStateAction<T | undefined>>;
   callback?: TFnVoid<C>;
   openDialog: TFnVoid<T>;
   closeDialog: TCloseDialog;
@@ -34,12 +34,15 @@ export const useDialog = <T,>(initialState: boolean): DialogControl<T> => {
     setOpen(true);
   };
   const closeDialog: TCloseDialog = (event, reason) => {
-    if (reason === "postDelete") setData(undefined);
-    if (reason !== "backdropClick") setOpen(false);
+    if (reason === "postDelete") {
+      setData(undefined);
+    }
+    if (reason !== "backdropClick") {
+      setOpen(false);
+    }
   };
-  const saveData: TFnVoid<T> = (data) => setData(data);
 
-  return { open, openDialog, closeDialog, data, saveData };
+  return { open, openDialog, closeDialog, data, setData };
 };
 
 export const getDisplayDialogs = <T,>({