Add the possibility to send attachments

This commit is contained in:
sfigato 2020-05-22 16:10:21 +02:00
parent e56df6d825
commit 08f79e2e0f
Signed by: blallo
GPG Key ID: 0CBE577C9B72DC3F
3 changed files with 57 additions and 10 deletions

View File

@ -3,6 +3,7 @@ package main
import (
"errors"
"fmt"
"os"
toml "github.com/pelletier/go-toml"
)
@ -40,11 +41,12 @@ type Config struct {
Bcc []string `toml:bcc,omitempty`
Subject string `toml:subject,omitempty`
Text string `toml:text,omitempty`
Attachments []string `toml:attachments,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",
msg := fmt.Sprintf(
"From: %s\nTo: %s\nCc: %s\nBcc: %s\nSubject: %s\nText:\n%s\nServer:\n%s\n",
c.From,
c.To,
c.Cc,
@ -53,6 +55,11 @@ func (c Config) String() string {
c.Text,
c.Server,
)
msg += "Attachments:"
for _, attachment := range c.Attachments {
msg += fmt.Sprintf(" - %s\n", attachment)
}
return msg
}
func NewConfig() *Config {
@ -118,6 +125,15 @@ func (c *Config) Validate() error {
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)
}
}
}
if len(c.Attachments) > 0 {
for _, file := range c.Attachments {
if _, err := os.Lstat(file); err != nil {
msg += fmt.Sprintf("Error with attachment: %s -> %s\n", file, err)
}
}
}
if msg != "" {
return errors.New(msg)
}
return nil

View File

@ -19,6 +19,10 @@ func formatMessage(config *Config) *mail.Message {
}
m.SetHeader("Subject", config.Subject)
m.SetBody("text/plain", config.Text)
for _, attachment := range config.Attachments {
Debug.F("Attacching: %s", attachment)
m.Attach(attachment)
}
Debug.F("Message to deliver:\n%s", m)
return m

29
main.go
View File

@ -2,6 +2,7 @@ package main
import (
"bufio"
"errors"
"flag"
"fmt"
"os"
@ -12,6 +13,27 @@ import (
var noVersion = "dev"
var version string
type attachmentList []string
func (a *attachmentList) String() string {
var str string
for _, attachment := range *a {
str += fmt.Sprintf("%s\n", attachment)
}
return str
}
func (a *attachmentList) Set(file string) error {
if _, err := os.Lstat(file); err != nil {
return err
}
if file == "" {
return errors.New("no file provided")
}
*a = append(*a, file)
return nil
}
func readFromConsole(result chan string) {
var text, line string
var err error
@ -73,6 +95,7 @@ func main() {
var encryption, dbg, versionFlag, interactive bool
var serverPortAux int
var serverPort int64
var attachments attachmentList
flag.BoolVar(&versionFlag, "version", false, "Prints the version and exits")
flag.StringVar(&configPath, "conf", "", "Path to a config file (defaults to /etc/sendmail.toml)")
@ -89,6 +112,7 @@ func main() {
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.Var(&attachments, "attach", "Attachment to the mail (may be repeated)")
flag.Parse()
LogInit(dbg)
@ -136,7 +160,8 @@ parameters:
password: %s
to: %s
from: %s
subject: %s`,
subject: %s
attachments: %s`,
configPath,
dbg,
serverAddress,
@ -147,6 +172,7 @@ parameters:
to,
from,
subject,
attachments,
)
config := initializeConfig(configPath, section)
@ -161,6 +187,7 @@ parameters:
config.Merge("Bcc", toList(bcc))
config.Merge("Subject", subject)
config.Merge("Text", text)
config.Merge("Attachments", attachments)
Debug.F("---\nPre-validation config\n%s", config)
err = config.Validate()