Fixing the issue with updating the TrayMenu upon connection

رفع مشکل بروزشدن TrayMenu هنگام اتصال
bepass-org · Jan 1, 2025 · 7e1d6fc · 7e1d6fc
1 parent fc89671
commit 7e1d6fc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/main/lib/wpManager.ts b/src/main/lib/wpManager.ts
@@ -162,7 +162,7 @@ class WarpPlusManager {
                 break;
 
             case ConnectionState.DISCONNECTING:
-                customEvent.emit('tray-icon', 'connecting');
+                customEvent.emit('tray-icon', 'disconnecting');
                 break;
 
             case ConnectionState.CONNECTED:

diff --git a/src/main/main.ts b/src/main/main.ts
@@ -470,7 +470,14 @@ class OblivionDesktop {
                 label: connectLabel,
                 type: 'normal',
                 enabled: canToggleConnection,
-                click: () => this.handleConnectionToggle()
+                click: () => {
+                    this.handleConnectionToggle();
+                    this.state.connectionStatus =
+                        this.state.connectionStatus === 'disconnected'
+                            ? 'connecting'
+                            : 'disconnecting';
+                    this.updateTrayMenu();
+                }
             },
             {
                 label: this.state.appLang.systemTray.settings,