package config

import (
	"atom/container"
	"atom/utils"
	"atom/utils/fs"
	"log"

	"github.com/pkg/errors"
	"github.com/rogeecn/fabfile"
	"github.com/spf13/viper"
)

type Config struct {
	App      App
	Captcha  Captcha
	Http     Http
	Log      Log
	Database Database
	Storage  Storage
}

func init() {
	if err := container.Container.Provide(Load); err != nil {
		log.Fatal(err)
	}
}

func Load() (*Config, error) {
	var err error
	confFile := utils.ShareConfigFile
	if confFile == "" {
		confFile, err = fabfile.Find("config.toml")
		if err != nil {
			return nil, err
		}
	}
	path, name, _ := fs.FilePathInfo(confFile)

	viper.SetConfigName(name)     // name of config file (without extension)
	viper.SetConfigType("toml")   // REQUIRED if the config file does not have the extension in the name
	viper.AddConfigPath("$HOME/") // call multiple times to add many search paths
	viper.AddConfigPath(path)     // optionally look for config in the working directory
	viper.AddConfigPath(".")      // optionally look for config in the working directory
	// Find and read the config file
	if err := viper.ReadInConfig(); err != nil { // Handle errors reading the config file
		return nil, errors.Wrapf(err, "read config failed, %s", confFile)
	}

	config := &Config{}
	if err := viper.Unmarshal(&config); err != nil {
		return nil, errors.Wrapf(err, "unmarshal data failed, %s", confFile)
	}

	return config, nil
}