From 7ca1020ab615e9f8b569d0b5a7fdecb6721b537d Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Tue, 6 Jul 2021 23:31:37 +0800
Subject: fix: Fix popup menu bug.

---
 FrontEnd/src/utilities/useClickOutside.ts | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

(limited to 'FrontEnd/src/utilities/useClickOutside.ts')

diff --git a/FrontEnd/src/utilities/useClickOutside.ts b/FrontEnd/src/utilities/useClickOutside.ts
index d4c53aa8..4b2f0e5c 100644
--- a/FrontEnd/src/utilities/useClickOutside.ts
+++ b/FrontEnd/src/utilities/useClickOutside.ts
@@ -2,7 +2,8 @@ import React from "react";
 
 export default function useClickOutside(
   element: HTMLElement | null | undefined,
-  onClickOutside: () => void
+  onClickOutside: () => void,
+  nextTick?: boolean
 ): void {
   const onClickOutsideRef = React.useRef<() => void>(onClickOutside);
 
@@ -22,10 +23,16 @@ export default function useClickOutside(
         }
         onClickOutsideRef.current();
       };
-      document.addEventListener("click", handler);
+      if (nextTick) {
+        setTimeout(() => {
+          document.addEventListener("click", handler);
+        });
+      } else {
+        document.addEventListener("click", handler);
+      }
       return () => {
         document.removeEventListener("click", handler);
       };
     }
-  }, [element]);
+  }, [element, nextTick]);
 }
-- 
cgit v1.2.3