lens/src/common/cluster-store.ts

import type { WorkspaceId } from "./workspace-store";
import path from "path";
import filenamify from "filenamify";
import { app, ipcRenderer } from "electron";
import { copyFile, ensureDir, unlink } from "fs-extra";
import { action, computed, observable, toJS } from "mobx";
import { appProto } from "./vars";
import { BaseStore } from "./base-store";
import { Cluster, ClusterState } from "../main/cluster";
import migrations from "../migrations/cluster-store"
import logger from "../main/logger";
import { tracker } from "./tracker";

export interface ClusterIconUpload {
  clusterId: string;
  name: string;
  path: string;
}

export interface ClusterStoreModel {
  activeCluster?: ClusterId; // last opened cluster
  clusters?: ClusterModel[]
}

export type ClusterId = string;

export interface ClusterModel {
  id: ClusterId;
  workspace?: WorkspaceId;
  contextName?: string;
  preferences?: ClusterPreferences;
  kubeConfigPath: string;

  /** @deprecated */
  kubeConfig?: string; // yaml
}

export interface ClusterPreferences {
  terminalCWD?: string;
  clusterName?: string;
  prometheus?: {
    namespace: string;
    service: string;
    port: number;
    prefix: string;
  };
  prometheusProvider?: {
    type: string;
  };
  icon?: string;
  httpsProxy?: string;
}

export class ClusterStore extends BaseStore<ClusterStoreModel> {
  static get iconsDir() {
    return path.join(app.getPath("userData"), "icons");
  }

  private constructor() {
    super({
      configName: "lens-cluster-store",
      accessPropertiesByDotNotation: false, // To make dots safe in cluster context names
      migrations: migrations,
    });
    if (ipcRenderer) {
      ipcRenderer.on("cluster:state", (event, clusterState: ClusterState) => {
        this.applyWithoutSync(() => {
          logger.info(`[CLUSTER-STORE]: received cluster(${clusterState.id}) update`, clusterState);
          const cluster = this.getById(clusterState.id);
          if (cluster) cluster.updateModel(clusterState)
        })
      })
    }
  }

  @observable activeClusterId: ClusterId;
  @observable removedClusters = observable.map<ClusterId, Cluster>();
  @observable clusters = observable.map<ClusterId, Cluster>();

  @computed get activeCluster(): Cluster | null {
    return this.getById(this.activeClusterId);
  }

  @computed get clustersList(): Cluster[] {
    return Array.from(this.clusters.values());
  }

  getById(id: ClusterId): Cluster {
    return this.clusters.get(id);
  }

  getByWorkspaceId(workspaceId: string): Cluster[] {
    return this.clustersList.filter(cluster => cluster.workspace === workspaceId)
  }

  @action
  addCluster(model: ClusterModel): Cluster {
    const cluster = new Cluster(model);
    this.clusters.set(model.id, cluster);
    return cluster;
  }

  @action
  removeById(clusterId: ClusterId): void {
    if (this.activeClusterId === clusterId) {
      this.activeClusterId = null;
    }
    this.clusters.delete(clusterId);
  }

  @action
  removeByWorkspaceId(workspaceId: string) {
    this.getByWorkspaceId(workspaceId).forEach(cluster => {
      this.removeById(cluster.id)
    })
  }

  @action
  protected async uploadClusterIcon({ clusterId, ...upload }: ClusterIconUpload): Promise<string> {
    const cluster = this.getById(clusterId);
    if (cluster) {
      tracker.event("cluster", "upload-icon");
      const fileDest = path.join(ClusterStore.iconsDir, filenamify(cluster.contextName + "-" + upload.name))
      await ensureDir(path.dirname(fileDest));
      await copyFile(upload.path, fileDest)
      cluster.preferences.icon = `${appProto}:///icons/${fileDest}`
      return cluster.preferences.icon;
    }
  }

  @action
  protected resetClusterIcon(clusterId: ClusterId) {
    const cluster = this.getById(clusterId);
    if (cluster) {
      tracker.event("cluster", "reset-icon")
      const iconPath = path.join(ClusterStore.iconsDir, path.basename(cluster.preferences.icon));
      unlink(iconPath).catch(() => null); // remove file
      delete cluster.preferences.icon;
    }
  }

  @action
  protected fromStore({ activeCluster, clusters = [] }: ClusterStoreModel = {}) {
    const currentClusters = this.clusters.toJS();
    const newClusters = new Map<ClusterId, Cluster>();
    const removedClusters = new Map<ClusterId, Cluster>();

    // update new clusters
    clusters.forEach(clusterModel => {
      let cluster = currentClusters.get(clusterModel.id);
      if (cluster) {
        cluster.updateModel(clusterModel);
      } else {
        cluster = new Cluster(clusterModel);
      }
      newClusters.set(clusterModel.id, cluster);
    });

    // update removed clusters
    currentClusters.forEach(cluster => {
      if (!newClusters.has(cluster.id)) {
        removedClusters.set(cluster.id, cluster);
      }
    });

    this.activeClusterId = newClusters.has(activeCluster) ? activeCluster : null;
    this.clusters.replace(newClusters);
    this.removedClusters.replace(removedClusters);

    // "auto-select" first cluster if not available or invalid from config file
    if (!this.activeClusterId && newClusters.size) {
      this.activeClusterId = Array.from(newClusters.values())[0].id;
    }
  }

  toJSON(): ClusterStoreModel {
    return toJS({
      activeCluster: this.activeClusterId,
      clusters: this.clustersList.map(cluster => cluster.toJSON()),
    }, {
      recurseEverything: true
    })
  }
}

export const clusterStore: ClusterStore = ClusterStore.getInstance();