lens/src/common/catalog-entities/kubernetes-cluster.ts

import { EventEmitter } from "events";
import { observable } from "mobx";
import { catalogCategoryRegistry } from "../catalog-category-registry";
import { CatalogCategory, CatalogEntity, CatalogEntityActionContext, CatalogEntityAddMenuContext, CatalogEntityContextMenuContext, CatalogEntityData, CatalogEntityMetadata, CatalogEntityStatus } from "../catalog-entity";
import { clusterDisconnectHandler } from "../cluster-ipc";
import { clusterStore } from "../cluster-store";
import { requestMain } from "../ipc";

export type KubernetesClusterSpec = {
  kubeconfigPath: string;
  kubeconfigContext: string;
};

export interface KubernetesClusterStatus extends CatalogEntityStatus {
  phase: "connected" | "disconnected";
}

export class KubernetesCluster implements CatalogEntity {
  public readonly apiVersion = "entity.k8slens.dev/v1alpha1";
  public readonly kind = "KubernetesCluster";
  @observable public metadata: CatalogEntityMetadata;
  @observable public status: KubernetesClusterStatus;
  @observable public spec: KubernetesClusterSpec;

  constructor(data: CatalogEntityData) {
    this.metadata = data.metadata;
    this.status = data.status as KubernetesClusterStatus;
    this.spec = data.spec as KubernetesClusterSpec;
  }

  getId() {
    return this.metadata.uid;
  }

  getName() {
    return this.metadata.name;
  }

  async onRun(context: CatalogEntityActionContext) {
    context.navigate(`/cluster/${this.metadata.uid}`);
  }

  async onDetailsOpen() {
    //
  }

  async onContextMenuOpen(context: CatalogEntityContextMenuContext) {
    context.menuItems = [
      {
        icon: "settings",
        title: "Settings",
        onlyVisibleForSource: "local",
        onClick: async () => context.navigate(`/cluster/${this.metadata.uid}/settings`)
      },
      {
        icon: "delete",
        title: "Delete",
        onlyVisibleForSource: "local",
        onClick: async () => clusterStore.removeById(this.metadata.uid),
        confirm: {
          message: `Remove Kubernetes Cluster "${this.metadata.name} from Lens?`
        }
      },
    ];

    if (this.status.active) {
      context.menuItems.unshift({
        icon: "link_off",
        title: "Disconnect",
        onClick: async () => {
          clusterStore.deactivate(this.metadata.uid);
          requestMain(clusterDisconnectHandler, this.metadata.uid);
        }
      });
    }

    const category = catalogCategoryRegistry.getCategoryForEntity<KubernetesClusterCategory>(this);

    if (category) category.emit("contextMenuOpen", this, context);
  }
}

export class KubernetesClusterCategory extends EventEmitter implements CatalogCategory {
  public readonly apiVersion = "catalog.k8slens.dev/v1alpha1";
  public readonly kind = "CatalogCategory";
  public metadata = {
    name: "Kubernetes Clusters"
  };
  public spec = {
    group: "entity.k8slens.dev",
    versions: [
      {
        name: "v1alpha1",
        entityClass: KubernetesCluster
      }
    ],
    names: {
      kind: "KubernetesCluster"
    }
  };

  constructor() {
    super();

    this.on("onCatalogAddMenu", (ctx: CatalogEntityAddMenuContext) => {
      ctx.menuItems.push({
        icon: "text_snippet",
        title: "Add from kubeconfig",
        onClick: async () => {
          ctx.navigate("/add-cluster");
        }
      });
    });
  }

  getId() {
    return `${this.spec.group}/${this.spec.names.kind}`;
  }
}

catalogCategoryRegistry.add(new KubernetesClusterCategory());