From 1710c9054898e820700d673e21e7c7a90a3f67b1 Mon Sep 17 00:00:00 2001 From: emersion Date: Sun, 14 Jan 2018 11:30:11 +0100 Subject: [PATCH] Connect to IMAP server, login and idle --- ui/account.go | 1 + worker/imap/worker.go | 122 +++++++++++++++++++++++++++++++++++---- worker/types/messages.go | 2 + 3 files changed, 114 insertions(+), 11 deletions(-) diff --git a/ui/account.go b/ui/account.go index 50f41e4..8353db3 100644 --- a/ui/account.go +++ b/ui/account.go @@ -26,6 +26,7 @@ func NewAccountTab(conf *config.AccountConfig) (*AccountTab, error) { } go work.Run() work.PostAction(types.Configure{Config: conf}) + work.PostAction(types.Connect{}) return &AccountTab{ Config: conf, Worker: work, diff --git a/worker/imap/worker.go b/worker/imap/worker.go index 080927d..6971f8c 100644 --- a/worker/imap/worker.go +++ b/worker/imap/worker.go @@ -1,20 +1,43 @@ package imap import ( - "time" + "fmt" + "net/url" + "strings" "git.sr.ht/~sircmpwn/aerc2/worker/types" + "github.com/emersion/go-imap" + "github.com/emersion/go-imap/client" + "github.com/emersion/go-imap-idle" ) +var errUnsupported = fmt.Errorf("unsupported command") + +type imapClient struct { + *client.Client + *idle.IdleClient +} + type IMAPWorker struct { messages chan types.WorkerMessage actions chan types.WorkerMessage + + config struct { + scheme string + insecure bool + addr string + user *url.Userinfo + } + + client *imapClient + updates chan client.Update } func NewIMAPWorker() *IMAPWorker { return &IMAPWorker{ messages: make(chan types.WorkerMessage, 50), actions: make(chan types.WorkerMessage, 50), + updates: make(chan client.Update, 50), } } @@ -26,27 +49,104 @@ func (w *IMAPWorker) PostAction(msg types.WorkerMessage) { w.actions <- msg } -func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) { +func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error { switch msg := msg.(type) { case types.Ping: - w.messages <- types.Ack{ - Message: types.RespondTo(msg), + // No-op + case types.Configure: + u, err := url.Parse(msg.Config.Source) + if err != nil { + return err } + + 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 + } + + w.config.addr = u.Host + if !strings.ContainsRune(w.config.addr, ':') { + w.config.addr += ":" + u.Scheme + } + + w.config.scheme = u.Scheme + w.config.user = u.User + case types.Connect: + // TODO: populate TLS config + + 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 { + if err := c.StartTLS(nil); err != nil { + return err + } + } + case "imaps": + c, err = client.DialTLS(w.config.addr, nil) + if err != nil { + return err + } + default: + return fmt.Errorf("Unknown IMAP scheme %s", w.config.scheme) + } + + if w.config.user != nil { + username := w.config.user.Username() + password, hasPassword := w.config.user.Password() + if !hasPassword { + // TODO: ask password + } + if err := c.Login(username, password); err != nil { + return err + } + } + + if _, err := c.Select(imap.InboxName, false); err != nil { + return err + } + + c.Updates = w.updates + w.client = &imapClient{c, idle.NewClient(c)} + + // TODO: don't idle right away + go w.client.IdleWithFallback(nil, 0) default: - w.messages <- types.Unsupported{ - Message: types.RespondTo(msg), - } + return errUnsupported } + return nil } func (w *IMAPWorker) Run() { - // TODO: IMAP shit for { select { case msg := <-w.actions: - w.handleMessage(msg) - default: - time.Sleep(100 * time.Millisecond) + fmt.Printf("<= %T\n", msg) + if err := w.handleMessage(msg); err == errUnsupported { + w.messages <- types.Unsupported{ + Message: types.RespondTo(msg), + } + } else if err != nil { + w.messages <- types.Error{ + Message: types.RespondTo(msg), + Error: err, + } + } else { + w.messages <- types.Ack{ + Message: types.RespondTo(msg), + } + } + case update := <-w.updates: + fmt.Printf("<= %T\n", update) } } } diff --git a/worker/types/messages.go b/worker/types/messages.go index a4d8ddb..e83bd6b 100644 --- a/worker/types/messages.go +++ b/worker/types/messages.go @@ -13,6 +13,7 @@ type Message struct { } // Meta-messages + type Ack struct { Message } @@ -27,6 +28,7 @@ type Unsupported struct { } // Commands + type Ping struct { Message }