From e7a72a80c4424b74755f1717ec181b208ea01aeb Mon Sep 17 00:00:00 2001
From: Austin Abro <AustinAbro321@gmail.com>
Date: Sun, 16 Feb 2025 20:38:28 +0000
Subject: [PATCH] set command line flags

Signed-off-by: Austin Abro <AustinAbro321@gmail.com>
---
 pkg/action/install.go  | 4 ++++
 pkg/action/rollback.go | 4 ++++
 pkg/action/upgrade.go  | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/pkg/action/install.go b/pkg/action/install.go
index c6d4f723c..12304d938 100644
--- a/pkg/action/install.go
+++ b/pkg/action/install.go
@@ -175,6 +175,10 @@ func (i *Install) GetRegistryClient() *registry.Client {
 	return i.registryClient
 }
 
+func (i *Install) shouldWait() bool {
+	return i.Wait != ""
+}
+
 func (i *Install) installCRDs(crds []chart.CRD) error {
 	// We do these one file at a time in the order they were read.
 	totalItems := []*resource.Info{}
diff --git a/pkg/action/rollback.go b/pkg/action/rollback.go
index f56052988..920ea58a9 100644
--- a/pkg/action/rollback.go
+++ b/pkg/action/rollback.go
@@ -98,6 +98,10 @@ func (r *Rollback) Run(name string) error {
 	return nil
 }
 
+func (r *Rollback) shouldWait() bool {
+	return !(r.Wait == "")
+}
+
 // prepareRollback finds the previous release and prepares a new release object with
 // the previous release's configuration
 func (r *Rollback) prepareRollback(name string) (*release.Release, *release.Release, bool, error) {
diff --git a/pkg/action/upgrade.go b/pkg/action/upgrade.go
index 3c84570b2..222458e76 100644
--- a/pkg/action/upgrade.go
+++ b/pkg/action/upgrade.go
@@ -208,6 +208,10 @@ func (u *Upgrade) RunWithContext(ctx context.Context, name string, ch chart.Char
 	return res, nil
 }
 
+func (u *Upgrade) shouldWait() bool {
+	return u.Wait != ""
+}
+
 // isDryRun returns true if Upgrade is set to run as a DryRun
 func (u *Upgrade) isDryRun() bool {
 	if u.DryRun || u.DryRunOption == "client" || u.DryRunOption == "server" || u.DryRunOption == "true" {