diff options
author | Simon Ser <contact@emersion.fr> | 2019-12-04 20:55:08 +0100 |
---|---|---|
committer | Simon Ser <contact@emersion.fr> | 2019-12-04 20:55:08 +0100 |
commit | d1d43809577f85d2feec4bc61601ca9ade378b5b (patch) | |
tree | e7e20b142b4472f2316cc400ef2f5cf94c83e0be /server.go | |
parent | 7c0b90609298735e4f6837f3d8bd1b4fdc464fcd (diff) | |
download | alps-d1d43809577f85d2feec4bc61601ca9ade378b5b.tar.gz alps-d1d43809577f85d2feec4bc61601ca9ade378b5b.zip |
Extract HTTP handlers in separate file
Diffstat (limited to 'server.go')
-rw-r--r-- | server.go | 265 |
1 files changed, 2 insertions, 263 deletions
@@ -2,18 +2,11 @@ package koushin import ( "fmt" - "io/ioutil" - "mime" "net/http" "net/url" - "strconv" "strings" "time" - "github.com/emersion/go-imap" - imapclient "github.com/emersion/go-imap/client" - "github.com/emersion/go-message" - "github.com/emersion/go-sasl" "github.com/labstack/echo/v4" ) @@ -117,199 +110,6 @@ func (c *context) setToken(token string) { c.SetCookie(&cookie) } -func handleLogin(ectx echo.Context) error { - ctx := ectx.(*context) - username := ctx.FormValue("username") - password := ctx.FormValue("password") - if username != "" && password != "" { - conn, err := ctx.server.connectIMAP() - if err != nil { - return err - } - - if err := conn.Login(username, password); err != nil { - conn.Logout() - return ctx.Render(http.StatusOK, "login.html", nil) - } - - token, err := ctx.server.imap.pool.Put(conn, username, password) - if err != nil { - return fmt.Errorf("failed to put connection in pool: %v", err) - } - ctx.setToken(token) - - return ctx.Redirect(http.StatusFound, "/mailbox/INBOX") - } - - return ctx.Render(http.StatusOK, "login.html", nil) -} - -func handleGetPart(ctx *context, raw bool) error { - mboxName, uid, err := parseMboxAndUid(ctx.Param("mbox"), ctx.Param("uid")) - if err != nil { - return echo.NewHTTPError(http.StatusBadRequest, err) - } - partPathString := ctx.QueryParam("part") - partPath, err := parsePartPath(partPathString) - if err != nil { - return echo.NewHTTPError(http.StatusBadRequest, err) - } - - var msg *imapMessage - var part *message.Entity - var mbox *imap.MailboxStatus - err = ctx.session.Do(func(c *imapclient.Client) error { - var err error - msg, part, err = getMessagePart(c, mboxName, uid, partPath) - mbox = c.Mailbox() - return err - }) - if err != nil { - return err - } - - mimeType, _, err := part.Header.ContentType() - if err != nil { - return fmt.Errorf("failed to parse part Content-Type: %v", err) - } - if len(partPath) == 0 { - mimeType = "message/rfc822" - } - - if raw { - 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 != "" { - dispParams["filename"] = filename - } - disp := mime.FormatMediaType("attachment", dispParams) - ctx.Response().Header().Set("Content-Disposition", disp) - } - return ctx.Stream(http.StatusOK, mimeType, part.Body) - } - - var body string - if strings.HasPrefix(strings.ToLower(mimeType), "text/") { - b, err := ioutil.ReadAll(part.Body) - if err != nil { - return fmt.Errorf("failed to read part body: %v", err) - } - body = string(b) - } - - return ctx.Render(http.StatusOK, "message.html", map[string]interface{}{ - "Mailbox": mbox, - "Message": msg, - "Body": body, - "PartPath": partPathString, - "MailboxPage": (mbox.Messages - msg.SeqNum) / messagesPerPage, - }) -} - -func handleCompose(ectx echo.Context) error { - ctx := ectx.(*context) - - var msg OutgoingMessage - if strings.ContainsRune(ctx.session.username, '@') { - msg.From = ctx.session.username - } - - if ctx.Request().Method == http.MethodGet && ctx.Param("uid") != "" { - // This is a reply - mboxName, uid, err := parseMboxAndUid(ctx.Param("mbox"), ctx.Param("uid")) - if err != nil { - return echo.NewHTTPError(http.StatusBadRequest, err) - } - partPath, err := parsePartPath(ctx.QueryParam("part")) - if err != nil { - return echo.NewHTTPError(http.StatusBadRequest, err) - } - - var inReplyTo *imapMessage - var part *message.Entity - err = ctx.session.Do(func(c *imapclient.Client) error { - var err error - inReplyTo, part, err = getMessagePart(c, mboxName, uid, partPath) - return err - }) - if err != nil { - return err - } - - mimeType, _, err := part.Header.ContentType() - if err != nil { - return fmt.Errorf("failed to parse part Content-Type: %v", err) - } - - if !strings.HasPrefix(strings.ToLower(mimeType), "text/") { - err := fmt.Errorf("cannot reply to \"%v\" part", mimeType) - return echo.NewHTTPError(http.StatusBadRequest, err) - } - - msg.Text, err = quote(part.Body) - if err != nil { - return err - } - - msg.InReplyTo = inReplyTo.Envelope.MessageId - // TODO: populate From from known user addresses and inReplyTo.Envelope.To - replyTo := inReplyTo.Envelope.ReplyTo - if len(replyTo) == 0 { - replyTo = inReplyTo.Envelope.From - } - if len(replyTo) > 0 { - msg.To = make([]string, len(replyTo)) - for i, to := range replyTo { - msg.To[i] = to.MailboxName + "@" + to.HostName - } - } - msg.Subject = inReplyTo.Envelope.Subject - if !strings.HasPrefix(strings.ToLower(msg.Subject), "re:") { - msg.Subject = "Re: " + msg.Subject - } - } - - if ctx.Request().Method == http.MethodPost { - msg.From = ctx.FormValue("from") - msg.To = parseAddressList(ctx.FormValue("to")) - msg.Subject = ctx.FormValue("subject") - msg.Text = ctx.FormValue("text") - msg.InReplyTo = ctx.FormValue("in_reply_to") - - c, err := ctx.server.connectSMTP() - if err != nil { - return err - } - defer c.Close() - - auth := sasl.NewPlainClient("", ctx.session.username, ctx.session.password) - if err := c.Auth(auth); err != nil { - return echo.NewHTTPError(http.StatusForbidden, err) - } - - if err := sendMessage(c, &msg); err != nil { - return err - } - - if err := c.Quit(); err != nil { - return fmt.Errorf("QUIT failed: %v", err) - } - - // TODO: append to IMAP Sent mailbox - - return ctx.Redirect(http.StatusFound, "/mailbox/INBOX") - } - - return ctx.Render(http.StatusOK, "compose.html", map[string]interface{}{ - "Message": &msg, - }) -} - func isPublic(path string) bool { return path == "/login" || strings.HasPrefix(path, "/assets/") || strings.HasPrefix(path, "/themes/") @@ -370,56 +170,7 @@ func New(e *echo.Echo, options *Options) error { } }) - e.GET("/mailbox/:mbox", func(ectx echo.Context) error { - ctx := ectx.(*context) - - mboxName, err := url.PathUnescape(ctx.Param("mbox")) - if err != nil { - return echo.NewHTTPError(http.StatusBadRequest, err) - } - - page := 0 - if pageStr := ctx.QueryParam("page"); pageStr != "" { - var err error - if page, err = strconv.Atoi(pageStr); err != nil || page < 0 { - return echo.NewHTTPError(http.StatusBadRequest, "invalid page index") - } - } - - var mailboxes []*imap.MailboxInfo - var msgs []imapMessage - var mbox *imap.MailboxStatus - err = ctx.session.Do(func(c *imapclient.Client) error { - var err error - if mailboxes, err = listMailboxes(c); err != nil { - return err - } - if msgs, err = listMessages(c, mboxName, page); err != nil { - return err - } - mbox = c.Mailbox() - return nil - }) - if err != nil { - return err - } - - prevPage, nextPage := -1, -1 - if page > 0 { - prevPage = page - 1 - } - if (page+1)*messagesPerPage < int(mbox.Messages) { - nextPage = page + 1 - } - - return ctx.Render(http.StatusOK, "mailbox.html", map[string]interface{}{ - "Mailbox": mbox, - "Mailboxes": mailboxes, - "Messages": msgs, - "PrevPage": prevPage, - "NextPage": nextPage, - }) - }) + e.GET("/mailbox/:mbox", handleGetMailbox) e.GET("/message/:mbox/:uid", func(ectx echo.Context) error { ctx := ectx.(*context) @@ -433,19 +184,7 @@ func New(e *echo.Echo, options *Options) error { e.GET("/login", handleLogin) e.POST("/login", handleLogin) - e.GET("/logout", func(ectx echo.Context) error { - ctx := ectx.(*context) - - err := ctx.session.Do(func(c *imapclient.Client) error { - return c.Logout() - }) - if err != nil { - return fmt.Errorf("failed to logout: %v", err) - } - - ctx.setToken("") - return ctx.Redirect(http.StatusFound, "/login") - }) + e.GET("/logout", handleLogout) e.GET("/compose", handleCompose) e.POST("/compose", handleCompose) |