aerc/commands/account/reply.go

254 lines
6 KiB
Go
Raw Normal View History

2019-05-16 18:15:34 +02:00
package account
import (
2019-05-16 18:39:22 +02:00
"bufio"
2019-05-16 18:15:34 +02:00
"errors"
"fmt"
2019-05-16 18:39:22 +02:00
"io"
gomail "net/mail"
"regexp"
2019-05-16 18:15:34 +02:00
"strings"
2019-05-18 21:34:16 +02:00
"git.sr.ht/~sircmpwn/getopt"
"github.com/emersion/go-imap"
"github.com/emersion/go-message"
_ "github.com/emersion/go-message/charset"
"github.com/emersion/go-message/mail"
2019-05-16 18:15:34 +02:00
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/widgets"
2019-05-16 18:15:34 +02:00
)
func init() {
register("reply", Reply)
2019-05-25 20:52:57 +02:00
register("forward", Reply)
2019-05-16 18:15:34 +02:00
}
var (
atom *regexp.Regexp = regexp.MustCompile("^[a-z0-9!#$%7'*+-/=?^_`{}|~ ]+$")
)
func formatAddress(addr *imap.Address) string {
if addr.PersonalName != "" {
if atom.MatchString(addr.PersonalName) {
return fmt.Sprintf("%s <%s@%s>",
addr.PersonalName, addr.MailboxName, addr.HostName)
} else {
return fmt.Sprintf("\"%s\" <%s@%s>",
strings.ReplaceAll(addr.PersonalName, "\"", "'"),
addr.MailboxName, addr.HostName)
}
} else {
return fmt.Sprintf("<%s@%s>", addr.MailboxName, addr.HostName)
}
}
2019-05-16 18:15:34 +02:00
func Reply(aerc *widgets.Aerc, args []string) error {
2019-05-16 18:39:22 +02:00
opts, optind, err := getopt.Getopts(args[1:], "aq")
if err != nil {
return err
}
2019-05-18 21:34:16 +02:00
if optind != len(args)-1 {
2019-05-16 18:15:34 +02:00
return errors.New("Usage: reply [-aq]")
}
2019-05-16 18:39:22 +02:00
var (
quote bool
replyAll bool
)
for _, opt := range opts {
switch opt.Option {
case 'a':
replyAll = true
case 'q':
quote = true
}
}
2019-05-16 18:15:34 +02:00
acct := aerc.SelectedAccount()
conf := acct.AccountConfig()
us, _ := gomail.ParseAddress(conf.From)
2019-05-16 18:39:22 +02:00
store := acct.Messages().Store()
2019-05-16 18:15:34 +02:00
msg := acct.Messages().Selected()
acct.Logger().Println("Replying to email " + msg.Envelope.MessageId)
var (
to []string
cc []string
toList []*imap.Address
)
2019-05-25 20:52:57 +02:00
if args[0] == "reply" {
if len(msg.Envelope.ReplyTo) != 0 {
toList = msg.Envelope.ReplyTo
2019-05-16 18:15:34 +02:00
} else {
2019-05-25 20:52:57 +02:00
toList = msg.Envelope.From
2019-05-16 18:15:34 +02:00
}
2019-05-25 20:52:57 +02:00
for _, addr := range toList {
if addr.PersonalName != "" {
to = append(to, fmt.Sprintf("%s <%s@%s>",
addr.PersonalName, addr.MailboxName, addr.HostName))
} else {
to = append(to, fmt.Sprintf("<%s@%s>",
addr.MailboxName, addr.HostName))
}
2019-05-16 18:15:34 +02:00
}
2019-05-25 20:52:57 +02:00
if replyAll {
for _, addr := range msg.Envelope.Cc {
cc = append(cc, formatAddress(addr))
}
for _, addr := range msg.Envelope.To {
address := fmt.Sprintf("%s@%s", addr.MailboxName, addr.HostName)
if address == us.Address {
continue
}
to = append(to, formatAddress(addr))
}
}
2019-05-16 18:15:34 +02:00
}
var subject string
2019-05-25 20:52:57 +02:00
if args[0] == "forward" {
subject = "Fwd: " + msg.Envelope.Subject
} else {
2019-05-25 20:52:57 +02:00
if !strings.HasPrefix(msg.Envelope.Subject, "Re: ") {
subject = "Re: " + msg.Envelope.Subject
} else {
subject = msg.Envelope.Subject
}
}
2019-05-16 18:15:34 +02:00
composer := widgets.NewComposer(
aerc.Config(), acct.AccountConfig(), acct.Worker()).
Defaults(map[string]string{
"To": strings.Join(to, ", "),
"Cc": strings.Join(cc, ", "),
2019-05-18 21:34:16 +02:00
"Subject": subject,
2019-05-16 18:15:34 +02:00
"In-Reply-To": msg.Envelope.MessageId,
2019-05-25 20:52:57 +02:00
})
if args[0] == "reply" {
composer.FocusTerminal()
}
2019-05-16 18:15:34 +02:00
2019-05-16 18:39:22 +02:00
addTab := func() {
tab := aerc.NewTab(composer, subject)
composer.OnSubjectChange(func(subject string) {
if subject == "" {
tab.Name = "New email"
} else {
tab.Name = subject
}
tab.Content.Invalidate()
})
}
2019-05-16 18:15:34 +02:00
2019-05-25 20:52:57 +02:00
if args[0] == "forward" {
2019-05-20 22:42:44 +02:00
// TODO: something more intelligent than fetching the 1st part
2019-05-25 20:52:57 +02:00
// TODO: add attachments!
2019-05-20 22:42:44 +02:00
store.FetchBodyPart(msg.Uid, []int{1}, func(reader io.Reader) {
header := message.Header{}
header.SetText(
"Content-Transfer-Encoding", msg.BodyStructure.Encoding)
header.SetContentType(
msg.BodyStructure.MIMEType, msg.BodyStructure.Params)
header.SetText("Content-Description", msg.BodyStructure.Description)
entity, err := message.New(header, reader)
if err != nil {
// TODO: Do something with the error
addTab()
return
}
mreader := mail.NewReader(entity)
part, err := mreader.NextPart()
if err != nil {
// TODO: Do something with the error
addTab()
return
}
2019-05-16 18:39:22 +02:00
pipeout, pipein := io.Pipe()
scanner := bufio.NewScanner(part.Body)
2019-05-16 18:39:22 +02:00
go composer.SetContents(pipeout)
// TODO: Let user customize the date format used here
2019-05-25 20:52:57 +02:00
io.WriteString(pipein, fmt.Sprintf("Forwarded message from %s on %s:\n\n",
msg.Envelope.From[0].PersonalName,
msg.Envelope.Date.Format("Mon Jan 2, 2006 at 3:04 PM")))
2019-05-16 18:39:22 +02:00
for scanner.Scan() {
2019-05-25 20:52:57 +02:00
io.WriteString(pipein, fmt.Sprintf("%s\n", scanner.Text()))
2019-05-16 18:39:22 +02:00
}
pipein.Close()
pipeout.Close()
addTab()
})
} else {
2019-05-25 20:52:57 +02:00
if quote {
2019-06-02 01:47:09 +02:00
var (
path []int
part *imap.BodyStructure
)
if len(msg.BodyStructure.Parts) != 0 {
part, path = findPlaintext(msg.BodyStructure, path)
}
if part == nil {
part = msg.BodyStructure
path = []int{1}
}
store.FetchBodyPart(msg.Uid, path, func(reader io.Reader) {
2019-05-25 20:52:57 +02:00
header := message.Header{}
header.SetText(
2019-06-02 01:47:09 +02:00
"Content-Transfer-Encoding", part.Encoding)
header.SetContentType(part.MIMEType, part.Params)
header.SetText("Content-Description", part.Description)
2019-05-25 20:52:57 +02:00
entity, err := message.New(header, reader)
if err != nil {
// TODO: Do something with the error
addTab()
return
}
mreader := mail.NewReader(entity)
part, err := mreader.NextPart()
if err != nil {
// TODO: Do something with the error
addTab()
return
}
pipeout, pipein := io.Pipe()
scanner := bufio.NewScanner(part.Body)
go composer.SetContents(pipeout)
// TODO: Let user customize the date format used here
io.WriteString(pipein, fmt.Sprintf("On %s %s wrote:\n",
msg.Envelope.Date.Format("Mon Jan 2, 2006 at 3:04 PM"),
msg.Envelope.From[0].PersonalName))
for scanner.Scan() {
io.WriteString(pipein, fmt.Sprintf("> %s\n", scanner.Text()))
}
pipein.Close()
pipeout.Close()
addTab()
})
} else {
addTab()
}
2019-05-16 18:39:22 +02:00
}
2019-05-16 18:15:34 +02:00
return nil
}
2019-06-02 01:47:09 +02:00
func findPlaintext(bs *imap.BodyStructure,
path []int) (*imap.BodyStructure, []int) {
for i, part := range bs.Parts {
cur := append(path, i+1)
if part.MIMEType == "text" && part.MIMESubType == "plain" {
return part, cur
}
if part.MIMEType == "multipart" {
if part, path := findPlaintext(part, cur); path != nil {
2019-06-02 01:47:09 +02:00
return part, path
}
}
}
return nil, nil
}