Adding upstream version 1.1.0.upstream/1.1.0upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 135 insertions, 0 deletions

diff --git a/pkg/config/config.go b/pkg/config/config.go
new file mode 100644
index 0000000..1388ba9
--- /dev/null
+++ b/pkg/config/config.go
@@ -0,0 +1,135 @@
+package config
+
+import (
+	"crypto/tls"
+	"crypto/x509"
+	"github.com/creasty/defaults"
+	"github.com/goccy/go-yaml"
+	"github.com/jessevdk/go-flags"
+	"github.com/pkg/errors"
+	"io/ioutil"
+	"os"
+)
+
+// Config defines Icinga DB config.
+type Config struct {
+	Database  Database  `yaml:"database"`
+	Redis     Redis     `yaml:"redis"`
+	Logging   Logging   `yaml:"logging"`
+	Retention Retention `yaml:"retention"`
+}
+
+// Validate checks constraints in the supplied configuration and returns an error if they are violated.
+func (c *Config) Validate() error {
+	if err := c.Database.Validate(); err != nil {
+		return err
+	}
+	if err := c.Redis.Validate(); err != nil {
+		return err
+	}
+	if err := c.Logging.Validate(); err != nil {
+		return err
+	}
+	if err := c.Retention.Validate(); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+// Flags defines CLI flags.
+type Flags struct {
+	// Version decides whether to just print the version and exit.
+	Version bool `long:"version" description:"print version and exit"`
+	// Config is the path to the config file
+	Config string `short:"c" long:"config" description:"path to config file" required:"true" default:"/etc/icingadb/config.yml"`
+}
+
+// FromYAMLFile returns a new Config value created from the given YAML config file.
+func FromYAMLFile(name string) (*Config, error) {
+	f, err := os.Open(name)
+	if err != nil {
+		return nil, errors.Wrap(err, "can't open YAML file "+name)
+	}
+	defer f.Close()
+
+	c := &Config{}
+	d := yaml.NewDecoder(f)
+
+	if err := defaults.Set(c); err != nil {
+		return nil, errors.Wrap(err, "can't set config defaults")
+	}
+
+	if err := d.Decode(c); err != nil {
+		return nil, errors.Wrap(err, "can't parse YAML file "+name)
+	}
+
+	if err := c.Validate(); err != nil {
+		return nil, errors.Wrap(err, "invalid configuration")
+	}
+
+	return c, nil
+}
+
+// ParseFlags parses CLI flags and
+// returns a Flags value created from them.
+func ParseFlags() (*Flags, error) {
+	f := &Flags{}
+	parser := flags.NewParser(f, flags.Default)
+
+	if _, err := parser.Parse(); err != nil {
+		return nil, errors.Wrap(err, "can't parse CLI flags")
+	}
+
+	return f, nil
+}
+
+// TLS provides TLS configuration options for Redis and Database.
+type TLS struct {
+	Enable   bool   `yaml:"tls"`
+	Cert     string `yaml:"cert"`
+	Key      string `yaml:"key"`
+	Ca       string `yaml:"ca"`
+	Insecure bool   `yaml:"insecure"`
+}
+
+// MakeConfig assembles a tls.Config from t and serverName.
+func (t *TLS) MakeConfig(serverName string) (*tls.Config, error) {
+	if !t.Enable {
+		return nil, nil
+	}
+
+	tlsConfig := &tls.Config{}
+	if t.Cert == "" {
+		if t.Key != "" {
+			return nil, errors.New("private key given, but client certificate missing")
+		}
+	} else if t.Key == "" {
+		return nil, errors.New("client certificate given, but private key missing")
+	} else {
+		crt, err := tls.LoadX509KeyPair(t.Cert, t.Key)
+		if err != nil {
+			return nil, errors.Wrap(err, "can't load X.509 key pair")
+		}
+
+		tlsConfig.Certificates = []tls.Certificate{crt}
+	}
+
+	if t.Insecure {
+		tlsConfig.InsecureSkipVerify = true
+	} else if t.Ca != "" {
+		raw, err := ioutil.ReadFile(t.Ca)
+		if err != nil {
+			return nil, errors.Wrap(err, "can't read CA file")
+		}
+
+		tlsConfig.RootCAs = x509.NewCertPool()
+		if !tlsConfig.RootCAs.AppendCertsFromPEM(raw) {
+			return nil, errors.New("can't parse CA file")
+		}
+	}
+
+	tlsConfig.ServerName = serverName
+
+	return tlsConfig, nil
+}