From 5da58c669ce516f7ebe2f6db7b9b30d43e31311d Mon Sep 17 00:00:00 2001 From: Reto Brunner Date: Wed, 18 Dec 2019 06:34:01 +0100 Subject: [PATCH] add mark command --- commands/msg/mark.go | 98 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 98 insertions(+) create mode 100644 commands/msg/mark.go diff --git a/commands/msg/mark.go b/commands/msg/mark.go new file mode 100644 index 0000000..6738a5e --- /dev/null +++ b/commands/msg/mark.go @@ -0,0 +1,98 @@ +package msg + +import ( + "fmt" + + "git.sr.ht/~sircmpwn/aerc/widgets" + "git.sr.ht/~sircmpwn/getopt" +) + +type Mark struct{} + +func init() { + register(Mark{}) +} + +func (Mark) Aliases() []string { + return []string{"mark", "unmark"} +} + +func (Mark) Complete(aerc *widgets.Aerc, args []string) []string { + return nil +} + +func (Mark) Execute(aerc *widgets.Aerc, args []string) error { + h := newHelper(aerc) + selected, err := h.msgProvider.SelectedMessage() + if err != nil { + return err + } + store, err := h.store() + if err != nil { + return err + } + opts, _, err := getopt.Getopts(args, "atv") + if err != nil { + return err + } + var all bool + var toggle bool + var visual bool + for _, opt := range opts { + switch opt.Option { + case 'a': + all = true + case 'v': + visual = true + case 't': + toggle = true + } + } + + switch args[0] { + case "mark": + if all && visual { + return fmt.Errorf("-a and -v are mutually exclusive") + } + + var modFunc func(uint32) + if toggle { + modFunc = store.ToggleMark + } else { + modFunc = store.Mark + } + if all { + uids := store.Uids() + for _, uid := range uids { + modFunc(uid) + } + return nil + } else if visual { + store.ToggleVisualMark() + return nil + } else { + modFunc(selected.Uid) + return nil + } + + case "unmark": + if visual { + return fmt.Errorf("visual mode not suported for this command") + } + + if all && toggle { + uids := store.Uids() + for _, uid := range uids { + store.ToggleMark(uid) + } + return nil + } else if all && !toggle { + store.ClearVisualMark() + return nil + } else { + store.Unmark(selected.Uid) + return nil + } + } + return nil // never reached +}