79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package account
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"git.sr.ht/~sircmpwn/getopt"
|
|
"github.com/emersion/go-imap"
|
|
|
|
"git.sr.ht/~sircmpwn/aerc/widgets"
|
|
)
|
|
|
|
type SearchFilter struct{}
|
|
|
|
func init() {
|
|
register(SearchFilter{})
|
|
}
|
|
|
|
func (_ SearchFilter) Aliases() []string {
|
|
return []string{"search", "filter"}
|
|
}
|
|
|
|
func (_ SearchFilter) Complete(aerc *widgets.Aerc, args []string) []string {
|
|
return nil
|
|
}
|
|
|
|
func (_ SearchFilter) Execute(aerc *widgets.Aerc, args []string) error {
|
|
var (
|
|
criteria *imap.SearchCriteria = imap.NewSearchCriteria()
|
|
)
|
|
|
|
opts, optind, err := getopt.Getopts(args, "ruH:")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
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 'H':
|
|
// TODO
|
|
}
|
|
}
|
|
for _, arg := range args[optind:] {
|
|
criteria.Header.Add("Subject", arg)
|
|
}
|
|
|
|
acct := aerc.SelectedAccount()
|
|
if acct == nil {
|
|
return errors.New("No account selected")
|
|
}
|
|
store := acct.Store()
|
|
if store == nil {
|
|
return errors.New("Cannot perform action. Messages still loading")
|
|
}
|
|
|
|
var cb func([]uint32)
|
|
if args[0] == "filter" {
|
|
aerc.SetStatus("Filtering...")
|
|
cb = func(uids []uint32) {
|
|
aerc.SetStatus("Filter complete.")
|
|
acct.Logger().Printf("Filter results: %v", uids)
|
|
store.ApplyFilter(uids)
|
|
}
|
|
} else {
|
|
aerc.SetStatus("Searching...")
|
|
cb = func(uids []uint32) {
|
|
aerc.SetStatus("Search complete.")
|
|
acct.Logger().Printf("Search results: %v", uids)
|
|
store.ApplySearch(uids)
|
|
// TODO: Remove when stores have multiple OnUpdate handlers
|
|
acct.Messages().Scroll()
|
|
}
|
|
}
|
|
store.Search(criteria, cb)
|
|
return nil
|
|
}
|