Add prefix to env variable lookup, replace . with _

e.g. $PACKWIZ_INIT_REINIT is equivalent to the -r flag
packwiz · Jul 6, 2023 · beef8c6 · beef8c6
1 parent 99d0af1
commit beef8c6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmd/root.go b/cmd/root.go
@@ -6,6 +6,7 @@ import (
 	"github.com/spf13/pflag"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
@@ -92,7 +93,10 @@ func initConfig() {
 		viper.SetConfigName(".packwiz")
 	}
 
-	viper.AutomaticEnv() // read in environment variables that match
+	// Read in environment variables that match
+	viper.SetEnvPrefix("packwiz")
+	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
+	viper.AutomaticEnv()
 
 	// If a config file is found, read it in.
 	if err := viper.ReadInConfig(); err == nil {