aboutsummaryrefslogtreecommitdiff
path: root/plugins/base
diff options
context:
space:
mode:
Diffstat (limited to 'plugins/base')
-rw-r--r--plugins/base/imap.go16
-rw-r--r--plugins/base/routes.go74
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,
})
}