Merge pull request #36 from hhollenstain/exit-early

feat: exit-early flag when provider or secret not found

Author: alexei-led

main.go

@@ -49,6 +49,10 @@ func main() {
 				Usage: "supported secrets manager provider ['aws', 'google']",
 				Value: "aws",
 			},
+			&cli.BoolFlag{
+				Name:    "exit-early",
+				Usage:   "exit when a provider fails or a secret is not found",
+				EnvVars: []string{"EXIT_EARLY"},
 			&cli.StringFlag{
 				Name:  "google-project",
 				Usage: "the google cloud project for secrets without a project prefix",
@@ -137,11 +141,14 @@ func mainCmd(c *cli.Context) error {
 	}
 	if err != nil {
 		log.WithField("provider", c.String("provider")).WithError(err).Error("failed to initialize secrets provider")
+		if c.Bool("exit-early") {
+			os.Exit(1)
+		}
 	}
 
 	// Launch main command
 	var childPid int
-	childPid, err = run(ctx, provider, c.Args().Slice())
+	childPid, err = run(ctx, provider, c.Bool("exit-early"), c.Args().Slice())
 	if err != nil {
 		log.WithError(err).Error("failed to run")
 		os.Exit(1)
@@ -181,7 +188,7 @@ func removeZombies(childPid int) {
 }
 
 // run passed command
-func run(ctx context.Context, provider secrets.Provider, commandSlice []string) (childPid int, err error) {
+func run(ctx context.Context, provider secrets.Provider, exitEarly bool, commandSlice []string) (childPid int, err error) {
 	var commandStr string
 	var argsSlice []string
 
@@ -213,6 +220,10 @@ func run(ctx context.Context, provider secrets.Provider, commandSlice []string)
 		cmd.Env, err = provider.ResolveSecrets(ctx, os.Environ())
 		if err != nil {
 			log.WithError(err).Error("failed to resolve secrets")
+			if exitEarly {
+				log.Error("Exiting early unable to retrieve secrets")
+				os.Exit(1)
+			}
 		}
 	} else {
 		log.Warn("no secrets provider available; using environment without resolving secrets")