From 28fc9fa53da1449498392f83d63a8502a5a958a3 Mon Sep 17 00:00:00 2001
From: Galen Abell <galen@galenabell.com>
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 <cmd> [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 <path>")
+	}
+
+	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