Save username/password in session

This is required for authenticating with the SMTP server when composing
a new message.

1 files changed, 6 insertions, 2 deletions

diff --git a/server.go b/server.go
index 6e800d0..be174b8 100644
--- a/server.go
+++ b/server.go
@@ -93,6 +93,7 @@ func NewServer(imapURL, smtpURL string) (*Server, error) {
 type context struct {
 	echo.Context
 	server *Server
+	session *Session
 	conn   *imapclient.Client
 }
 
@@ -126,7 +127,7 @@ func handleLogin(ectx echo.Context) error {
 			return ctx.Render(http.StatusOK, "login.html", nil)
 		}
 
-		token, err := ctx.server.imap.pool.Put(conn)
+		token, err := ctx.server.imap.pool.Put(conn, username, password)
 		if err != nil {
 			return fmt.Errorf("failed to put connection in pool: %v", err)
 		}
@@ -167,6 +168,8 @@ func handleGetPart(ctx *context, raw bool) error {
 		disp, dispParams, _ := part.Header.ContentDisposition()
 		filename := dispParams["filename"]
 
+		// TODO: set Content-Length if possible
+
 		if !strings.EqualFold(mimeType, "text/plain") || strings.EqualFold(disp, "attachment") {
 			dispParams := make(map[string]string)
 			if filename != "" {
@@ -235,13 +238,14 @@ func New(imapURL, smtpURL string) *echo.Echo {
 				return err
 			}
 
-			ctx.conn, err = ctx.server.imap.pool.Get(cookie.Value)
+			ctx.session, err = ctx.server.imap.pool.Get(cookie.Value)
 			if err == ErrSessionExpired {
 				ctx.setToken("")
 				return ctx.Redirect(http.StatusFound, "/login")
 			} else if err != nil {
 				return err
 			}
+			ctx.conn = ctx.session.imapConn
 
 			return next(ctx)
 		}