lens/src/main/initializers/ipc.ts

/**
 * Copyright (c) 2021 OpenLens Authors
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of
 * this software and associated documentation files (the "Software"), to deal in
 * the Software without restriction, including without limitation the rights to
 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
 * the Software, and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
 * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
 * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

import AwaitLock from "await-lock";
import type { IpcMainInvokeEvent } from "electron";
import { when } from "mobx";
import { KubernetesCluster } from "../../common/catalog-entities";
import { ClusterFrames } from "../../common/cluster-frames";
import { clusterActivateHandler, clusterSetFrameIdHandler, clusterVisibilityHandler, clusterRefreshHandler, clusterDisconnectHandler, clusterKubectlApplyAllHandler, clusterKubectlDeleteAllHandler, clusterDeleteHandler, navigateToClusterHandler, claimClusterFrameHandler } from "../../common/cluster-ipc";
import { ClusterId, ClusterStore } from "../../common/cluster-store";
import { appEventBus } from "../../common/event-bus";
import { ipcMainHandle } from "../../common/ipc";
import { getOrInsert } from "../../common/utils";
import { catalogEntityRegistry } from "../catalog";
import { ClusterManager } from "../cluster-manager";
import { bundledKubectlPath } from "../kubectl";
import logger from "../logger";
import { promiseExecFile } from "../promise-exec";
import { ResourceApplier } from "../resource-applier";
import { NavigateFrameInfoSpecifier, WindowManager } from "../window-manager";

export function initIpcMainHandlers() {
  ipcMainHandle(clusterActivateHandler, (event, clusterId: ClusterId, force = false) => {
    return ClusterStore.getInstance()
      .getById(clusterId)
      ?.activate(force);
  });

  ipcMainHandle(clusterSetFrameIdHandler, (event: IpcMainInvokeEvent, clusterId: ClusterId) => {
    const cluster = ClusterStore.getInstance().getById(clusterId);

    if (cluster) {
      ClusterFrames.getInstance().set(cluster.id, {
        frameId: event.frameId,
        processId: event.processId,
        windowId: event.sender.getProcessId(),
      });
      cluster.pushState();
    }
  });

  ipcMainHandle(clusterVisibilityHandler, (event: IpcMainInvokeEvent, clusterId: ClusterId, visible: boolean) => {
    const entity = catalogEntityRegistry.getById(clusterId);

    for (const kubeEntity of catalogEntityRegistry.getItemsByEntityClass(KubernetesCluster)) {
      kubeEntity.status.active = false;
    }

    if (entity) {
      entity.status.active = visible;
    }
  });

  ipcMainHandle(clusterRefreshHandler, (event, clusterId: ClusterId) => {
    return ClusterStore.getInstance()
      .getById(clusterId)
      ?.refresh({ refreshMetadata: true });
  });

  ipcMainHandle(clusterDisconnectHandler, (event, clusterId: ClusterId) => {
    appEventBus.emit({ name: "cluster", action: "stop" });
    const cluster = ClusterStore.getInstance().getById(clusterId);

    if (cluster) {
      cluster.disconnect();
      ClusterFrames.getInstance().clearInfoForCluster(cluster.id);
    }
  });

  ipcMainHandle(clusterDeleteHandler, async (event, clusterId: ClusterId) => {
    appEventBus.emit({ name: "cluster", action: "remove" });
    const cluster = ClusterStore.getInstance().getById(clusterId);

    if (!cluster) {
      return;
    }

    ClusterManager.getInstance().deleting.add(clusterId);
    cluster.disconnect();
    ClusterFrames.getInstance().clearInfoForCluster(cluster.id);
    const kubectlPath = bundledKubectlPath();
    const args = ["config", "delete-context", cluster.contextName, "--kubeconfig", cluster.kubeConfigPath];

    try {
      await promiseExecFile(kubectlPath, args);
    } catch ({ stderr }) {
      logger.error(`[CLUSTER-REMOVE]: failed to remove cluster: ${stderr}`, { clusterId, context: cluster.contextName });

      throw `Failed to remove cluster: ${stderr}`;
    }
  });

  ipcMainHandle(clusterKubectlApplyAllHandler, async (event, clusterId: ClusterId, resources: string[], extraArgs: string[]) => {
    appEventBus.emit({ name: "cluster", action: "kubectl-apply-all" });
    const cluster = ClusterStore.getInstance().getById(clusterId);

    if (cluster) {
      const applier = new ResourceApplier(cluster);

      try {
        const stdout = await applier.kubectlApplyAll(resources, extraArgs);

        return { stdout };
      } catch (error: any) {
        return { stderr: error };
      }
    } else {
      throw `${clusterId} is not a valid cluster id`;
    }
  });

  ipcMainHandle(clusterKubectlDeleteAllHandler, async (event, clusterId: ClusterId, resources: string[], extraArgs: string[]) => {
    appEventBus.emit({ name: "cluster", action: "kubectl-delete-all" });
    const cluster = ClusterStore.getInstance().getById(clusterId);

    if (cluster) {
      const applier = new ResourceApplier(cluster);

      try {
        const stdout = await applier.kubectlDeleteAll(resources, extraArgs);

        return { stdout };
      } catch (error: any) {
        return { stderr: error };
      }
    } else {
      throw `${clusterId} is not a valid cluster id`;
    }
  });

  ipcMainHandle(claimClusterFrameHandler, async (event, clusterId: ClusterId) => {
    return ClusterFrames.getInstance().claimCluster(clusterId, event.sender.getProcessId());
  });

  const navigateLocks = new Map<ClusterId, AwaitLock>();

  ipcMainHandle(navigateToClusterHandler, async (event, clusterId: ClusterId, newWindow?: boolean) => {
    appEventBus.emit({ name: "cluster", action: "navigate" });
    const cluster = ClusterStore.getInstance().getById(clusterId);

    if (!cluster) {
      return void logger.warn("[NAVIGATE-TO-CLUSTER]: unknown cluster", { clusterId });
    }

    const lock = getOrInsert(navigateLocks, clusterId, new AwaitLock());

    try {
      console.log("trying to acquire lock for", clusterId, "in", navigateToClusterHandler);
      await lock.acquireAsync();
      console.log("acquired lock for", clusterId, "in", navigateToClusterHandler);
      const specifics: NavigateFrameInfoSpecifier[] = [{ clusterId }];

      if (newWindow) {
        specifics.push({ windowId: true });
      } else {
        specifics.push({ windowId: event.sender.getProcessId() });
      }

      await WindowManager.getInstance().navigate(`/cluster/${clusterId}`, ...specifics);
      await when(() => Boolean(ClusterFrames.getInstance().getFrameInfoByClusterId(clusterId)));
    } catch (error) {
      logger.error("Failed to load url for new cluster window", error);
    } finally {
      lock.release();
    }
  });
}