refactor: replace hyphens and dots in environment variable keys (#25)

appleboy · web-flow · commit ae81a78d2f23 · 2023-03-12T15:22:46.000+08:00
- Added import for the "strings" package - Defined a replacer for "-" and "." with "_" in variable "replacer" - Added a call to "SetEnvKeyReplacer" for the "viper" object in the "initConfig" function. fix #16
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -6,6 +6,7 @@ import (
 	"log"
 	"os"
 	"path"
+	"strings"
 
 	"github.com/appleboy/com/file"
 	"github.com/spf13/cobra"
@@ -20,7 +21,8 @@ var rootCmd = &cobra.Command{
 
 // Used for flags.
 var (
-	cfgFile string
+	cfgFile  string
+	replacer = strings.NewReplacer("-", "_", ".", "_")
 )
 
 func init() {
@@ -66,6 +68,7 @@ func initConfig() {
 		}
 	}
 
+	viper.SetEnvKeyReplacer(replacer)
 	viper.AutomaticEnv()
 
 	if err := viper.ReadInConfig(); err != nil {

Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,7 @@ import (`
`6`	`6`	`"log"`
`7`	`7`	`"os"`
`8`	`8`	`"path"`
	`9`	`+ "strings"`
`9`	`10`
`10`	`11`	`"github.com/appleboy/com/file"`
`11`	`12`	`"github.com/spf13/cobra"`
`@@ -20,7 +21,8 @@ var rootCmd = &cobra.Command{`
`20`	`21`
`21`	`22`	`// Used for flags.`
`22`	`23`	`var (`
`23`		`- cfgFile string`
	`24`	`+ cfgFile string`
	`25`	`+ replacer = strings.NewReplacer("-", "_", ".", "_")`
`24`	`26`	`)`
`25`	`27`
`26`	`28`	`func init() {`
`@@ -66,6 +68,7 @@ func initConfig() {`
`66`	`68`	`}`
`67`	`69`	`}`
`68`	`70`
	`71`	`+ viper.SetEnvKeyReplacer(replacer)`
`69`	`72`	`viper.AutomaticEnv()`
`70`	`73`
`71`	`74`	`if err := viper.ReadInConfig(); err != nil {`