FIXUP: Fix location cache in Electron app

mullvad · Dec 8, 2023 · d7baa41 · d7baa41
1 parent 95bd4b8
commit d7baa41
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/gui/src/main/tunnel-state.ts b/gui/src/main/tunnel-state.ts
@@ -39,7 +39,9 @@ export default class TunnelStateHandler {
     this.tunnelStateFallback = this.tunnelState;
 
     this.setTunnelState(
-      state === 'disconnecting' ? { state, details: 'nothing' as const } : { state },
+      state === 'disconnecting'
+        ? { state, details: 'nothing' as const, location: this.lastKnownDisconnectedLocation }
+        : { state },
     );
 
     this.tunnelStateFallbackScheduler.schedule(() => {
@@ -69,18 +71,18 @@ export default class TunnelStateHandler {
       this.expectNextTunnelState('connecting');
       this.tunnelStateFallback = newState;
     } else {
-      this.setTunnelState(newState);
-    }
+      if (newState.state === 'disconnected' && newState.location !== undefined) {
+        this.lastKnownDisconnectedLocation = newState.location;
+      }
 
-    if (newState.state === 'disconnected' && newState.location !== undefined) {
-      this.lastKnownDisconnectedLocation = newState.location;
-    }
+      if (
+        newState.state === 'disconnecting' ||
+        (newState.state === 'disconnected' && newState.location === undefined)
+      ) {
+        newState.location = this.lastKnownDisconnectedLocation;
+      }
 
-    if (
-      newState.state === 'disconnecting' ||
-      (newState.state === 'disconnected' && newState.location === undefined)
-    ) {
-      newState.location = this.lastKnownDisconnectedLocation;
+      this.setTunnelState(newState);
     }
   }