delphi.market/server/router/middleware/session.go

package middleware

import (
	"database/sql"
	"net/http"
	"time"

	"git.ekzyis.com/ekzyis/delphi.market/db"
	"git.ekzyis.com/ekzyis/delphi.market/server/router/context"
	"github.com/labstack/echo/v4"
)

func Session(sc context.ServerContext) echo.MiddlewareFunc {
	return func(next echo.HandlerFunc) echo.HandlerFunc {
		return func(c echo.Context) error {
			var (
				cookie *http.Cookie
				err    error
				s      *db.Session
				u      *db.User
			)
			if cookie, err = c.Cookie("session"); err != nil {
				// cookie not found
				return next(c)
			}
			s = &db.Session{SessionId: cookie.Value}
			if err = sc.Db.FetchSession(s); err == nil {
				// session found
				u = &db.User{Pubkey: s.Pubkey, Msats: s.Msats, LastSeen: time.Now()}
				if err = sc.Db.UpdateUser(u); err != nil {
					return err
				}
				c.Set("session", *u)
			} else if err != sql.ErrNoRows {
				return err
			}
			return next(c)
		}
	}
}

func SessionGuard(sc context.ServerContext) echo.MiddlewareFunc {
	return func(next echo.HandlerFunc) echo.HandlerFunc {
		return func(c echo.Context) error {
			session := c.Get("session")
			if session == nil {
				return c.Redirect(http.StatusTemporaryRedirect, "/login")
			}
			return next(c)
		}
	}
}