aboutsummaryrefslogtreecommitdiff
path: root/commands/msg/forward.go
diff options
context:
space:
mode:
authorSrivathsan Murali <sri@vathsan.com>2019-11-03 13:51:14 +0100
committerDrew DeVault <sir@cmpwn.com>2019-11-10 10:15:49 -0500
commit3ba69edab5f0c787424dac9649e43a7743da13ca (patch)
treebccbdd4e1844cc89f011839f0d6557012a14d1d0 /commands/msg/forward.go
parentad68a9e4e471eb708893ad16601ab14a4672a2da (diff)
Add Templates with Parsing
+ Changes NewComposer to return error. + Add lib to handle templates using "text/template". + Add -T option to following commands - compose. - reply - forward + Quoted replies using templates. + Forwards as body using templates + Default templates are installed similar to filters. + Templates Config in aerc.conf. - Required templates are parsed while loading config. + Add aerc-templates.7 manual for using template data.
Diffstat (limited to 'commands/msg/forward.go')
-rw-r--r--commands/msg/forward.go155
1 files changed, 77 insertions, 78 deletions
diff --git a/commands/msg/forward.go b/commands/msg/forward.go
index 494072d..7570177 100644
--- a/commands/msg/forward.go
+++ b/commands/msg/forward.go
@@ -1,20 +1,21 @@
package msg
import (
- "bufio"
+ "bytes"
"errors"
"fmt"
- "git.sr.ht/~sircmpwn/aerc/lib"
- "git.sr.ht/~sircmpwn/aerc/models"
- "git.sr.ht/~sircmpwn/aerc/widgets"
- "git.sr.ht/~sircmpwn/getopt"
- "github.com/emersion/go-message"
- "github.com/emersion/go-message/mail"
"io"
"io/ioutil"
"os"
"path"
"strings"
+
+ "github.com/emersion/go-message"
+ "github.com/emersion/go-message/mail"
+
+ "git.sr.ht/~sircmpwn/aerc/models"
+ "git.sr.ht/~sircmpwn/aerc/widgets"
+ "git.sr.ht/~sircmpwn/getopt"
)
type forward struct{}
@@ -32,15 +33,18 @@ func (forward) Complete(aerc *widgets.Aerc, args []string) []string {
}
func (forward) Execute(aerc *widgets.Aerc, args []string) error {
- opts, optind, err := getopt.Getopts(args, "A")
+ opts, optind, err := getopt.Getopts(args, "AT:")
if err != nil {
return err
}
attach := false
+ template := ""
for _, opt := range opts {
switch opt.Option {
case 'A':
attach = true
+ case 'T':
+ template = opt.Value
}
}
@@ -69,10 +73,20 @@ func (forward) Execute(aerc *widgets.Aerc, args []string) error {
"To": to,
"Subject": subject,
}
- composer := widgets.NewComposer(aerc, aerc.Config(), acct.AccountConfig(),
- acct.Worker(), defaults)
- addTab := func() {
+ addTab := func() (*widgets.Composer, error) {
+ if template != "" {
+ defaults["OriginalFrom"] = models.FormatAddresses(msg.Envelope.From)
+ defaults["OriginalDate"] = msg.Envelope.Date.Format("Mon Jan 2, 2006 at 3:04 PM")
+ }
+
+ composer, err := widgets.NewComposer(aerc, aerc.Config(), acct.AccountConfig(),
+ acct.Worker(), template, defaults)
+ if err != nil {
+ aerc.PushError("Error: " + err.Error())
+ return nil, err
+ }
+
tab := aerc.NewTab(composer, subject)
if to == "" {
composer.FocusRecipient()
@@ -87,83 +101,68 @@ func (forward) Execute(aerc *widgets.Aerc, args []string) error {
}
tab.Content.Invalidate()
})
+ return composer, nil
}
if attach {
- forwardAttach(store, composer, msg, addTab)
- } else {
- forwardBodyPart(store, composer, msg, addTab)
- }
- return nil
-}
-
-func forwardAttach(store *lib.MessageStore, composer *widgets.Composer,
- msg *models.MessageInfo, addTab func()) {
-
- store.FetchFull([]uint32{msg.Uid}, func(reader io.Reader) {
tmpDir, err := ioutil.TempDir("", "aerc-tmp-attachment")
if err != nil {
- // TODO: Do something with the error
- addTab()
- return
+ return err
}
tmpFileName := path.Join(tmpDir,
strings.ReplaceAll(fmt.Sprintf("%s.eml", msg.Envelope.Subject), "/", "-"))
- tmpFile, err := os.Create(tmpFileName)
- if err != nil {
- println(err)
- // TODO: Do something with the error
- addTab()
- return
- }
+ store.FetchFull([]uint32{msg.Uid}, func(reader io.Reader) {
+ tmpFile, err := os.Create(tmpFileName)
+ if err != nil {
+ println(err)
+ // TODO: Do something with the error
+ addTab()
+ return
+ }
- defer tmpFile.Close()
- io.Copy(tmpFile, reader)
- composer.AddAttachment(tmpFileName)
- composer.OnClose(func(composer *widgets.Composer) {
- os.RemoveAll(tmpDir)
+ defer tmpFile.Close()
+ io.Copy(tmpFile, reader)
+ composer, err := addTab()
+ if err != nil {
+ return
+ }
+ composer.AddAttachment(tmpFileName)
+ composer.OnClose(func(composer *widgets.Composer) {
+ os.RemoveAll(tmpDir)
+ })
})
- addTab()
- })
-}
-
-func forwardBodyPart(store *lib.MessageStore, composer *widgets.Composer,
- msg *models.MessageInfo, addTab func()) {
- // 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
+ } else {
+ if template == "" {
+ template = aerc.Config().Templates.Forwards
}
- pipeout, pipein := io.Pipe()
- scanner := bufio.NewScanner(part.Body)
- go composer.PrependContents(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()
- })
+ // 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
+ }
+ buf := new(bytes.Buffer)
+ buf.ReadFrom(part.Body)
+ defaults["Original"] = buf.String()
+ addTab()
+ })
+ }
+ return nil
}