From cac3dbad6f2b6cacc8c5de90128b54b7d3ceff03 Mon Sep 17 00:00:00 2001
From: Noah Lackstein <noah@lackstein.com>
Date: Fri, 25 Sep 2020 11:11:32 -0400
Subject: [PATCH] feat(helm): Add pre-ready hook to upgrades

Signed-off-by: Noah Lackstein <noah@lackstein.com>
---
 pkg/action/upgrade.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pkg/action/upgrade.go b/pkg/action/upgrade.go
index e7c2aec25..c98494e96 100644
--- a/pkg/action/upgrade.go
+++ b/pkg/action/upgrade.go
@@ -340,6 +340,12 @@ func (u *Upgrade) performUpgrade(originalRelease, upgradedRelease *release.Relea
 		}
 	}
 
+	if !u.DisableHooks {
+		if err := u.cfg.execHook(upgradedRelease, release.HookPreReady, u.Timeout); err != nil {
+			return u.failRelease(upgradedRelease, results.Created, fmt.Errorf("failed pre-ready: %s", err))
+		}
+	}
+
 	if u.Wait {
 		if err := u.cfg.KubeClient.Wait(target, u.Timeout); err != nil {
 			u.cfg.recordRelease(originalRelease)