52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"gitea.pena/PenaSide/codeword/internal/models"
|
|
"strings"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
const (
|
|
prefix = "Bearer "
|
|
)
|
|
|
|
func NewAuthenticator(jwtUtil *JWT) func(*fiber.Ctx) error {
|
|
return func(c *fiber.Ctx) error {
|
|
if jwtUtil == nil {
|
|
return fmt.Errorf("jwt util is nil")
|
|
}
|
|
|
|
if err := authenticate(c, jwtUtil); err != nil {
|
|
return fmt.Errorf("authentication error:%d", err)
|
|
}
|
|
|
|
return c.Next()
|
|
}
|
|
}
|
|
|
|
func authenticate(c *fiber.Ctx, jwtUtil *JWT) error {
|
|
authHeader := c.Get("Authorization")
|
|
if authHeader == "" || !strings.HasPrefix(authHeader, prefix) {
|
|
return fmt.Errorf("failed to parse jws from request header: %s", authHeader)
|
|
}
|
|
|
|
jws := strings.TrimPrefix(authHeader, prefix)
|
|
|
|
userID, validateErr := jwtUtil.Validate(jws)
|
|
if validateErr != nil {
|
|
return validateErr
|
|
}
|
|
|
|
c.Locals(models.AuthJWTDecodedUserIDKey, userID)
|
|
c.Locals(models.AuthJWTDecodedAccessTokenKey, jws)
|
|
|
|
return nil
|
|
}
|
|
|
|
func ExtractUserID(c *fiber.Ctx) string {
|
|
userID := c.Context().UserValue(models.AuthJWTDecodedUserIDKey).(string)
|
|
return userID
|
|
}
|