From 702719c072dea73ed5ec046fa640b3151ebcbc55 Mon Sep 17 00:00:00 2001 From: Simon Ser Date: Tue, 3 Dec 2019 15:44:20 +0100 Subject: Save username/password in session This is required for authenticating with the SMTP server when composing a new message. --- server.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'server.go') 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) } -- cgit v1.2.3-59-g8ed1b