package config import ( "fmt" "log/slog" "github.com/joho/godotenv" "github.com/spf13/viper" ) // Reads in dotenv file for configuration func loadEnv() error { if err := godotenv.Load(); err != nil { return fmt.Errorf("invalid .env file: %w", err) } return nil } // Reads in config file and ENV variables if set to form a configuration. func loadConfig(cfgFile string) error { if cfgFile != "" { // Use config file from the flag. viper.SetConfigFile(cfgFile) } else { // Search config in /etc or local directory. viper.AddConfigPath(EtcConfigPath) viper.AddConfigPath(".") viper.SetConfigName("config") } // Load default configuration LoadDefaultConfig() viper.SetEnvPrefix(EnvPrefix) viper.AutomaticEnv() // read in environment variables that match err := viper.ReadInConfig() if err != nil { if _, ok := err.(viper.ConfigFileNotFoundError); ok { // Indicates an explicitly set config file is not found or // that no config file was found in any search path slog.Warn("No config file found!") } else { // Config file was found but another error was produced return err } } else { // Config file found and successfully parsed slog.Info("Using config file: " + viper.ConfigFileUsed()) } return nil } // Initilises environment file configuration and general app configuration func Init(cfgFile string, conf *JobSearchConfig) error { if err := loadEnv(); err != nil { return err } if conf.Logging.LogLevel() <= slog.LevelDebug { printEnv() } if err := loadConfig(cfgFile); err != nil { return err } // Load configuration from Viper, ensure -debug and -quiet are retained oldLogLevel := conf.Logging.LogLevel() LoadFromViper(conf) if oldLogLevel == slog.LevelDebug { conf.Logging.Level = "debug" } if oldLogLevel == slog.LevelError { conf.Logging.Level = "error" } if conf.Logging.LogLevel() <= slog.LevelDebug { printLoadedConfig() printConfig(*conf) } return nil }