directory: add IANA mailbox roles

Add IANA registered mailbox role, and a custom aerc role "query". This
will be used in subsequent commits which add the Role field to
templates, allowing users to style mailbox by IANA role, or style
notmuch queries differently than maildir dirs when using the notmuch
worker + maildir option.

Signed-off-by: Tim Culverhouse <tim@timculverhouse.com>
Acked-by: Robin Jarry<robin@jarry.cc>

2 files changed, 20 insertions, 3 deletions

diff --git a/worker/imap/list.go b/worker/imap/list.go
index 657779e5..41924d86 100644
--- a/worker/imap/list.go
+++ b/worker/imap/list.go
@@ -1,6 +1,8 @@
 package imap
 
 import (
+	"strings"
+
 	"github.com/emersion/go-imap"
 
 	"git.sr.ht/~rjarry/aerc/log"
@@ -21,11 +23,24 @@ func (imapw *IMAPWorker) handleListDirectories(msg *types.ListDirectories) {
 				// no need to pass this to handlers if it can't be opened
 				continue
 			}
+			dir := &models.Directory{
+				Name: mbox.Name,
+			}
+			for _, attr := range mbox.Attributes {
+				attr = strings.TrimPrefix(attr, "\\")
+				attr = strings.ToLower(attr)
+				role, ok := models.Roles[attr]
+				if !ok {
+					continue
+				}
+				dir.Role = role
+			}
+			if mbox.Name == "INBOX" {
+				dir.Role = models.InboxRole
+			}
 			imapw.worker.PostMessage(&types.Directory{
 				Message: types.RespondTo(msg),
-				Dir: &models.Directory{
-					Name: mbox.Name,
-				},
+				Dir:     dir,
 			}, nil)
 		}
 		done <- nil
diff --git a/worker/notmuch/worker.go b/worker/notmuch/worker.go
index b25a8786..6b1b64d9 100644
--- a/worker/notmuch/worker.go
+++ b/worker/notmuch/worker.go
@@ -259,6 +259,7 @@ func (w *worker) handleListDirectories(msg *types.ListDirectories) error {
 			Message: types.RespondTo(msg),
 			Dir: &models.Directory{
 				Name: name,
+				Role: models.QueryRole,
 			},
 		}, nil)
 	}
@@ -317,6 +318,7 @@ func (w *worker) handleOpenDirectory(msg *types.OpenDirectory) error {
 				Message: types.RespondTo(msg),
 				Dir: &models.Directory{
 					Name: q,
+					Role: models.QueryRole,
 				},
 			}, nil)
 		}