pcmt/handlers/middleware.go

// Copyright 2023 wanderer <a_mirre at utb dot cz>
// SPDX-License-Identifier: AGPL-3.0-only

package handlers

import (
	"context"
	"net/http"
	"strings"

	moduser "git.dotya.ml/mirre-mt/pcmt/modules/user"
	"github.com/CAFxX/httpcompression"
	"github.com/CAFxX/httpcompression/contrib/andybalholm/brotli"
	"github.com/CAFxX/httpcompression/contrib/compress/gzip"
	"github.com/labstack/echo-contrib/session"
	"github.com/labstack/echo/v4"
)

func MiddlewareSession(next echo.HandlerFunc) echo.HandlerFunc {
	return func(c echo.Context) error {
		sess, _ := session.Get(setting.SessionCookieName(), c)

		var username string

		uname, ok := sess.Values["username"].(string)

		if ok {
			username = uname

			log.Info("gorilla session", "username", username)

			refreshSession(
				sess,
				"/",
				setting.SessionMaxAge(),
				true,
				c.Request().URL.Scheme == "https", //nolint:goconst
				http.SameSiteStrictMode,
			)

			if err := sess.Save(c.Request(), c.Response()); err != nil {
				c.Logger().Error("failed to save session")

				return renderErrorPage(
					c,
					http.StatusInternalServerError,
					http.StatusText(http.StatusInternalServerError)+" (make sure you've got cookies enabled)",
					err.Error(),
				)
			}

			c.Set("sess", sess)

			var u moduser.User

			ctx := context.WithValue(context.Background(), moduser.CtxKey{}, slogger)
			if usr, err := moduser.QueryUser(ctx, dbclient, username); err == nil && usr != nil {
				u.ID = usr.ID
				u.Username = usr.Username
				u.IsAdmin = usr.IsAdmin
				u.CreatedAt = usr.CreatedAt
				u.IsActive = usr.IsActive
				u.IsLoggedIn = true
			} else {
				c.Logger().Error(http.StatusText(http.StatusInternalServerError) + " - " + err.Error())

				return renderErrorPage(
					c,
					http.StatusInternalServerError,
					http.StatusText(http.StatusInternalServerError),
					err.Error(),
				)
			}

			c.Set("sloggerCtx", ctx)
			c.Set("sessUsr", u)

			return next(c)
		}

		if !sess.IsNew {
			c.Logger().Debugf("%d - %s", http.StatusUnauthorized, "you need to log in")
		}

		return renderErrorPage(
			c,
			http.StatusUnauthorized,
			http.StatusText(http.StatusUnauthorized),
			ErrNoSession.Error(),
		)
	}
}

var cacheExtensions = [2]string{".png", ".svg"}

func MiddlewareCache(next echo.HandlerFunc) echo.HandlerFunc {
	return func(c echo.Context) error {
		cache := false

		for _, v := range cacheExtensions {
			if strings.HasSuffix(c.Request().URL.Path, v) {
				cache = true
				break
			}
		}

		if cache {
			c.Response().Header().Set(echo.HeaderCacheControl, "300")
		}

		return next(c)
	}
}

func WrapMiddlewareCompress() (echo.MiddlewareFunc, error) {
	brEnc, err := brotli.New(brotli.Options{})
	if err != nil {
		return nil, err
	}

	gzEnc, err := gzip.New(gzip.Options{})
	if err != nil {
		return nil, err
	}

	blocklist := true
	a, _ := httpcompression.Adapter(
		httpcompression.Compressor(brotli.Encoding, 1, brEnc),
		httpcompression.Compressor(gzip.Encoding, 0, gzEnc),
		httpcompression.Prefer(httpcompression.PreferServer),
		httpcompression.MinSize(100),
		httpcompression.ContentTypes([]string{
			"image/jpeg",
			"image/gif",
			"image/png",
		}, blocklist),
	)

	return echo.WrapMiddleware(a), nil
}