sendmail/config.go

125 lines
3.2 KiB
Go
Raw Normal View History

package main
import (
"errors"
"fmt"
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, error) {
config := NewConfig()
tree, err := toml.LoadFile(configPath)
if err != nil {
return config, err
}
keys := tree.Keys()
if !IsPresent(keys, section) {
return config, errors.New("missing section")
}
sub := tree.Get(section).(*toml.Tree)
err = sub.Unmarshal(config)
return config, err
}
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
}