李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
82 lines
2.6 KiB
Go
82 lines
2.6 KiB
Go
package repository
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
|
||
captcha "github.com/alibabacloud-go/captcha-20230305/client"
|
||
openapiutil "github.com/alibabacloud-go/darabonba-openapi/v2/utils"
|
||
"github.com/alibabacloud-go/tea/dara"
|
||
"github.com/alibabacloud-go/tea/tea"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/service"
|
||
)
|
||
|
||
const aliyunCaptchaTimeoutMillis = 10_000
|
||
|
||
type aliyunCaptchaVerifier struct {
|
||
protocol string // "HTTPS"；测试注入 "HTTP" 指向 httptest.Server
|
||
timeoutMillis int
|
||
}
|
||
|
||
func NewAliyunCaptchaVerifier() service.AliyunCaptchaVerifier {
|
||
return &aliyunCaptchaVerifier{
|
||
protocol: "HTTPS",
|
||
timeoutMillis: aliyunCaptchaTimeoutMillis,
|
||
}
|
||
}
|
||
|
||
// VerifyCaptcha 调用阿里云验证码 2.0 VerifyIntelligentCaptcha。
|
||
// AK/SK 是可热更的后台设置，每次调用按当前凭证新建 client。
|
||
func (v *aliyunCaptchaVerifier) VerifyCaptcha(ctx context.Context, cred service.AliyunCaptchaCredentials, captchaVerifyParam string) (*service.AliyunCaptchaVerifyResult, error) {
|
||
client, err := captcha.NewClient(&openapiutil.Config{
|
||
AccessKeyId: dara.String(cred.AccessKeyID),
|
||
AccessKeySecret: dara.String(cred.AccessKeySecret),
|
||
Endpoint: dara.String(cred.Endpoint),
|
||
Protocol: dara.String(v.protocol),
|
||
ConnectTimeout: dara.Int(v.timeoutMillis),
|
||
ReadTimeout: dara.Int(v.timeoutMillis),
|
||
})
|
||
if err != nil {
|
||
return nil, fmt.Errorf("create aliyun captcha client: %w", err)
|
||
}
|
||
|
||
request := &captcha.VerifyIntelligentCaptchaRequest{
|
||
CaptchaVerifyParam: dara.String(captchaVerifyParam),
|
||
SceneId: dara.String(cred.SceneID),
|
||
}
|
||
|
||
response, err := client.VerifyIntelligentCaptchaWithContext(ctx, request, &dara.RuntimeOptions{})
|
||
if err != nil {
|
||
return nil, normalizeAliyunCaptchaError(err)
|
||
}
|
||
|
||
result := &service.AliyunCaptchaVerifyResult{}
|
||
if body := response.Body; body != nil && body.Result != nil {
|
||
result.VerifyResult = dara.BoolValue(body.Result.VerifyResult)
|
||
result.VerifyCode = dara.StringValue(body.Result.VerifyCode)
|
||
}
|
||
return result, nil
|
||
}
|
||
|
||
// normalizeAliyunCaptchaError 把 SDK 的两种错误类型归一化为 service.AliyunCaptchaAPIError，
|
||
// 其余错误（网络/超时等）原样返回。
|
||
func normalizeAliyunCaptchaError(err error) error {
|
||
var teaErr *tea.SDKError
|
||
if errors.As(err, &teaErr) {
|
||
return &service.AliyunCaptchaAPIError{
|
||
Code: tea.StringValue(teaErr.Code),
|
||
Message: tea.StringValue(teaErr.Message),
|
||
}
|
||
}
|
||
var daraErr *dara.SDKError
|
||
if errors.As(err, &daraErr) {
|
||
return &service.AliyunCaptchaAPIError{
|
||
Code: dara.StringValue(daraErr.Code),
|
||
Message: dara.StringValue(daraErr.Message),
|
||
}
|
||
}
|
||
return err
|
||
}
|