first commit

app/main/api/internal/logic/auth/sendsmslogic.go

@@ -0,0 +1,105 @@
+package auth
+
+import (
+	"context"
+	"fmt"
+	"math/rand"
+	"time"
+	"znc-server/common/xerr"
+	"znc-server/pkg/lzkit/crypto"
+
+	"github.com/pkg/errors"
+
+	"znc-server/app/main/api/internal/svc"
+	"znc-server/app/main/api/internal/types"
+
+	openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
+	dysmsapi "github.com/alibabacloud-go/dysmsapi-20170525/v3/client"
+	"github.com/alibabacloud-go/tea-utils/v2/service"
+	"github.com/alibabacloud-go/tea/tea"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type SendSmsLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewSendSmsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SendSmsLogic {
+	return &SendSmsLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *SendSmsLogic) SendSms(req *types.SendSmsReq) error {
+	secretKey := l.svcCtx.Config.Encrypt.SecretKey
+	encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey)
+	if err != nil {
+		return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 加密手机号失败: %v", err)
+	}
+	// 检查手机号是否在一分钟内已发送过验证码
+	limitCodeKey := fmt.Sprintf("limit:%s:%s", req.ActionType, encryptedMobile)
+	exists, err := l.svcCtx.Redis.Exists(limitCodeKey)
+	if err != nil {
+		return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 读取redis缓存失败: %s", encryptedMobile)
+	}
+
+	if exists {
+		// 如果 Redis 中已经存在标记，说明在 1 分钟内请求过，返回错误
+		return errors.Wrapf(xerr.NewErrMsg("一分钟内不能重复发送验证码"), "短信发送, 手机号1分钟内重复请求发送验证码: %s", encryptedMobile)
+	}
+
+	code := fmt.Sprintf("%06d", rand.New(rand.NewSource(time.Now().UnixNano())).Intn(1000000))
+
+	// 发送短信
+	smsResp, err := l.sendSmsRequest(req.Mobile, code)
+	if err != nil {
+		return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 调用阿里客户端失败: %v", err)
+	}
+	if *smsResp.Body.Code != "OK" {
+		return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 阿里客户端响应失败: %s", *smsResp.Body.Message)
+	}
+	codeKey := fmt.Sprintf("%s:%s", req.ActionType, encryptedMobile)
+	// 将验证码保存到 Redis，设置过期时间
+	err = l.svcCtx.Redis.Setex(codeKey, code, l.svcCtx.Config.VerifyCode.ValidTime) // 验证码有效期5分钟
+	if err != nil {
+		return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 验证码设置过期时间失败: %v", err)
+	}
+	// 在 Redis 中设置 1 分钟的标记，限制重复请求
+	err = l.svcCtx.Redis.Setex(limitCodeKey, code, 60) // 标记 1 分钟内不能重复请求
+	if err != nil {
+		return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 验证码设置限制重复请求失败: %v", err)
+	}
+	return nil
+}
+
+// CreateClient 创建阿里云短信客户端
+func (l *SendSmsLogic) CreateClient() (*dysmsapi.Client, error) {
+	config := &openapi.Config{
+		AccessKeyId:     &l.svcCtx.Config.VerifyCode.AccessKeyID,
+		AccessKeySecret: &l.svcCtx.Config.VerifyCode.AccessKeySecret,
+	}
+	config.Endpoint = tea.String(l.svcCtx.Config.VerifyCode.EndpointURL)
+	return dysmsapi.NewClient(config)
+}
+
+// sendSmsRequest 发送短信请求
+func (l *SendSmsLogic) sendSmsRequest(mobile, code string) (*dysmsapi.SendSmsResponse, error) {
+	// 初始化阿里云短信客户端
+	cli, err := l.CreateClient()
+	if err != nil {
+		return nil, err
+	}
+
+	request := &dysmsapi.SendSmsRequest{
+		SignName:      tea.String(l.svcCtx.Config.VerifyCode.SignName),
+		TemplateCode:  tea.String(l.svcCtx.Config.VerifyCode.TemplateCode),
+		PhoneNumbers:  tea.String(mobile),
+		TemplateParam: tea.String(fmt.Sprintf("{\"code\":\"%s\"}", code)),
+	}
+	runtime := &service.RuntimeOptions{}
+	return cli.SendSmsWithOptions(request, runtime)
+}