From 710ff9d3d2e55113798c39b0595f8f71b12091ef Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Mon, 31 Jul 2023 21:16:27 +0800
Subject: ...

---
 FrontEnd/src/views/common/dialog/index.ts | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'FrontEnd/src/views/common/dialog/index.ts')

diff --git a/FrontEnd/src/views/common/dialog/index.ts b/FrontEnd/src/views/common/dialog/index.ts
index e37b9ed2..59f15791 100644
--- a/FrontEnd/src/views/common/dialog/index.ts
+++ b/FrontEnd/src/views/common/dialog/index.ts
@@ -18,14 +18,19 @@ type DialogPropsMap<D extends string> = DialogMap<
 
 export function useDialog<D extends string>(
   dialogs: D[],
-  initDialog?: D | null,
+  options?: {
+    initDialog?: D | null;
+    onClose?: {
+      [K in D]?: () => void;
+    };
+  },
 ): {
   dialog: D | null;
   switchDialog: (newDialog: D | null) => void;
   dialogPropsMap: DialogPropsMap<D>;
   createDialogSwitch: (newDialog: D | null) => () => void;
 } {
-  const [dialog, setDialog] = useState<D | null>(initDialog ?? null);
+  const [dialog, setDialog] = useState<D | null>(options?.initDialog ?? null);
 
   const [dialogKeys, setDialogKeys] = useState<DialogKeyMap<D>>(
     () => Object.fromEntries(dialogs.map((d) => [d, 0])) as DialogKeyMap<D>,
@@ -47,7 +52,10 @@ export function useDialog<D extends string>(
         {
           key: `${d}-${dialogKeys[d]}`,
           open: dialog === d,
-          onClose: () => switchDialog(null),
+          onClose: () => {
+            switchDialog(null);
+            options?.onClose?.[d]?.();
+          },
         },
       ]),
     ) as DialogPropsMap<D>,
-- 
cgit v1.2.3