Merge pull request #9 from shipyard/use-env-var-prefix

Use a special prefix for env vars and add a dash replacer for flags
shipyard · Feb 24, 2023 · 44f8603 · 44f8603
2 parents dd88de8 + 6860ce6
commit 44f8603
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -6,6 +6,7 @@ import (
  "log"
  "os"
  "path/filepath"
+ "strings"
 
  "github.com/fatih/color"
  "github.com/spf13/cobra"
@@ -46,6 +47,10 @@ func Execute() {
 }
 
 func init() {
+ replacer := strings.NewReplacer("-", "_")
+ viper.SetEnvKeyReplacer(replacer)
+ viper.SetEnvPrefix("shipyard")
+ viper.AutomaticEnv()
  cobra.OnInitialize(initConfig)
 
  rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.shipyard/config.yaml)")
@@ -79,8 +84,6 @@ func setupCommands() {
 }
 
 func initConfig() {
- viper.AutomaticEnv()
-
  if cfgFile != "" {
  viper.SetConfigFile(cfgFile)
  if err := viper.ReadInConfig(); err != nil {