fix: connectToggle

bepass-org · Jun 17, 2024 · 907dd01 · 907dd01
1 parent 6550d62
commit 907dd01
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/src/main/main.ts b/src/main/main.ts
@@ -269,7 +269,7 @@ if (!gotTheLock) {
                     });*/
                     ipcMain.on('tray-menu', (event) => {
                         event.reply('tray-menu', {
-                            key: 'connectToggle',
+                            key: 'connect',
                             msg: 'Connect Tray Click!'
                         });
                     });
@@ -304,7 +304,7 @@ if (!gotTheLock) {
                             connectStatus === 'disconnected'
                         ) {
                             trayMenuEvent?.reply('tray-menu', {
-                                key: 'connectToggle',
+                                key: connectStatus === 'disconnected' ? 'connect' : 'disconnect',
                                 msg: 'Connect Tray Click!'
                             });
                         }

diff --git a/src/renderer/pages/Landing/useLanding.ts b/src/renderer/pages/Landing/useLanding.ts
@@ -135,8 +135,19 @@ const useLanding = () => {
         });
 
         ipcRenderer.on('tray-menu', (args: any) => {
-            if (args.key === 'connectToggle' && !isLoading) {
-                onChange();
+            if (args.key === 'connect' && !isLoading) {
+                setIpInfo({
+                    countryCode: false,
+                    ip: ''
+                });
+                setProxyStatus(proxyMode);
+                ipcRenderer.sendMessage('wp-start');
+                setIsLoading(true);
+                setPing(0);
+            }
+            if (args.key === 'disconnect' && !isLoading) {
+                ipcRenderer.sendMessage('wp-end');
+                setIsLoading(true);
             }
             if (args.key === 'changePage') {
                 navigate(args.msg);