aerc/commands/compose/send.go

204 lines
4.4 KiB
Go
Raw Normal View History

package compose
import (
2019-05-14 20:05:29 +02:00
"crypto/tls"
"errors"
2019-05-14 20:05:29 +02:00
"fmt"
"io"
2019-05-14 20:05:29 +02:00
"net/mail"
"net/url"
"strings"
2019-05-14 20:18:01 +02:00
"time"
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/miolini/datacounter"
"git.sr.ht/~sircmpwn/aerc2/widgets"
"git.sr.ht/~sircmpwn/aerc2/worker/types"
)
func init() {
register("send", SendMessage)
}
func SendMessage(aerc *widgets.Aerc, args []string) error {
if len(args) > 1 {
return errors.New("Usage: send-message")
}
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")
}
uri, err := url.Parse(config.Outgoing)
if err != nil {
2019-05-14 20:05:29 +02:00
return err
}
var (
scheme string
auth string = "plain"
)
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)
}
header, rcpts, err := composer.PrepareHeader()
2019-05-14 20:05:29 +02:00
if err != nil {
return err
}
if config.From == "" {
return errors.New("No 'From' configured for this account")
}
from, err := mail.ParseAddress(config.From)
if err != nil {
return err
}
var (
saslClient sasl.Client
conn *smtp.Client
)
switch auth {
case "":
fallthrough
case "none":
saslClient = nil
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"
}
sendAsync := func() (int, error) {
2019-05-14 20:18:01 +02:00
tlsConfig := &tls.Config{
// TODO: ask user first
InsecureSkipVerify: true,
2019-05-14 20:05:29 +02:00
}
2019-05-14 20:18:01 +02:00
switch scheme {
case "smtp":
host := uri.Host
if !strings.ContainsRune(host, ':') {
host = host + ":587" // Default to submission port
}
conn, err = smtp.Dial(host)
if err != nil {
return 0, err
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-14 20:18:01 +02:00
if err = conn.StartTLS(tlsConfig); err != nil {
return 0, err
}
} 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
if !strings.ContainsRune(host, ':') {
host = host + ":465" // Default to smtps port
}
conn, err = smtp.DialTLS(host, tlsConfig)
if err != nil {
return 0, err
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
// TODO: sendmail
if saslClient != nil {
if err = conn.Auth(saslClient); err != nil {
return 0, err
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, err
2019-05-14 20:18:01 +02:00
}
for _, rcpt := range rcpts {
if err = conn.Rcpt(rcpt); err != nil {
return 0, err
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, err
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-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,
Flags: []string{},
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
}