基础架构

internal/domains/user/services/sms_code_service.go

@@ -5,31 +5,32 @@ import (
 	"fmt"
 	"time"
 
-	"github.com/google/uuid"
 	"go.uber.org/zap"
 
 	"tyapi-server/internal/config"
 	"tyapi-server/internal/domains/user/entities"
 	"tyapi-server/internal/domains/user/repositories"
+	"tyapi-server/internal/infrastructure/external/sms"
 	"tyapi-server/internal/shared/interfaces"
-	"tyapi-server/internal/shared/sms"
 )
 
 // SMSCodeService 短信验证码服务
 type SMSCodeService struct {
-	repo      *repositories.SMSCodeRepository
-	smsClient sms.Service
+	repo      repositories.SMSCodeRepository
+	smsClient *sms.AliSMSService
 	cache     interfaces.CacheService
 	config    config.SMSConfig
+	appConfig config.AppConfig
 	logger    *zap.Logger
 }
 
 // NewSMSCodeService 创建短信验证码服务
 func NewSMSCodeService(
-	repo *repositories.SMSCodeRepository,
-	smsClient sms.Service,
+	repo repositories.SMSCodeRepository,
+	smsClient *sms.AliSMSService,
 	cache interfaces.CacheService,
 	config config.SMSConfig,
+	appConfig config.AppConfig,
 	logger *zap.Logger,
 ) *SMSCodeService {
 	return &SMSCodeService{
@@ -37,31 +38,25 @@ func NewSMSCodeService(
 		smsClient: smsClient,
 		cache:     cache,
 		config:    config,
+		appConfig: appConfig,
 		logger:    logger,
 	}
 }
 
 // SendCode 发送验证码
 func (s *SMSCodeService) SendCode(ctx context.Context, phone string, scene entities.SMSScene, clientIP, userAgent string) error {
-	// 1. 检查频率限制
-	if err := s.checkRateLimit(ctx, phone); err != nil {
-		return err
-	}
-
-	// 2. 生成验证码
+	// 1. 生成验证码
 	code := s.smsClient.GenerateCode(s.config.CodeLength)
 
-	// 3. 使用工厂方法创建SMS验证码记录
+	// 2. 使用工厂方法创建SMS验证码记录
 	smsCode, err := entities.NewSMSCode(phone, code, scene, s.config.ExpireTime, clientIP, userAgent)
 	if err != nil {
 		return fmt.Errorf("创建验证码记录失败: %w", err)
 	}
 
-	// 4. 设置ID
-	smsCode.ID = uuid.New().String()
-
-	// 5. 保存验证码
-	if err := s.repo.Create(ctx, smsCode); err != nil {
+	// 4. 保存验证码
+	*smsCode, err = s.repo.Create(ctx, *smsCode)
+	if err != nil {
 		s.logger.Error("保存短信验证码失败",
 			zap.String("phone", smsCode.GetMaskedPhone()),
 			zap.String("scene", smsCode.GetSceneName()),
@@ -69,7 +64,7 @@ func (s *SMSCodeService) SendCode(ctx context.Context, phone string, scene entit
 		return fmt.Errorf("保存验证码失败: %w", err)
 	}
 
-	// 6. 发送短信
+	// 5. 发送短信
 	if err := s.smsClient.SendVerificationCode(ctx, phone, code); err != nil {
 		// 记录发送失败但不删除验证码记录，让其自然过期
 		s.logger.Error("发送短信验证码失败",
@@ -79,8 +74,8 @@ func (s *SMSCodeService) SendCode(ctx context.Context, phone string, scene entit
 		return fmt.Errorf("短信发送失败: %w", err)
 	}
 
-	// 7. 更新发送记录缓存
-	s.updateSendRecord(ctx, phone)
+	// 6. 更新发送记录缓存
+	s.updateSendRecord(ctx, phone, scene)
 
 	s.logger.Info("短信验证码发送成功",
 		zap.String("phone", smsCode.GetMaskedPhone()),
@@ -92,19 +87,33 @@ func (s *SMSCodeService) SendCode(ctx context.Context, phone string, scene entit
 
 // VerifyCode 验证验证码
 func (s *SMSCodeService) VerifyCode(ctx context.Context, phone, code string, scene entities.SMSScene) error {
+	// 开发模式下跳过验证码校验
+	if s.appConfig.IsDevelopment() {
+		s.logger.Info("开发模式：验证码校验已跳过",
+			zap.String("phone", phone),
+			zap.String("scene", string(scene)),
+			zap.String("code", code))
+		return nil
+	}
+
 	// 1. 根据手机号和场景获取有效的验证码记录
-	smsCode, err := s.repo.GetValidCode(ctx, phone, scene)
+	smsCode, err := s.repo.GetValidByPhoneAndScene(ctx, phone, scene)
 	if err != nil {
 		return fmt.Errorf("验证码无效或已过期")
 	}
 
-	// 2. 使用实体的验证方法
+	// 2. 检查场景是否匹配
+	if smsCode.Scene != scene {
+		return fmt.Errorf("验证码错误或已过期")
+	}
+
+	// 3. 使用实体的验证方法
 	if err := smsCode.VerifyCode(code); err != nil {
 		return err
 	}
 
-	// 3. 保存更新后的验证码状态
-	if err := s.repo.Update(ctx, smsCode); err != nil {
+	// 4. 保存更新后的验证码状态
+	if err := s.repo.Update(ctx, *smsCode); err != nil {
 		s.logger.Error("更新验证码状态失败",
 			zap.String("code_id", smsCode.ID),
 			zap.Error(err))
@@ -120,10 +129,10 @@ func (s *SMSCodeService) VerifyCode(ctx context.Context, phone, code string, sce
 
 // CanResendCode 检查是否可以重新发送验证码
 func (s *SMSCodeService) CanResendCode(ctx context.Context, phone string, scene entities.SMSScene) (bool, error) {
-	// 1. 获取最近的验证码记录
-	recentCode, err := s.repo.GetRecentCode(ctx, phone, scene)
+	// 1. 获取最近的验证码记录（按场景）
+	recentCode, err := s.repo.GetValidByPhoneAndScene(ctx, phone, scene)
 	if err != nil {
-		// 如果没有记录，可以发送
+		// 如果没有该场景的记录，可以发送
 		return true, nil
 	}
 
@@ -144,8 +153,8 @@ func (s *SMSCodeService) CanResendCode(ctx context.Context, phone string, scene
 
 // GetCodeStatus 获取验证码状态信息
 func (s *SMSCodeService) GetCodeStatus(ctx context.Context, phone string, scene entities.SMSScene) (map[string]interface{}, error) {
-	// 1. 获取最近的验证码记录
-	recentCode, err := s.repo.GetRecentCode(ctx, phone, scene)
+	// 1. 获取最近的验证码记录（按场景）
+	recentCode, err := s.repo.GetValidByPhoneAndScene(ctx, phone, scene)
 	if err != nil {
 		return map[string]interface{}{
 			"has_code": false,
@@ -170,11 +179,11 @@ func (s *SMSCodeService) GetCodeStatus(ctx context.Context, phone string, scene
 }
 
 // checkRateLimit 检查发送频率限制
-func (s *SMSCodeService) checkRateLimit(ctx context.Context, phone string) error {
+func (s *SMSCodeService) CheckRateLimit(ctx context.Context, phone string, scene entities.SMSScene) error {
 	now := time.Now()
 
 	// 检查最小发送间隔
-	lastSentKey := fmt.Sprintf("sms:last_sent:%s", phone)
+	lastSentKey := fmt.Sprintf("sms:last_sent:%s:%s", scene, phone)
 	var lastSent time.Time
 	if err := s.cache.Get(ctx, lastSentKey, &lastSent); err == nil {
 		if now.Sub(lastSent) < s.config.RateLimit.MinInterval {
@@ -204,11 +213,11 @@ func (s *SMSCodeService) checkRateLimit(ctx context.Context, phone string) error
 }
 
 // updateSendRecord 更新发送记录
-func (s *SMSCodeService) updateSendRecord(ctx context.Context, phone string) {
+func (s *SMSCodeService) updateSendRecord(ctx context.Context, phone string, scene entities.SMSScene) {
 	now := time.Now()
 
 	// 更新最后发送时间
-	lastSentKey := fmt.Sprintf("sms:last_sent:%s", phone)
+	lastSentKey := fmt.Sprintf("sms:last_sent:%s:%s", scene, phone)
 	s.cache.Set(ctx, lastSentKey, now, s.config.RateLimit.MinInterval)
 
 	// 更新每小时计数
@@ -232,5 +241,5 @@ func (s *SMSCodeService) updateSendRecord(ctx context.Context, phone string) {
 
 // CleanExpiredCodes 清理过期验证码
 func (s *SMSCodeService) CleanExpiredCodes(ctx context.Context) error {
-	return s.repo.CleanupExpired(ctx)
+	return s.repo.DeleteBatch(ctx, []string{})
 }