lens/src/main/cluster-detectors/distribution-detector.ts

import { BaseClusterDetector } from "./base-cluster-detector";
import { ClusterMetadataKey } from "../cluster";

export class DistributionDetector extends BaseClusterDetector {
  key = ClusterMetadataKey.DISTRIBUTION
  version: string

  public async detect() {
    this.version = await this.getKubernetesVersion()
    if (await this.isRancher()) {
      return { value: "rancher", accuracy: 80}
    }
    if (this.isGKE()) {
      return { value: "gke", accuracy: 80}
    }
    if (this.isEKS()) {
      return { value: "eks", accuracy: 80}
    }
    if (this.isIKS()) {
      return { value: "iks", accuracy: 80}
    }
    if (this.isAKS()) {
      return { value: "aks", accuracy: 80}
    }
    if (this.isDigitalOcean()) {
      return { value: "digitalocean", accuracy: 90}
    }
    if (this.isMinikube()) {
      return { value: "minikube", accuracy: 80}
    }
    if (this.isCustom()) {
      return { value: "custom", accuracy: 10}
    }
    return { value: "vanilla", accuracy: 10}
  }

  public async getKubernetesVersion() {
    if (this.cluster.version) return this.cluster.version

    const response = await this.k8sRequest("/version")
    return response.gitVersion
  }

  protected isGKE() {
    return this.version.includes("gke")
  }

  protected isEKS() {
    return this.version.includes("eks")
  }

  protected isIKS() {
    return this.version.includes("IKS")
  }

  protected isAKS() {
    return this.cluster.apiUrl.endsWith("azmk8s.io")
  }

  protected isDigitalOcean() {
    return this.cluster.apiUrl.endsWith("k8s.ondigitalocean.com")
  }

  protected isMinikube() {
    return this.cluster.contextName.startsWith("minikube")
  }

  protected isCustom() {
    return this.version.includes("+")
  }

  protected async isRancher() {
    try {
      const response = await this.k8sRequest("")
      return response.data.find((api: any) => api?.apiVersion?.group === "meta.cattle.io") !== undefined
    } catch (e) {
      return false
    }
  }
}