First CLI implementation.
This commit is contained in:
commit
4d4454a14c
2 changed files with 263 additions and 0 deletions
213
main.go
Normal file
213
main.go
Normal file
|
@ -0,0 +1,213 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"reflect"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
toml "github.com/pelletier/go-toml"
|
||||||
|
// mail "gopkg.in/gomail.v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ServerConfig struct {
|
||||||
|
Address string `toml:address,omitempty`
|
||||||
|
Port int64 `toml:port,omitempty`
|
||||||
|
Encryption string `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: %s\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 debug(active bool, fmt string, msg ...interface{}) {
|
||||||
|
if active {
|
||||||
|
log.Printf(fmt, msg...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func isPresent(slice []string, val string) bool {
|
||||||
|
for _, item := range slice {
|
||||||
|
if item == val {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func isEmpty(value interface{}) bool {
|
||||||
|
t := reflect.TypeOf(value)
|
||||||
|
v := reflect.ValueOf(value)
|
||||||
|
switch v.Kind() {
|
||||||
|
case reflect.Slice:
|
||||||
|
return v.Len() == 1
|
||||||
|
default:
|
||||||
|
return value == reflect.Zero(t).Interface()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func readConfig(configPath, section string) *Config {
|
||||||
|
config := NewConfig()
|
||||||
|
tree, err := toml.LoadFile(configPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Panicln(err)
|
||||||
|
}
|
||||||
|
keys := tree.Keys()
|
||||||
|
|
||||||
|
if !isPresent(keys, section) {
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
sub := tree.Get(section).(*toml.Tree)
|
||||||
|
err = sub.Unmarshal(config)
|
||||||
|
return config
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Config) Validate() error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func merge(key string, value, obj interface{}) {
|
||||||
|
if isEmpty(value) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var elem reflect.Value
|
||||||
|
if reflect.TypeOf(obj) != reflect.TypeOf(reflect.Value{}) {
|
||||||
|
log.Println("Not Value")
|
||||||
|
elem = reflect.ValueOf(obj).Elem()
|
||||||
|
} else {
|
||||||
|
log.Println("Value")
|
||||||
|
elem = obj.(reflect.Value)
|
||||||
|
}
|
||||||
|
field := elem.FieldByName(key)
|
||||||
|
|
||||||
|
if field.CanSet() {
|
||||||
|
switch field.Kind() {
|
||||||
|
case reflect.Int64:
|
||||||
|
field.SetInt(value.(int64))
|
||||||
|
case reflect.Bool:
|
||||||
|
field.SetBool(value.(bool))
|
||||||
|
case reflect.String:
|
||||||
|
field.SetString(value.(string))
|
||||||
|
case reflect.Slice:
|
||||||
|
field.Set(reflect.ValueOf(value))
|
||||||
|
default:
|
||||||
|
merge(key, reflect.ValueOf(value), field)
|
||||||
|
//merge(key, value, field.Pointer())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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 main() {
|
||||||
|
//var err error
|
||||||
|
var configPath, section, serverAddress, encryption, user, password, to, cc, bcc, from, subject string
|
||||||
|
var dbg bool
|
||||||
|
var serverPort_ int
|
||||||
|
var serverPort int64
|
||||||
|
flag.StringVar(&configPath, "conf", "/etc/sendmail.toml", "Path to a config file (defaults to /etc/sendmail.toml)")
|
||||||
|
flag.StringVar(§ion, "section", "default", "Section of the conf to read (defaults to \"default\")")
|
||||||
|
flag.BoolVar(&dbg, "dbg", false, "Enable debugging output")
|
||||||
|
flag.StringVar(&serverAddress, "server-address", "", "The SMTP server address")
|
||||||
|
flag.IntVar(&serverPort_, "server-port", 0, "The SMTP server")
|
||||||
|
flag.StringVar(&encryption, "encryption", "", "The encryption type (no, starttls, tls)")
|
||||||
|
flag.StringVar(&user, "user", "", "The user to authenticate with to the server")
|
||||||
|
flag.StringVar(&password, "password", "", "The password to authenticate with to the server")
|
||||||
|
flag.StringVar(&to, "to", "", "Comma-separated list of recipient(s)")
|
||||||
|
flag.StringVar(&cc, "cc", "", "Comma-separated list of carbon-copy recipient(s)")
|
||||||
|
flag.StringVar(&bcc, "bcc", "", "Comma-separated list of blind carbon-copy recipient(s)")
|
||||||
|
flag.StringVar(&from, "from", "", "Sender of the mail (used as default account user to log in on the SMTP server)")
|
||||||
|
flag.StringVar(&subject, "sub", "", "Subject of the mail")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
serverPort = int64(serverPort_)
|
||||||
|
debug(
|
||||||
|
dbg,
|
||||||
|
`parameters:
|
||||||
|
conf: %s
|
||||||
|
dbg: %t
|
||||||
|
address: %s
|
||||||
|
port: %d
|
||||||
|
encryption: %s
|
||||||
|
user: %s
|
||||||
|
password: %s
|
||||||
|
to: %s
|
||||||
|
from: %s
|
||||||
|
subject: %s`,
|
||||||
|
configPath,
|
||||||
|
dbg,
|
||||||
|
serverAddress,
|
||||||
|
serverPort,
|
||||||
|
encryption,
|
||||||
|
user,
|
||||||
|
password,
|
||||||
|
to,
|
||||||
|
from,
|
||||||
|
subject,
|
||||||
|
)
|
||||||
|
config := readConfig(configPath, section)
|
||||||
|
debug(dbg, "\n%s", *config)
|
||||||
|
|
||||||
|
// config.Server.Merge("Address", serverAddress)
|
||||||
|
// config.Server.Merge("Port", serverPort)
|
||||||
|
// config.Server.Merge("Encryption", encryption)
|
||||||
|
// config.Server.Merge("User", user)
|
||||||
|
// config.Server.Merge("Password", password)
|
||||||
|
config.Merge("Server", &ServerConfig{serverAddress, serverPort, encryption, user, password})
|
||||||
|
config.Merge("From", from)
|
||||||
|
config.Merge("To", strings.Split(to, ","))
|
||||||
|
config.Merge("Cc", strings.Split(cc, ","))
|
||||||
|
config.Merge("Bcc", strings.Split(bcc, ","))
|
||||||
|
config.Merge("Subject", subject)
|
||||||
|
// config.Merge("Text", text)
|
||||||
|
|
||||||
|
debug(dbg, "\n%s", config)
|
||||||
|
|
||||||
|
}
|
50
test_conf.toml
Normal file
50
test_conf.toml
Normal file
|
@ -0,0 +1,50 @@
|
||||||
|
[default]
|
||||||
|
from = "me@domain.com"
|
||||||
|
to = ["test@dom.in","test@examle.com"]
|
||||||
|
cc = ["other@foo.com","name@bar.org"]
|
||||||
|
bcc = ["forever@alone.com"]
|
||||||
|
subject = "My favourite topic"
|
||||||
|
text = '''
|
||||||
|
Hello,
|
||||||
|
I am the crown prince of Nigeria.
|
||||||
|
I need to launder some money, would you help?
|
||||||
|
|
||||||
|
|
||||||
|
Greetings
|
||||||
|
'''
|
||||||
|
|
||||||
|
[default.server]
|
||||||
|
address = "1.3.1.2"
|
||||||
|
port = 587
|
||||||
|
encryption = "tls"
|
||||||
|
user = "faggiano@uccelli.net"
|
||||||
|
password = "yougetit"
|
||||||
|
|
||||||
|
[personal]
|
||||||
|
from = "notme@domain.com"
|
||||||
|
to = ["test@dom.in","test@examle.com"]
|
||||||
|
cc = ["other@foo.com","name@bar.org"]
|
||||||
|
bcc = ["forever@alone.com"]
|
||||||
|
subject = "My favourite topic"
|
||||||
|
text = '''
|
||||||
|
I am an h4X0r. I infected your computer.
|
||||||
|
|
||||||
|
You will follow this instructions or I will publish your
|
||||||
|
most personal pictures.
|
||||||
|
|
||||||
|
https://followthewhiterabbit.com
|
||||||
|
'''
|
||||||
|
|
||||||
|
[personal.server]
|
||||||
|
address = "10.13.12.10"
|
||||||
|
port = 25
|
||||||
|
encryption = "starttls"
|
||||||
|
user = "master@domain.local"
|
||||||
|
password = "yougetitagain"
|
||||||
|
|
||||||
|
[secret]
|
||||||
|
[secret.server]
|
||||||
|
address = "my.server.org"
|
||||||
|
port = 12345
|
||||||
|
encryption = "tls"
|
||||||
|
user = "me"
|
Loading…
Reference in a new issue