From 28fc9fa53da1449498392f83d63a8502a5a958a3 Mon Sep 17 00:00:00 2001 From: Galen Abell Date: Sun, 26 May 2019 17:37:39 -0400 Subject: Add :save and :pipe commands to viewer * :save takes a path and saves the current message part to that location * :pipe is the same as pipe on the account page, but uses the current message part rather than the whole email (ie :pipe gzip -d) * Refactored account:pipe and extracted common pipe code to commands.util.QuickTerm * Added helper command aerc.PushError --- commands/msgview/pipe.go | 45 ++++++++++++++++++++++++++++++++++++ commands/msgview/save.go | 59 ++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 104 insertions(+) create mode 100644 commands/msgview/pipe.go create mode 100644 commands/msgview/save.go (limited to 'commands/msgview') diff --git a/commands/msgview/pipe.go b/commands/msgview/pipe.go new file mode 100644 index 0000000..81cef7d --- /dev/null +++ b/commands/msgview/pipe.go @@ -0,0 +1,45 @@ +package msgview + +import ( + "encoding/base64" + "errors" + "fmt" + "io" + "mime/quotedprintable" + + "git.sr.ht/~sircmpwn/aerc/commands" + "git.sr.ht/~sircmpwn/aerc/widgets" +) + +func init() { + register("pipe", Pipe) +} + +func Pipe(aerc *widgets.Aerc, args []string) error { + if len(args) < 2 { + return errors.New("Usage: :pipe [args...]") + } + + mv := aerc.SelectedTab().(*widgets.MessageViewer) + p := mv.CurrentPart() + + p.Store.FetchBodyPart(p.Msg.Uid, p.Index, func(reader io.Reader) { + // email parts are encoded as 7bit (plaintext), quoted-printable, or base64 + switch p.Part.Encoding { + case "base64": + reader = base64.NewDecoder(base64.StdEncoding, reader) + case "quoted-printable": + reader = quotedprintable.NewReader(reader) + } + + term, err := commands.QuickTerm(aerc, args[1:], reader) + if err != nil { + aerc.PushError(" " + err.Error()) + return + } + name := fmt.Sprintf("%s <%s/[%d]", args[1], p.Msg.Envelope.Subject, p.Index) + aerc.NewTab(term, name) + }) + + return nil +} diff --git a/commands/msgview/save.go b/commands/msgview/save.go new file mode 100644 index 0000000..4dabd52 --- /dev/null +++ b/commands/msgview/save.go @@ -0,0 +1,59 @@ +package msgview + +import ( + "encoding/base64" + "errors" + "io" + "mime/quotedprintable" + "os" + "time" + + "git.sr.ht/~sircmpwn/aerc/widgets" + "github.com/mitchellh/go-homedir" +) + +func init() { + register("save", Save) +} + +func Save(aerc *widgets.Aerc, args []string) error { + if len(args) < 2 { + return errors.New("Usage: :save ") + } + + mv := aerc.SelectedTab().(*widgets.MessageViewer) + p := mv.CurrentPart() + + p.Store.FetchBodyPart(p.Msg.Uid, p.Index, func(reader io.Reader) { + // email parts are encoded as 7bit (plaintext), quoted-printable, or base64 + switch p.Part.Encoding { + case "base64": + reader = base64.NewDecoder(base64.StdEncoding, reader) + case "quoted-printable": + reader = quotedprintable.NewReader(reader) + } + + target, err := homedir.Expand(args[1]) + if err != nil { + aerc.PushError(" " + err.Error()) + return + } + + f, err := os.Create(target) + if err != nil { + aerc.PushError(" " + err.Error()) + return + } + defer f.Close() + + _, err = io.Copy(f, reader) + if err != nil { + aerc.PushError(" " + err.Error()) + return + } + + aerc.PushStatus("Saved to "+target, 10*time.Second) + }) + + return nil +} -- cgit v1.2.3