init: pristine aerc 0.20.0 source

commands/msg/mark.go

@@ -0,0 +1,131 @@
+package msg
+
+import (
+	"fmt"
+
+	"git.sr.ht/~rjarry/aerc/commands"
+	"git.sr.ht/~rjarry/aerc/models"
+)
+
+type Mark struct {
+	All         bool `opt:"-a" aliases:"mark,unmark" desc:"Mark all messages in current folder."`
+	Toggle      bool `opt:"-t" aliases:"mark,unmark" desc:"Toggle the marked state."`
+	Visual      bool `opt:"-v" aliases:"mark,unmark" desc:"Enter / leave visual mark mode."`
+	VisualClear bool `opt:"-V" aliases:"mark,unmark" desc:"Same as -v but does not clear existing selection."`
+	Thread      bool `opt:"-T" aliases:"mark,unmark" desc:"Mark all messages from the selected thread."`
+}
+
+func init() {
+	commands.Register(Mark{})
+}
+
+func (Mark) Description() string {
+	return "Mark, unmark or remark messages."
+}
+
+func (Mark) Context() commands.CommandContext {
+	return commands.MESSAGE_LIST | commands.MESSAGE_VIEWER
+}
+
+func (Mark) Aliases() []string {
+	return []string{"mark", "unmark", "remark"}
+}
+
+func (m Mark) Execute(args []string) error {
+	h := newHelper()
+	OnSelectedMessage := func(fn func(models.UID)) 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
+	}
+	store, err := h.store()
+	if err != nil {
+		return err
+	}
+	marker := store.Marker()
+
+	if m.Thread && m.All {
+		return fmt.Errorf("-a and -T are mutually exclusive")
+	}
+
+	if m.Thread && (m.Visual || m.VisualClear) {
+		return fmt.Errorf("-v and -T are mutually exclusive")
+	}
+	if m.Visual && m.All {
+		return fmt.Errorf("-a and -v are mutually exclusive")
+	}
+
+	switch args[0] {
+	case "mark":
+		var modFunc func(models.UID)
+		if m.Toggle {
+			modFunc = marker.ToggleMark
+		} else {
+			modFunc = marker.Mark
+		}
+		switch {
+		case m.All:
+			uids := store.Uids()
+			for _, uid := range uids {
+				modFunc(uid)
+			}
+			return nil
+		case m.Visual || m.VisualClear:
+			marker.ToggleVisualMark(m.VisualClear)
+			return nil
+		default:
+			if m.Thread {
+				threadPtr, err := store.SelectedThread()
+				if err != nil {
+					return err
+				}
+				for _, uid := range threadPtr.Root().Uids() {
+					modFunc(uid)
+				}
+			} else {
+				return OnSelectedMessage(modFunc)
+			}
+			return nil
+		}
+
+	case "unmark":
+		if m.Visual || m.VisualClear {
+			return fmt.Errorf("visual mode not supported for this command")
+		}
+
+		switch {
+		case m.All && m.Toggle:
+			uids := store.Uids()
+			for _, uid := range uids {
+				marker.ToggleMark(uid)
+			}
+			return nil
+		case m.All && !m.Toggle:
+			marker.ClearVisualMark()
+			return nil
+		default:
+			if m.Thread {
+				threadPtr, err := store.SelectedThread()
+				if err != nil {
+					return err
+				}
+				for _, uid := range threadPtr.Root().Uids() {
+					marker.Unmark(uid)
+				}
+			} else {
+				return OnSelectedMessage(marker.Unmark)
+			}
+			return nil
+		}
+	case "remark":
+		marker.Remark()
+		return nil
+	}
+	return nil // never reached
+}