47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/TicketsBot/GoPanel/config"
|
|
"github.com/TicketsBot/GoPanel/utils"
|
|
"github.com/dgrijalva/jwt-go"
|
|
"github.com/gin-gonic/gin"
|
|
"strconv"
|
|
)
|
|
|
|
func AuthenticateToken(ctx *gin.Context) {
|
|
header := ctx.GetHeader("Authorization")
|
|
|
|
token, err := jwt.Parse(header, func(token *jwt.Token) (interface{}, error) {
|
|
if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
|
|
return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"])
|
|
}
|
|
|
|
return []byte(config.Conf.Server.Secret), nil
|
|
})
|
|
|
|
if err != nil {
|
|
ctx.AbortWithStatusJSON(401, utils.ErrorJson(err))
|
|
return
|
|
}
|
|
|
|
if claims, ok := token.Claims.(jwt.MapClaims); ok && token.Valid {
|
|
userId, hasUserId := claims["userid"]
|
|
if !hasUserId {
|
|
ctx.AbortWithStatusJSON(401, utils.ErrorStr("Token is invalid"))
|
|
return
|
|
}
|
|
|
|
parsedId, err := strconv.ParseUint(userId.(string), 10, 64)
|
|
if err != nil {
|
|
ctx.AbortWithStatusJSON(401, utils.ErrorStr("Token is invalid"))
|
|
return
|
|
}
|
|
|
|
ctx.Keys["userid"] = parsedId
|
|
} else {
|
|
ctx.AbortWithStatusJSON(401, utils.ErrorStr("Token is invalid"))
|
|
return
|
|
}
|
|
}
|