fadd

pkg/captcha/aliyun.go

@@ -0,0 +1,53 @@
+package captcha
+
+import (
+	"errors"
+	"os"
+
+	openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
+
+	"github.com/alibabacloud-go/captcha-20230305/client"
+	"github.com/alibabacloud-go/tea/tea"
+)
+
+type Config struct {
+	AccessKeyID     string
+	AccessKeySecret string
+	EndpointURL     string
+	SceneID         string
+}
+
+func Verify(cfg Config, captchaVerifyParam string) error {
+	if os.Getenv("ENV") == "development" {
+		return nil
+	}
+	if captchaVerifyParam == "" {
+		return errors.New("图形验证码校验失败")
+	}
+
+	clientCfg := &openapi.Config{
+		AccessKeyId:     tea.String(cfg.AccessKeyID),
+		AccessKeySecret: tea.String(cfg.AccessKeySecret),
+		Endpoint:        tea.String(cfg.EndpointURL),
+	}
+	captchaClient, err := client.NewClient(clientCfg)
+	if err != nil {
+		return errors.New("图形验证码校验失败")
+	}
+
+	req := &client.VerifyIntelligentCaptchaRequest{
+		SceneId:            tea.String(cfg.SceneID),
+		CaptchaVerifyParam: tea.String(captchaVerifyParam),
+	}
+
+	resp, err := captchaClient.VerifyIntelligentCaptcha(req)
+	if err != nil {
+		return errors.New("图形验证码校验失败")
+	}
+
+	if resp.Body == nil || !tea.BoolValue(resp.Body.Result.VerifyResult) {
+		return errors.New("图形验证码校验失败")
+	}
+
+	return nil
+}