58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package commands
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"strings"
|
|
|
|
"github.com/emersion/go-imap"
|
|
)
|
|
|
|
// Search is a SEARCH command, as defined in RFC 3501 section 6.4.4.
|
|
type Search struct {
|
|
Charset string
|
|
Criteria *imap.SearchCriteria
|
|
}
|
|
|
|
func (cmd *Search) Command() *imap.Command {
|
|
var args []interface{}
|
|
if cmd.Charset != "" {
|
|
args = append(args, imap.RawString("CHARSET"), cmd.Charset)
|
|
}
|
|
args = append(args, cmd.Criteria.Format()...)
|
|
|
|
return &imap.Command{
|
|
Name: "SEARCH",
|
|
Arguments: args,
|
|
}
|
|
}
|
|
|
|
func (cmd *Search) Parse(fields []interface{}) error {
|
|
if len(fields) == 0 {
|
|
return errors.New("Missing search criteria")
|
|
}
|
|
|
|
// Parse charset
|
|
if f, ok := fields[0].(string); ok && strings.EqualFold(f, "CHARSET") {
|
|
if len(fields) < 2 {
|
|
return errors.New("Missing CHARSET value")
|
|
}
|
|
if cmd.Charset, ok = fields[1].(string); !ok {
|
|
return errors.New("Charset must be a string")
|
|
}
|
|
fields = fields[2:]
|
|
}
|
|
|
|
var charsetReader func(io.Reader) io.Reader
|
|
charset := strings.ToLower(cmd.Charset)
|
|
if charset != "utf-8" && charset != "us-ascii" && charset != "" {
|
|
charsetReader = func(r io.Reader) io.Reader {
|
|
r, _ = imap.CharsetReader(charset, r)
|
|
return r
|
|
}
|
|
}
|
|
|
|
cmd.Criteria = new(imap.SearchCriteria)
|
|
return cmd.Criteria.ParseWithCharset(fields, charsetReader)
|
|
}
|