diff options
author | 2019-12-17 11:36:08 +0100 | |
---|---|---|
committer | 2019-12-17 11:36:08 +0100 | |
commit | 3aea768cad3a5c9bb2a16039beee4b3dc8ee9d48 (patch) | |
tree | 0cf5866e9027eedf30de1e22af6ddc3d4427565c /plugins/base/handlers.go | |
parent | b8407569f0a3807ab6bee31c6b0c5f07455f7cb9 (diff) | |
download | alps-3aea768cad3a5c9bb2a16039beee4b3dc8ee9d48.tar.gz alps-3aea768cad3a5c9bb2a16039beee4b3dc8ee9d48.tar.bz2 alps-3aea768cad3a5c9bb2a16039beee4b3dc8ee9d48.zip |
plugins/base: extract route registration into function
Diffstat (limited to 'plugins/base/handlers.go')
-rw-r--r-- | plugins/base/handlers.go | 381 |
1 files changed, 0 insertions, 381 deletions
diff --git a/plugins/base/handlers.go b/plugins/base/handlers.go deleted file mode 100644 index 923ab4f..0000000 --- a/plugins/base/handlers.go +++ /dev/null @@ -1,381 +0,0 @@ -package koushinbase - -import ( - "fmt" - "io/ioutil" - "mime" - "net/http" - "net/url" - "strconv" - "strings" - - "git.sr.ht/~emersion/koushin" - "github.com/emersion/go-imap" - imapmove "github.com/emersion/go-imap-move" - imapclient "github.com/emersion/go-imap/client" - "github.com/emersion/go-message" - "github.com/emersion/go-smtp" - "github.com/labstack/echo/v4" -) - -type MailboxRenderData struct { - koushin.RenderData - Mailbox *imap.MailboxStatus - Mailboxes []*imap.MailboxInfo - Messages []IMAPMessage - PrevPage, NextPage int - Query string -} - -func handleGetMailbox(ectx echo.Context) error { - ctx := ectx.(*koushin.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") - } - } - - query := ctx.FormValue("query") - - var mailboxes []*imap.MailboxInfo - var msgs []IMAPMessage - var mbox *imap.MailboxStatus - err = ctx.Session.DoIMAP(func(c *imapclient.Client) error { - var err error - if mailboxes, err = listMailboxes(c); err != nil { - return err - } - if query != "" { - msgs, err = searchMessages(c, mboxName, query) - } else { - msgs, err = listMessages(c, mboxName, page) - } - if err != nil { - return err - } - mbox = c.Mailbox() - return nil - }) - if err != nil { - return err - } - - prevPage, nextPage := -1, -1 - if query == "" { - // TODO: paging for search - if page > 0 { - prevPage = page - 1 - } - if (page+1)*messagesPerPage < int(mbox.Messages) { - nextPage = page + 1 - } - } - - return ctx.Render(http.StatusOK, "mailbox.html", &MailboxRenderData{ - RenderData: *koushin.NewRenderData(ctx), - Mailbox: mbox, - Mailboxes: mailboxes, - Messages: msgs, - PrevPage: prevPage, - NextPage: nextPage, - Query: query, - }) -} - -func handleLogin(ectx echo.Context) error { - ctx := ectx.(*koushin.Context) - - username := ctx.FormValue("username") - password := ctx.FormValue("password") - if username != "" && password != "" { - s, err := ctx.Server.Sessions.Put(username, password) - if err != nil { - if _, ok := err.(koushin.AuthError); ok { - return ctx.Render(http.StatusOK, "login.html", nil) - } - return fmt.Errorf("failed to put connection in pool: %v", err) - } - ctx.SetSession(s) - - return ctx.Redirect(http.StatusFound, "/mailbox/INBOX") - } - - return ctx.Render(http.StatusOK, "login.html", koushin.NewRenderData(ctx)) -} - -func handleLogout(ectx echo.Context) error { - ctx := ectx.(*koushin.Context) - - ctx.Session.Close() - ctx.SetSession(nil) - return ctx.Redirect(http.StatusFound, "/login") -} - -type MessageRenderData struct { - koushin.RenderData - Mailboxes []*imap.MailboxInfo - Mailbox *imap.MailboxStatus - Message *IMAPMessage - Body string - PartPath string - MailboxPage int -} - -func handleGetPart(ctx *koushin.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 mailboxes []*imap.MailboxInfo - var msg *IMAPMessage - var part *message.Entity - var mbox *imap.MailboxStatus - err = ctx.Session.DoIMAP(func(c *imapclient.Client) error { - var err error - if mailboxes, err = listMailboxes(c); err != nil { - return err - } - if msg, part, err = getMessagePart(c, mboxName, uid, partPath); err != nil { - return err - } - mbox = c.Mailbox() - return nil - }) - 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", &MessageRenderData{ - RenderData: *koushin.NewRenderData(ctx), - Mailboxes: mailboxes, - Mailbox: mbox, - Message: msg, - Body: body, - PartPath: partPathString, - MailboxPage: int(mbox.Messages-msg.SeqNum) / messagesPerPage, - }) -} - -type ComposeRenderData struct { - koushin.RenderData - Message *OutgoingMessage -} - -func handleCompose(ectx echo.Context) error { - ctx := ectx.(*koushin.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.DoIMAP(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.Address() - } - } - 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") - - form, err := ctx.MultipartForm() - if err != nil { - return fmt.Errorf("failed to get multipart form: %v", err) - } - msg.Attachments = form.File["attachments"] - - err = ctx.Session.DoSMTP(func(c *smtp.Client) error { - return sendMessage(c, &msg) - }) - if err != nil { - if _, ok := err.(koushin.AuthError); ok { - return echo.NewHTTPError(http.StatusForbidden, err) - } - return err - } - - // TODO: append to IMAP Sent mailbox - - return ctx.Redirect(http.StatusFound, "/mailbox/INBOX") - } - - return ctx.Render(http.StatusOK, "compose.html", &ComposeRenderData{ - RenderData: *koushin.NewRenderData(ctx), - Message: &msg, - }) -} - -func handleMove(ectx echo.Context) error { - ctx := ectx.(*koushin.Context) - - mboxName, uid, err := parseMboxAndUid(ctx.Param("mbox"), ctx.Param("uid")) - if err != nil { - return echo.NewHTTPError(http.StatusBadRequest, err) - } - - to := ctx.FormValue("to") - - err = ctx.Session.DoIMAP(func(c *imapclient.Client) error { - mc := imapmove.NewClient(c) - - if err := ensureMailboxSelected(c, mboxName); err != nil { - return err - } - - var seqSet imap.SeqSet - seqSet.AddNum(uid) - if err := mc.UidMoveWithFallback(&seqSet, to); err != nil { - return fmt.Errorf("failed to move message: %v", err) - } - - // TODO: get the UID of the message in the destination mailbox with UIDPLUS - return nil - }) - if err != nil { - return err - } - - return ctx.Redirect(http.StatusFound, fmt.Sprintf("/mailbox/%v", to)) -} - -func handleDelete(ectx echo.Context) error { - ctx := ectx.(*koushin.Context) - - mboxName, uid, err := parseMboxAndUid(ctx.Param("mbox"), ctx.Param("uid")) - if err != nil { - return echo.NewHTTPError(http.StatusBadRequest, err) - } - - err = ctx.Session.DoIMAP(func(c *imapclient.Client) error { - if err := ensureMailboxSelected(c, mboxName); err != nil { - return err - } - - var seqSet imap.SeqSet - seqSet.AddNum(uid) - - item := imap.FormatFlagsOp(imap.AddFlags, true) - flags := []interface{}{imap.DeletedFlag} - if err := c.UidStore(&seqSet, item, flags, nil); err != nil { - return fmt.Errorf("failed to add deleted flag: %v", err) - } - - if err := c.Expunge(nil); err != nil { - return fmt.Errorf("failed to expunge mailbox: %v", err) - } - - // Deleting a message invalidates our cached message count - // TODO: listen to async updates instead - if _, err := c.Select(mboxName, false); err != nil { - return fmt.Errorf("failed to select mailbox: %v", err) - } - - return nil - }) - if err != nil { - return err - } - - return ctx.Redirect(http.StatusFound, fmt.Sprintf("/mailbox/%v", mboxName)) -} |