jira-bot/pkg/config/config.go

package config

import (
	"errors"
	"log"

	"github.com/spf13/viper"
)

// Load config file from given path
func LoadConfig(file string) (*viper.Viper, error) {

	v := viper.New()
	v.SetConfigName("./" + file)
	v.AddConfigPath(".")
	v.AutomaticEnv()
	if err := v.ReadInConfig(); err != nil {
		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
			return nil, errors.New("config file not found")
		}
		return nil, err
	}

	return v, nil
}

// Parse config file
func ParseConfig(v *viper.Viper) (*Config, error) {
	var c Config
	err := v.Unmarshal(&c)
	if err != nil {
		log.Printf("unable to decode into struct, %v", err)
		return nil, err
	}
	return &c, nil
}

// Get config
func GetConfig(filename string) (*Config, error) {
	cfgFile, err := LoadConfig(filename)
	if err != nil {
		return nil, err
	}
	cfg, err := ParseConfig(cfgFile)
	return cfg, err
}

func GetReply() (*Reply, error) {
	cfgFile, err := LoadConfig("reply")
	if err != nil {
		return nil, err
	}
	reply := Reply{}
	_err := cfgFile.Unmarshal(&reply)
	if _err != nil {
		log.Printf("unable to decode into struct, %v", err)
		return nil, err
	}
	return &reply, nil
}