ryanc/bb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Respect the debug flag in the environment

Browse Source
This commit is contained in:
Ryan Cavicchioni 2022-08-28 09:45:34 -05:00
parent 7d071ebe0b
commit fbd655b2da
Signed by: ryanc
GPG Key ID: 877EEDAF9245103D
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
bot/config.go
View File

@ -10,6 +10,7 @@ var (
type ( type (
Config struct { Config struct {
Debug bool `mapstructure:"debug"`
Handler HandlerConfig `mapstructure:"handler"` Handler HandlerConfig `mapstructure:"handler"`
Prefix string `mapstructure:"prefix"` Prefix string `mapstructure:"prefix"`
DiscordToken string `mapstructure:"discord_token"` DiscordToken string `mapstructure:"discord_token"`

9
cmd/bb/main.go
View File

@ -83,10 +83,6 @@ func setupConfig() {
pflag.Parse() pflag.Parse()
viper.BindPFlags(pflag.CommandLine) viper.BindPFlags(pflag.CommandLine)
if viper.GetBool("debug") {
log.SetLevel(log.DebugLevel)
}
viper.SetEnvPrefix("BEEPBOOP") viper.SetEnvPrefix("BEEPBOOP")
viper.AutomaticEnv() viper.AutomaticEnv()
@ -96,6 +92,7 @@ func setupConfig() {
err = viper.ReadInConfig() err = viper.ReadInConfig()
viper.BindEnv("DEBUG")
viper.BindEnv("DISCORD_TOKEN") viper.BindEnv("DISCORD_TOKEN")
viper.BindEnv("OPEN_WEATHER_MAP_TOKEN") viper.BindEnv("OPEN_WEATHER_MAP_TOKEN")
@ -107,4 +104,8 @@ func setupConfig() {
if err != nil { if err != nil {
log.Fatalf("unable to decode into struct: %v", err) log.Fatalf("unable to decode into struct: %v", err)
} }
if viper.GetBool("debug") {
log.SetLevel(log.DebugLevel)
}
} }