163 lines
3.1 KiB
Go
163 lines
3.1 KiB
Go
package koushin
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
imapclient "github.com/emersion/go-imap/client"
|
|
)
|
|
|
|
const cookieName = "koushin_session"
|
|
|
|
type Server struct {
|
|
imap struct {
|
|
host string
|
|
tls bool
|
|
insecure bool
|
|
|
|
pool *ConnPool
|
|
}
|
|
}
|
|
|
|
func NewServer(imapURL string) (*Server, error) {
|
|
u, err := url.Parse(imapURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
s := &Server{}
|
|
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 nil, fmt.Errorf("unrecognized IMAP URL scheme: %s", u.Scheme)
|
|
}
|
|
|
|
s.imap.pool = NewConnPool()
|
|
|
|
return s, nil
|
|
}
|
|
|
|
type context struct {
|
|
echo.Context
|
|
server *Server
|
|
conn *imapclient.Client
|
|
}
|
|
|
|
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)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ctx.setToken(token)
|
|
|
|
return ctx.Redirect(http.StatusFound, "/mailbox/INBOX")
|
|
}
|
|
|
|
return ctx.Render(http.StatusOK, "login.html", nil)
|
|
}
|
|
|
|
func New(imapURL string) *echo.Echo {
|
|
e := echo.New()
|
|
|
|
s, err := NewServer(imapURL)
|
|
if err != nil {
|
|
e.Logger.Fatal(err)
|
|
}
|
|
|
|
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 ctx.Path() == "/login" {
|
|
return next(ctx)
|
|
} else {
|
|
return ctx.Redirect(http.StatusFound, "/login")
|
|
}
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx.conn, 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.Renderer, err = loadTemplates()
|
|
if err != nil {
|
|
e.Logger.Fatal("Failed to load templates:", err)
|
|
}
|
|
|
|
e.GET("/mailbox/:mbox", func(ectx echo.Context) error {
|
|
ctx := ectx.(*context)
|
|
|
|
mailboxes, err := listMailboxes(ctx.conn)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
msgs, err := listMessages(ctx.conn, ctx.Param("mbox"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return ctx.Render(http.StatusOK, "mailbox.html", map[string]interface{}{
|
|
"Mailboxes": mailboxes,
|
|
"Messages": msgs,
|
|
})
|
|
})
|
|
|
|
e.GET("/login", handleLogin)
|
|
e.POST("/login", handleLogin)
|
|
|
|
e.Static("/assets", "public/assets")
|
|
|
|
return e
|
|
}
|