ycc-proxy-server/pkg/captcha/aliyun.go

package captcha

import (
	"os"

	captcha20230305 "github.com/alibabacloud-go/captcha-20230305/client"
	openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
	"github.com/alibabacloud-go/tea/tea"
	"github.com/pkg/errors"
	"github.com/zeromicro/go-zero/core/logx"

	"ycc-server/common/xerr"
)

// Config 验证码配置
type Config struct {
	AccessKeyID     string
	AccessKeySecret string
	EndpointURL     string
	SceneID         string
}

// Verify 验证阿里云滑块验证码
func Verify(cfg Config, captchaVerifyParam string) error {
	// 开发环境跳过验证
	if os.Getenv("ENV") == "development" {
		logx.Info("[Captcha] 开发环境，跳过验证码校验")
		return nil
	}

	// 检查参数
	if captchaVerifyParam == "" {
		logx.Info("[Captcha] 空参数，跳过验证码校验，兼容未做防盗刷验证码的场景")
		return nil
	}

	// 创建客户端配置
	clientCfg := &openapi.Config{
		AccessKeyId:     tea.String(cfg.AccessKeyID),
		AccessKeySecret: tea.String(cfg.AccessKeySecret),
	}
	clientCfg.Endpoint = tea.String(cfg.EndpointURL)

	// 创建客户端
	client, err := captcha20230305.NewClient(clientCfg)
	if err != nil {
		logx.Errorf("[Captcha] 创建阿里云验证码客户端失败: %+v", err)
		// 客户端创建失败时，为了不影响业务可用性，记录日志但视为通过
		// 可根据风险偏好调整此策略
		return nil
	}

	// 构建验证请求
	req := &captcha20230305.VerifyIntelligentCaptchaRequest{
		SceneId:            tea.String(cfg.SceneID),
		CaptchaVerifyParam: tea.String(captchaVerifyParam),
	}

	// 调用验证接口
	resp, err := client.VerifyIntelligentCaptcha(req)
	if err != nil {
		logx.Errorf("[Captcha] 调用阿里云验证码接口失败: %+v", err)
		// 接口调用失败时，为了不影响业务可用性，记录日志但视为通过
		// 可根据风险偏好调整此策略
		return nil
	}

	// 检查验证结果
	if resp == nil || resp.Body == nil || resp.Body.Result == nil {
		logx.Errorf("[Captcha] 阿里云验证码响应异常: resp=%+v", resp)
		return errors.Wrapf(xerr.NewErrMsg("图形验证码校验失败"), "invalid response")
	}

	if tea.BoolValue(resp.Body.Result.VerifyResult) {
		logx.Info("[Captcha] 验证码校验通过")
		return nil
	}

	// 验证失败
	logx.Errorf("[Captcha] 验证码校验失败: code=%s", tea.StringValue(resp.Body.Result.VerifyCode))
	return errors.Wrapf(xerr.NewErrMsg("图形验证码校验失败"), "verify failed: %s", tea.StringValue(resp.Body.Result.VerifyCode))
}