92 lines
2.3 KiB
Go
92 lines
2.3 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
"strings"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// loadEnv loads environment variables from a .env file in the working
|
|
// directory.
|
|
func loadEnv() error {
|
|
if err := godotenv.Load(); err != nil {
|
|
return fmt.Errorf("invalid .env file: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// loadConfig sets up Viper with the defaults, cfgFile (or config.* found in
|
|
// EtcConfigPath or the working directory) and JOBSEARCH_* environment
|
|
// variables. A missing config file is only a warning.
|
|
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.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
|
|
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
|
|
}
|
|
|
|
// Init loads the .env file and the configuration (see loadConfig) into conf.
|
|
// A debug or error log level already set in conf, from --debug or --quiet,
|
|
// is kept over the loaded one.
|
|
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
|
|
}
|