aerc/commands/compose/send.go

265 lines
6 KiB
Go
Raw Normal View History

package compose
import (
2019-05-14 20:05:29 +02:00
"crypto/tls"
"fmt"
"io"
2019-05-14 20:05:29 +02:00
"net/mail"
"net/url"
2019-07-27 17:19:49 +02:00
"os/exec"
2019-05-14 20:05:29 +02:00
"strings"
2019-05-14 20:18:01 +02:00
"time"
2019-05-14 20:05:29 +02:00
2019-06-08 17:51:26 +02:00
"github.com/emersion/go-imap"
2019-05-14 20:05:29 +02:00
"github.com/emersion/go-sasl"
"github.com/emersion/go-smtp"
2019-05-14 20:18:01 +02:00
"github.com/gdamore/tcell"
"github.com/google/shlex"
"github.com/miolini/datacounter"
"github.com/pkg/errors"
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/widgets"
"git.sr.ht/~sircmpwn/aerc/worker/types"
)
type Send struct{}
func init() {
register(Send{})
}
func (Send) Aliases() []string {
return []string{"send"}
}
func (Send) Complete(aerc *widgets.Aerc, args []string) []string {
return nil
}
func (Send) Execute(aerc *widgets.Aerc, args []string) error {
if len(args) > 1 {
return errors.New("Usage: send")
}
composer, _ := aerc.SelectedTab().(*widgets.Composer)
2019-05-14 20:05:29 +02:00
config := composer.Config()
if config.Outgoing == "" {
return errors.New(
"No outgoing mail transport configured for this account")
}
2019-06-21 20:33:09 +02:00
aerc.Logger().Println("Sending mail")
2019-05-14 20:05:29 +02:00
uri, err := url.Parse(config.Outgoing)
if err != nil {
return errors.Wrap(err, "url.Parse(outgoing)")
2019-05-14 20:05:29 +02:00
}
var (
scheme string
auth string = "plain"
)
2019-07-27 17:19:49 +02:00
if uri.Scheme != "" {
parts := strings.Split(uri.Scheme, "+")
if len(parts) == 1 {
scheme = parts[0]
} else if len(parts) == 2 {
scheme = parts[0]
auth = parts[1]
} else {
return fmt.Errorf("Unknown transfer protocol %s", uri.Scheme)
}
2019-05-14 20:05:29 +02:00
}
header, rcpts, err := composer.PrepareHeader()
2019-05-14 20:05:29 +02:00
if err != nil {
return errors.Wrap(err, "PrepareHeader")
2019-05-14 20:05:29 +02:00
}
if config.From == "" {
return errors.New("No 'From' configured for this account")
}
from, err := mail.ParseAddress(config.From)
if err != nil {
return errors.Wrap(err, "ParseAddress(config.From)")
2019-05-14 20:05:29 +02:00
}
var (
saslClient sasl.Client
conn *smtp.Client
)
switch auth {
case "":
fallthrough
case "none":
saslClient = nil
case "login":
password, _ := uri.User.Password()
saslClient = sasl.NewLoginClient(uri.User.Username(), password)
2019-05-14 20:05:29 +02:00
case "plain":
password, _ := uri.User.Password()
saslClient = sasl.NewPlainClient("", uri.User.Username(), password)
default:
return fmt.Errorf("Unsupported auth mechanism %s", auth)
}
aerc.RemoveTab(composer)
2019-05-14 20:18:01 +02:00
var starttls bool
if starttls_, ok := config.Params["smtp-starttls"]; ok {
starttls = starttls_ == "yes"
}
2019-07-27 17:19:49 +02:00
smtpAsync := func() (int, error) {
2019-05-14 20:18:01 +02:00
switch scheme {
case "smtp":
host := uri.Host
2019-05-20 23:25:12 +02:00
serverName := uri.Host
2019-05-14 20:18:01 +02:00
if !strings.ContainsRune(host, ':') {
host = host + ":587" // Default to submission port
2019-05-20 23:25:12 +02:00
} else {
serverName = host[:strings.IndexRune(host, ':')]
2019-05-14 20:18:01 +02:00
}
conn, err = smtp.Dial(host)
if err != nil {
return 0, errors.Wrap(err, "smtp.Dial")
2019-05-14 20:18:01 +02:00
}
defer conn.Close()
if sup, _ := conn.Extension("STARTTLS"); sup {
if !starttls {
err := errors.New("STARTTLS is supported by this server, " +
"but not set in accounts.conf. " +
"Add smtp-starttls=yes")
return 0, err
}
2019-05-20 23:25:12 +02:00
if err = conn.StartTLS(&tls.Config{
ServerName: serverName,
}); err != nil {
return 0, errors.Wrap(err, "StartTLS")
}
} else {
if starttls {
err := errors.New("STARTTLS requested, but not supported " +
"by this SMTP server. Is someone tampering with your " +
"connection?")
return 0, err
2019-05-14 20:18:01 +02:00
}
}
case "smtps":
host := uri.Host
2019-05-20 23:25:12 +02:00
serverName := uri.Host
2019-05-14 20:18:01 +02:00
if !strings.ContainsRune(host, ':') {
host = host + ":465" // Default to smtps port
2019-05-20 23:25:12 +02:00
} else {
serverName = host[:strings.IndexRune(host, ':')]
2019-05-14 20:18:01 +02:00
}
2019-05-20 23:25:12 +02:00
conn, err = smtp.DialTLS(host, &tls.Config{
ServerName: serverName,
})
2019-05-14 20:18:01 +02:00
if err != nil {
return 0, errors.Wrap(err, "smtp.DialTLS")
2019-05-14 20:18:01 +02:00
}
defer conn.Close()
2019-05-14 20:05:29 +02:00
}
2019-05-14 20:18:01 +02:00
if saslClient != nil {
if err = conn.Auth(saslClient); err != nil {
return 0, errors.Wrap(err, "conn.Auth")
2019-05-14 20:05:29 +02:00
}
}
2019-05-14 20:18:01 +02:00
// TODO: the user could conceivably want to use a different From and sender
if err = conn.Mail(from.Address); err != nil {
return 0, errors.Wrap(err, "conn.Mail")
2019-05-14 20:18:01 +02:00
}
2019-06-21 20:33:09 +02:00
aerc.Logger().Printf("rcpt to: %v", rcpts)
2019-05-14 20:18:01 +02:00
for _, rcpt := range rcpts {
if err = conn.Rcpt(rcpt); err != nil {
return 0, errors.Wrap(err, "conn.Rcpt")
2019-05-14 20:18:01 +02:00
}
2019-05-14 20:05:29 +02:00
}
2019-05-14 20:18:01 +02:00
wc, err := conn.Data()
2019-05-14 20:05:29 +02:00
if err != nil {
return 0, errors.Wrap(err, "conn.Data")
2019-05-14 20:05:29 +02:00
}
2019-05-14 20:18:01 +02:00
defer wc.Close()
ctr := datacounter.NewWriterCounter(wc)
composer.WriteMessage(header, ctr)
return int(ctr.Count()), nil
2019-05-14 20:05:29 +02:00
}
2019-07-27 17:19:49 +02:00
sendmailAsync := func() (int, error) {
args, err := shlex.Split(uri.Path)
if err != nil {
return 0, err
}
if len(args) == 0 {
return 0, fmt.Errorf("no command specified")
}
bin := args[0]
args = append(args[1:], rcpts...)
cmd := exec.Command(bin, args...)
2019-07-27 17:19:49 +02:00
wc, err := cmd.StdinPipe()
if err != nil {
return 0, errors.Wrap(err, "cmd.StdinPipe")
}
defer wc.Close()
go cmd.Run()
ctr := datacounter.NewWriterCounter(wc)
composer.WriteMessage(header, ctr)
return int(ctr.Count()), nil
}
sendAsync := func() (int, error) {
2019-07-27 18:36:30 +02:00
fmt.Println(scheme)
2019-07-27 17:19:49 +02:00
switch scheme {
case "smtp":
2019-07-27 18:36:30 +02:00
fallthrough
2019-07-27 17:19:49 +02:00
case "smtps":
return smtpAsync()
case "":
return sendmailAsync()
}
return 0, errors.New("Unknown scheme")
}
2019-05-14 20:18:01 +02:00
go func() {
aerc.SetStatus("Sending...")
nbytes, err := sendAsync()
if err != nil {
aerc.SetStatus(" "+err.Error()).
Color(tcell.ColorDefault, tcell.ColorRed)
return
}
if config.CopyTo != "" {
aerc.SetStatus("Copying to " + config.CopyTo)
worker := composer.Worker()
r, w := io.Pipe()
worker.PostAction(&types.AppendMessage{
Destination: config.CopyTo,
2019-06-08 17:51:26 +02:00
Flags: []string{imap.SeenFlag},
2019-05-18 21:34:16 +02:00
Date: time.Now(),
Reader: r,
Length: nbytes,
}, func(msg types.WorkerMessage) {
switch msg := msg.(type) {
case *types.Done:
2019-05-17 05:57:38 +02:00
aerc.SetStatus("Message sent.")
r.Close()
composer.Close()
case *types.Error:
aerc.PushStatus(" "+msg.Error.Error(), 10*time.Second).
Color(tcell.ColorDefault, tcell.ColorRed)
r.Close()
composer.Close()
}
})
header, _, _ := composer.PrepareHeader()
composer.WriteMessage(header, w)
w.Close()
} else {
2019-05-17 05:57:38 +02:00
aerc.SetStatus("Message sent.")
composer.Close()
}
2019-05-14 20:18:01 +02:00
}()
return nil
}