feat: allow configuration overrides from environment

noodnik2 · Feb 16, 2024 · e116c5a · e116c5a
1 parent 9b17d5f
commit e116c5a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -1,6 +1,8 @@
 package config
 
 import (
+	"strings"
+
 	"github.com/noodnik2/gochat/internal/adapter/chatter"
 	"github.com/noodnik2/gochat/internal/adapter/scriber"
 	"github.com/spf13/viper"
@@ -40,9 +42,15 @@ type Config struct {
 }
 
 func Load() (Config, error) {
+	viper.SetEnvPrefix("gochat")
 	viper.SetConfigName("config-local")
 	viper.AddConfigPath("config")
 
+	// the following enables you to override configuration values with environment variables
+	// prefixed with "GOCHAT_"; e.g., "make run-chat GOCHAT_CHATTER_ADAPTER=Gemini"
+	viper.AutomaticEnv()
+	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
+
 	var cfg Config
 
 	errParse := viper.ReadInConfig()