ycc-proxy-server/common/ctxdata/ctxData.go

package ctxdata

import (
	"context"
	"ycc-server/app/main/model"
	jwtx "ycc-server/common/jwt"
	"errors"
	"fmt"
)

const CtxKeyJwtUserId = "userId"

// 定义错误类型
var (
	ErrNoInCtx       = errors.New("上下文中没有相关数据")
	ErrInvalidUserId = errors.New("用户ID格式无效") // 数据异常
)

// GetUidFromCtx 从 context 中获取用户 ID（字符串）
func GetUidFromCtx(ctx context.Context) (string, error) {
	// 尝试从上下文中获取 jwtUserId
	value := ctx.Value(CtxKeyJwtUserId)
    if value == nil {
        claims, err := GetClaimsFromCtx(ctx)
        if err != nil {
            return "", err
        }
        return claims.UserId, nil
    }

	// 根据值的类型进行不同处理
    switch v := value.(type) {
    case string:
        return v, nil
    default:
        return "", fmt.Errorf("%w: 期望类型 string, 实际类型 %T", ErrInvalidUserId, value)
    }
}

func GetClaimsFromCtx(ctx context.Context) (*jwtx.JwtClaims, error) {
	value := ctx.Value(jwtx.ExtraKey)
	if value == nil {
		return nil, ErrNoInCtx
	}

	// 首先尝试直接断言为 *jwtx.JwtClaims
	if claims, ok := value.(*jwtx.JwtClaims); ok {
		return claims, nil
	}

	// 如果直接断言失败，尝试从 map[string]interface{} 中解析
	if claimsMap, ok := value.(map[string]interface{}); ok {
		return jwtx.MapToJwtClaims(claimsMap)
	}

	return nil, ErrNoInCtx
}

// IsNoUserIdError 判断是否是未登录错误
func IsNoUserIdError(err error) bool {
	return errors.Is(err, ErrNoInCtx)
}

// IsInvalidUserIdError 判断是否是用户ID格式错误
func IsInvalidUserIdError(err error) bool {
	return errors.Is(err, ErrInvalidUserId)
}

// GetPlatformFromCtx 从 context 中获取平台
func GetPlatformFromCtx(ctx context.Context) (string, error) {
	platform, platformOk := ctx.Value("platform").(string)
	if !platformOk {
		return "", fmt.Errorf("平台不存在: %s", platform)
	}

	switch platform {
	case model.PlatformWxMini:
		return model.PlatformWxMini, nil
	case model.PlatformWxH5:
		return model.PlatformWxH5, nil
	case model.PlatformApp:
		return model.PlatformApp, nil
	case model.PlatformH5:
		return model.PlatformH5, nil
	default:
		return "", fmt.Errorf("不支持的支付平台: %s", platform)
	}
}