set command line flags

Signed-off-by: Austin Abro <AustinAbro321@gmail.com>

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{}

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) {

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" {