Move forward command to it's own file

The two commands did not have much code in common
This commit is contained in:
Jelle Besseling 2019-08-18 11:33:13 +02:00 committed by Drew DeVault
parent 17aede55dd
commit 217e8855f2
2 changed files with 130 additions and 70 deletions

104
commands/msg/forward.go Normal file
View File

@ -0,0 +1,104 @@
package msg
import (
"bufio"
"errors"
"fmt"
"git.sr.ht/~sircmpwn/aerc/widgets"
"github.com/emersion/go-message"
"github.com/emersion/go-message/mail"
"io"
)
type forward struct{}
func init() {
register(forward{})
}
func (_ forward) Aliases() []string {
return []string{"forward"}
}
func (_ forward) Complete(aerc *widgets.Aerc, args []string) []string {
return nil
}
func (_ forward) Execute(aerc *widgets.Aerc, args []string) error {
if len(args) != 1 {
return errors.New("Usage: forward")
}
widget := aerc.SelectedTab().(widgets.ProvidesMessage)
acct := widget.SelectedAccount()
if acct == nil {
return errors.New("No account selected")
}
store := widget.Store()
if store == nil {
return errors.New("Cannot perform action. Messages still loading")
}
msg, err := widget.SelectedMessage()
if err != nil {
return err
}
acct.Logger().Println("Forwarding email " + msg.Envelope.MessageId)
subject := "Fwd: " + msg.Envelope.Subject
defaults := map[string]string{
"Subject": subject,
}
composer := widgets.NewComposer(aerc.Config(), acct.AccountConfig(),
acct.Worker(), defaults)
addTab := func() {
tab := aerc.NewTab(composer, subject)
composer.OnHeaderChange("Subject", func(subject string) {
if subject == "" {
tab.Name = "New email"
} else {
tab.Name = subject
}
tab.Content.Invalidate()
})
}
// TODO: something more intelligent than fetching the 1st part
// TODO: add attachments!
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
}
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("Forwarded message from %s on %s:\n\n",
msg.Envelope.From[0].Name,
msg.Envelope.Date.Format("Mon Jan 2, 2006 at 3:04 PM")))
for scanner.Scan() {
io.WriteString(pipein, fmt.Sprintf("%s\n", scanner.Text()))
}
pipein.Close()
pipeout.Close()
addTab()
})
return nil
}

View File

@ -24,7 +24,7 @@ func init() {
} }
func (_ reply) Aliases() []string { func (_ reply) Aliases() []string {
return []string{"reply", "forward"} return []string{"reply"}
} }
func (_ reply) Complete(aerc *widgets.Aerc, args []string) []string { func (_ reply) Complete(aerc *widgets.Aerc, args []string) []string {
@ -103,14 +103,10 @@ func (_ reply) Execute(aerc *widgets.Aerc, args []string) error {
} }
var subject string var subject string
if args[0] == "forward" { if !strings.HasPrefix(msg.Envelope.Subject, "Re: ") {
subject = "Fwd: " + msg.Envelope.Subject subject = "Re: " + msg.Envelope.Subject
} else { } else {
if !strings.HasPrefix(msg.Envelope.Subject, "Re: ") { subject = msg.Envelope.Subject
subject = "Re: " + msg.Envelope.Subject
} else {
subject = msg.Envelope.Subject
}
} }
defaults := map[string]string{ defaults := map[string]string{
@ -139,16 +135,25 @@ func (_ reply) Execute(aerc *widgets.Aerc, args []string) error {
}) })
} }
if args[0] == "forward" { if quote {
// TODO: something more intelligent than fetching the 1st part var (
// TODO: add attachments! path []int
store.FetchBodyPart(msg.Uid, []int{1}, func(reader io.Reader) { part *models.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) {
header := message.Header{} header := message.Header{}
header.SetText( header.SetText(
"Content-Transfer-Encoding", msg.BodyStructure.Encoding) "Content-Transfer-Encoding", part.Encoding)
header.SetContentType( header.SetContentType(part.MIMEType, part.Params)
msg.BodyStructure.MIMEType, msg.BodyStructure.Params) header.SetText("Content-Description", part.Description)
header.SetText("Content-Description", msg.BodyStructure.Description)
entity, err := message.New(header, reader) entity, err := message.New(header, reader)
if err != nil { if err != nil {
// TODO: Do something with the error // TODO: Do something with the error
@ -167,67 +172,18 @@ func (_ reply) Execute(aerc *widgets.Aerc, args []string) error {
scanner := bufio.NewScanner(part.Body) scanner := bufio.NewScanner(part.Body)
go composer.SetContents(pipeout) go composer.SetContents(pipeout)
// TODO: Let user customize the date format used here // TODO: Let user customize the date format used here
io.WriteString(pipein, fmt.Sprintf("Forwarded message from %s on %s:\n\n", io.WriteString(pipein, fmt.Sprintf("On %s %s wrote:\n",
msg.Envelope.From[0].Name, msg.Envelope.Date.Format("Mon Jan 2, 2006 at 3:04 PM"),
msg.Envelope.Date.Format("Mon Jan 2, 2006 at 3:04 PM"))) msg.Envelope.From[0].Name))
for scanner.Scan() { for scanner.Scan() {
io.WriteString(pipein, fmt.Sprintf("%s\n", scanner.Text())) io.WriteString(pipein, fmt.Sprintf("> %s\n", scanner.Text()))
} }
pipein.Close() pipein.Close()
pipeout.Close() pipeout.Close()
addTab() addTab()
}) })
} else { } else {
if quote { addTab()
var (
path []int
part *models.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) {
header := message.Header{}
header.SetText(
"Content-Transfer-Encoding", part.Encoding)
header.SetContentType(part.MIMEType, part.Params)
header.SetText("Content-Description", part.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
}
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].Name))
for scanner.Scan() {
io.WriteString(pipein, fmt.Sprintf("> %s\n", scanner.Text()))
}
pipein.Close()
pipeout.Close()
addTab()
})
} else {
addTab()
}
} }
return nil return nil