2018-01-10 01:39:00 +00:00
|
|
|
package imap
|
|
|
|
|
|
|
|
import (
|
2018-02-01 02:54:52 +00:00
|
|
|
"crypto/tls"
|
2018-01-14 10:30:11 +00:00
|
|
|
"fmt"
|
|
|
|
"net/url"
|
|
|
|
"strings"
|
2018-01-10 16:19:45 +00:00
|
|
|
|
2018-01-14 10:30:11 +00:00
|
|
|
"github.com/emersion/go-imap"
|
2020-09-12 13:05:02 +00:00
|
|
|
sortthread "github.com/emersion/go-imap-sortthread"
|
2018-02-01 02:18:21 +00:00
|
|
|
"github.com/emersion/go-imap/client"
|
2019-07-10 19:00:28 +00:00
|
|
|
"golang.org/x/oauth2"
|
2018-02-01 02:18:21 +00:00
|
|
|
|
2019-07-10 19:00:28 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/lib"
|
2019-07-08 02:43:56 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/models"
|
2019-07-18 04:25:42 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/worker/handlers"
|
2019-05-18 00:57:10 +00:00
|
|
|
"git.sr.ht/~sircmpwn/aerc/worker/types"
|
2018-01-10 01:39:00 +00:00
|
|
|
)
|
|
|
|
|
2019-07-18 04:25:42 +00:00
|
|
|
func init() {
|
|
|
|
handlers.RegisterWorkerFactory("imap", NewIMAPWorker)
|
|
|
|
handlers.RegisterWorkerFactory("imaps", NewIMAPWorker)
|
|
|
|
}
|
|
|
|
|
2018-01-14 10:30:11 +00:00
|
|
|
var errUnsupported = fmt.Errorf("unsupported command")
|
|
|
|
|
|
|
|
type imapClient struct {
|
|
|
|
*client.Client
|
2020-09-12 13:05:02 +00:00
|
|
|
sort *sortthread.SortClient
|
2018-01-14 10:30:11 +00:00
|
|
|
}
|
|
|
|
|
2018-01-10 01:39:00 +00:00
|
|
|
type IMAPWorker struct {
|
2018-01-14 10:30:11 +00:00
|
|
|
config struct {
|
2019-07-10 19:00:28 +00:00
|
|
|
scheme string
|
|
|
|
insecure bool
|
|
|
|
addr string
|
|
|
|
user *url.Userinfo
|
|
|
|
folders []string
|
|
|
|
oauthBearer lib.OAuthBearer
|
2018-01-14 10:30:11 +00:00
|
|
|
}
|
|
|
|
|
2019-03-11 03:45:00 +00:00
|
|
|
client *imapClient
|
2019-05-14 00:16:55 +00:00
|
|
|
idleStop chan struct{}
|
|
|
|
idleDone chan error
|
2019-08-22 12:53:27 +00:00
|
|
|
selected *imap.MailboxStatus
|
2019-03-11 03:45:00 +00:00
|
|
|
updates chan client.Update
|
|
|
|
worker *types.Worker
|
2019-03-21 03:23:38 +00:00
|
|
|
// Map of sequence numbers to UIDs, index 0 is seq number 1
|
|
|
|
seqMap []uint32
|
2018-01-10 01:39:00 +00:00
|
|
|
}
|
|
|
|
|
2019-07-18 04:25:42 +00:00
|
|
|
func NewIMAPWorker(worker *types.Worker) (types.Backend, error) {
|
2018-01-10 01:39:00 +00:00
|
|
|
return &IMAPWorker{
|
2019-05-14 00:16:55 +00:00
|
|
|
idleDone: make(chan error),
|
|
|
|
updates: make(chan client.Update, 50),
|
|
|
|
worker: worker,
|
2019-08-22 12:53:27 +00:00
|
|
|
selected: &imap.MailboxStatus{},
|
2019-07-18 04:25:42 +00:00
|
|
|
}, nil
|
2018-01-10 01:39:00 +00:00
|
|
|
}
|
|
|
|
|
2018-01-14 10:30:11 +00:00
|
|
|
func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error {
|
2019-05-14 00:16:55 +00:00
|
|
|
if w.idleStop != nil {
|
|
|
|
close(w.idleStop)
|
|
|
|
if err := <-w.idleDone; err != nil {
|
|
|
|
w.worker.PostMessage(&types.Error{Error: err}, nil)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-15 11:37:18 +00:00
|
|
|
var reterr error // will be returned at the end, needed to support idle
|
|
|
|
|
2018-01-10 16:19:45 +00:00
|
|
|
switch msg := msg.(type) {
|
2018-02-02 00:54:19 +00:00
|
|
|
case *types.Unsupported:
|
2018-01-14 10:30:11 +00:00
|
|
|
// No-op
|
2018-02-02 00:54:19 +00:00
|
|
|
case *types.Configure:
|
2018-01-14 10:30:11 +00:00
|
|
|
u, err := url.Parse(msg.Config.Source)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
2018-01-10 01:39:00 +00:00
|
|
|
}
|
2018-01-14 10:30:11 +00:00
|
|
|
|
|
|
|
w.config.scheme = u.Scheme
|
|
|
|
if strings.HasSuffix(w.config.scheme, "+insecure") {
|
|
|
|
w.config.scheme = strings.TrimSuffix(w.config.scheme, "+insecure")
|
|
|
|
w.config.insecure = true
|
|
|
|
}
|
|
|
|
|
2019-07-10 19:00:28 +00:00
|
|
|
if strings.HasSuffix(w.config.scheme, "+oauthbearer") {
|
|
|
|
w.config.scheme = strings.TrimSuffix(w.config.scheme, "+oauthbearer")
|
|
|
|
w.config.oauthBearer.Enabled = true
|
|
|
|
q := u.Query()
|
2019-10-13 18:01:16 +00:00
|
|
|
|
|
|
|
oauth2 := &oauth2.Config{}
|
2019-07-10 19:00:28 +00:00
|
|
|
if q.Get("token_endpoint") != "" {
|
2019-10-13 18:01:16 +00:00
|
|
|
oauth2.ClientID = q.Get("client_id")
|
|
|
|
oauth2.ClientSecret = q.Get("client_secret")
|
|
|
|
oauth2.Scopes = []string{q.Get("scope")}
|
|
|
|
oauth2.Endpoint.TokenURL = q.Get("token_endpoint")
|
2019-07-10 19:00:28 +00:00
|
|
|
}
|
2019-10-13 18:01:16 +00:00
|
|
|
w.config.oauthBearer.OAuth2 = oauth2
|
2019-07-10 19:00:28 +00:00
|
|
|
}
|
|
|
|
|
2018-01-14 10:30:11 +00:00
|
|
|
w.config.addr = u.Host
|
|
|
|
if !strings.ContainsRune(w.config.addr, ':') {
|
2019-05-20 23:20:20 +00:00
|
|
|
w.config.addr += ":" + w.config.scheme
|
2018-01-10 01:39:00 +00:00
|
|
|
}
|
2018-01-14 10:30:11 +00:00
|
|
|
|
|
|
|
w.config.user = u.User
|
2019-06-12 06:31:51 +00:00
|
|
|
w.config.folders = msg.Config.Folders
|
2018-02-02 00:54:19 +00:00
|
|
|
case *types.Connect:
|
2018-01-14 10:30:11 +00:00
|
|
|
var (
|
|
|
|
c *client.Client
|
|
|
|
err error
|
|
|
|
)
|
|
|
|
switch w.config.scheme {
|
|
|
|
case "imap":
|
|
|
|
c, err = client.Dial(w.config.addr)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if !w.config.insecure {
|
2019-05-20 18:01:59 +00:00
|
|
|
if err := c.StartTLS(&tls.Config{}); err != nil {
|
2018-01-14 10:30:11 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
case "imaps":
|
2019-05-20 18:01:59 +00:00
|
|
|
c, err = client.DialTLS(w.config.addr, &tls.Config{})
|
2018-01-14 10:30:11 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return fmt.Errorf("Unknown IMAP scheme %s", w.config.scheme)
|
|
|
|
}
|
2019-08-04 20:02:01 +00:00
|
|
|
c.ErrorLog = w.worker.Logger
|
2018-01-14 10:30:11 +00:00
|
|
|
|
|
|
|
if w.config.user != nil {
|
|
|
|
username := w.config.user.Username()
|
|
|
|
password, hasPassword := w.config.user.Password()
|
|
|
|
if !hasPassword {
|
|
|
|
// TODO: ask password
|
|
|
|
}
|
2019-07-10 19:00:28 +00:00
|
|
|
|
|
|
|
if w.config.oauthBearer.Enabled {
|
|
|
|
if err := w.config.oauthBearer.Authenticate(username, password, c); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
} else if err := c.Login(username, password); err != nil {
|
2018-01-14 10:30:11 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-05-15 23:41:21 +00:00
|
|
|
c.SetDebug(w.worker.Logger.Writer())
|
|
|
|
|
2018-01-14 10:30:11 +00:00
|
|
|
if _, err := c.Select(imap.InboxName, false); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
c.Updates = w.updates
|
2021-11-01 20:19:17 +00:00
|
|
|
w.client = &imapClient{c, sortthread.NewSortClient(c)}
|
2018-02-02 00:54:19 +00:00
|
|
|
w.worker.PostMessage(&types.Done{types.RespondTo(msg)}, nil)
|
|
|
|
case *types.ListDirectories:
|
2018-02-02 00:34:08 +00:00
|
|
|
w.handleListDirectories(msg)
|
2019-01-13 21:18:10 +00:00
|
|
|
case *types.OpenDirectory:
|
|
|
|
w.handleOpenDirectory(msg)
|
2019-03-11 03:45:00 +00:00
|
|
|
case *types.FetchDirectoryContents:
|
|
|
|
w.handleFetchDirectoryContents(msg)
|
2019-06-08 17:41:56 +00:00
|
|
|
case *types.CreateDirectory:
|
|
|
|
w.handleCreateDirectory(msg)
|
2020-08-18 20:27:23 +00:00
|
|
|
case *types.RemoveDirectory:
|
|
|
|
w.handleRemoveDirectory(msg)
|
2019-03-15 02:19:04 +00:00
|
|
|
case *types.FetchMessageHeaders:
|
|
|
|
w.handleFetchMessageHeaders(msg)
|
2019-03-31 16:14:37 +00:00
|
|
|
case *types.FetchMessageBodyPart:
|
|
|
|
w.handleFetchMessageBodyPart(msg)
|
2019-03-31 16:17:57 +00:00
|
|
|
case *types.FetchFullMessages:
|
|
|
|
w.handleFetchFullMessages(msg)
|
2019-03-21 03:23:38 +00:00
|
|
|
case *types.DeleteMessages:
|
|
|
|
w.handleDeleteMessages(msg)
|
2020-07-05 14:29:52 +00:00
|
|
|
case *types.FlagMessages:
|
|
|
|
w.handleFlagMessages(msg)
|
2020-05-25 14:59:48 +00:00
|
|
|
case *types.AnsweredMessages:
|
|
|
|
w.handleAnsweredMessages(msg)
|
2019-05-14 20:34:42 +00:00
|
|
|
case *types.CopyMessages:
|
|
|
|
w.handleCopyMessages(msg)
|
2019-05-15 23:41:21 +00:00
|
|
|
case *types.AppendMessage:
|
|
|
|
w.handleAppendMessage(msg)
|
2019-06-24 20:31:37 +00:00
|
|
|
case *types.SearchDirectory:
|
|
|
|
w.handleSearchDirectory(msg)
|
2018-01-14 10:30:11 +00:00
|
|
|
default:
|
2020-02-15 11:37:18 +00:00
|
|
|
reterr = errUnsupported
|
2018-01-10 01:39:00 +00:00
|
|
|
}
|
2019-05-14 00:16:55 +00:00
|
|
|
|
|
|
|
if w.idleStop != nil {
|
|
|
|
w.idleStop = make(chan struct{})
|
|
|
|
go func() {
|
2021-11-01 20:19:17 +00:00
|
|
|
w.idleDone <- w.client.Idle(w.idleStop, &client.IdleOptions{0, 0})
|
2019-05-14 00:16:55 +00:00
|
|
|
}()
|
|
|
|
}
|
2020-02-15 11:37:18 +00:00
|
|
|
return reterr
|
2018-01-10 01:39:00 +00:00
|
|
|
}
|
|
|
|
|
2019-01-13 21:18:10 +00:00
|
|
|
func (w *IMAPWorker) handleImapUpdate(update client.Update) {
|
|
|
|
w.worker.Logger.Printf("(= %T", update)
|
|
|
|
switch update := update.(type) {
|
|
|
|
case *client.MailboxUpdate:
|
|
|
|
status := update.Mailbox
|
2019-03-11 03:45:00 +00:00
|
|
|
if w.selected.Name == status.Name {
|
2019-08-22 12:53:27 +00:00
|
|
|
w.selected = status
|
2019-03-11 03:45:00 +00:00
|
|
|
}
|
2019-01-13 21:18:10 +00:00
|
|
|
w.worker.PostMessage(&types.DirectoryInfo{
|
2019-07-08 02:43:56 +00:00
|
|
|
Info: &models.DirectoryInfo{
|
|
|
|
Flags: status.Flags,
|
|
|
|
Name: status.Name,
|
|
|
|
ReadOnly: status.ReadOnly,
|
|
|
|
|
|
|
|
Exists: int(status.Messages),
|
|
|
|
Recent: int(status.Recent),
|
|
|
|
Unseen: int(status.Unseen),
|
|
|
|
},
|
2019-01-13 21:18:10 +00:00
|
|
|
}, nil)
|
2019-06-09 18:55:04 +00:00
|
|
|
case *client.MessageUpdate:
|
|
|
|
msg := update.Message
|
|
|
|
if msg.Uid == 0 {
|
|
|
|
msg.Uid = w.seqMap[msg.SeqNum-1]
|
|
|
|
}
|
|
|
|
w.worker.PostMessage(&types.MessageInfo{
|
2019-07-08 02:43:56 +00:00
|
|
|
Info: &models.MessageInfo{
|
2019-07-08 02:43:58 +00:00
|
|
|
BodyStructure: translateBodyStructure(msg.BodyStructure),
|
|
|
|
Envelope: translateEnvelope(msg.Envelope),
|
2020-03-03 13:45:06 +00:00
|
|
|
Flags: translateImapFlags(msg.Flags),
|
2019-07-08 02:43:56 +00:00
|
|
|
InternalDate: msg.InternalDate,
|
|
|
|
Uid: msg.Uid,
|
|
|
|
},
|
2019-06-09 18:55:04 +00:00
|
|
|
}, nil)
|
2019-05-14 00:23:23 +00:00
|
|
|
case *client.ExpungeUpdate:
|
|
|
|
i := update.SeqNum - 1
|
|
|
|
uid := w.seqMap[i]
|
|
|
|
w.seqMap = append(w.seqMap[:i], w.seqMap[i+1:]...)
|
|
|
|
w.worker.PostMessage(&types.MessagesDeleted{
|
|
|
|
Uids: []uint32{uid},
|
|
|
|
}, nil)
|
2019-01-13 21:18:10 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-01-10 01:39:00 +00:00
|
|
|
func (w *IMAPWorker) Run() {
|
|
|
|
for {
|
|
|
|
select {
|
2018-02-01 23:42:03 +00:00
|
|
|
case msg := <-w.worker.Actions:
|
|
|
|
msg = w.worker.ProcessAction(msg)
|
2018-01-14 10:30:11 +00:00
|
|
|
if err := w.handleMessage(msg); err == errUnsupported {
|
2018-02-02 00:54:19 +00:00
|
|
|
w.worker.PostMessage(&types.Unsupported{
|
2018-01-14 10:30:11 +00:00
|
|
|
Message: types.RespondTo(msg),
|
2018-02-01 23:42:03 +00:00
|
|
|
}, nil)
|
2018-01-14 10:30:11 +00:00
|
|
|
} else if err != nil {
|
2018-02-02 00:54:19 +00:00
|
|
|
w.worker.PostMessage(&types.Error{
|
2018-01-14 10:30:11 +00:00
|
|
|
Message: types.RespondTo(msg),
|
|
|
|
Error: err,
|
2018-02-01 23:42:03 +00:00
|
|
|
}, nil)
|
2018-01-14 10:30:11 +00:00
|
|
|
}
|
|
|
|
case update := <-w.updates:
|
2019-01-13 21:18:10 +00:00
|
|
|
w.handleImapUpdate(update)
|
2018-01-10 01:39:00 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|