package main import ( "errors" "fmt" "os" toml "github.com/pelletier/go-toml" ) type Validation struct { Param string CmdFlag string ConfFlag string } type ServerConfig struct { Address string `toml:address,omitempty` Port int64 `toml:port,omitempty` Encryption bool `toml:encryption,omitempty` User string `toml:user,omitempty` Password string `toml:password,omitempty` } func (s ServerConfig) String() string { return fmt.Sprintf( "\tAddress: %s\n\tPort: %d\n\tEncryption: %t\n\tUser: %s\n\tPassword: %s\n", s.Address, s.Port, s.Encryption, s.User, s.Password, ) } type Config struct { Server *ServerConfig `toml:server,omitempty` From string `toml:from,omitempty` To []string `toml:to,omitempty` Cc []string `toml:cc,omitempty` Bcc []string `toml:bcc,omitempty` Subject string `toml:subject,omitempty` Text string `toml:text,omitempty` } func (c Config) String() string { return fmt.Sprintf( "From: %s\nTo: %s\nCc: %s\nBcc: %s\nSubject: %s\nText:\n%s\nServer:\n%s", c.From, c.To, c.Cc, c.Bcc, c.Subject, c.Text, c.Server, ) } func NewConfig() *Config { server := &ServerConfig{} config := &Config{} config.Server = server return config } func (s *ServerConfig) Merge(key string, value interface{}) { Merge(key, value, s) } func (c *Config) Merge(key string, value interface{}) { Merge(key, value, c) } func readConfig(configPath, section string) *Config { config := NewConfig() tree, err := toml.LoadFile(configPath) if err != nil { Error.F("Error in parsing the configuration\n%s", err) os.Exit(2) } keys := tree.Keys() if !IsPresent(keys, section) { return config } sub := tree.Get(section).(*toml.Tree) err = sub.Unmarshal(config) return config } func checkValidity(validations *[]Validation, obj interface{}, name, cmd, param string) { if IsEmpty(obj) { *validations = append(*validations, Validation{name, cmd, param}) } } func (c *Config) Validate() error { var validations = []Validation{} var msg string checkValidity(&validations, c.Server.Address, "server address", "server-address", "server.address") checkValidity(&validations, c.Server.Port, "server port", "server-port", "server.port") if !c.Server.Encryption { Warning.Ln("Warning: not using encryption!") } checkValidity(&validations, c.Server.User, "username", "user", "server.user") checkValidity(&validations, c.Server.Password, "password", "password", "server.password") checkValidity(&validations, c.From, "from", "from", "from") checkValidity(&validations, c.To, "to", "to", "to") checkValidity(&validations, c.Subject, "subject", "sub", "subject") checkValidity(&validations, c.Text, "body", "", "text") Debug.F("Lengths:\n\tTo: %d\n\tCc: %d\n\tBcc: %d", len(c.To), len(c.Cc), len(c.Bcc)) if len(validations) > 0 { for _, v := range validations { if v.CmdFlag == "" { msg += fmt.Sprintf("%s: pass a value either via stdin or in configuration file section (%s)\n", v.Param, v.ConfFlag) } else { msg += fmt.Sprintf("%s: pass a value either via command line (-%s) or in configuration file section (%s)\n", v.Param, v.CmdFlag, v.ConfFlag) } } return errors.New(msg) } return nil }