aerc/worker/imap/worker.go

package imap

import (
	"fmt"
	"log"
	"net/url"
	"strings"

	"github.com/davecgh/go-spew/spew"
	"github.com/emersion/go-imap"
	"github.com/emersion/go-imap-idle"
	"github.com/emersion/go-imap/client"

	"git.sr.ht/~sircmpwn/aerc2/worker/types"
)

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
	logger  *log.Logger
}

func NewIMAPWorker(logger *log.Logger) *IMAPWorker {
	return &IMAPWorker{
		messages: make(chan types.WorkerMessage, 50),
		actions:  make(chan types.WorkerMessage, 50),
		updates:  make(chan client.Update, 50),
		logger:   logger,
	}
}

func (w *IMAPWorker) GetMessages() chan types.WorkerMessage {
	return w.messages
}

func (w *IMAPWorker) PostAction(msg types.WorkerMessage) {
	w.actions <- msg
}

func (w *IMAPWorker) handleMessage(msg types.WorkerMessage) error {
	switch msg := msg.(type) {
	case types.Ping:
		// 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:
		return errUnsupported
	}
	return nil
}

// Logs an action but censors passwords
func (w *IMAPWorker) logAction(msg types.WorkerMessage) {
	switch msg := msg.(type) {
	case types.Configure:
		src := msg.Config.Source
		msg.Config.Source = "[obsfucated]"
		w.logger.Printf("<= %s", spew.Sdump(msg))
		msg.Config.Source = src
	default:
		w.logger.Printf("<= %s", spew.Sdump(msg))
	}
}

func (w *IMAPWorker) Run() {
	for {
		select {
		case msg := <-w.actions:
			w.logAction(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:
			w.logger.Printf("[= %s", spew.Sdump(update))
		}
	}
}