lens/src/main/context-handler.ts

import type { PrometheusProvider, PrometheusService } from "./prometheus/provider-registry"
import type { ClusterPreferences } from "../common/cluster-store";
import type { Cluster } from "./cluster"
import type httpProxy from "http-proxy"
import url, { UrlWithStringQuery } from "url";
import { CoreV1Api } from "@kubernetes/client-node"
import { prometheusProviders } from "../common/prometheus-providers"
import logger from "./logger"
import { getFreePort } from "./port"
import { KubeAuthProxy } from "./kube-auth-proxy"

export class ContextHandler {
  public proxyPort: number;
  public clusterUrl: UrlWithStringQuery;
  protected kubeAuthProxy: KubeAuthProxy
  protected apiTarget: httpProxy.ServerOptions
  protected prometheusProvider: string
  protected prometheusPath: string

  constructor(protected cluster: Cluster) {
    this.clusterUrl = url.parse(cluster.apiUrl);
    this.setupPrometheus(cluster.preferences);
  }

  protected setupPrometheus(preferences: ClusterPreferences = {}) {
    this.prometheusProvider = preferences.prometheusProvider?.type;
    this.prometheusPath = null;
    if (preferences.prometheus) {
      const { namespace, service, port } = preferences.prometheus
      this.prometheusPath = `${namespace}/services/${service}:${port}`
    }
  }

  protected async resolvePrometheusPath(): Promise<string> {
    const { service, namespace, port } = await this.getPrometheusService()
    return `${namespace}/services/${service}:${port}`
  }

  async getPrometheusProvider() {
    if (!this.prometheusProvider) {
      const service = await this.getPrometheusService()
      logger.info(`using ${service.id} as prometheus provider`)
      this.prometheusProvider = service.id
    }
    return prometheusProviders.find(p => p.id === this.prometheusProvider)
  }

  async getPrometheusService(): Promise<PrometheusService> {
    const providers = this.prometheusProvider ? prometheusProviders.filter(provider => provider.id == this.prometheusProvider) : prometheusProviders;
    const prometheusPromises: Promise<PrometheusService>[] = providers.map(async (provider: PrometheusProvider): Promise<PrometheusService> => {
      const apiClient = this.cluster.getProxyKubeconfig().makeApiClient(CoreV1Api)
      return await provider.getPrometheusService(apiClient)
    })
    const resolvedPrometheusServices = await Promise.all(prometheusPromises)
    const service = resolvedPrometheusServices.filter(n => n)[0];
    return service || {
      id: "lens",
      namespace: "lens-metrics",
      service: "prometheus",
      port: 80
    }
  }

  async getPrometheusPath(): Promise<string> {
    if (!this.prometheusPath) {
      this.prometheusPath = await this.resolvePrometheusPath()
    }
    return this.prometheusPath;
  }

  async resolveAuthProxyUrl() {
    const proxyPort = await this.ensurePort();
    const path = this.clusterUrl.path !== "/" ? this.clusterUrl.path : ""
    return `http://127.0.0.1:${proxyPort}${path}`;
  }

  async getApiTarget(isWatchRequest = false): Promise<httpProxy.ServerOptions> {
    if (this.apiTarget && !isWatchRequest) {
      return this.apiTarget
    }
    const timeout = isWatchRequest ? 4 * 60 * 60 * 1000 : 30000 // 4 hours for watch request, 30 seconds for the rest
    const apiTarget = await this.newApiTarget(timeout)
    if (!isWatchRequest) {
      this.apiTarget = apiTarget
    }
    return apiTarget
  }

  protected async newApiTarget(timeout: number): Promise<httpProxy.ServerOptions> {
    const proxyUrl = await this.resolveAuthProxyUrl();
    return {
      target: proxyUrl,
      changeOrigin: true,
      timeout: timeout,
      headers: {
        "Host": this.clusterUrl.hostname,
      },
    }
  }

  async ensurePort(): Promise<number> {
    if (!this.proxyPort) {
      this.proxyPort = await getFreePort();
    }
    return this.proxyPort
  }

  async ensureServer() {
    if (!this.kubeAuthProxy) {
      await this.ensurePort();
      const proxyEnv = Object.assign({}, process.env)
      if (this.cluster.preferences.httpsProxy) {
        proxyEnv.HTTPS_PROXY = this.cluster.preferences.httpsProxy
      }
      this.kubeAuthProxy = new KubeAuthProxy(this.cluster, this.proxyPort, proxyEnv)
      await this.kubeAuthProxy.run()
    }
  }

  stopServer() {
    if (this.kubeAuthProxy) {
      this.kubeAuthProxy.exit()
      this.kubeAuthProxy = null
    }
  }

  get proxyLastError(): string {
    return this.kubeAuthProxy?.lastError || ""
  }
}