package main

import (
	"flag"
	"fmt"

	"github.com/spf13/viper"
)

type MQTTConfig struct {
	MainConf      *viper.Viper
	RateLimitConf *viper.Viper
}

func (conf *MQTTConfig) GetViper(flags *flag.FlagSet, name string) (*viper.Viper, error) {
	v := viper.New()
	file := flags.String(name, "", name+" config file path")
	if *file != "" {
		v.SetConfigFile(*file)
	} else {
		v.SetConfigName(name)
		v.AddConfigPath("conf")
		v.SetConfigType("toml")
	}
	if err := v.ReadInConfig(); err != nil {
		return nil, fmt.Errorf("%s:%s", name, err.Error())
	}

	return v, nil
}

func (conf *MQTTConfig) LoadConfigFile(flags *flag.FlagSet) error {
	var err error

	conf.MainConf, err = conf.GetViper(flags, "main")
	if err != nil {
		return err
	}

	conf.RateLimitConf, err = conf.GetViper(flags, "ratelimit")
	if err != nil {
		return err
	}

	return nil
}

func (conf *MQTTConfig) GetConf(name string) *viper.Viper {
	item := conf.MainConf
	switch name {
	case "main":
		item = conf.MainConf
	case "ratelimit":
		item = conf.RateLimitConf
	}

	return item
}