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" ) const cookieName = "koushin_session" const messagesPerPage = 50 type Server struct { imap struct { host string tls bool insecure bool pool *ConnPool } smtp struct { host string tls bool insecure bool } } func (s *Server) parseIMAPURL(imapURL string) error { u, err := url.Parse(imapURL) if err != nil { return fmt.Errorf("failed to parse IMAP server URL: %v", err) } s.imap.host = u.Host switch u.Scheme { case "imap": // This space is intentionally left blank case "imaps": s.imap.tls = true case "imap+insecure": s.imap.insecure = true default: return fmt.Errorf("unrecognized IMAP URL scheme: %s", u.Scheme) } return nil } func (s *Server) parseSMTPURL(smtpURL string) error { u, err := url.Parse(smtpURL) if err != nil { return fmt.Errorf("failed to parse SMTP server URL: %v", err) } s.smtp.host = u.Host switch u.Scheme { case "smtp": // This space is intentionally left blank case "smtps": s.smtp.tls = true case "smtp+insecure": s.smtp.insecure = true default: return fmt.Errorf("unrecognized SMTP URL scheme: %s", u.Scheme) } return nil } func newServer(imapURL, smtpURL string) (*Server, error) { s := &Server{} if err := s.parseIMAPURL(imapURL); err != nil { return nil, err } s.imap.pool = NewConnPool() if smtpURL != "" { if err := s.parseSMTPURL(smtpURL); err != nil { return nil, err } } return s, nil } type context struct { echo.Context server *Server session *Session } var aLongTimeAgo = time.Unix(233431200, 0) func (c *context) setToken(token string) { cookie := http.Cookie{ Name: cookieName, Value: token, HttpOnly: true, // TODO: domain, secure } if token == "" { cookie.Expires = aLongTimeAgo // unset the cookie } 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/") } type Options struct { IMAPURL, SMTPURL string Theme string } func New(e *echo.Echo, options *Options) error { s, err := newServer(options.IMAPURL, options.SMTPURL) if err != nil { return err } e.Renderer, err = loadTemplates(e.Logger, options.Theme) if err != nil { return fmt.Errorf("failed to load templates: %v", err) } e.HTTPErrorHandler = func(err error, c echo.Context) { code := http.StatusInternalServerError if he, ok := err.(*echo.HTTPError); ok { code = he.Code } else { c.Logger().Error(err) } // TODO: hide internal errors c.String(code, err.Error()) } e.Use(func(next echo.HandlerFunc) echo.HandlerFunc { return func(ectx echo.Context) error { ctx := &context{Context: ectx, server: s} cookie, err := ctx.Cookie(cookieName) if err == http.ErrNoCookie { // Require auth for all pages except /login if isPublic(ctx.Path()) { return next(ctx) } else { return ctx.Redirect(http.StatusFound, "/login") } } else if err != nil { return err } 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 } return next(ctx) } }) 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("/message/:mbox/:uid", func(ectx echo.Context) error { ctx := ectx.(*context) return handleGetPart(ctx, false) }) e.GET("/message/:mbox/:uid/raw", func(ectx echo.Context) error { ctx := ectx.(*context) return handleGetPart(ctx, true) }) 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("/compose", handleCompose) e.POST("/compose", handleCompose) e.GET("/message/:mbox/:uid/reply", handleCompose) e.POST("/message/:mbox/:uid/reply", handleCompose) e.Static("/assets", "public/assets") e.Static("/themes", "public/themes") return nil }