Adding flag for dot-env creation on download.

Author: Otávio Fernandes

cmd/vault-handler/download.go

@@ -3,10 +3,11 @@ package main
 import (
 	"os"
 
-	vh "github.com/otaviof/vault-handler/pkg/vault-handler"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
+
+	vh "github.com/otaviof/vault-handler/pkg/vault-handler"
 )
 
 var downloadCmd = &cobra.Command{
@@ -39,6 +40,7 @@ func init() {
 	flags := downloadCmd.PersistentFlags()
 
 	flags.String("output-dir", ".", "Output directory.")
+	flags.Bool("dot-env", false, "Create a dot-env file with downloaded secrets")
 
 	rootCmd.AddCommand(downloadCmd)
 

cmd/vault-handler/vault_handler.go

@@ -4,10 +4,11 @@ import (
 	"os"
 	"strings"
 
-	vh "github.com/otaviof/vault-handler/pkg/vault-handler"
 	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
+
+	vh "github.com/otaviof/vault-handler/pkg/vault-handler"
 )
 
 var rootCmd = &cobra.Command{
@@ -65,6 +66,7 @@ func configFromEnv() *vh.Config {
 	return &vh.Config{
 		DryRun:        viper.GetBool("dry-run"),
 		OutputDir:     viper.GetString("output-dir"),
+		DotEnv:        viper.GetBool("dot-env"),
 		InputDir:      viper.GetString("input-dir"),
 		VaultAddr:     viper.GetString("vault-addr"),
 		VaultToken:    viper.GetString("vault-token"),

pkg/vault-handler/config.go

@@ -9,6 +9,7 @@ type Config struct {
 	DryRun        bool   // dry-run flag
 	OutputDir     string // output directory path
 	InputDir      string // input directory, when uploading
+	DotEnv        bool   // create a dot-env file with secrets
 	VaultAddr     string // vault api endpoint
 	VaultToken    string // vault token
 	VaultRoleID   string // vault approle role-id
@@ -47,7 +48,7 @@ func (c *Config) ValidateKubernetes() error {
 	if c.Namespace == "" {
 		return fmt.Errorf("namespace is not informed")
 	}
-	if c.KubeConfig != "" && !fileExists(c.KubeConfig) {
+	if c.KubeConfig != "" && !FileExists(c.KubeConfig) {
 		return fmt.Errorf("can't find kube-config file at '%s'", c.KubeConfig)
 	}
 	return nil