feat(cli): kuba show now support -o/--output flag (#47)

Supoorted values:

- dotenv (default)
- json
- shell (can be sourced with `source <(kuba show -o shell --env prod)

Author: gorillamoe

cmd/kuba/show.go

@@ -2,6 +2,7 @@ package kuba
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"regexp"
 	"sort"
@@ -17,6 +18,7 @@ var (
 	showEnvironment string
 	showConfigFile  string
 	showSensitive   bool
+	showOutput      string
 )
 
 const (
@@ -62,6 +64,7 @@ func init() {
 	showCmd.Flags().StringVarP(&showEnvironment, "env", "e", "default", "Environment to use (default: default). Provide without value to list available environments.")
 	showCmd.Flags().StringVarP(&showConfigFile, "config", "c", "", "Path to kuba.yaml configuration file")
 	showCmd.Flags().BoolVar(&showSensitive, "sensitive", false, "Redact sensitive values")
+	showCmd.Flags().StringVarP(&showOutput, "output", "o", "dotenv", "Output format: dotenv (default), json, shell")
 	envFlag := showCmd.Flags().Lookup("env")
 	if envFlag != nil {
 		envFlag.NoOptDefVal = showListEnvironmentsValue
@@ -132,13 +135,33 @@ func runShowCommand(patterns []string, listEnvironments bool) error {
 	filteredSecrets := filterSecrets(secrets, patterns)
 	logger.Debug("Filtered secrets", "original_count", len(secrets), "filtered_count", len(filteredSecrets))
 
-	// Display secrets
+	// Prepare secrets for output
+	displaySecrets := make(map[string]string, len(filteredSecrets))
 	for key, value := range filteredSecrets {
 		displayValue := value
 		if showSensitive {
 			displayValue = maskSecret(value)
 		}
-		fmt.Printf("%s=%s\n", key, displayValue)
+		displaySecrets[key] = displayValue
+	}
+
+	switch showOutput {
+	case "dotenv":
+		for _, key := range getSortedKeys(displaySecrets) {
+			fmt.Printf("%s=%s\n", key, displaySecrets[key])
+		}
+	case "shell":
+		for _, key := range getSortedKeys(displaySecrets) {
+			fmt.Printf("export %s=%s\n", key, displaySecrets[key])
+		}
+	case "json":
+		payload, err := json.MarshalIndent(displaySecrets, "", "  ")
+		if err != nil {
+			return fmt.Errorf("failed to format secrets as json: %w", err)
+		}
+		fmt.Println(string(payload))
+	default:
+		return fmt.Errorf("invalid output format '%s': must be one of: dotenv, json, shell", showOutput)
 	}
 
 	return nil
@@ -193,3 +216,12 @@ func getSortedEnvironmentNames(cfg *config.KubaConfig) []string {
 	sort.Strings(names)
 	return names
 }
+
+func getSortedKeys(values map[string]string) []string {
+	keys := make([]string, 0, len(values))
+	for key := range values {
+		keys = append(keys, key)
+	}
+	sort.Strings(keys)
+	return keys
+}

cmd/kuba/show_test.go

@@ -1,6 +1,7 @@
 package kuba
 
 import (
+	"encoding/json"
 	"io"
 	"os"
 	"strings"
@@ -15,6 +16,7 @@ func TestRunShowCommandListsEnvironments(t *testing.T) {
 		showEnvironment = "default"
 		showConfigFile = ""
 		showSensitive = false
+		showOutput = "dotenv"
 	})
 
 	tmpFile, err := os.CreateTemp("", "kuba-show-*.yaml")
@@ -66,6 +68,7 @@ func TestRunShowCommandUsesProvidedEnvironment(t *testing.T) {
 		showEnvironment = "default"
 		showConfigFile = ""
 		showSensitive = false
+		showOutput = "dotenv"
 	})
 
 	tmpFile, err := os.CreateTemp("", "kuba-show-*.yaml")
@@ -117,6 +120,7 @@ func TestRunShowCommandConsumesArgWhenEnvFlagNoOptSet(t *testing.T) {
 		showEnvironment = "default"
 		showConfigFile = ""
 		showSensitive = false
+		showOutput = "dotenv"
 	})
 
 	tmpFile, err := os.CreateTemp("", "kuba-show-*.yaml")
@@ -162,3 +166,107 @@ staging:
 	output := strings.TrimSpace(string(outputBytes))
 	assert.Equal(t, "BAR=bar", output)
 }
+
+func TestRunShowCommandOutputsJSON(t *testing.T) {
+	t.Cleanup(func() {
+		showEnvironment = "default"
+		showConfigFile = ""
+		showSensitive = false
+		showOutput = "dotenv"
+	})
+
+	tmpFile, err := os.CreateTemp("", "kuba-show-*.yaml")
+	require.NoError(t, err)
+	t.Cleanup(func() { _ = os.Remove(tmpFile.Name()) })
+
+	configContent := `
+default:
+  provider: local
+  env:
+    FOO:
+      value: foo
+    BAR:
+      value: bar
+`
+	_, err = tmpFile.WriteString(configContent)
+	require.NoError(t, err)
+	require.NoError(t, tmpFile.Close())
+
+	showEnvironment = "default"
+	showConfigFile = tmpFile.Name()
+	showOutput = "json"
+
+	originalStdout := os.Stdout
+	r, w, err := os.Pipe()
+	require.NoError(t, err)
+	os.Stdout = w
+
+	runErr := runShowCommand(nil, false)
+	require.NoError(t, runErr)
+
+	require.NoError(t, w.Close())
+	os.Stdout = originalStdout
+
+	outputBytes, err := io.ReadAll(r)
+	require.NoError(t, err)
+	require.NoError(t, r.Close())
+
+	var output map[string]string
+	require.NoError(t, json.Unmarshal(outputBytes, &output))
+
+	assert.Equal(t, map[string]string{
+		"BAR": "bar",
+		"FOO": "foo",
+	}, output)
+}
+
+func TestRunShowCommandOutputsShell(t *testing.T) {
+	t.Cleanup(func() {
+		showEnvironment = "default"
+		showConfigFile = ""
+		showSensitive = false
+		showOutput = "dotenv"
+	})
+
+	tmpFile, err := os.CreateTemp("", "kuba-show-*.yaml")
+	require.NoError(t, err)
+	t.Cleanup(func() { _ = os.Remove(tmpFile.Name()) })
+
+	configContent := `
+default:
+  provider: local
+  env:
+    FOO:
+      value: foo
+    BAR:
+      value: bar
+`
+	_, err = tmpFile.WriteString(configContent)
+	require.NoError(t, err)
+	require.NoError(t, tmpFile.Close())
+
+	showEnvironment = "default"
+	showConfigFile = tmpFile.Name()
+	showOutput = "shell"
+
+	originalStdout := os.Stdout
+	r, w, err := os.Pipe()
+	require.NoError(t, err)
+	os.Stdout = w
+
+	runErr := runShowCommand(nil, false)
+	require.NoError(t, runErr)
+
+	require.NoError(t, w.Close())
+	os.Stdout = originalStdout
+
+	outputBytes, err := io.ReadAll(r)
+	require.NoError(t, err)
+	require.NoError(t, r.Close())
+
+	output := strings.Split(strings.TrimSpace(string(outputBytes)), "\n")
+	assert.Equal(t, []string{
+		"export BAR=bar",
+		"export FOO=foo",
+	}, output)
+}