fix: auth login ways

internal/utils/token.go

@@ -8,16 +8,18 @@ import (
 )
 
 type Claims struct {
-	UserID   string `json:"user_id"`
-	Username string `json:"username"`
+	UserID       string `json:"user_id"`
+	Username     string `json:"username"`
+	TokenVersion int    `json:"version"`
 	jwt.RegisteredClaims
 }
 
 // GenerateToken 生成 JWT token
-func GenerateToken(userID string, username string, expireDays int, secret string) (string, error) {
+func GenerateToken(userID string, username string, version int, expireDays int, secret string) (string, error) {
 	claims := Claims{
-		UserID:   userID,
-		Username: username,
+		UserID:       userID,
+		Username:     username,
+		TokenVersion: version,
 		RegisteredClaims: jwt.RegisteredClaims{
 			ExpiresAt: jwt.NewNumericDate(time.Now().Add(time.Duration(expireDays) * 24 * time.Hour)),
 			IssuedAt:  jwt.NewNumericDate(time.Now()),
@@ -29,18 +31,22 @@ func GenerateToken(userID string, username string, expireDays int, secret string
 }
 
 // ParseToken 解析 JWT token
-func ParseToken(tokenString string, secret string) (string, string, error) {
+func ParseToken(tokenString string, secret string) (string, string, int, error) {
 	token, err := jwt.ParseWithClaims(tokenString, &Claims{}, func(token *jwt.Token) (any, error) {
+		// 校验算法
+		if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
+			return nil, errors.New("unexpected signing method")
+		}
 		return []byte(secret), nil
 	})
 
 	if err != nil {
-		return "", "", err
+		return "", "", 0, err
 	}
 
 	if claims, ok := token.Claims.(*Claims); ok && token.Valid {
-		return claims.UserID, claims.Username, nil
+		return claims.UserID, claims.Username, claims.TokenVersion, nil
 	}
 
-	return "", "", errors.New("invalid token")
+	return "", "", 0, errors.New("invalid token")
 }