44 lines
1.3 KiB
Go
44 lines
1.3 KiB
Go
package auth
|
||
|
||
import (
|
||
"net/http"
|
||
"strings"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// Middleware пропускает запрос только с валидным Bearer-токеном и кладёт
|
||
// в контекст user_id и role; без токена или с ошибкой — 401.
|
||
func Middleware(secret string) gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
header := c.GetHeader("Authorization")
|
||
if !strings.HasPrefix(header, "Bearer ") {
|
||
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "no token"})
|
||
return
|
||
}
|
||
claims, err := ParseToken(secret, strings.TrimPrefix(header, "Bearer "))
|
||
if err != nil {
|
||
c.AbortWithStatusJSON(http.StatusUnauthorized, gin.H{"error": "invalid token"})
|
||
return
|
||
}
|
||
c.Set("user_id", claims.UserID)
|
||
c.Set("role", claims.Role)
|
||
c.Next()
|
||
}
|
||
}
|
||
|
||
// RequireRole пропускает запрос, если роль в списке разрешённых.
|
||
// Ставится после Middleware: роль берётся из уже проверенного токена.
|
||
func RequireRole(roles ...string) gin.HandlerFunc {
|
||
return func(c *gin.Context) {
|
||
role := c.GetString("role")
|
||
for _, r := range roles {
|
||
if r == role {
|
||
c.Next()
|
||
return
|
||
}
|
||
}
|
||
c.AbortWithStatusJSON(http.StatusForbidden, gin.H{"error": "forbidden"})
|
||
}
|
||
}
|