lens/src/extensions/cluster-feature.ts

import fs from "fs";
import path from "path"
import hb from "handlebars"
import { observable } from "mobx"
import { ResourceApplier } from "../main/resource-applier"
import { Cluster } from "../main/cluster";
import logger from "../main/logger";
import { app } from "electron"
import { requestMain } from "../common/ipc";
import { clusterKubectlApplyAllHandler } from "../common/cluster-ipc";

export interface ClusterFeatureStatus {
  currentVersion: string;
  installed: boolean;
  latestVersion: string;
  canUpgrade: boolean;
}

export abstract class ClusterFeature {
  name: string;
  latestVersion: string;
  config: any;

  @observable status: ClusterFeatureStatus = {
    currentVersion: null,
    installed: false,
    latestVersion: null,
    canUpgrade: false
  }

  abstract async install(cluster: Cluster): Promise<void>;

  abstract async upgrade(cluster: Cluster): Promise<void>;

  abstract async uninstall(cluster: Cluster): Promise<void>;

  abstract async updateStatus(cluster: Cluster): Promise<ClusterFeatureStatus>;

  protected async applyResources(cluster: Cluster, resources: string[]) {
    if (app) {
      await new ResourceApplier(cluster).kubectlApplyAll(resources)
    } else {
      await requestMain(clusterKubectlApplyAllHandler, cluster.id, resources)
    }
  }

  protected renderTemplates(folderPath: string): string[] {
    const resources: string[] = [];
    logger.info(`[FEATURE]: render templates from ${folderPath}`);
    fs.readdirSync(folderPath).forEach(filename => {
      const file = path.join(folderPath, filename);
      const raw = fs.readFileSync(file);
      if (filename.endsWith('.hb')) {
        const template = hb.compile(raw.toString());
        resources.push(template(this.config));
      } else {
        resources.push(raw.toString());
      }
    });

    return resources;
  }
}