add helm download

kkpan11 · pull · Oct 22, 2023 · Oct 22, 2023 · Nov 3, 2023 · Nov 3, 2023
commit 4b345883940b04d323152d382e053ed906d5b5ae
diff --git a/src/_nebari/constants.py b/src/_nebari/constants.py
@@ -1,12 +1,12 @@
 CURRENT_RELEASE = "2024.7.1"
 
+HELM_VERSION = "v3.15.3"
+KUSTOMIZE_VERSION = "5.4.3"
 # NOTE: Terraform cannot be upgraded further due to Hashicorp licensing changes
 # implemented in August 2023.
 # https://www.hashicorp.com/license-faq
 TERRAFORM_VERSION = "1.5.7"
 
-KUSTOMIZE_VERSION = "5.4.3"
-
 # 04-kubernetes-ingress
 DEFAULT_TRAEFIK_IMAGE_TAG = "2.9.1"
 

diff --git a/src/_nebari/provider/helm.py b/src/_nebari/provider/helm.py
@@ -0,0 +1,68 @@
+import logging
+import subprocess
+import tempfile
+from pathlib import Path
+
+from _nebari import constants
+from _nebari.utils import run_subprocess_cmd
+
+logger = logging.getLogger(__name__)
+
+
+class HelmException(Exception):
+    pass
+
+
+def download_helm_binary(version=constants.HELM_VERSION) -> Path:
+    filename_directory = Path(tempfile.gettempdir()) / "helm" / version
+    filename_path = filename_directory / "helm"
+
+    if not filename_directory.is_dir():
+        filename_directory.mkdir(parents=True)
+
+    if not filename_path.is_file():
+        logger.info(
+            "downloading and extracting Helm binary version %s to path=%s",
+            constants.HELM_VERSION,
+            filename_path,
+        )
+        install_script = subprocess.run(
+            [
+                "curl",
+                "-s",
+                "https://raw.githubusercontent.com/helm/helm/main/scripts/get-helm-3",
+            ],
+            stdout=subprocess.PIPE,
+            check=True,
+        )
+        subprocess.run(
+            [
+                "bash",
+                "-s",
+                "--",
+                "-v",
+                constants.HELM_VERSION,
+                "--no-sudo",
+            ],
+            input=install_script.stdout,
+            check=True,
+            env={"HELM_INSTALL_DIR": str(filename_directory)},
+        )
+
+    filename_path.chmod(0o555)
+    return filename_path
+
+
+def run_helm_subprocess(processargs, **kwargs) -> None:
+    helm_path = download_helm_binary()
+    logger.info("helm at %s", helm_path)
+    if run_subprocess_cmd([helm_path] + processargs, **kwargs):
+        raise HelmException("Helm returned an error")
+
+
+def version() -> str:
+    helm_path = download_helm_binary()
+    logger.info("checking helm=%s version", helm_path)
+
+    version_output = subprocess.check_output([helm_path, "version"]).decode("utf-8")
+    return version_output
diff --git a/src/_nebari/provider/kustomize.py b/src/_nebari/provider/kustomize.py
@@ -47,7 +47,6 @@ def download_kustomize_binary(version=constants.KUSTOMIZE_VERSION) -> Path:
 
 def run_kustomize_subprocess(processargs, **kwargs) -> None:
     kustomize_path = download_kustomize_binary()
-    logger.info("kustomize at %s", kustomize_path)
     if run_subprocess_cmd([kustomize_path] + processargs, **kwargs):
         raise KustomizeException("Kustomize returned an error")
 

diff --git a/src/_nebari/stages/base.py b/src/_nebari/stages/base.py
@@ -9,7 +9,7 @@
 from kubernetes import client, config
 from kubernetes.client.rest import ApiException
 
-from _nebari.provider import kubernetes, kustomize, terraform
+from _nebari.provider import helm, kubernetes, kustomize, terraform
 from _nebari.stages.tf_objects import NebariTerraformState
 from nebari.hookspecs import NebariStage
 
@@ -81,6 +81,8 @@ def render(self) -> Dict[pathlib.Path, str]:
                     "-o",
                     f"{temp_dir}",
                     "--enable-helm",
+                    "--helm-command",
+                    f"{helm.download_helm_binary()}",
                     f"{self.template_directory}",
                 ]
             )