aerc/commands/msg/mark.go

146 lines
2.6 KiB
Go
Raw Normal View History

2019-12-18 06:34:01 +01:00
package msg
import (
"fmt"
"git.sr.ht/~rjarry/aerc/widgets"
2019-12-18 06:34:01 +01:00
"git.sr.ht/~sircmpwn/getopt"
)
type Mark struct{}
func init() {
register(Mark{})
}
func (Mark) Aliases() []string {
return []string{"mark", "unmark", "remark"}
2019-12-18 06:34:01 +01:00
}
func (Mark) Complete(aerc *widgets.Aerc, args []string) []string {
return nil
}
func (Mark) Execute(aerc *widgets.Aerc, args []string) error {
h := newHelper(aerc)
OnSelectedMessage := func(fn func(uint32)) error {
if fn == nil {
return fmt.Errorf("no operation selected")
}
selected, err := h.msgProvider.SelectedMessage()
if err != nil {
return err
}
fn(selected.Uid)
return nil
2019-12-18 06:34:01 +01:00
}
store, err := h.store()
if err != nil {
return err
}
marker := store.Marker()
opts, _, err := getopt.Getopts(args, "atvVT")
2019-12-18 06:34:01 +01:00
if err != nil {
return err
}
var all bool
var toggle bool
var visual bool
var clearVisual bool
var thread bool
2019-12-18 06:34:01 +01:00
for _, opt := range opts {
switch opt.Option {
case 'a':
all = true
case 'v':
visual = true
clearVisual = true
case 'V':
visual = true
2019-12-18 06:34:01 +01:00
case 't':
toggle = true
case 'T':
thread = true
2019-12-18 06:34:01 +01:00
}
}
if thread && len(store.Threads()) == 0 {
return fmt.Errorf("No threads found")
}
if thread && all {
return fmt.Errorf("-a and -T are mutually exclusive")
}
if thread && visual {
return fmt.Errorf("-v and -T are mutually exclusive")
}
2019-12-18 06:34:01 +01:00
switch args[0] {
case "mark":
if all && visual {
return fmt.Errorf("-a and -v are mutually exclusive")
}
var modFunc func(uint32)
if toggle {
modFunc = marker.ToggleMark
2019-12-18 06:34:01 +01:00
} else {
modFunc = marker.Mark
2019-12-18 06:34:01 +01:00
}
switch {
case all:
2019-12-18 06:34:01 +01:00
uids := store.Uids()
for _, uid := range uids {
modFunc(uid)
}
return nil
case visual:
marker.ToggleVisualMark(clearVisual)
2019-12-18 06:34:01 +01:00
return nil
default:
if thread {
for _, uid := range store.SelectedThread().Root().Uids() {
modFunc(uid)
}
} else {
return OnSelectedMessage(modFunc)
}
2019-12-18 06:34:01 +01:00
return nil
}
case "unmark":
if visual {
2020-05-20 02:48:23 +02:00
return fmt.Errorf("visual mode not supported for this command")
2019-12-18 06:34:01 +01:00
}
switch {
case all && toggle:
2019-12-18 06:34:01 +01:00
uids := store.Uids()
for _, uid := range uids {
marker.ToggleMark(uid)
2019-12-18 06:34:01 +01:00
}
return nil
case all && !toggle:
marker.ClearVisualMark()
2019-12-18 06:34:01 +01:00
return nil
default:
if thread {
for _, uid := range store.SelectedThread().Root().Uids() {
marker.Unmark(uid)
}
} else {
return OnSelectedMessage(marker.Unmark)
}
2019-12-18 06:34:01 +01:00
return nil
}
case "remark":
if all || visual || toggle || thread {
return fmt.Errorf("Usage: :remark")
}
marker.Remark()
return nil
2019-12-18 06:34:01 +01:00
}
return nil // never reached
}