Broadcast cluster's disconnected state before stopping KubeAuthProxy

Signed-off-by: Sebastian Malton <sebastian@malton.name>

src/main/__test__/cluster.test.ts

@@ -36,6 +36,8 @@ jest.mock("winston", () => ({
     label: jest.fn(),
     timestamp: jest.fn(),
     printf: jest.fn(),
+    padLevels: jest.fn(),
+    ms: jest.fn(),
   },
   createLogger: jest.fn().mockReturnValue(logger),
   transports: {
@@ -49,12 +51,31 @@ jest.mock("../context-handler");
 jest.mock("request");
 jest.mock("request-promise-native");
 
+jest.mock("electron", () => ({
+  app: {
+    getVersion: () => "99.99.99",
+    getName: () => "lens",
+    setName: jest.fn(),
+    setPath: jest.fn(),
+    getPath: () => "tmp",
+    getLocale: () => "en",
+    setLoginItemSettings: jest.fn(),
+  },
+  ipcMain: {
+    on: jest.fn(),
+    once: jest.fn(),
+    handle: jest.fn(),
+  },
+}));
+
 import { Console } from "console";
 import mockFs from "mock-fs";
+import { AppPaths } from "../../common/app-paths";
 import { Cluster } from "../cluster";
 import { Kubectl } from "../kubectl";
 
 console = new Console(process.stdout, process.stderr); // fix mockFS
+AppPaths.init();
 
 describe("create clusters", () => {
   beforeEach(() => {

src/main/catalog-sources/__test__/kubeconfig-sync.test.ts

@@ -42,6 +42,7 @@ jest.mock("electron", () => ({
   },
   ipcMain: {
     on: jest.fn(),
+    once: jest.fn(),
     handle: jest.fn(),
   },
 }));

src/main/cluster.ts

@@ -34,7 +34,7 @@ import { DetectorRegistry } from "./cluster-detectors/detector-registry";
 import plimit from "p-limit";
 import type { ClusterState, ClusterRefreshOptions, ClusterMetricsResourceType, ClusterId, ClusterMetadata, ClusterModel, ClusterPreferences, ClusterPrometheusPreferences, UpdateClusterModel, KubeAuthUpdate } from "../common/cluster-types";
 import { ClusterMetadataKey, initialNodeShellImage, ClusterStatus } from "../common/cluster-types";
-import { storedKubeConfigFolder, toJS } from "../common/utils";
+import { disposer, storedKubeConfigFolder, toJS } from "../common/utils";
 import type { Response } from "request";
 
 /**
@@ -53,7 +53,7 @@ export class Cluster implements ClusterModel, ClusterState {
    */
   public contextHandler: ContextHandler;
   protected kubeconfigManager: KubeconfigManager;
-  protected eventDisposers: Function[] = [];
+  protected eventDisposers = disposer();
   protected activated = false;
   private resourceAccessStatuses: Map<KubeApiResource, boolean> = new Map();
 
@@ -324,15 +324,6 @@ export class Cluster implements ClusterModel, ClusterState {
     this.getProxyKubeconfig();
   }
 
-  /**
-   * internal
-   */
-  protected unbindEvents() {
-    logger.info(`[CLUSTER]: unbind events`, this.getMeta());
-    this.eventDisposers.forEach(dispose => dispose());
-    this.eventDisposers.length = 0;
-  }
-
   /**
    * @param force force activation
    * @internal
@@ -395,8 +386,13 @@ export class Cluster implements ClusterModel, ClusterState {
    * @internal
    */
   @action disconnect() {
-    this.unbindEvents();
+    logger.info(`[CLUSTER]: disconnecting cluster`, { id: this.id });
-    this.contextHandler?.stopServer();
+
+    ipcMain.once(`cluster:${this.id}:frame-removed`, () => {
+      this.contextHandler?.stopServer();
+    });
+
+    this.eventDisposers();
     this.disconnected = true;
     this.online = false;
     this.accessible = false;
@@ -405,7 +401,6 @@ export class Cluster implements ClusterModel, ClusterState {
     this.allowedNamespaces = [];
     this.resourceAccessStatuses.clear();
     this.pushState();
-    logger.info(`[CLUSTER]: disconnect`, this.getMeta());
   }
 
   /**

src/main/kube-auth-proxy.ts

@@ -129,11 +129,11 @@ export class KubeAuthProxy {
     this.ready = false;
 
     if (this.proxyProcess) {
-      logger.debug("[KUBE-AUTH]: stopping local proxy", this.cluster.getMeta());
+      logger.info("[KUBE-AUTH]: stopping local proxy", { clusterId: this.cluster.id });
+      this.proxyProcess.kill();
       this.proxyProcess.removeAllListeners();
       this.proxyProcess.stderr.removeAllListeners();
       this.proxyProcess.stdout.removeAllListeners();
-      this.proxyProcess.kill();
       this.proxyProcess = null;
     }
   }

src/renderer/components/cluster-manager/lens-views.ts

@@ -21,6 +21,7 @@
 
 import { action, IReactionDisposer, makeObservable, observable, reaction, when } from "mobx";
 import logger from "../../../main/logger";
+import { broadcastMessage } from "../../../common/ipc";
 import { clusterVisibilityHandler } from "../../../common/cluster-ipc";
 import { ClusterStore } from "../../../common/cluster-store";
 import type { ClusterId } from "../../../common/cluster-types";
@@ -90,6 +91,7 @@ export class ClusterFrameHandler extends Singleton {
             logger.info(`[LENS-VIEW]: remove dashboard, clusterId=${clusterId}`);
             this.views.delete(clusterId);
             iframe.parentNode.removeChild(iframe);
+            broadcastMessage(`cluster:${clusterId}:frame-removed`);
             dispose();
           },
         );