120 lines
3.2 KiB
Go
120 lines
3.2 KiB
Go
package config
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io/fs"
|
|
"log/slog"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// envKeyReplacer turns a dotted key into its environment variable's suffix
|
|
// (logging.level to LOGGING_LEVEL, once upper-cased).
|
|
var envKeyReplacer = strings.NewReplacer(".", "_")
|
|
|
|
// EnvVarName returns the environment variable that sets key, e.g.
|
|
// JOBSEARCH_LOGGING_LEVEL for logging.level.
|
|
func EnvVarName(key string) string {
|
|
return EnvPrefix + "_" + strings.ToUpper(envKeyReplacer.Replace(key))
|
|
}
|
|
|
|
// isExplicitlySet reports whether key's value in Viper was set by the config
|
|
// file or by its environment variable, rather than coming from its default.
|
|
// As Viper does, it ignores an empty environment variable. Values set with
|
|
// viper.Set (e.g. by StoreValue) are not counted.
|
|
func isExplicitlySet(key string) bool {
|
|
if viper.InConfig(key) {
|
|
return true
|
|
}
|
|
value, ok := os.LookupEnv(EnvVarName(key))
|
|
return ok && value != ""
|
|
}
|
|
|
|
// loadEnv loads environment variables from a .env file in the working
|
|
// directory, if there is one. One that can't be read or parsed is an error.
|
|
func loadEnv() error {
|
|
err := godotenv.Load()
|
|
if errors.Is(err, fs.ErrNotExist) {
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return fmt.Errorf("invalid .env file: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// loadConfig sets up Viper with the defaults, configFile (or config.* found in
|
|
// EtcConfigPath or the working directory) and JOBSEARCH_* environment
|
|
// variables. A missing config file is only a warning.
|
|
func loadConfig(configFile string) error {
|
|
if configFile != "" {
|
|
// Use config file from the flag.
|
|
viper.SetConfigFile(configFile)
|
|
} 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(envKeyReplacer)
|
|
viper.AutomaticEnv() // read in environment variables that match
|
|
|
|
err := viper.ReadInConfig()
|
|
var notFound viper.ConfigFileNotFoundError
|
|
switch {
|
|
case errors.As(err, ¬Found):
|
|
// No config file was found in any search path.
|
|
slog.Warn("no config file found; using defaults")
|
|
case err != nil:
|
|
// A config file was found but couldn't be read or parsed.
|
|
return err
|
|
default:
|
|
slog.Info("using config file", "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(configFile string, conf *JobSearchConfig) error {
|
|
if err := loadEnv(); err != nil {
|
|
return err
|
|
}
|
|
if conf.Logging.LogLevel() <= slog.LevelDebug {
|
|
printEnv()
|
|
}
|
|
|
|
if err := loadConfig(configFile); 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
|
|
}
|