Add the possibility to send attachments
This commit is contained in:
parent
e56df6d825
commit
08f79e2e0f
20
config.go
20
config.go
|
@ -3,6 +3,7 @@ package main
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
toml "github.com/pelletier/go-toml"
|
toml "github.com/pelletier/go-toml"
|
||||||
)
|
)
|
||||||
|
@ -40,11 +41,12 @@ type Config struct {
|
||||||
Bcc []string `toml:bcc,omitempty`
|
Bcc []string `toml:bcc,omitempty`
|
||||||
Subject string `toml:subject,omitempty`
|
Subject string `toml:subject,omitempty`
|
||||||
Text string `toml:text,omitempty`
|
Text string `toml:text,omitempty`
|
||||||
|
Attachments []string `toml:attachments,omitempty`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Config) String() string {
|
func (c Config) String() string {
|
||||||
return fmt.Sprintf(
|
msg := fmt.Sprintf(
|
||||||
"From: %s\nTo: %s\nCc: %s\nBcc: %s\nSubject: %s\nText:\n%s\nServer:\n%s",
|
"From: %s\nTo: %s\nCc: %s\nBcc: %s\nSubject: %s\nText:\n%s\nServer:\n%s\n",
|
||||||
c.From,
|
c.From,
|
||||||
c.To,
|
c.To,
|
||||||
c.Cc,
|
c.Cc,
|
||||||
|
@ -53,6 +55,11 @@ func (c Config) String() string {
|
||||||
c.Text,
|
c.Text,
|
||||||
c.Server,
|
c.Server,
|
||||||
)
|
)
|
||||||
|
msg += "Attachments:"
|
||||||
|
for _, attachment := range c.Attachments {
|
||||||
|
msg += fmt.Sprintf(" - %s\n", attachment)
|
||||||
|
}
|
||||||
|
return msg
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewConfig() *Config {
|
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)
|
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 errors.New(msg)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|
4
mail.go
4
mail.go
|
@ -19,6 +19,10 @@ func formatMessage(config *Config) *mail.Message {
|
||||||
}
|
}
|
||||||
m.SetHeader("Subject", config.Subject)
|
m.SetHeader("Subject", config.Subject)
|
||||||
m.SetBody("text/plain", config.Text)
|
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)
|
Debug.F("Message to deliver:\n%s", m)
|
||||||
|
|
||||||
return m
|
return m
|
||||||
|
|
29
main.go
29
main.go
|
@ -2,6 +2,7 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
|
"errors"
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
|
@ -12,6 +13,27 @@ import (
|
||||||
var noVersion = "dev"
|
var noVersion = "dev"
|
||||||
var version string
|
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) {
|
func readFromConsole(result chan string) {
|
||||||
var text, line string
|
var text, line string
|
||||||
var err error
|
var err error
|
||||||
|
@ -73,6 +95,7 @@ func main() {
|
||||||
var encryption, dbg, versionFlag, interactive bool
|
var encryption, dbg, versionFlag, interactive bool
|
||||||
var serverPortAux int
|
var serverPortAux int
|
||||||
var serverPort int64
|
var serverPort int64
|
||||||
|
var attachments attachmentList
|
||||||
|
|
||||||
flag.BoolVar(&versionFlag, "version", false, "Prints the version and exits")
|
flag.BoolVar(&versionFlag, "version", false, "Prints the version and exits")
|
||||||
flag.StringVar(&configPath, "conf", "", "Path to a config file (defaults to /etc/sendmail.toml)")
|
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(&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(&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.StringVar(&subject, "sub", "", "Subject of the mail")
|
||||||
|
flag.Var(&attachments, "attach", "Attachment to the mail (may be repeated)")
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
LogInit(dbg)
|
LogInit(dbg)
|
||||||
|
@ -136,7 +160,8 @@ parameters:
|
||||||
password: %s
|
password: %s
|
||||||
to: %s
|
to: %s
|
||||||
from: %s
|
from: %s
|
||||||
subject: %s`,
|
subject: %s
|
||||||
|
attachments: %s`,
|
||||||
configPath,
|
configPath,
|
||||||
dbg,
|
dbg,
|
||||||
serverAddress,
|
serverAddress,
|
||||||
|
@ -147,6 +172,7 @@ parameters:
|
||||||
to,
|
to,
|
||||||
from,
|
from,
|
||||||
subject,
|
subject,
|
||||||
|
attachments,
|
||||||
)
|
)
|
||||||
|
|
||||||
config := initializeConfig(configPath, section)
|
config := initializeConfig(configPath, section)
|
||||||
|
@ -161,6 +187,7 @@ parameters:
|
||||||
config.Merge("Bcc", toList(bcc))
|
config.Merge("Bcc", toList(bcc))
|
||||||
config.Merge("Subject", subject)
|
config.Merge("Subject", subject)
|
||||||
config.Merge("Text", text)
|
config.Merge("Text", text)
|
||||||
|
config.Merge("Attachments", attachments)
|
||||||
|
|
||||||
Debug.F("---\nPre-validation config\n%s", config)
|
Debug.F("---\nPre-validation config\n%s", config)
|
||||||
err = config.Validate()
|
err = config.Validate()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user