Adding format options for env

Author: morfien101

build.sh

@@ -2,8 +2,8 @@
 
 # Set version
 VERSION_MAJOR=0
-VERSION_MINOR=3
-VERSION_PATCH=2
+VERSION_MINOR=4
+VERSION_PATCH=0
 VERSION_SPECIAL=
 VERSION=""
 

main.go

@@ -23,6 +23,8 @@ func main() {
 	flagPath := flag.String("path", "", "Parameter Store path.")
 	flagBase64 := flag.Bool("base64", false, "Base64 encode collected values.")
 	flagFormat := flag.String("format", "line", fmt.Sprintf("Format for output. Supported values: %s.", strings.Join(parameterstore.ValidFormats(), ",")))
+	flagUpperCase := flag.Bool("upper-case", false, "Upper case the path. Only works with format 'env'.")
+	flagPrefix := flag.String("prefix", "", "Prefix the keys with this value. Only works with format 'env'.")
 	flagRecursive := flag.Bool("recursive", false, "Look up all keys in branch.")
 	flagDecrypt := flag.Bool("decrypt", false, "Request decrypted keys.")
 	flagAccessKey := flag.String("access-key", "", "Access key for AWS API.")
@@ -34,6 +36,7 @@ func main() {
 	flagFileOutput := flag.String("f", "", "Output to specified file.")
 	flagHelp := flag.Bool("h", false, "Help menu.")
 	flagVersion := flag.Bool("v", false, "Show application Version.")
+
 	flag.Parse()
 
 	if *flagVersion {
@@ -81,12 +84,16 @@ func main() {
 	var output string
 
 	if *flagRecursive {
-		psMap, err := ps.CollectPath()
+		psMap, err := ps.CollectPath(*flagUpperCase)
 		if err != nil {
 			log.Fatalf("Failed to read from parameter store. Error: %s", err)
 		}
-
-		formattedOutput, err := ps.FormatOutput(psMap, *flagFormat)
+		formatOptions := parameterstore.FormatOptions{
+			Format:    *flagFormat,
+			Prefix:    *flagPrefix,
+			UpperCase: *flagUpperCase,
+		}
+		formattedOutput, err := ps.FormatOutput(psMap, formatOptions)
 		if err != nil {
 			log.Fatal(err)
 		}

parameterStore/formatting.go

@@ -6,9 +6,16 @@ import (
 )
 
 var (
-	formats = []string{"line", "json", "pretty-json", "yaml", "env"}
+	formats             = []string{"line", "json", "pretty-json", "yaml", "env"}
+	globalformatOptions = FormatOptions{}
 )
 
+type FormatOptions struct {
+	Format    string
+	Prefix    string
+	UpperCase bool
+}
+
 func ValidFormats() []string {
 	return formats
 }
@@ -25,6 +32,20 @@ func FormatValidation(formatString string) bool {
 	return valid
 }
 
+func upperIfNeeded(s string) string {
+	if globalformatOptions.UpperCase {
+		return strings.ToUpper(s)
+	}
+	return s
+}
+
+func prefixIfNeeded(s string) string {
+	if globalformatOptions.Prefix != "" {
+		return fmt.Sprintf("%s%s", globalformatOptions.Prefix, s)
+	}
+	return s
+}
+
 func lineFormat(data map[string]string) []byte {
 	output := []string{}
 	for key, value := range data {
@@ -39,7 +60,7 @@ func envFormat(data map[string]string) []byte {
 
 	for key, value := range data {
 		brokenKeys := strings.Split(key, "/")
-		out = append(out, fmt.Sprintf("%s=%s", brokenKeys[len(brokenKeys)-1], value))
+		out = append(out, fmt.Sprintf("%s=%s", upperIfNeeded(prefixIfNeeded(brokenKeys[len(brokenKeys)-1])), value))
 	}
 
 	return []byte(strings.Join(out, "\n"))

parameterStore/parameterstore.go

@@ -39,7 +39,7 @@ func New(session *session.Session, path string, recursive, decrypt, includePath,
 	return ps
 }
 
-func (ps *ParameterStore) CollectPath() (map[string]string, error) {
+func (ps *ParameterStore) CollectPath(upperCase bool) (map[string]string, error) {
 	inputObjects := ps.pathInput("")
 	values, err := ps.values(inputObjects)
 	if err != nil {
@@ -145,8 +145,9 @@ func (ps *ParameterStore) b64(data string) string {
 	return base64.StdEncoding.EncodeToString([]byte(data))
 }
 
-func (ps ParameterStore) FormatOutput(data map[string]string, format string) ([]byte, error) {
-	switch format {
+func (ps ParameterStore) FormatOutput(data map[string]string, formatOptions FormatOptions) ([]byte, error) {
+	globalformatOptions = formatOptions
+	switch globalformatOptions.Format {
 	case "json":
 		return json.Marshal(convertTree(data))
 	case "pretty-json":
@@ -158,6 +159,6 @@ func (ps ParameterStore) FormatOutput(data map[string]string, format string) ([]
 	case "env":
 		return envFormat(data), nil
 	default:
-		return []byte{}, fmt.Errorf("output format '%s' is not valid", format)
+		return []byte{}, fmt.Errorf("output format '%s' is not valid", globalformatOptions.Format)
 	}
 }