From 640bd497317891eae8c34255c077956183d526ae Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Mon, 2 Dec 2019 17:31:34 +0100
Subject: [PATCH] List mailboxes

---
 public/index.html |  7 +++++++
 server.go         | 28 ++++++++++++++++++++++++----
 2 files changed, 31 insertions(+), 4 deletions(-)

diff --git a/public/index.html b/public/index.html
index 9dbc3fe..4a9566a 100644
--- a/public/index.html
+++ b/public/index.html
@@ -2,4 +2,11 @@
 
 <h1>koushin</h1>
 
+<p>Mailboxes:</p>
+<ul>
+	{{range .Mailboxes}}
+		<li>{{.Name}}</li>
+	{{end}}
+</ul>
+
 {{template "foot"}}
diff --git a/server.go b/server.go
index cd03c5d..14b9892 100644
--- a/server.go
+++ b/server.go
@@ -7,6 +7,7 @@ import (
 	"time"
 
 	"github.com/labstack/echo/v4"
+	"github.com/emersion/go-imap"
 	imapclient "github.com/emersion/go-imap/client"
 )
 
@@ -132,7 +133,12 @@ func New(imapURL string) *echo.Echo {
 
 			cookie, err := ctx.Cookie(cookieName)
 			if err == http.ErrNoCookie {
-				return next(ctx)
+				// Require auth for all pages except /login
+				if ctx.Path() == "/login" {
+					return next(ctx)
+				} else {
+					return ctx.Redirect(http.StatusFound, "/login")
+				}
 			} else if err != nil {
 				return err
 			}
@@ -156,11 +162,25 @@ func New(imapURL string) *echo.Echo {
 
 	e.GET("/", func(ectx echo.Context) error {
 		ctx := ectx.(*context)
-		if ctx.conn == nil {
-			return ctx.Redirect(http.StatusFound, "/login")
+
+		ch := make(chan *imap.MailboxInfo, 10)
+		done := make(chan error, 1)
+		go func () {
+			done <- ctx.conn.List("", "*", ch)
+		}()
+
+		var mailboxes []*imap.MailboxInfo
+		for mbox := range ch {
+			mailboxes = append(mailboxes, mbox)
 		}
 
-		return ctx.Render(http.StatusOK, "index.html", nil)
+		if err := <-done; err != nil {
+			return err
+		}
+
+		return ctx.Render(http.StatusOK, "index.html", map[string]interface{}{
+			"Mailboxes": mailboxes,
+		})
 	})
 
 	e.GET("/login", handleLogin)