From 700dea23fa75f213af2f99449db994008eab9d21 Mon Sep 17 00:00:00 2001 From: Drew DeVault Date: Sat, 30 Mar 2019 11:58:24 -0400 Subject: Implement :pipe --- commands/account/pipe.go | 73 ++++++++++++++++++++++++++++++++++++++++++++++ commands/terminal/close.go | 1 + 2 files changed, 74 insertions(+) create mode 100644 commands/account/pipe.go (limited to 'commands') 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 [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 } -- cgit v1.2.3