Add flag based search options

Provide search and filter with the option to specify more flag based
conditions.
Use '-x <flag>' to search for messages with a flag (seen, answered,
flagged) and '-X <flag>' to search for messages without a flag.

2 files changed, 42 insertions, 2 deletions

diff --git a/worker/imap/search.go b/worker/imap/search.go
index 42e155b6..f866b1cc 100644
--- a/worker/imap/search.go
+++ b/worker/imap/search.go
@@ -1,6 +1,9 @@
 package imap
 
 import (
+	"errors"
+	"strings"
+
 	"github.com/emersion/go-imap"
 
 	"git.sr.ht/~sircmpwn/getopt"
@@ -9,7 +12,7 @@ import (
 func parseSearch(args []string) (*imap.SearchCriteria, error) {
 	criteria := imap.NewSearchCriteria()
 
-	opts, optind, err := getopt.Getopts(args, "rubat:H:f:c:")
+	opts, optind, err := getopt.Getopts(args, "rubax:X:t:H:f:c:")
 	if err != nil {
 		return nil, err
 	}
@@ -21,6 +24,14 @@ func parseSearch(args []string) (*imap.SearchCriteria, error) {
 			criteria.WithFlags = append(criteria.WithFlags, imap.SeenFlag)
 		case 'u':
 			criteria.WithoutFlags = append(criteria.WithoutFlags, imap.SeenFlag)
+		case 'x':
+			if f, err := getParsedFlag(opt.Value); err == nil {
+				criteria.WithFlags = append(criteria.WithFlags, f)
+			}
+		case 'X':
+			if f, err := getParsedFlag(opt.Value); err == nil {
+				criteria.WithoutFlags = append(criteria.WithoutFlags, f)
+			}
 		case 'H':
 			// TODO
 		case 'f':
@@ -46,3 +57,15 @@ func parseSearch(args []string) (*imap.SearchCriteria, error) {
 	}
 	return criteria, nil
 }
+
+func getParsedFlag(name string) (string, error) {
+	switch strings.ToLower(name) {
+	case "seen":
+		return imap.SeenFlag, nil
+	case "flagged":
+		return imap.FlaggedFlag, nil
+	case "answered":
+		return imap.AnsweredFlag, nil
+	}
+	return imap.FlaggedFlag, errors.New("Flag not suppored")
+}
diff --git a/worker/maildir/search.go b/worker/maildir/search.go
index f658c024..005f6faa 100644
--- a/worker/maildir/search.go
+++ b/worker/maildir/search.go
@@ -29,7 +29,7 @@ func newSearchCriteria() *searchCriteria {
 func parseSearch(args []string) (*searchCriteria, error) {
 	criteria := newSearchCriteria()
 
-	opts, optind, err := getopt.Getopts(args, "rubat:H:f:c:")
+	opts, optind, err := getopt.Getopts(args, "rux:X:bat:H:f:c:")
 	if err != nil {
 		return nil, err
 	}
@@ -41,6 +41,10 @@ func parseSearch(args []string) (*searchCriteria, error) {
 			criteria.WithFlags = append(criteria.WithFlags, maildir.FlagSeen)
 		case 'u':
 			criteria.WithoutFlags = append(criteria.WithoutFlags, maildir.FlagSeen)
+		case 'x':
+			criteria.WithFlags = append(criteria.WithFlags, getParsedFlag(opt.Value))
+		case 'X':
+			criteria.WithoutFlags = append(criteria.WithoutFlags, getParsedFlag(opt.Value))
 		case 'H':
 			// TODO
 		case 'f':
@@ -67,6 +71,19 @@ func parseSearch(args []string) (*searchCriteria, error) {
 	return criteria, nil
 }
 
+func getParsedFlag(name string) maildir.Flag {
+	var f maildir.Flag
+	switch strings.ToLower(name) {
+	case "seen":
+		f = maildir.FlagSeen
+	case "answered":
+		f = maildir.FlagReplied
+	case "flagged":
+		f = maildir.FlagFlagged
+	}
+	return f
+}
+
 func (w *Worker) search(criteria *searchCriteria) ([]uint32, error) {
 	requiredParts := getRequiredParts(criteria)
 	w.worker.Logger.Printf("Required parts bitmask for search: %b", requiredParts)