aerc/commands/msg/reply.go

235 lines
5.6 KiB
Go
Raw Normal View History

package msg
2019-05-16 18:15:34 +02:00
import (
"bytes"
2019-05-16 18:15:34 +02:00
"errors"
"fmt"
2019-05-16 18:39:22 +02:00
"io"
gomail "net/mail"
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-message"
_ "github.com/emersion/go-message/charset"
"github.com/emersion/go-message/mail"
2019-05-16 18:15:34 +02:00
"git.sr.ht/~sircmpwn/aerc/models"
2019-05-18 02:57:10 +02:00
"git.sr.ht/~sircmpwn/aerc/widgets"
2019-05-16 18:15:34 +02:00
)
type reply struct{}
2019-05-16 18:15:34 +02:00
func init() {
register(reply{})
}
func (reply) Aliases() []string {
return []string{"reply"}
}
func (reply) Complete(aerc *widgets.Aerc, args []string) []string {
return nil
2019-05-16 18:15:34 +02:00
}
func (reply) Execute(aerc *widgets.Aerc, args []string) error {
opts, optind, err := getopt.Getopts(args, "aqT:")
2019-05-16 18:39:22 +02:00
if err != nil {
return err
}
if optind != len(args) {
return errors.New("Usage: reply [-aq -T <template>]")
2019-05-16 18:15:34 +02:00
}
2019-05-16 18:39:22 +02:00
var (
quote bool
replyAll bool
template string
2019-05-16 18:39:22 +02:00
)
for _, opt := range opts {
switch opt.Option {
case 'a':
replyAll = true
case 'q':
quote = true
case 'T':
template = opt.Value
2019-05-16 18:39:22 +02:00
}
}
2019-05-16 18:15:34 +02:00
widget := aerc.SelectedTab().(widgets.ProvidesMessage)
acct := widget.SelectedAccount()
if acct == nil {
return errors.New("No account selected")
}
conf := acct.AccountConfig()
us, _ := gomail.ParseAddress(conf.From)
store := widget.Store()
if store == nil {
return errors.New("Cannot perform action. Messages still loading")
}
msg, err := widget.SelectedMessage()
if err != nil {
return err
}
2019-05-16 18:15:34 +02:00
acct.Logger().Println("Replying to email " + msg.Envelope.MessageId)
var (
to []string
cc []string
toList []*models.Address
2019-05-16 18:15:34 +02:00
)
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.Name != "" {
2019-05-25 20:52:57 +02:00
to = append(to, fmt.Sprintf("%s <%s@%s>",
addr.Name, addr.Mailbox, addr.Host))
2019-05-25 20:52:57 +02:00
} else {
to = append(to, fmt.Sprintf("<%s@%s>", addr.Mailbox, addr.Host))
2019-05-25 20:52:57 +02:00
}
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, addr.Format())
2019-05-25 20:52:57 +02:00
}
for _, addr := range msg.Envelope.To {
address := fmt.Sprintf("%s@%s", addr.Mailbox, addr.Host)
2019-05-25 20:52:57 +02:00
if address == us.Address {
continue
}
to = append(to, addr.Format())
}
}
2019-05-16 18:15:34 +02:00
}
var subject string
if !strings.HasPrefix(strings.ToLower(msg.Envelope.Subject), "re: ") {
subject = "Re: " + msg.Envelope.Subject
} else {
subject = msg.Envelope.Subject
}
2019-05-16 18:15:34 +02:00
defaults := map[string]string{
"To": strings.Join(to, ", "),
"Cc": strings.Join(cc, ", "),
"Subject": subject,
"In-Reply-To": msg.Envelope.MessageId,
}
addTab := func() error {
if template != "" {
defaults["OriginalFrom"] = models.FormatAddresses(msg.Envelope.From)
defaults["OriginalDate"] = msg.Envelope.Date.Format("Mon Jan 2, 2006 at 3:04 PM")
}
2019-05-25 20:52:57 +02:00
composer, err := widgets.NewComposer(aerc, aerc.Config(),
acct.AccountConfig(), acct.Worker(), template, defaults)
if err != nil {
aerc.PushError("Error: " + err.Error())
return err
}
if args[0] == "reply" {
composer.FocusTerminal()
}
2019-05-16 18:15:34 +02:00
2019-05-16 18:39:22 +02:00
tab := aerc.NewTab(composer, subject)
composer.OnHeaderChange("Subject", func(subject string) {
2019-05-16 18:39:22 +02:00
if subject == "" {
tab.Name = "New email"
} else {
tab.Name = subject
}
tab.Content.Invalidate()
})
return nil
2019-05-16 18:39:22 +02:00
}
2019-05-16 18:15:34 +02:00
if quote {
if template == "" {
template = aerc.Config().Templates.QuotedReply
}
store.FetchBodyPart(msg.Uid, []int{1}, func(reader io.Reader) {
header := message.Header{}
if len(msg.BodyStructure.Parts) > 0 {
partID := 0 // TODO: will we always choose first msg part?
header.SetText(
"Content-Transfer-Encoding", msg.BodyStructure.Parts[partID].Encoding)
if msg.BodyStructure.Parts[partID].MIMESubType == "" {
header.SetContentType(
msg.BodyStructure.Parts[partID].MIMEType,
msg.BodyStructure.Parts[partID].Params)
} else {
// include SubType if defined (text/plain, text/html, ...)
header.SetContentType(
fmt.Sprintf("%s/%s", msg.BodyStructure.Parts[partID].MIMEType,
msg.BodyStructure.Parts[partID].MIMESubType),
msg.BodyStructure.Parts[partID].Params)
}
header.SetText("Content-Description", msg.BodyStructure.Parts[partID].Description)
} else { // Parts has no headers, so we use global headers info
header.SetText(
"Content-Transfer-Encoding", msg.BodyStructure.Encoding)
if msg.BodyStructure.MIMESubType == "" {
header.SetContentType(
msg.BodyStructure.MIMEType,
msg.BodyStructure.Params)
} else {
// include SubType if defined (text/plain, text/html, ...)
header.SetContentType(
fmt.Sprintf("%s/%s", msg.BodyStructure.MIMEType,
msg.BodyStructure.MIMESubType),
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
}
buf := new(bytes.Buffer)
buf.ReadFrom(part.Body)
defaults["Original"] = buf.String()
2019-05-16 18:39:22 +02:00
addTab()
})
return nil
2019-05-16 18:39:22 +02:00
} else {
return addTab()
2019-05-16 18:39:22 +02:00
}
2019-05-16 18:15:34 +02:00
}
2019-06-02 01:47:09 +02:00
func findPlaintext(bs *models.BodyStructure,
path []int) (*models.BodyStructure, []int) {
2019-06-02 01:47:09 +02:00
for i, part := range bs.Parts {
cur := append(path, i+1)
if strings.ToLower(part.MIMEType) == "text" &&
strings.ToLower(part.MIMESubType) == "plain" {
2019-06-02 01:47:09 +02:00
return part, cur
}
if strings.ToLower(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
}