From bef7dd7cb7e2cbee52de22004cceae86db3f4790 Mon Sep 17 00:00:00 2001
From: Brad Davidson <brad.davidson@rancher.com>
Date: Wed, 16 Jun 2021 09:38:36 -0700
Subject: [PATCH] Add helm-set-status plugin to handle pending releases (#28)

* Add helm-set-status plugin to handle pending releases

Signed-off-by: Brad Davidson <brad.davidson@rancher.com>
---
 entry              | 21 +++++++++++++++++++--
 package/Dockerfile |  9 +++++++++
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/entry b/entry
index 911ff1f..83fb78c 100755
--- a/entry
+++ b/entry
@@ -33,8 +33,25 @@ helm_update() {
 		exit
 	fi
 
-	# Upgrade only if the status is already deployed, or if a previous upgrade was interrupted
-	if [[ "${STATUS}" =~ ^(deployed|pending-upgrade)$ ]]; then
+	# If a previous helm_v3 operation was interrupted unexpectedly, set it to failed.
+	if [[ "${STATUS}" =~ ^(pending-install|pending-upgrade|pending-rollback)$ ]] && [[ "${HELM}" == "helm_v3" ]]; then
+		echo Previous helm job was interrupted, updating status from ${STATUS} to failed
+		${HELM} set-status ${NAME} failed --namespace ${TARGET_NAMESPACE}
+
+		# Upgrades can be retried; install and rollback will be handled as failure below.
+		# If this upgrade also fails, retries of the job will handle that as failure due to the release state change.
+		if [[ "${STATUS}" == "pending-upgrade" ]]; then
+			echo "Retrying upgrade of ${NAME}"
+			shift 1
+			${HELM} upgrade "$@" ${NAME} ${CHART} ${TIMEOUT_ARG} ${VALUES}
+			exit
+		else
+			STATUS=failed
+		fi
+	fi
+
+	# Upgrade only if the chart is already deployed
+	if [[ "${STATUS}" == "deployed" ]]; then
 		echo Already installed ${NAME}, upgrading
 		shift 1
 		${HELM} upgrade "$@" ${NAME} ${CHART} ${TIMEOUT_ARG} ${VALUES}
diff --git a/package/Dockerfile b/package/Dockerfile
index eeba336..9a5df06 100644
--- a/package/Dockerfile
+++ b/package/Dockerfile
@@ -6,9 +6,18 @@ RUN mv /usr/bin/helm /usr/bin/helm_v2
 RUN curl https://get.helm.sh/helm-v3.5.4-linux-${ARCH}.tar.gz | tar xvzf - --strip-components=1 -C /usr/bin
 RUN mv /usr/bin/helm /usr/bin/helm_v3
 
+FROM golang:1.16-alpine3.12 as plugins
+RUN apk add -U curl ca-certificates build-base binutils-gold
+ARG ARCH
+COPY --from=extract /usr/bin/helm_v3 /usr/bin/helm
+RUN mkdir -p /go/src/github.com/k3s-io/helm-set-status && \
+    curl -L https://github.com/k3s-io/helm-set-status/archive/refs/tags/v0.1.1.tar.gz | tar xvzf - --strip-components=1 -C /go/src/github.com/k3s-io/helm-set-status && \
+    make -C /go/src/github.com/k3s-io/helm-set-status install
+
 FROM alpine:3.12
 RUN apk add -U --no-cache ca-certificates jq bash git
 COPY --from=extract /usr/bin/helm_v2 /usr/bin/helm_v3 /usr/bin/tiller /usr/bin/
+COPY --from=plugins /root/.local/share/helm/plugins/ /root/.local/share/helm/plugins/
 COPY entry /usr/bin/
 ENTRYPOINT ["entry"]
 ENV STABLE_REPO_URL=https://charts.helm.sh/stable/