// getEmails will fetch the full bodies of all emails listed in the given command.
func getEmails(client *imap.Client, cmd *imap.Command) ([]map[string]interface{}, error) {
var emails []map[string]interface{}
seq := new(imap.SeqSet)
for _, rsp := range cmd.Data {
for _, uid := range rsp.SearchResults() {
seq.AddNum(uid)
}
}
if seq.Empty() {
return emails, nil
}
fCmd, err := imap.Wait(client.UIDFetch(seq, "INTERNALDATE", "BODY[]", "UID", "RFC822.HEADER"))
if err != nil {
return emails, err
}
var email map[string]interface{}
for _, msgData := range fCmd.Data {
msgFields := msgData.MessageInfo().Attrs
email, err = newEmailMessage(msgFields)
if err != nil {
return emails, err
}
emails = append(emails, email)
// mark message as read
fSeq := new(imap.SeqSet)
fSeq.AddNum(imap.AsNumber(msgFields["UID"]))
_, err = imap.Wait(client.UIDStore(fSeq, "+FLAGS", "\\SEEN"))
if err != nil {
return emails, err
}
}
return emails, nil
}
// loadMessages will search and fetch message body
func loadMessages(client *imap.Client, config *Config) error {
cmd, err := imap.Wait(client.UIDSearch("FROM", client.Quote(config.From), "UNSEEN"))
if err != nil {
return err
}
seq := new(imap.SeqSet)
if len(cmd.Data) > 0 {
seq.AddNum(cmd.Data[0].SearchResults()...)
}
if seq.Empty() {
return nil
}
logger.Infof("Fetched UIDs %v", seq)
cmd, err = imap.Wait(client.UIDFetch(seq, "FLAGS", "INTERNALDATE", "RFC822.SIZE", "BODY[]"))
if err != nil {
return err
}
for _, rsp := range cmd.Data {
body, err := fetchMessage(client, rsp.MessageInfo().UID)
if err != nil {
logger.Fatal(err)
return err
}
err = writeFileFromBody(bytes.NewReader(body), config)
if err != nil {
logger.Fatal(err)
return err
}
}
if !seq.Empty() {
_, err = imap.Wait(client.UIDStore(seq, "+FLAGS.SILENT", imap.NewFlagSet(`\Seen`)))
}
cmd.Data = nil
return err
}