From 617c24ffc53d9ba99731519ee1983647411073bd Mon Sep 17 00:00:00 2001
From: Yonas Berhe <yberhe@Yonass-MacBook-Pro.local>
Date: Tue, 10 Sep 2024 09:28:46 -0700
Subject: [PATCH] add support for additional rancher helm repos

---
 cypress/jenkins/init.sh | 29 +++++++++++++++++++++++++----
 1 file changed, 25 insertions(+), 4 deletions(-)

diff --git a/cypress/jenkins/init.sh b/cypress/jenkins/init.sh
index 862c31b0d6..2a35ca0df3 100755
--- a/cypress/jenkins/init.sh
+++ b/cypress/jenkins/init.sh
@@ -30,6 +30,7 @@ DASHBOARD_REPO="${DASHBOARD_REPO:-rancher/dashboard.git}"
 DASHBOARD_BRANCH="${DASHBOARD_BRANCH:-master}"
 GITHUB_URL="https://github.com/"
 RANCHER_TYPE="${RANCHER_TYPE:-local}"
+RANCHER_REPO="${RANCHER_REPO:-latest}"
 HELM_VERSION="${HELM_VERSION:-3.13.2}"
 NODEJS_VERSION="${NODEJS_VERSION:-14.19.1}"
 CYPRESS_VERSION="${CYPRESS_VERSION:-13.2.0}"
@@ -92,13 +93,33 @@ create_initial_clusters() {
     TARFILE="helm-v${HELM_VERSION}-linux-amd64.tar.gz"
     curl -L -o "${TARFILE}" "https://get.helm.sh/${TARFILE}"
     tar -C "${WORKSPACE}/bin" --strip-components=1 -xzf "${TARFILE}"
-    helm repo add rancher-latest https://releases.rancher.com/server-charts/latest
-    helm repo update
+    if [[ -n "${RANCHER_REPO}" ]]; then
+      if [[ "${RANCHER_REPO}" == "prime" ]]; then
+        RANCHER_CHART_URL=https://charts.rancher.com/server-charts/"${RANCHER_REPO}"
+        helm repo add rancher-"${RANCHER_REPO}" "${RANCHER_CHART_URL}"
+        helm repo update
+      elif [[ "${RANCHER_REPO}" == "alpha" ]]; then
+        RANCHER_CHART_URL=https://releases.rancher.com/server-charts/"${RANCHER_REPO}"
+        helm repo add rancher-"${RANCHER_REPO}" "${RANCHER_CHART_URL}"
+        helm repo update
+      elif [[ "${RANCHER_REPO}" == "stable" ]]; then
+        RANCHER_CHART_URL=https://releases.rancher.com/server-charts/"${RANCHER_REPO}"
+        helm repo add rancher-"${RANCHER_REPO}" "${RANCHER_CHART_URL}"
+        helm repo update
+      else
+        RANCHER_CHART_URL=https://releases.rancher.com/server-charts/latest
+        helm repo add rancher-latest "${RANCHER_CHART_URL}"
+        helm repo update
+      fi
+      corral config vars set rancher_chart_repo "${RANCHER_REPO}"
+      url_string=$(echo "${RANCHER_CHART_URL}" | grep -o '.*s')
+      corral config vars set rancher_chart_url "${url_string}"
+    fi
     version_string=$(echo "${RANCHER_IMAGE_TAG}" | cut -f1 -d"-")
     if [[ "${RANCHER_IMAGE_TAG}" == "head" ]]; then
-        RANCHER_VERSION=$(helm search repo rancher-latest --devel --versions | sed -n '1!p' | head -1 | cut -f2 | tr -d '[:space:]')
+      RANCHER_VERSION=$(helm search repo "${RANCHER_REPO}" --devel --versions | sed -n '1!p' | head -1 | cut -f2 | tr -d '[:space:]')
     else
-        RANCHER_VERSION=$(helm search repo rancher-latest --devel --versions | grep "${version_string}" | head -n 1 | cut -f2 | tr -d '[:space:]')
+      RANCHER_VERSION=$(helm search repo "${RANCHER_REPO}" --devel --versions | grep "${version_string}" | head -n 1 | cut -f2 | tr -d '[:space:]')
     fi
     corral config vars set rancher_image_tag "${RANCHER_IMAGE_TAG}"
   fi