aerc/commands/msg/pipe.go

239 lines
5.3 KiB
Go
Raw Normal View History

2019-07-05 16:21:12 +00:00
package msg
import (
"errors"
"fmt"
"io"
2019-07-08 22:19:08 +00:00
"os/exec"
"sort"
"time"
2019-07-05 16:21:12 +00:00
"git.sr.ht/~rjarry/aerc/commands"
"git.sr.ht/~rjarry/aerc/widgets"
"git.sr.ht/~rjarry/aerc/worker/types"
2019-07-08 22:19:08 +00:00
"git.sr.ht/~sircmpwn/getopt"
2019-07-05 16:21:12 +00:00
)
type Pipe struct{}
func init() {
register(Pipe{})
}
func (Pipe) Aliases() []string {
2019-07-05 16:21:12 +00:00
return []string{"pipe"}
}
func (Pipe) Complete(aerc *widgets.Aerc, args []string) []string {
2019-07-05 16:21:12 +00:00
return nil
}
func (Pipe) Execute(aerc *widgets.Aerc, args []string) error {
2019-07-05 16:21:12 +00:00
var (
2019-07-08 22:19:08 +00:00
background bool
pipeFull bool
pipePart bool
2019-07-05 16:21:12 +00:00
)
// TODO: let user specify part by index or preferred mimetype
2019-07-08 22:19:08 +00:00
opts, optind, err := getopt.Getopts(args, "bmp")
2019-07-05 16:21:12 +00:00
if err != nil {
return err
}
for _, opt := range opts {
switch opt.Option {
2019-07-08 22:19:08 +00:00
case 'b':
background = true
2019-07-05 16:21:12 +00:00
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")
}
}
2019-07-08 22:19:08 +00:00
doTerm := func(reader io.Reader, name string) {
term, err := commands.QuickTerm(aerc, cmd, reader)
if err != nil {
aerc.PushError(err.Error())
2019-07-08 22:19:08 +00:00
return
}
aerc.NewTab(term, name)
}
doExec := func(reader io.Reader) {
ecmd := exec.Command(cmd[0], cmd[1:]...)
2019-07-08 22:50:40 +00:00
pipe, err := ecmd.StdinPipe()
if err != nil {
return
}
go func() {
defer pipe.Close()
io.Copy(pipe, reader)
}()
err = ecmd.Run()
2019-07-08 22:19:08 +00:00
if err != nil {
aerc.PushError(err.Error())
2019-07-08 22:19:08 +00:00
} else {
2019-07-08 22:32:31 +00:00
if ecmd.ProcessState.ExitCode() != 0 {
aerc.PushError(fmt.Sprintf(
"%s: completed with status %d", cmd[0],
ecmd.ProcessState.ExitCode()))
} else {
aerc.PushStatus(fmt.Sprintf(
"%s: completed with status %d", cmd[0],
ecmd.ProcessState.ExitCode()), 10*time.Second)
2019-07-08 22:32:31 +00:00
}
2019-07-08 22:19:08 +00:00
}
}
2019-07-05 16:21:12 +00:00
if pipeFull {
var uids []uint32
var title string
h := newHelper(aerc)
store, err := h.store()
if err != nil {
return err
}
uids, err = h.markedOrSelectedUids()
if err != nil {
return err
}
if len(uids) == 1 {
info := store.Messages[uids[0]]
if info != nil {
envelope := info.Envelope
if envelope != nil {
title = envelope.Subject
}
}
}
if title == "" {
title = fmt.Sprintf("%d messages", len(uids))
}
var messages []*types.FullMessage
done := make(chan bool, 1)
store.FetchFull(uids, func(fm *types.FullMessage) {
messages = append(messages, fm)
if len(messages) == len(uids) {
done <- true
}
})
go func() {
select {
case <-done:
break
case <-time.After(30 * time.Second):
// TODO: find a better way to determine if store.FetchFull()
// has finished with some errors.
aerc.PushError("Failed to fetch all messages")
if len(messages) == 0 {
return
}
}
// Sort all messages by increasing Message-Id header.
// This will ensure that patch series are applied in order.
sort.Slice(messages, func(i, j int) bool {
infoi := store.Messages[messages[i].Content.Uid]
infoj := store.Messages[messages[j].Content.Uid]
if infoi == nil || infoj == nil {
return false
}
return infoi.Envelope.MessageId < infoj.Envelope.MessageId
})
reader := newMessagesReader(messages)
2019-07-08 22:19:08 +00:00
if background {
doExec(reader)
2019-07-08 22:19:08 +00:00
} else {
doTerm(reader, fmt.Sprintf("%s <%s", cmd[0], title))
2019-07-05 16:21:12 +00:00
}
}()
2019-07-05 16:21:12 +00:00
} else if pipePart {
p := provider.SelectedMessagePart()
2020-07-05 12:27:21 +00:00
if p == nil {
return fmt.Errorf("could not fetch message part")
}
store := provider.Store()
store.FetchBodyPart(p.Msg.Uid, p.Index, func(reader io.Reader) {
2019-07-08 22:19:08 +00:00
if background {
doExec(reader)
} else {
name := fmt.Sprintf("%s <%s/[%d]",
cmd[0], p.Msg.Envelope.Subject, p.Index)
doTerm(reader, name)
2019-07-05 16:21:12 +00:00
}
})
}
return nil
}
// The actual sender address does not matter, nor does the date. This is mostly indended
// for git am which requires separators to look like something valid.
// https://github.com/git/git/blame/v2.35.1/builtin/mailsplit.c#L15-L44
var mboxSeparator []byte = []byte("From ???@??? Tue Jun 23 16:32:49 1981\n")
type messagesReader struct {
messages []*types.FullMessage
mbox bool
separatorNeeded bool
}
func newMessagesReader(messages []*types.FullMessage) io.Reader {
needMboxSeparator := len(messages) > 1
return &messagesReader{messages, needMboxSeparator, needMboxSeparator}
}
func (mr *messagesReader) Read(p []byte) (n int, err error) {
for len(mr.messages) > 0 {
if mr.separatorNeeded {
offset := copy(p, mboxSeparator)
n, err = mr.messages[0].Content.Reader.Read(p[offset:])
n += offset
mr.separatorNeeded = false
} else {
n, err = mr.messages[0].Content.Reader.Read(p)
}
if err == io.EOF {
mr.messages = mr.messages[1:]
mr.separatorNeeded = mr.mbox
}
if n > 0 || err != io.EOF {
if err == io.EOF && len(mr.messages) > 0 {
// Don't return EOF yet. More messages remain.
err = nil
}
return n, err
}
}
return 0, io.EOF
}