import { KubernetesWorkload } from "../types/kubernetesTypes";

export function getImagePullSecrets(inputObject: any) {
  if (!inputObject?.spec) return null;

  if (
    inputObject.kind.toLowerCase() === KubernetesWorkload.CRON_JOB.toLowerCase()
  )
    return inputObject?.spec?.jobTemplate?.spec?.template?.spec
      ?.imagePullSecrets;

  if (inputObject.kind.toLowerCase() === KubernetesWorkload.POD.toLowerCase())
    return inputObject.spec.imagePullSecrets;

  if (inputObject?.spec?.template?.spec) {
    return inputObject.spec.template.spec.imagePullSecrets;
  }
}

export function setImagePullSecrets(
  inputObject: any,
  newImagePullSecrets: any
) {
  if (!inputObject || !inputObject.spec || !newImagePullSecrets) return;

  if (inputObject.kind.toLowerCase() === KubernetesWorkload.POD.toLowerCase()) {
    inputObject.spec.imagePullSecrets = newImagePullSecrets;
    return;
  }

  if (
    inputObject.kind.toLowerCase() === KubernetesWorkload.CRON_JOB.toLowerCase()
  ) {
    if (inputObject?.spec?.jobTemplate?.spec?.template?.spec)
      inputObject.spec.jobTemplate.spec.template.spec.imagePullSecrets =
        newImagePullSecrets;
    return;
  }

  if (inputObject?.spec?.template?.spec) {
    inputObject.spec.template.spec.imagePullSecrets = newImagePullSecrets;
    return;
  }
}