From a0903274297edacd33dc92202fbc468eb22f91a7 Mon Sep 17 00:00:00 2001
From: FatttSnake <fatttsnake@gmail.com>
Date: Mon, 16 Oct 2023 15:31:58 +0800
Subject: [PATCH] Add exit waiting to ToolsFramework

---
 src/pages/ToolsFramework.tsx | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/pages/ToolsFramework.tsx b/src/pages/ToolsFramework.tsx
index ebaa939..c997eff 100644
--- a/src/pages/ToolsFramework.tsx
+++ b/src/pages/ToolsFramework.tsx
@@ -17,6 +17,7 @@ const ToolsFramework: React.FC = () => {
     const [submenuTop, setSubmenuTop] = useState(0)
     const [submenuLeft, setSubmenuLeft] = useState(0)
     const [hideSidebar, setHideSidebar] = useState(getLocalStorage('hideSidebar') === 'false')
+    const [exiting, setExiting] = useState(false)
 
     const switchSidebar = () => {
         setHideSidebar(!hideSidebar)
@@ -52,6 +53,11 @@ const ToolsFramework: React.FC = () => {
     }
 
     const handleLogout = () => {
+        if (exiting) {
+            return
+        }
+
+        setExiting(true)
         void logout().finally(() => {
             setTimeout(() => {
                 window.location.reload()
@@ -232,7 +238,10 @@ const ToolsFramework: React.FC = () => {
                                     className={'icon-exit'}
                                     onClick={handleLogout}
                                 >
-                                    <Icon component={IconFatwebExit} />
+                                    <Icon
+                                        component={exiting ? IconFatwebLoading : IconFatwebExit}
+                                        spin={exiting}
+                                    />
                                 </span>
                             </div>
                         </div>