Make :pipe command more generic

5 files changed, 107 insertions, 99 deletions

diff --git a/commands/account/pipe.go b/commands/account/pipe.go
deleted file mode 100644
index a68ef64e..00000000
--- a/commands/account/pipe.go
+++ /dev/null
@@ -1,42 +0,0 @@
-package account
-
-import (
-	"errors"
-	"io"
-
-	"git.sr.ht/~sircmpwn/aerc/commands"
-	"git.sr.ht/~sircmpwn/aerc/widgets"
-)
-
-type Pipe struct{}
-
-func init() {
-	register(Pipe{})
-}
-
-func (_ Pipe) Aliases() []string {
-	return []string{"pipe"}
-}
-
-func (_ Pipe) Complete(aerc *widgets.Aerc, args []string) []string {
-	return nil
-}
-
-func (_ Pipe) Execute(aerc *widgets.Aerc, args []string) error {
-	if len(args) < 2 {
-		return errors.New("Usage: :pipe <cmd> [args...]")
-	}
-	acct := aerc.SelectedAccount()
-	store := acct.Messages().Store()
-	msg := acct.Messages().Selected()
-	store.FetchFull([]uint32{msg.Uid}, func(reader io.Reader) {
-		term, err := commands.QuickTerm(aerc, args[1:], reader)
-		if err != nil {
-			aerc.PushError(" " + err.Error())
-			return
-		}
-		name := args[1] + " <" + msg.Envelope.Subject
-		aerc.NewTab(term, name)
-	})
-	return nil
-}
diff --git a/commands/msg/pipe.go b/commands/msg/pipe.go
new file mode 100644
index 00000000..949bc95e
--- /dev/null
+++ b/commands/msg/pipe.go
@@ -0,0 +1,105 @@
+package msg
+
+import (
+	"encoding/base64"
+	"errors"
+	"fmt"
+	"io"
+	"mime/quotedprintable"
+	"strings"
+
+	"git.sr.ht/~sircmpwn/getopt"
+
+	"git.sr.ht/~sircmpwn/aerc/commands"
+	"git.sr.ht/~sircmpwn/aerc/widgets"
+)
+
+type Pipe struct{}
+
+func init() {
+	register(Pipe{})
+}
+
+func (_ Pipe) Aliases() []string {
+	return []string{"pipe"}
+}
+
+func (_ Pipe) Complete(aerc *widgets.Aerc, args []string) []string {
+	return nil
+}
+
+func (_ Pipe) Execute(aerc *widgets.Aerc, args []string) error {
+	var (
+		pipeFull bool
+		pipePart bool
+	)
+	// TODO: let user specify part by index or preferred mimetype
+	opts, optind, err := getopt.Getopts(args, "mp")
+	if err != nil {
+		return err
+	}
+	for _, opt := range opts {
+		switch opt.Option {
+		case 'm':
+			if pipePart {
+				return errors.New("-m and -p are mutually exclusive")
+			}
+			pipeFull = true
+		case 'p':
+			if pipeFull {
+				return errors.New("-m and -p are mutually exclusive")
+			}
+			pipePart = true
+		}
+	}
+	cmd := args[optind:]
+	if len(cmd) == 0 {
+		return errors.New("Usage: pipe [-mp] <cmd> [args...]")
+	}
+
+	provider := aerc.SelectedTab().(widgets.ProvidesMessage)
+	if !pipeFull && !pipePart {
+		if _, ok := provider.(*widgets.MessageViewer); ok {
+			pipePart = true
+		} else if _, ok := provider.(*widgets.AccountView); ok {
+			pipeFull = true
+		} else {
+			return errors.New(
+				"Neither -m nor -p specified and cannot infer default")
+		}
+	}
+
+	if pipeFull {
+		store := provider.Store()
+		msg := provider.SelectedMessage()
+		store.FetchFull([]uint32{msg.Uid}, func(reader io.Reader) {
+			term, err := commands.QuickTerm(aerc, cmd, reader)
+			if err != nil {
+				aerc.PushError(" " + err.Error())
+				return
+			}
+			name := cmd[0] + " <" + msg.Envelope.Subject
+			aerc.NewTab(term, name)
+		})
+	} else if pipePart {
+		p := provider.SelectedMessagePart()
+		p.Store.FetchBodyPart(p.Msg.Uid, p.Index, func(reader io.Reader) {
+			// email parts are encoded as 7bit (plaintext), quoted-printable, or base64
+			if strings.EqualFold(p.Part.Encoding, "base64") {
+				reader = base64.NewDecoder(base64.StdEncoding, reader)
+			} else if strings.EqualFold(p.Part.Encoding, "quoted-printable") {
+				reader = quotedprintable.NewReader(reader)
+			}
+
+			term, err := commands.QuickTerm(aerc, cmd, reader)
+			if err != nil {
+				aerc.PushError(" " + err.Error())
+				return
+			}
+			name := fmt.Sprintf("%s <%s/[%d]", cmd[0], p.Msg.Envelope.Subject, p.Index)
+			aerc.NewTab(term, name)
+		})
+	}
+
+	return nil
+}
diff --git a/commands/msgview/open.go b/commands/msgview/open.go
index d25fa678..f4a09318 100644
--- a/commands/msgview/open.go
+++ b/commands/msgview/open.go
@@ -34,7 +34,7 @@ func (_ Open) Execute(aerc *widgets.Aerc, args []string) error {
 	}
 
 	mv := aerc.SelectedTab().(*widgets.MessageViewer)
-	p := mv.CurrentPart()
+	p := mv.SelectedMessagePart()
 
 	p.Store.FetchBodyPart(p.Msg.Uid, p.Index, func(reader io.Reader) {
 		// email parts are encoded as 7bit (plaintext), quoted-printable, or base64
diff --git a/commands/msgview/pipe.go b/commands/msgview/pipe.go
deleted file mode 100644
index 56c125bf..00000000
--- a/commands/msgview/pipe.go
+++ /dev/null
@@ -1,55 +0,0 @@
-package msgview
-
-import (
-	"encoding/base64"
-	"errors"
-	"fmt"
-	"io"
-	"mime/quotedprintable"
-	"strings"
-
-	"git.sr.ht/~sircmpwn/aerc/commands"
-	"git.sr.ht/~sircmpwn/aerc/widgets"
-)
-
-type Pipe struct{}
-
-func init() {
-	register(Pipe{})
-}
-
-func (_ Pipe) Aliases() []string {
-	return []string{"pipe"}
-}
-
-func (_ Pipe) Complete(aerc *widgets.Aerc, args []string) []string {
-	return nil
-}
-
-func (_ Pipe) Execute(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
-		if strings.EqualFold(p.Part.Encoding, "base64") {
-			reader = base64.NewDecoder(base64.StdEncoding, reader)
-		} else if strings.EqualFold(p.Part.Encoding, "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
index 93fa83f8..75ba56f3 100644
--- a/commands/msgview/save.go
+++ b/commands/msgview/save.go
@@ -56,7 +56,7 @@ func (_ Save) Execute(aerc *widgets.Aerc, args []string) error {
 	}
 
 	mv := aerc.SelectedTab().(*widgets.MessageViewer)
-	p := mv.CurrentPart()
+	p := mv.SelectedMessagePart()
 
 	p.Store.FetchBodyPart(p.Msg.Uid, p.Index, func(reader io.Reader) {
 		// email parts are encoded as 7bit (plaintext), quoted-printable, or base64