From f14f3b3287cfc29ad91babb6193d2aa306e442e4 Mon Sep 17 00:00:00 2001
From: Jari Kolehmainen <jari.kolehmainen@gmail.com>
Date: Thu, 11 Feb 2021 17:36:47 +0200
Subject: [PATCH] Cleanup iframe when cluster is disconnected (#2120)

Signed-off-by: Jari Kolehmainen <jari.kolehmainen@gmail.com>
---
 src/renderer/components/cluster-manager/lens-views.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/renderer/components/cluster-manager/lens-views.ts b/src/renderer/components/cluster-manager/lens-views.ts
index 17e6c9f00d..da7efb377c 100644
--- a/src/renderer/components/cluster-manager/lens-views.ts
+++ b/src/renderer/components/cluster-manager/lens-views.ts
@@ -40,7 +40,15 @@ export async function initView(clusterId: ClusterId) {
 }
 
 export async function autoCleanOnRemove(clusterId: ClusterId, iframe: HTMLIFrameElement) {
-  await when(() => !clusterStore.getById(clusterId));
+  await when(() => {
+    const cluster = clusterStore.getById(clusterId);
+
+    if (!cluster) return true;
+
+    const view = lensViews.get(clusterId);
+
+    return cluster.disconnected && view?.isLoaded;
+  });
   logger.info(`[LENS-VIEW]: remove dashboard, clusterId=${clusterId}`);
   lensViews.delete(clusterId);