978d35d356
Apply GoDoc comment policy (comments for humans should have a space after the //; machine-readable comments shouldn't) Use strings.ReplaceAll instead of strings.Replace when appropriate Remove if/else chains by replacing them with switches Use short assignment/increment notation Replace single case switches with if statements Combine else and if when appropriate Signed-off-by: Moritz Poldrack <moritz@poldrack.dev> Acked-by: Robin Jarry <robin@jarry.cc>
75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package imap
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
|
|
"github.com/emersion/go-imap"
|
|
|
|
"git.sr.ht/~sircmpwn/getopt"
|
|
)
|
|
|
|
func parseSearch(args []string) (*imap.SearchCriteria, error) {
|
|
criteria := imap.NewSearchCriteria()
|
|
if len(args) == 0 {
|
|
return criteria, nil
|
|
}
|
|
|
|
opts, optind, err := getopt.Getopts(args, "rubax:X:t:H:f:c:")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
body := false
|
|
text := false
|
|
for _, opt := range opts {
|
|
switch opt.Option {
|
|
case 'r':
|
|
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':
|
|
criteria.Header.Add("From", opt.Value)
|
|
case 't':
|
|
criteria.Header.Add("To", opt.Value)
|
|
case 'c':
|
|
criteria.Header.Add("Cc", opt.Value)
|
|
case 'b':
|
|
body = true
|
|
case 'a':
|
|
text = true
|
|
}
|
|
}
|
|
switch {
|
|
case text:
|
|
criteria.Text = args[optind:]
|
|
case body:
|
|
criteria.Body = args[optind:]
|
|
default:
|
|
for _, arg := range args[optind:] {
|
|
criteria.Header.Add("Subject", arg)
|
|
}
|
|
}
|
|
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")
|
|
}
|