config.go

package main

import (
	"fmt"

	"github.com/spf13/viper"
)

// Config used to load configuration
type Config struct {
	storageDirectory string
	authEnabled      bool
	username         string
	password         string
	port             int
	ip               string
}

func (c *Config) loadConfig(envfile string) {
	viper.SetConfigFile(".env.dist")
	viper.SetConfigType("env")
	viper.AutomaticEnv()

	viper.SetDefault("tf_storage_dir", "./store")
	viper.SetDefault("tf_auth_enabled", false)
	viper.SetDefault("tf_username", "admin")
	viper.SetDefault("tf_password", "admin")
	viper.SetDefault("tf_port", 8080)
	viper.SetDefault("tf_ip", "127.0.0.1")

	if err := viper.ReadInConfig(); err != nil {
		logger.Debugf("Error while reading config file %s", err)
	}

	viper.SetConfigFile(envfile)
	if err := viper.MergeInConfig(); err != nil {
		logger.Debugf("Error while reading config file %s", err)
	}

	c.storageDirectory = viper.GetString("tf_storage_dir")
	c.authEnabled = viper.GetBool("tf_auth_enabled")
	c.username = viper.GetString("tf_username")
	c.password = viper.GetString("tf_password")
	c.port = viper.GetInt("tf_port")
	c.ip = viper.GetString("tf_ip")
}

func (c *Config) getAuthMap() map[string]string {
	authData := make(map[string]string)

	authData[c.username] = c.password

	return authData
}

func (c *Config) getAddr() string {
	return fmt.Sprintf("%s:%d", c.ip, c.port)
}