44 lines
914 B
Go
44 lines
914 B
Go
package server
|
|
|
|
import (
|
|
"strings"
|
|
"github.com/gofiber/fiber/v2"
|
|
"penahub.gitlab.yandexcloud.net/backend/penahub_common/jwt_adapter"
|
|
)
|
|
|
|
func LocalJwt() fiber.Handler {
|
|
return func(c *fiber.Ctx) error {
|
|
if c.Get("LocalUserID") != "" {
|
|
c.Locals("userID", c.Get("LocalUserID"))
|
|
}
|
|
|
|
return c.Next()
|
|
}
|
|
}
|
|
|
|
func Jwt() fiber.Handler {
|
|
return func(c *fiber.Ctx) error {
|
|
// check LocalJwt
|
|
if c.Locals("userID") != nil {
|
|
return c.Next()
|
|
}
|
|
|
|
jwt, err := jwt_adapter.Decode(strings.ReplaceAll(c.Get(jwt_adapter.DefaultHeaderKey), "Bearer ", ""))
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusUnauthorized, err.Error())
|
|
}
|
|
|
|
err = jwt.Validate()
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusUnauthorized, err.Error())
|
|
}
|
|
|
|
if jwt.GetUserID() == "" {
|
|
return fiber.NewError(fiber.StatusUnauthorized, "empty user")
|
|
}
|
|
|
|
c.Locals("userID", jwt.GetUserID())
|
|
return c.Next()
|
|
}
|
|
}
|