mg-transport-core/core/config/config.go

package config

import (
	"io/ioutil"
	"path/filepath"
	"time"

	"github.com/op/go-logging"
	"gopkg.in/yaml.v2"
)

// Configuration settings data structure.
type Configuration interface {
	GetVersion() string
	GetSentryDSN() string
	GetLogLevel() logging.Level
	GetHTTPConfig() HTTPServerConfig
	GetZabbixConfig() ZabbixConfig
	GetDBConfig() DatabaseConfig
	GetAWSConfig() AWS
	GetTransportInfo() InfoInterface
	GetHTTPClientConfig() *HTTPClientConfig
	GetUpdateInterval() int
	IsDebug() bool
}

// InfoInterface transport settings data structure.
type InfoInterface interface {
	GetName() string
	GetCode() string
	GetLogoPath() string
	GetSecret() string
}

// Config struct.
type Config struct {
	HTTPClientConfig *HTTPClientConfig `yaml:"http_client"`
	ConfigAWS        AWS               `yaml:"config_aws"`
	TransportInfo    Info              `yaml:"transport_info"`
	HTTPServer       HTTPServerConfig  `yaml:"http_server"`
	ZabbixConfig     ZabbixConfig      `yaml:"zabbix"`
	Version          string            `yaml:"version"`
	SentryDSN        string            `yaml:"sentry_dsn"`
	Database         DatabaseConfig    `yaml:"database"`
	UpdateInterval   int               `yaml:"update_interval"`
	LogLevel         logging.Level     `yaml:"log_level"`
	Debug            bool              `yaml:"debug"`
}

// Info struct.
type Info struct {
	Name     string `yaml:"name"`
	Code     string `yaml:"code"`
	LogoPath string `yaml:"logo_path"`
	Secret   string `yaml:"secret"`
}

// AWS struct.
type AWS struct {
	AccessKeyID     string `yaml:"access_key_id"`
	SecretAccessKey string `yaml:"secret_access_key"`
	Endpoint        string `yaml:"endpoint"`
	Region          string `yaml:"region"`
	Bucket          string `yaml:"bucket"`
	FolderName      string `yaml:"folder_name"`
	ContentType     string `yaml:"content_type"`
}

// DatabaseConfig struct.
type DatabaseConfig struct {
	Connection         interface{} `yaml:"connection"`
	TablePrefix        string      `yaml:"table_prefix"`
	MaxOpenConnections int         `yaml:"max_open_connections"`
	MaxIdleConnections int         `yaml:"max_idle_connections"`
	ConnectionLifetime int         `yaml:"connection_lifetime"`
	Logging            bool        `yaml:"logging"`
}

// HTTPClientConfig struct.
type HTTPClientConfig struct {
	SSLVerification *bool         `yaml:"ssl_verification"`
	MockAddress     string        `yaml:"mock_address"`
	MockedDomains   []string      `yaml:"mocked_domains"`
	Timeout         time.Duration `yaml:"timeout"`
}

// HTTPServerConfig struct.
type HTTPServerConfig struct {
	Host   string `yaml:"host"`
	Listen string `yaml:"listen"`
}

// ZabbixConfig contains information about Zabbix connection.
type ZabbixConfig struct {
	ServerHost   string `yaml:"server_host"`
	Host         string `yaml:"host"`
	ServerPort   int    `yaml:"server_port"`
	MetricPrefix string `yaml:"metric_prefix"`
	Interval     uint64 `yaml:"interval"`
}

// NewConfig reads configuration file and returns config instance
// Usage:
//
//	NewConfig("config.yml")
func NewConfig(path string) *Config {
	return (&Config{}).LoadConfig(path)
}

// LoadConfig read & load configuration file.
func (c *Config) LoadConfig(path string) *Config {
	return c.LoadConfigFromData(c.GetConfigData(path))
}

// LoadConfigFromData loads config from byte sequence.
func (c *Config) LoadConfigFromData(data []byte) *Config {
	if err := yaml.Unmarshal(data, c); err != nil {
		panic(err)
	}

	return c
}

// GetConfigData returns config file data in form of byte sequence.
func (c *Config) GetConfigData(path string) []byte {
	var err error

	path, err = filepath.Abs(path)
	if err != nil {
		panic(err)
	}

	source, err := ioutil.ReadFile(path)
	if err != nil {
		panic(err)
	}

	return source
}

// GetSentryDSN sentry connection dsn.
func (c Config) GetSentryDSN() string {
	return c.SentryDSN
}

// GetVersion transport version.
func (c Config) GetVersion() string {
	return c.Version
}

// GetLogLevel log level.
func (c Config) GetLogLevel() logging.Level {
	return c.LogLevel
}

// GetTransportInfo transport basic data.
func (c Config) GetTransportInfo() InfoInterface {
	return c.TransportInfo
}

// IsDebug debug flag.
func (c Config) IsDebug() bool {
	return c.Debug
}

// GetAWSConfig AWS configuration.
func (c Config) GetAWSConfig() AWS {
	return c.ConfigAWS
}

// GetDBConfig database configuration.
func (c Config) GetDBConfig() DatabaseConfig {
	return c.Database
}

// GetHTTPConfig server configuration.
func (c Config) GetHTTPConfig() HTTPServerConfig {
	return c.HTTPServer
}

// GetZabbixConfig returns zabbix configuration.
func (c Config) GetZabbixConfig() ZabbixConfig {
	return c.ZabbixConfig
}

// GetUpdateInterval user data update interval.
func (c Config) GetUpdateInterval() int {
	return c.UpdateInterval
}

// GetHTTPClientConfig returns http client config.
func (c Config) GetHTTPClientConfig() *HTTPClientConfig {
	return c.HTTPClientConfig
}

// GetName transport name.
func (t Info) GetName() string {
	return t.Name
}

// GetCode transport code.
func (t Info) GetCode() string {
	return t.Code
}

// GetLogoPath transport logo.
func (t Info) GetLogoPath() string {
	return t.LogoPath
}

// GetSecret returns secret.
func (t Info) GetSecret() string {
	return t.Secret
}

// IsSSLVerificationEnabled returns SSL verification flag (default is true).
func (h *HTTPClientConfig) IsSSLVerificationEnabled() bool {
	if h.SSLVerification == nil {
		return true
	}

	return *h.SSLVerification
}