2019-06-02 05:15:04 +00:00
|
|
|
package msg
|
2019-05-16 16:15:34 +00:00
|
|
|
|
|
|
|
import (
|
2019-11-03 12:51:14 +00:00
|
|
|
"bytes"
|
2019-05-16 16:15:34 +00:00
|
|
|
"errors"
|
|
|
|
"fmt"
|
2019-05-16 16:39:22 +00:00
|
|
|
"io"
|
2019-05-16 16:15:34 +00:00
|
|
|
"strings"
|
|
|
|
|
2019-05-18 19:34:16 +00:00
|
|
|
"git.sr.ht/~sircmpwn/getopt"
|
2019-05-16 16:15:34 +00:00
|
|
|
|
2020-06-19 15:58:08 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/lib"
|
2020-08-19 10:06:02 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/lib/format"
|
2019-07-08 02:43:58 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/models"
|
2019-05-18 00:57:10 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/widgets"
|
2020-11-10 18:57:09 +00:00
|
|
|
"github.com/emersion/go-message/mail"
|
2019-05-16 16:15:34 +00:00
|
|
|
)
|
|
|
|
|
2019-06-27 17:33:11 +00:00
|
|
|
type reply struct{}
|
|
|
|
|
2019-05-16 16:15:34 +00:00
|
|
|
func init() {
|
2019-06-27 17:33:11 +00:00
|
|
|
register(reply{})
|
|
|
|
}
|
|
|
|
|
2019-09-03 19:34:03 +00:00
|
|
|
func (reply) Aliases() []string {
|
2019-08-18 09:33:13 +00:00
|
|
|
return []string{"reply"}
|
2019-06-27 17:33:11 +00:00
|
|
|
}
|
|
|
|
|
2019-09-03 19:34:03 +00:00
|
|
|
func (reply) Complete(aerc *widgets.Aerc, args []string) []string {
|
2019-06-27 17:33:11 +00:00
|
|
|
return nil
|
2019-05-16 16:15:34 +00:00
|
|
|
}
|
|
|
|
|
2019-09-03 19:34:03 +00:00
|
|
|
func (reply) Execute(aerc *widgets.Aerc, args []string) error {
|
2019-11-03 12:51:14 +00:00
|
|
|
opts, optind, err := getopt.Getopts(args, "aqT:")
|
2019-05-16 16:39:22 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-06-08 18:57:56 +00:00
|
|
|
if optind != len(args) {
|
2019-11-03 12:51:14 +00:00
|
|
|
return errors.New("Usage: reply [-aq -T <template>]")
|
2019-05-16 16:15:34 +00:00
|
|
|
}
|
2019-05-16 16:39:22 +00:00
|
|
|
var (
|
|
|
|
quote bool
|
|
|
|
replyAll bool
|
2019-11-03 12:51:14 +00:00
|
|
|
template string
|
2019-05-16 16:39:22 +00:00
|
|
|
)
|
|
|
|
for _, opt := range opts {
|
|
|
|
switch opt.Option {
|
|
|
|
case 'a':
|
|
|
|
replyAll = true
|
|
|
|
case 'q':
|
|
|
|
quote = true
|
2019-11-03 12:51:14 +00:00
|
|
|
case 'T':
|
|
|
|
template = opt.Value
|
2019-05-16 16:39:22 +00:00
|
|
|
}
|
|
|
|
}
|
2019-05-16 16:15:34 +00:00
|
|
|
|
2019-06-02 05:15:04 +00:00
|
|
|
widget := aerc.SelectedTab().(widgets.ProvidesMessage)
|
|
|
|
acct := widget.SelectedAccount()
|
2019-11-03 12:51:14 +00:00
|
|
|
|
2019-06-02 05:15:04 +00:00
|
|
|
if acct == nil {
|
|
|
|
return errors.New("No account selected")
|
|
|
|
}
|
2019-05-16 18:16:45 +00:00
|
|
|
conf := acct.AccountConfig()
|
2020-08-19 10:06:02 +00:00
|
|
|
from, err := format.ParseAddress(conf.From)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-08-26 06:13:38 +00:00
|
|
|
aliases, err := format.ParseAddressList(conf.Aliases)
|
2020-08-20 17:22:50 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-08-26 06:13:38 +00:00
|
|
|
|
2019-06-02 05:15:04 +00:00
|
|
|
store := widget.Store()
|
2019-07-14 07:42:24 +00:00
|
|
|
if store == nil {
|
|
|
|
return errors.New("Cannot perform action. Messages still loading")
|
|
|
|
}
|
2019-07-10 00:04:21 +00:00
|
|
|
msg, err := widget.SelectedMessage()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2019-05-16 16:15:34 +00:00
|
|
|
|
2020-08-26 06:13:38 +00:00
|
|
|
// figure out the sending from address if we have aliases
|
|
|
|
if len(aliases) != 0 {
|
|
|
|
rec := newAddrSet()
|
|
|
|
rec.AddList(msg.Envelope.To)
|
|
|
|
rec.AddList(msg.Envelope.Cc)
|
|
|
|
// test the from first, it has priority over any present alias
|
|
|
|
if rec.Contains(from) {
|
|
|
|
// do nothing
|
|
|
|
} else {
|
|
|
|
for _, a := range aliases {
|
|
|
|
if rec.Contains(a) {
|
|
|
|
from = a
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-05-16 16:15:34 +00:00
|
|
|
var (
|
2020-11-10 18:57:09 +00:00
|
|
|
to []*mail.Address
|
|
|
|
cc []*mail.Address
|
2019-05-16 16:15:34 +00:00
|
|
|
)
|
2020-08-20 17:22:50 +00:00
|
|
|
|
2020-08-26 06:13:38 +00:00
|
|
|
recSet := newAddrSet() // used for de-duping
|
2020-08-20 17:22:50 +00:00
|
|
|
|
2020-08-26 06:13:38 +00:00
|
|
|
if len(msg.Envelope.ReplyTo) != 0 {
|
|
|
|
to = msg.Envelope.ReplyTo
|
|
|
|
} else {
|
|
|
|
to = msg.Envelope.From
|
|
|
|
}
|
|
|
|
recSet.AddList(to)
|
2020-08-20 17:22:50 +00:00
|
|
|
|
2020-08-26 06:13:38 +00:00
|
|
|
if replyAll {
|
|
|
|
// order matters, due to the deduping
|
|
|
|
// in order of importance, first parse the To, then the Cc header
|
|
|
|
|
|
|
|
// we add our from address, so that we don't self address ourselves
|
|
|
|
recSet.Add(from)
|
|
|
|
|
2020-11-10 18:57:09 +00:00
|
|
|
envTos := make([]*mail.Address, 0, len(msg.Envelope.To))
|
2020-08-26 06:13:38 +00:00
|
|
|
for _, addr := range msg.Envelope.To {
|
|
|
|
if recSet.Contains(addr) {
|
|
|
|
continue
|
2020-05-04 17:17:23 +00:00
|
|
|
}
|
2020-08-26 06:13:38 +00:00
|
|
|
envTos = append(envTos, addr)
|
2020-05-04 17:17:23 +00:00
|
|
|
}
|
2020-08-26 06:13:38 +00:00
|
|
|
recSet.AddList(envTos)
|
|
|
|
to = append(to, envTos...)
|
|
|
|
|
|
|
|
for _, addr := range msg.Envelope.Cc {
|
|
|
|
//dedupe stuff from the to/from headers
|
|
|
|
if recSet.Contains(addr) {
|
|
|
|
continue
|
2019-05-16 18:16:45 +00:00
|
|
|
}
|
2020-08-26 06:13:38 +00:00
|
|
|
cc = append(cc, addr)
|
2019-05-16 18:16:45 +00:00
|
|
|
}
|
2020-08-26 06:13:38 +00:00
|
|
|
recSet.AddList(cc)
|
2019-05-16 16:15:34 +00:00
|
|
|
}
|
|
|
|
|
2019-05-16 16:42:46 +00:00
|
|
|
var subject string
|
2019-09-07 18:42:17 +00:00
|
|
|
if !strings.HasPrefix(strings.ToLower(msg.Envelope.Subject), "re: ") {
|
2019-08-18 09:33:13 +00:00
|
|
|
subject = "Re: " + msg.Envelope.Subject
|
2019-05-16 16:42:46 +00:00
|
|
|
} else {
|
2019-08-18 09:33:13 +00:00
|
|
|
subject = msg.Envelope.Subject
|
2019-05-16 16:42:46 +00:00
|
|
|
}
|
2019-05-16 16:15:34 +00:00
|
|
|
|
2019-07-22 23:29:07 +00:00
|
|
|
defaults := map[string]string{
|
2020-08-19 10:06:02 +00:00
|
|
|
"To": format.FormatAddresses(to),
|
|
|
|
"Cc": format.FormatAddresses(cc),
|
2020-11-10 18:57:09 +00:00
|
|
|
"From": format.AddressForHumans(from),
|
2019-07-22 23:29:07 +00:00
|
|
|
"Subject": subject,
|
|
|
|
"In-Reply-To": msg.Envelope.MessageId,
|
|
|
|
}
|
2020-01-08 20:44:14 +00:00
|
|
|
original := models.OriginalMail{}
|
2019-07-22 23:29:07 +00:00
|
|
|
|
2019-11-03 12:51:14 +00:00
|
|
|
addTab := func() error {
|
|
|
|
if template != "" {
|
2020-08-19 10:06:02 +00:00
|
|
|
original.From = format.FormatAddresses(msg.Envelope.From)
|
2020-06-26 07:25:53 +00:00
|
|
|
original.Date = msg.Envelope.Date
|
2019-11-03 12:51:14 +00:00
|
|
|
}
|
2019-05-25 18:52:57 +00:00
|
|
|
|
2020-04-24 09:42:21 +00:00
|
|
|
composer, err := widgets.NewComposer(aerc, acct, aerc.Config(),
|
2020-01-08 20:44:14 +00:00
|
|
|
acct.AccountConfig(), acct.Worker(), template, defaults, original)
|
2019-11-03 12:51:14 +00:00
|
|
|
if err != nil {
|
2020-05-28 14:32:42 +00:00
|
|
|
aerc.PushError("Error: " + err.Error())
|
2019-11-03 12:51:14 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if args[0] == "reply" {
|
|
|
|
composer.FocusTerminal()
|
|
|
|
}
|
2019-05-16 16:15:34 +00:00
|
|
|
|
2019-05-16 16:39:22 +00:00
|
|
|
tab := aerc.NewTab(composer, subject)
|
2019-07-22 23:29:07 +00:00
|
|
|
composer.OnHeaderChange("Subject", func(subject string) {
|
2019-05-16 16:39:22 +00:00
|
|
|
if subject == "" {
|
|
|
|
tab.Name = "New email"
|
|
|
|
} else {
|
|
|
|
tab.Name = subject
|
|
|
|
}
|
|
|
|
tab.Content.Invalidate()
|
|
|
|
})
|
2019-11-03 12:51:14 +00:00
|
|
|
|
2020-05-25 14:59:48 +00:00
|
|
|
composer.OnClose(func(c *widgets.Composer) {
|
2020-09-21 18:43:22 +00:00
|
|
|
if c.Sent() {
|
|
|
|
store.Answered([]uint32{msg.Uid}, true, nil)
|
|
|
|
}
|
2020-05-25 14:59:48 +00:00
|
|
|
})
|
|
|
|
|
2019-11-03 12:51:14 +00:00
|
|
|
return nil
|
2019-05-16 16:39:22 +00:00
|
|
|
}
|
2019-05-16 16:15:34 +00:00
|
|
|
|
2019-08-18 09:33:13 +00:00
|
|
|
if quote {
|
2019-11-03 12:51:14 +00:00
|
|
|
if template == "" {
|
|
|
|
template = aerc.Config().Templates.QuotedReply
|
2019-08-18 09:33:13 +00:00
|
|
|
}
|
|
|
|
|
2020-06-19 15:58:08 +00:00
|
|
|
part := lib.FindPlaintext(msg.BodyStructure, nil)
|
2020-05-17 10:08:17 +00:00
|
|
|
if part == nil {
|
2020-06-19 15:58:08 +00:00
|
|
|
// mkey... let's get the first thing that isn't a container
|
|
|
|
// if that's still nil it's either not a multipart msg (ok) or
|
|
|
|
// broken (containers only)
|
|
|
|
part = lib.FindFirstNonMultipart(msg.BodyStructure, nil)
|
2020-05-17 10:08:17 +00:00
|
|
|
}
|
|
|
|
store.FetchBodyPart(msg.Uid, part, func(reader io.Reader) {
|
2019-11-03 12:51:14 +00:00
|
|
|
buf := new(bytes.Buffer)
|
2020-01-04 20:13:51 +00:00
|
|
|
buf.ReadFrom(reader)
|
2020-01-08 20:44:14 +00:00
|
|
|
original.Text = buf.String()
|
2020-01-08 20:44:16 +00:00
|
|
|
if len(msg.BodyStructure.Parts) == 0 {
|
|
|
|
original.MIMEType = fmt.Sprintf("%s/%s",
|
|
|
|
msg.BodyStructure.MIMEType, msg.BodyStructure.MIMESubType)
|
|
|
|
} else {
|
|
|
|
// TODO: still will be "multipart/mixed" for mixed mails with
|
|
|
|
// attachments, fix this after aerc could handle responding to
|
|
|
|
// such mails
|
|
|
|
original.MIMEType = fmt.Sprintf("%s/%s",
|
|
|
|
msg.BodyStructure.Parts[0].MIMEType,
|
|
|
|
msg.BodyStructure.Parts[0].MIMESubType)
|
|
|
|
}
|
2019-05-16 16:39:22 +00:00
|
|
|
addTab()
|
|
|
|
})
|
2019-11-03 12:51:14 +00:00
|
|
|
return nil
|
2019-05-16 16:39:22 +00:00
|
|
|
} else {
|
2019-11-03 12:51:14 +00:00
|
|
|
return addTab()
|
2019-05-16 16:39:22 +00:00
|
|
|
}
|
2019-05-16 16:15:34 +00:00
|
|
|
}
|
2020-08-26 06:13:38 +00:00
|
|
|
|
|
|
|
type addrSet map[string]struct{}
|
|
|
|
|
|
|
|
func newAddrSet() addrSet {
|
|
|
|
s := make(map[string]struct{})
|
|
|
|
return addrSet(s)
|
|
|
|
}
|
|
|
|
|
2020-11-10 18:57:09 +00:00
|
|
|
func (s addrSet) Add(a *mail.Address) {
|
2020-08-26 06:13:38 +00:00
|
|
|
s[a.Address] = struct{}{}
|
|
|
|
}
|
|
|
|
|
2020-11-10 18:57:09 +00:00
|
|
|
func (s addrSet) AddList(al []*mail.Address) {
|
2020-08-26 06:13:38 +00:00
|
|
|
for _, a := range al {
|
|
|
|
s[a.Address] = struct{}{}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-11-10 18:57:09 +00:00
|
|
|
func (s addrSet) Contains(a *mail.Address) bool {
|
2020-08-26 06:13:38 +00:00
|
|
|
_, ok := s[a.Address]
|
|
|
|
return ok
|
|
|
|
}
|