package run import ( "fmt" "github.com/pelotech/drone-helm3/internal/env" ) // Upgrade is an execution step that calls `helm upgrade` when executed. type Upgrade struct { *config chart string release string chartVersion string dryRun bool wait bool values string stringValues string valuesFiles []string reuseValues bool timeout string force bool atomic bool cleanupOnFail bool certs *repoCerts createNamespace bool skipCrds bool cmd cmd } // NewUpgrade creates an Upgrade using fields from the given Config. No validation is performed at this time. func NewUpgrade(cfg env.Config) *Upgrade { return &Upgrade{ config: newConfig(cfg), chart: cfg.Chart, release: cfg.Release, chartVersion: cfg.ChartVersion, dryRun: cfg.DryRun, wait: cfg.Wait, values: cfg.Values, stringValues: cfg.StringValues, valuesFiles: cfg.ValuesFiles, reuseValues: cfg.ReuseValues, timeout: cfg.Timeout, force: cfg.Force, atomic: cfg.AtomicUpgrade, cleanupOnFail: cfg.CleanupOnFail, certs: newRepoCerts(cfg), createNamespace: cfg.CreateNamespace, skipCrds: cfg.SkipCrds, } } // Execute executes the `helm upgrade` command. func (u *Upgrade) Execute() error { return u.cmd.Run() } // Prepare gets the Upgrade ready to execute. func (u *Upgrade) Prepare() error { if u.chart == "" { return fmt.Errorf("chart is required") } if u.release == "" { return fmt.Errorf("release is required") } args := u.globalFlags() args = append(args, "upgrade", "--install") if u.chartVersion != "" { args = append(args, "--version", u.chartVersion) } if u.dryRun { args = append(args, "--dry-run") } if u.wait { args = append(args, "--wait") } if u.reuseValues { args = append(args, "--reuse-values") } if u.timeout != "" { args = append(args, "--timeout", u.timeout) } if u.force { args = append(args, "--force") } if u.atomic { args = append(args, "--atomic") } if u.cleanupOnFail { args = append(args, "--cleanup-on-fail") } if u.values != "" { args = append(args, "--set", u.values) } if u.stringValues != "" { args = append(args, "--set-string", u.stringValues) } if u.createNamespace { args = append(args, "--create-namespace") } if u.skipCrds { args = append(args, "--skip-crds") } for _, vFile := range u.valuesFiles { args = append(args, "--values", vFile) } args = append(args, u.certs.flags()...) args = append(args, u.release, u.chart) u.cmd = command(helmBin, args...) u.cmd.Stdout(u.stdout) u.cmd.Stderr(u.stderr) if u.debug { fmt.Fprintf(u.stderr, "Generated command: '%s'\n", u.cmd.String()) } return nil }