config.go

package main

import (
	"os"
	"runtime"
	"strings"

	log "github.com/Sirupsen/logrus"
	gcfg "gopkg.in/gcfg.v1"
)

type GeneralConf struct {
	Debug bool
}

type Config struct {
	General  GeneralConf
	Mqtt     MqttConf     `gcfg:"mqforward-mqtt"`
	InfluxDB InfluxDBConf `gcfg:"mqforward-influxdb"`
}

func UserHomeDir() string {
	if runtime.GOOS == "windows" {
		home := os.Getenv("HOMEDRIVE") + os.Getenv("HOMEPATH")
		if home == "" {
			home = os.Getenv("USERPROFILE")
		}
		return home
	}
	return os.Getenv("HOME")
}

func ExpandPath(path string) string {
	return strings.Replace(path, "~", UserHomeDir(), 1)
}

func LoadConf(path string) (MqttConf, InfluxDBConf, error) {
	path = ExpandPath(path)

	var cfg Config
	err := gcfg.ReadFileInto(&cfg, path)
	if err != nil {
		return MqttConf{}, InfluxDBConf{}, err
	}

	if cfg.General.Debug {
		log.SetLevel(log.DebugLevel)
	}

	return cfg.Mqtt, cfg.InfluxDB, nil
}