cls/pkg/jwtfx/claims.go

package jwtfx

import (
	"fmt"
	"github.com/gin-gonic/gin"
)

const (
	JWTClaimsKey = "JWT_CLAIMS"
)

// Claims 定义了 JWT token 中的标准字段
type Claims struct {
	Phone   string `json:"phone,omitempty"`
	GuestID string `json:"guest_id,omitempty"`
}

// GetUserIdentifier 从上下文中获取用户标识（手机号或游客ID）
func GetUserIdentifier(c *gin.Context) string {
	value, exists := c.Get(JWTClaimsKey)
	if !exists {
		fmt.Println("未找到claims")
		return ""
	}

	claims, ok := value.(map[string]interface{})
	if !ok {
		fmt.Printf("claims类型转换失败: %T\n", value)
		return ""
	}

	fmt.Printf("从上下文中提取的claims: %+v\n", claims)

	if claims["username"] != nil {
		phone := claims["username"].(string)
		fmt.Printf("找到用户手机号: %s\n", phone)
		return phone
	}

	if claims["guest_id"] != nil {
		guestId := claims["guest_id"].(string)
		fmt.Printf("找到游客ID: %s\n", guestId)
		return guestId
	}

	fmt.Println("未找到用户标识")
	return ""
}

// IsGuest 判断当前用户是否为游客
func IsGuest(c *gin.Context) bool {
	value, exists := c.Get(JWTClaimsKey)
	if !exists {
		return true
	}

	claims, ok := value.(map[string]interface{})
	if !ok {
		return true
	}

	isGuest := claims["guest_id"] != nil
	fmt.Printf("判断是否为游客: %v\n", isGuest)
	return isGuest
}

// GetClaims 获取完整的 Claims 信息
func GetClaims(c *gin.Context) *Claims {
	value, exists := c.Get(JWTClaimsKey)
	if !exists {
		fmt.Println("未找到claims")
		return nil
	}

	claims, ok := value.(map[string]interface{})
	if !ok {
		fmt.Printf("claims类型转换失败: %T\n", value)
		return nil
	}

	fmt.Printf("获取完整的claims: %+v\n", claims)
	return &Claims{
		Phone:   getStringFromClaims(claims, "phone"),
		GuestID: getStringFromClaims(claims, "guest_id"),
	}
}

// getStringFromClaims 安全地从 claims 中获取字符串值
func getStringFromClaims(claims map[string]interface{}, key string) string {
	if claims[key] != nil {
		if str, ok := claims[key].(string); ok {
			fmt.Printf("从claims中获取到 %s: %s\n", key, str)
			return str
		}
		fmt.Printf("claims中的 %s 不是字符串类型\n", key)
	}
	fmt.Printf("claims中不存在 %s\n", key)
	return ""
}