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 GetLogFormat() string 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"` LogFormat string `yaml:"log_format"` 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 } func (c Config) GetLogFormat() string { return c.LogFormat } // 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 }