ultone/internal/middleware/auth/auth.go

package auth

import (
	"errors"
	"github.com/loveuer/nf"
	"github.com/loveuer/nf/nft/resp"
	"strings"
	"ultone/internal/controller"
	"ultone/internal/database/cache"
	"ultone/internal/log"
	"ultone/internal/opt"
)

var (
	tokenFunc = func(c *nf.Ctx) string {
		token := c.Get("Authorization")
		if token == "" {
			token = c.Cookies(opt.CookieName)
		}

		return token
	}
)

func NewAuth() nf.HandlerFunc {
	return func(c *nf.Ctx) error {
		token := tokenFunc(c)

		if token = strings.TrimPrefix(token, "Bearer "); token == "" {
			return resp.Resp401(c, token)
		}

		log.Debug(c.Context(), "middleware.NewAuth: token=%s", token)

		target, err := controller.UserController.GetUserByToken(c.Context(), token)
		if err != nil {
			log.Error(c.Context(), "middleware.NewAuth: get user by token=%s err=%v", token, err)
			if errors.Is(err, cache.ErrorKeyNotFound) {
				return resp.Resp401(c, err)
			}

			return resp.RespError(c, err)
		}

		if err = target.IsValid(true); err != nil {
			return resp.Resp401(c, err.Error(), err.Error())
		}

		c.Locals("user", target)
		c.Locals("token", token)

		return c.Next()
	}
}