91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
package captcha
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"tyass-server/common/xerr"
|
|
|
|
captcha20230305 "github.com/alibabacloud-go/captcha-20230305/client"
|
|
openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
|
|
"github.com/alibabacloud-go/tea/tea"
|
|
)
|
|
|
|
type Config struct {
|
|
AccessKeyID string
|
|
AccessKeySecret string
|
|
EndpointURL string
|
|
SceneID string
|
|
}
|
|
|
|
// Verify 验证图形验证码,支持微信环境跳过验证
|
|
func Verify(cfg Config, captchaVerifyParam string) error {
|
|
return VerifyWithContext(context.Background(), cfg, captchaVerifyParam)
|
|
}
|
|
|
|
// VerifyWithContext 带上下文的验证函数
|
|
func VerifyWithContext(ctx context.Context, cfg Config, captchaVerifyParam string) error {
|
|
// 开发环境直接通过
|
|
if os.Getenv("ENV") == "development" {
|
|
return nil
|
|
}
|
|
|
|
// 检查是否为微信环境
|
|
if isWechatEnvironment(ctx) {
|
|
return nil
|
|
}
|
|
|
|
if captchaVerifyParam == "" {
|
|
return errors.Wrapf(xerr.NewErrMsg("图形验证码校验失败"), "empty captchaVerifyParam")
|
|
}
|
|
|
|
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 {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "captcha client create error: %+v", err)
|
|
}
|
|
|
|
req := &captcha20230305.VerifyIntelligentCaptchaRequest{
|
|
SceneId: tea.String(cfg.SceneID),
|
|
CaptchaVerifyParam: tea.String(captchaVerifyParam),
|
|
}
|
|
resp, err := client.VerifyIntelligentCaptcha(req)
|
|
if err != nil {
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "captcha verify error: %+v", err)
|
|
}
|
|
if tea.BoolValue(resp.Body.Result.VerifyResult) {
|
|
return nil
|
|
}
|
|
return errors.Wrapf(xerr.NewErrMsg("图形验证码校验失败"), "verifyResult=false")
|
|
}
|
|
|
|
// isWechatEnvironment 检查是否为微信环境
|
|
func isWechatEnvironment(ctx context.Context) bool {
|
|
// 从上下文中获取请求头
|
|
userAgent := ""
|
|
if val := ctx.Value("user-agent"); val != nil {
|
|
if ua, ok := val.(string); ok {
|
|
userAgent = ua
|
|
}
|
|
}
|
|
if val := ctx.Value("User-Agent"); val != nil {
|
|
if ua, ok := val.(string); ok {
|
|
userAgent = ua
|
|
}
|
|
}
|
|
|
|
// 检查User-Agent是否包含微信标识
|
|
if strings.Contains(strings.ToLower(userAgent), "micromessenger") {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|