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
}