101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/roboll/helmfile/helmexec"
|
|
"github.com/roboll/helmfile/pkg/app"
|
|
"github.com/roboll/helmfile/state"
|
|
"github.com/urfave/cli"
|
|
"go.uber.org/zap"
|
|
"os/exec"
|
|
"strings"
|
|
"syscall"
|
|
)
|
|
|
|
func VisitAllDesiredStates(c *cli.Context, converge func(*state.HelmState, helmexec.Interface, app.Context) (bool, []error)) error {
|
|
a, fileOrDir, err := InitAppEntry(c, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx := app.NewContext()
|
|
|
|
convergeWithHelmBinary := func(st *state.HelmState, helm helmexec.Interface) (bool, []error) {
|
|
if c.GlobalString("helm-binary") != "" {
|
|
helm.SetHelmBinary(c.GlobalString("helm-binary"))
|
|
}
|
|
return converge(st, helm, ctx)
|
|
}
|
|
|
|
err = a.VisitDesiredStates(fileOrDir, a.Selectors, convergeWithHelmBinary)
|
|
|
|
return toCliError(err)
|
|
}
|
|
|
|
func InitAppEntry(c *cli.Context, reverse bool) (*app.App, string, error) {
|
|
if c.NArg() > 0 {
|
|
cli.ShowAppHelp(c)
|
|
return nil, "", fmt.Errorf("err: extraneous arguments: %s", strings.Join(c.Args(), ", "))
|
|
}
|
|
|
|
fileOrDir := c.GlobalString("file")
|
|
kubeContext := c.GlobalString("kube-context")
|
|
namespace := c.GlobalString("namespace")
|
|
selectors := c.GlobalStringSlice("selector")
|
|
logger := c.App.Metadata["logger"].(*zap.SugaredLogger)
|
|
|
|
env := c.GlobalString("environment")
|
|
if env == "" {
|
|
env = state.DefaultEnv
|
|
}
|
|
|
|
app := app.Init(&app.App{
|
|
KubeContext: kubeContext,
|
|
Logger: logger,
|
|
Reverse: reverse,
|
|
Env: env,
|
|
Namespace: namespace,
|
|
Selectors: selectors,
|
|
})
|
|
|
|
return app, fileOrDir, nil
|
|
}
|
|
|
|
func FindAndIterateOverDesiredStatesUsingFlagsWithReverse(c *cli.Context, reverse bool, converge func(*state.HelmState, helmexec.Interface, app.Context) []error) error {
|
|
a, fileOrDir, err := InitAppEntry(c, reverse)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx := app.NewContext()
|
|
|
|
convergeWithHelmBinary := func(st *state.HelmState, helm helmexec.Interface) []error {
|
|
if c.GlobalString("helm-binary") != "" {
|
|
helm.SetHelmBinary(c.GlobalString("helm-binary"))
|
|
}
|
|
return converge(st, helm, ctx)
|
|
}
|
|
|
|
err = a.VisitDesiredStatesWithReleasesFiltered(fileOrDir, convergeWithHelmBinary)
|
|
|
|
return toCliError(err)
|
|
}
|
|
|
|
func toCliError(err error) error {
|
|
if err != nil {
|
|
switch e := err.(type) {
|
|
case *app.NoMatchingHelmfileError:
|
|
return cli.NewExitError(e.Error(), 2)
|
|
case *exec.ExitError:
|
|
// Propagate any non-zero exit status from the external command like `helm` that is failed under the hood
|
|
status := e.Sys().(syscall.WaitStatus)
|
|
return cli.NewExitError(e.Error(), status.ExitStatus())
|
|
case *state.DiffError:
|
|
return cli.NewExitError(e.Error(), e.Code)
|
|
default:
|
|
return cli.NewExitError(e.Error(), 1)
|
|
}
|
|
}
|
|
return err
|
|
}
|