Implement :pipe

2 files changed, 74 insertions, 0 deletions

diff --git a/commands/account/pipe.go b/commands/account/pipe.go
new file mode 100644
index 0000000..5693d4a
--- /dev/null
+++ b/commands/account/pipe.go
@@ -0,0 +1,73 @@
+package account
+
+import (
+	"bytes"
+	"errors"
+	"io"
+	"os/exec"
+	"time"
+
+	"git.sr.ht/~sircmpwn/aerc2/widgets"
+
+	"github.com/gdamore/tcell"
+	"github.com/mohamedattahri/mail"
+)
+
+func init() {
+	register("pipe", Pipe)
+}
+
+func Pipe(aerc *widgets.Aerc, args []string) error {
+	if len(args) < 2 {
+		return errors.New("Usage: :pipe <cmd> [args...]")
+	}
+	acct := aerc.SelectedAccount()
+	if acct == nil {
+		return errors.New("No account selected")
+	}
+	store := acct.Messages().Store()
+	msg := acct.Messages().Selected()
+	store.FetchBodies([]uint32{msg.Uid}, func(msg *mail.Message) {
+		cmd := exec.Command(args[1], args[2:]...)
+		pipe, err := cmd.StdinPipe()
+		if err != nil {
+			aerc.PushStatus(" "+err.Error(), 10*time.Second).
+				Color(tcell.ColorRed, tcell.ColorWhite)
+			return
+		}
+		term, err := widgets.NewTerminal(cmd)
+		if err != nil {
+			aerc.PushStatus(" "+err.Error(), 10*time.Second).
+				Color(tcell.ColorRed, tcell.ColorWhite)
+			return
+		}
+		host := widgets.NewTermHost(term, aerc.Config())
+		name := msg.Subject()
+		if len(name) > 12 {
+			name = name[:12]
+		}
+		aerc.NewTab(host, args[1] + " <" + name)
+		term.OnClose = func(err error) {
+			if err != nil {
+				aerc.PushStatus(" "+err.Error(), 10*time.Second).
+					Color(tcell.ColorRed, tcell.ColorWhite)
+			} else {
+				// TODO: Tab-specific status stacks
+				aerc.PushStatus("Process complete, press any key to close.",
+					10*time.Second)
+			}
+		}
+		term.OnStart = func() {
+			go func() {
+				reader := bytes.NewBuffer(msg.Bytes())
+				_, err := io.Copy(pipe, reader)
+				if err != nil {
+					aerc.PushStatus(" "+err.Error(), 10*time.Second).
+						Color(tcell.ColorRed, tcell.ColorWhite)
+				}
+				pipe.Close()
+			}()
+		}
+	})
+	return nil
+}
diff --git a/commands/terminal/close.go b/commands/terminal/close.go
index 68be6ac..cb5702e 100644
--- a/commands/terminal/close.go
+++ b/commands/terminal/close.go
@@ -19,5 +19,6 @@ func CommandClose(aerc *widgets.Aerc, args []string) error {
 		return errors.New("Error: not a terminal")
 	}
 	thost.Terminal().Close(nil)
+	aerc.RemoveTab(thost)
 	return nil
 }