Categorize and normalize folder listing

2 files changed, 75 insertions, 15 deletions

diff --git a/plugins/base/imap.go b/plugins/base/imap.go
index 16bd2ef..a7f4ea1 100644
--- a/plugins/base/imap.go
+++ b/plugins/base/imap.go
@@ -19,6 +19,9 @@ import (
 
 type MailboxInfo struct {
 	*imap.MailboxInfo
+
+	Active bool
+	Unseen int
 }
 
 func (mbox *MailboxInfo) URL() *url.URL {
@@ -27,6 +30,15 @@ func (mbox *MailboxInfo) URL() *url.URL {
 	}
 }
 
+func (mbox *MailboxInfo) HasAttr(flag string) bool {
+	for _, attr := range mbox.Attributes {
+		if attr == flag {
+			return true
+		}
+	}
+	return false
+}
+
 func listMailboxes(conn *imapclient.Client) ([]MailboxInfo, error) {
 	ch := make(chan *imap.MailboxInfo, 10)
 	done := make(chan error, 1)
@@ -36,7 +48,7 @@ func listMailboxes(conn *imapclient.Client) ([]MailboxInfo, error) {
 
 	var mailboxes []MailboxInfo
 	for mbox := range ch {
-		mailboxes = append(mailboxes, MailboxInfo{mbox})
+		mailboxes = append(mailboxes, MailboxInfo{mbox, false, -1})
 	}
 
 	if err := <-done; err != nil {
@@ -133,7 +145,7 @@ func getMailboxByType(conn *imapclient.Client, mboxType mailboxType) (*MailboxIn
 	if best == nil {
 		return nil, nil
 	}
-	return &MailboxInfo{best}, nil
+	return &MailboxInfo{best, false, -1}, nil
 }
 
 func ensureMailboxSelected(conn *imapclient.Client, mboxName string) error {
diff --git a/plugins/base/routes.go b/plugins/base/routes.go
index b61b844..977b017 100644
--- a/plugins/base/routes.go
+++ b/plugins/base/routes.go
@@ -65,12 +65,57 @@ func registerRoutes(p *alps.GoPlugin) {
 
 type MailboxRenderData struct {
 	alps.BaseRenderData
-	Mailbox            *MailboxStatus
-	Inbox              *MailboxStatus
-	Mailboxes          []MailboxInfo
-	Messages           []IMAPMessage
-	PrevPage, NextPage int
-	Query              string
+	Mailbox              *MailboxStatus
+	Inbox                *MailboxStatus
+	CategorizedMailboxes CategorizedMailboxes
+	Mailboxes            []MailboxInfo
+	Messages             []IMAPMessage
+	PrevPage, NextPage   int
+	Query                string
+}
+
+// Organizes mailboxes into common/uncommon categories
+type CategorizedMailboxes struct {
+	Common struct {
+		Inbox   *MailboxInfo
+		Drafts  *MailboxInfo
+		Sent    *MailboxInfo
+		Junk    *MailboxInfo
+		Trash   *MailboxInfo
+		Archive *MailboxInfo
+	}
+	Additional []*MailboxInfo
+}
+
+func categorizeMailboxes(mailboxes []MailboxInfo,
+	inbox *MailboxStatus, active *MailboxStatus) CategorizedMailboxes {
+
+	var out CategorizedMailboxes
+	mmap := map[string]**MailboxInfo{
+		"INBOX": &out.Common.Inbox,
+		"Drafts": &out.Common.Drafts,
+		"Sent": &out.Common.Sent,
+		"Junk": &out.Common.Junk,
+		"Trash": &out.Common.Trash,
+		"Archive": &out.Common.Archive,
+	}
+	for i, _ := range mailboxes {
+		// Populate unseen & active states
+		if mailboxes[i].Name == active.Name {
+			mailboxes[i].Unseen = int(active.Unseen)
+			mailboxes[i].Active = true
+		}
+		if mailboxes[i].Name == inbox.Name {
+			mailboxes[i].Unseen = int(inbox.Unseen)
+		}
+
+		if ptr, ok := mmap[mailboxes[i].Name]; ok {
+			*ptr = &mailboxes[i]
+		} else {
+			out.Additional = append(out.Additional, &mailboxes[i])
+		}
+	}
+	return out
 }
 
 func handleGetMailbox(ctx *alps.Context) error {
@@ -154,15 +199,18 @@ func handleGetMailbox(ctx *alps.Context) error {
 		title = fmt.Sprintf("(%d) %s", mbox.Unseen, title)
 	}
 
+	categorized := categorizeMailboxes(mailboxes, inbox, mbox)
+
 	return ctx.Render(http.StatusOK, "mailbox.html", &MailboxRenderData{
 		BaseRenderData: *alps.NewBaseRenderData(ctx).WithTitle(title),
-		Mailbox:        mbox,
-		Inbox:          inbox,
-		Mailboxes:      mailboxes,
-		Messages:       msgs,
-		PrevPage:       prevPage,
-		NextPage:       nextPage,
-		Query:          query,
+		Mailbox:              mbox,
+		Inbox:                inbox,
+		CategorizedMailboxes: categorized,
+		Mailboxes:            mailboxes,
+		Messages:             msgs,
+		PrevPage:             prevPage,
+		NextPage:             nextPage,
+		Query:                query,
 	})
 }