aboutsummaryrefslogtreecommitdiff
path: root/worker/notmuch/worker.go
diff options
context:
space:
mode:
authorReto Brunner <reto@labrat.space>2019-08-05 09:16:09 +0200
committerDrew DeVault <sir@cmpwn.com>2019-08-08 10:10:34 +0900
commitc38ddf8d3069663b5ff6a0de36dba3e9efba912d (patch)
treef4ad4d39e1f8a4e841146a273930efcff4183ff9 /worker/notmuch/worker.go
parent2485d509838013351672db50b2cb57880b7c3e1c (diff)
Add notmuch backend
This commit introduces the notmuch backend. The backend is conditionally compiled in if the "notmuch" tag is provided. Most of the message types are implemented, with the notable exceptions of DeleteMessages as well as any copy / move / append type. Reason being, that those aren't normally applicable in a notmuch based workflow. Changes v2 --> v3, based on review comments * Use account config for configuration
Diffstat (limited to 'worker/notmuch/worker.go')
-rw-r--r--worker/notmuch/worker.go393
1 files changed, 393 insertions, 0 deletions
diff --git a/worker/notmuch/worker.go b/worker/notmuch/worker.go
new file mode 100644
index 0000000..6187b24
--- /dev/null
+++ b/worker/notmuch/worker.go
@@ -0,0 +1,393 @@
+//+build notmuch
+
+package notmuch
+
+import (
+ "bufio"
+ "fmt"
+ "net/url"
+ "os"
+ "path/filepath"
+ "strings"
+
+ "git.sr.ht/~sircmpwn/aerc/config"
+ "git.sr.ht/~sircmpwn/aerc/lib/uidstore"
+ "git.sr.ht/~sircmpwn/aerc/models"
+ "git.sr.ht/~sircmpwn/aerc/worker/handlers"
+ "git.sr.ht/~sircmpwn/aerc/worker/types"
+ "github.com/mitchellh/go-homedir"
+ notmuch "github.com/zenhack/go.notmuch"
+)
+
+func init() {
+ handlers.RegisterWorkerFactory("notmuch", NewWorker)
+}
+
+var errUnsupported = fmt.Errorf("unsupported command")
+
+type worker struct {
+ w *types.Worker
+ pathToDB string
+ db *notmuch.DB
+ selected *notmuch.Query
+ uidStore *uidstore.Store
+ excludedTags []string
+ nameQueryMap map[string]string
+}
+
+// NewWorker creates a new maildir worker with the provided worker.
+func NewWorker(w *types.Worker) (types.Backend, error) {
+ return &worker{w: w}, nil
+}
+
+// Run starts the worker's message handling loop.
+func (w *worker) Run() {
+ for {
+ action := <-w.w.Actions
+ msg := w.w.ProcessAction(action)
+ if err := w.handleMessage(msg); err == errUnsupported {
+ w.w.PostMessage(&types.Unsupported{
+ Message: types.RespondTo(msg),
+ }, nil)
+ } else if err != nil {
+ w.w.PostMessage(&types.Error{
+ Message: types.RespondTo(msg),
+ Error: err,
+ }, nil)
+ }
+ }
+}
+
+func (w *worker) done(msg types.WorkerMessage) {
+ w.w.PostMessage(&types.Done{types.RespondTo(msg)}, nil)
+}
+
+func (w *worker) err(msg types.WorkerMessage, err error) {
+ w.w.PostMessage(&types.Error{
+ Message: types.RespondTo(msg),
+ Error: err,
+ }, nil)
+}
+func (w *worker) handleMessage(msg types.WorkerMessage) error {
+ switch msg := msg.(type) {
+ case *types.Unsupported:
+ // No-op
+ case *types.Configure:
+ return w.handleConfigure(msg)
+ case *types.Connect:
+ return w.handleConnect(msg)
+ case *types.ListDirectories:
+ return w.handleListDirectories(msg)
+ case *types.OpenDirectory:
+ return w.handleOpenDirectory(msg)
+ case *types.FetchDirectoryContents:
+ return w.handleFetchDirectoryContents(msg)
+ case *types.FetchMessageHeaders:
+ return w.handleFetchMessageHeaders(msg)
+ case *types.FetchMessageBodyPart:
+ return w.handleFetchMessageBodyPart(msg)
+ case *types.FetchFullMessages:
+ return w.handleFetchFullMessages(msg)
+ case *types.ReadMessages:
+ return w.handleReadMessages(msg)
+ // TODO
+ // return w.handleSearchDirectory(msg)
+ // case *types.DeleteMessages:
+
+ // not implemented, they are generally not used
+ // in a notmuch based workflow
+ // case *types.CopyMessages:
+ // return w.handleCopyMessages(msg)
+ // case *types.AppendMessage:
+ // return w.handleAppendMessage(msg)
+ // case *types.CreateDirectory:
+ // return w.handleCreateDirectory(msg)
+ }
+ return errUnsupported
+}
+
+func (w *worker) handleConfigure(msg *types.Configure) error {
+ u, err := url.Parse(msg.Config.Source)
+ if err != nil {
+ w.w.Logger.Printf("error configuring notmuch worker: %v", err)
+ return err
+ }
+ home, err := homedir.Expand(u.Hostname())
+ if err != nil {
+ return fmt.Errorf("could not resolve home directory: %v", err)
+ }
+ w.pathToDB = filepath.Join(home, u.Path)
+ w.uidStore = uidstore.NewStore()
+
+ if err = w.loadQueryMap(msg.Config); err != nil {
+ return fmt.Errorf("could not load query map: %v", err)
+ }
+ if err = w.loadExcludeTags(msg.Config); err != nil {
+ return fmt.Errorf("could not load excluded tags: %v", err)
+ }
+ w.w.Logger.Printf("configured db directory: %s", w.pathToDB)
+ return nil
+}
+
+func (w *worker) handleConnect(msg *types.Connect) error {
+ var err error
+ w.db, err = notmuch.Open(w.pathToDB, notmuch.DBReadWrite)
+ if err != nil {
+ return fmt.Errorf("could not connect to notmuch db: %v", err)
+ }
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) handleListDirectories(msg *types.ListDirectories) error {
+ for name := range w.nameQueryMap {
+ w.w.PostMessage(&types.Directory{
+ Message: types.RespondTo(msg),
+ Dir: &models.Directory{
+ Name: name,
+ Attributes: []string{},
+ },
+ }, nil)
+ }
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) handleOpenDirectory(msg *types.OpenDirectory) error {
+ w.w.Logger.Printf("opening %s", msg.Directory)
+ // try the friendly name first, if that fails assume it's a query
+ query, ok := w.nameQueryMap[msg.Directory]
+ if !ok {
+ query = msg.Directory
+ }
+ w.selected = w.db.NewQuery(query)
+ w.selected.SetExcludeScheme(notmuch.EXCLUDE_TRUE)
+ w.selected.SetSortScheme(notmuch.SORT_OLDEST_FIRST)
+ for _, t := range w.excludedTags {
+ err := w.selected.AddTagExclude(t)
+ if err != nil && err != notmuch.ErrIgnored {
+ return err
+ }
+ }
+ //TODO: why does this need to be sent twice??
+ info := &types.DirectoryInfo{
+ Info: &models.DirectoryInfo{
+ Name: msg.Directory,
+ Flags: []string{},
+ ReadOnly: false,
+ // total messages
+ Exists: w.selected.CountMessages(),
+ // new messages since mailbox was last opened
+ Recent: 0,
+ // total unread
+ Unseen: 0,
+ },
+ }
+ w.w.PostMessage(info, nil)
+ w.w.PostMessage(info, nil)
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) handleFetchDirectoryContents(
+ msg *types.FetchDirectoryContents) error {
+ uids, err := w.uidsFromQuery(w.selected)
+ if err != nil {
+ w.w.Logger.Printf("error scanning uids: %v", err)
+ return err
+ }
+ w.w.PostMessage(&types.DirectoryContents{
+ Message: types.RespondTo(msg),
+ Uids: uids,
+ }, nil)
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) handleFetchMessageHeaders(
+ msg *types.FetchMessageHeaders) error {
+ for _, uid := range msg.Uids {
+ m, err := w.msgFromUid(uid)
+ if err != nil {
+ w.w.Logger.Printf("could not get message: %v", err)
+ w.err(msg, err)
+ continue
+ }
+ info, err := m.MessageInfo()
+ if err != nil {
+ w.w.Logger.Printf("could not get message info: %v", err)
+ w.err(msg, err)
+ continue
+ }
+ w.w.PostMessage(&types.MessageInfo{
+ Message: types.RespondTo(msg),
+ Info: info,
+ }, nil)
+ }
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) uidsFromQuery(query *notmuch.Query) ([]uint32, error) {
+ msgs, err := query.Messages()
+ if err != nil {
+ return nil, err
+ }
+ var msg *notmuch.Message
+ var uids []uint32
+ for msgs.Next(&msg) {
+ uid := w.uidStore.GetOrInsert(msg.ID())
+ uids = append(uids, uid)
+
+ }
+ return uids, nil
+}
+
+func (w *worker) msgFromUid(uid uint32) (*Message, error) {
+ key, ok := w.uidStore.GetKey(uid)
+ if !ok {
+ return nil, fmt.Errorf("Invalid uid: %v", uid)
+ }
+ nm, err := w.db.FindMessage(key)
+ if err != nil {
+ return nil, fmt.Errorf("Could not fetch message for key %q: %v", key, err)
+ }
+ msg := &Message{
+ key: key,
+ uid: uid,
+ msg: nm,
+ }
+ return msg, nil
+}
+
+func (w *worker) handleFetchMessageBodyPart(
+ msg *types.FetchMessageBodyPart) error {
+
+ m, err := w.msgFromUid(msg.Uid)
+ if err != nil {
+ w.w.Logger.Printf("could not get message %d: %v", msg.Uid, err)
+ return err
+ }
+ r, err := m.NewBodyPartReader(msg.Part)
+ if err != nil {
+ w.w.Logger.Printf(
+ "could not get body part reader for message=%d, parts=%#v: %v",
+ msg.Uid, msg.Part, err)
+ return err
+ }
+ w.w.PostMessage(&types.MessageBodyPart{
+ Message: types.RespondTo(msg),
+ Part: &models.MessageBodyPart{
+ Reader: r,
+ Uid: msg.Uid,
+ },
+ }, nil)
+
+ if err := m.MarkRead(true); err != nil {
+ w.w.Logger.Printf("could not mark message as read: %v", err)
+ return err
+ }
+
+ // send updated flags to ui
+ info, err := m.MessageInfo()
+ if err != nil {
+ w.w.Logger.Printf("could not fetch message info: %v", err)
+ return err
+ }
+ w.w.PostMessage(&types.MessageInfo{
+ Message: types.RespondTo(msg),
+ Info: info,
+ }, nil)
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) handleFetchFullMessages(msg *types.FetchFullMessages) error {
+ for _, uid := range msg.Uids {
+ m, err := w.msgFromUid(uid)
+ if err != nil {
+ w.w.Logger.Printf("could not get message %d: %v", uid, err)
+ return err
+ }
+ r, err := m.NewReader()
+ if err != nil {
+ w.w.Logger.Printf("could not get message reader: %v", err)
+ return err
+ }
+ w.w.PostMessage(&types.FullMessage{
+ Message: types.RespondTo(msg),
+ Content: &models.FullMessage{
+ Uid: uid,
+ Reader: r,
+ },
+ }, nil)
+ }
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) handleReadMessages(msg *types.ReadMessages) error {
+ for _, uid := range msg.Uids {
+ m, err := w.msgFromUid(uid)
+ if err != nil {
+ w.w.Logger.Printf("could not get message: %v", err)
+ w.err(msg, err)
+ continue
+ }
+ if err := m.MarkRead(msg.Read); err != nil {
+ w.w.Logger.Printf("could not mark message as read: %v", err)
+ w.err(msg, err)
+ continue
+ }
+ info, err := m.MessageInfo()
+ if err != nil {
+ w.w.Logger.Printf("could not get message info: %v", err)
+ w.err(msg, err)
+ continue
+ }
+ w.w.PostMessage(&types.MessageInfo{
+ Message: types.RespondTo(msg),
+ Info: info,
+ }, nil)
+ }
+ w.done(msg)
+ return nil
+}
+
+func (w *worker) loadQueryMap(acctConfig *config.AccountConfig) error {
+ raw, ok := acctConfig.Params["query-map"]
+ if !ok {
+ // nothing to do
+ return nil
+ }
+ file, err := homedir.Expand(raw)
+ if err != nil {
+ return err
+ }
+ f, err := os.Open(file)
+ if err != nil {
+ return err
+ }
+ defer f.Close()
+ w.nameQueryMap = make(map[string]string)
+ scanner := bufio.NewScanner(f)
+ for scanner.Scan() {
+ line := scanner.Text()
+ split := strings.SplitN(line, "=", 2)
+ if len(split) != 2 {
+ return fmt.Errorf("invalid line %q, want name=query", line)
+ }
+ w.nameQueryMap[split[0]] = split[1]
+ }
+ return nil
+}
+
+func (w *worker) loadExcludeTags(acctConfig *config.AccountConfig) error {
+ raw, ok := acctConfig.Params["exclude-tags"]
+ if !ok {
+ // nothing to do
+ return nil
+ }
+ w.excludedTags = strings.Split(raw, ",")
+ return nil
+}