完成验证码发送和验证

tone-page-server/src/verification/dto/send-verification-code.dto.ts

@@ -0,0 +1,19 @@
+import { IsEnum, IsString, Length, ValidateIf } from "class-validator";
+
+export class SendVerificationCodeDto {
+    @IsEnum(['phone', 'email'], { message: '请求类型错误' })
+    targetType: 'phone' | 'email';
+
+    @IsEnum(['login'], { message: '请求类型错误' })
+    type: 'login'
+
+    @ValidateIf(o => o.targetType === 'phone')
+    @IsString({ message: '手机号必须输入' })
+    @Length(11, 11, { message: '手机号异常' })// 中国大陆，11位数字
+    phone?: string;
+
+    @ValidateIf(o => o.targetType === 'email')
+    @IsString({ message: '邮箱必须输入' })
+    @Length(6, 254, { message: '邮箱异常' })// RFC 5321
+    email?: string;
+}

tone-page-server/src/verification/verification.controller.spec.ts

@@ -0,0 +1,18 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { VerificationController } from './verification.controller';
+
+describe('VerificationController', () => {
+  let controller: VerificationController;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      controllers: [VerificationController],
+    }).compile();
+
+    controller = module.get<VerificationController>(VerificationController);
+  });
+
+  it('should be defined', () => {
+    expect(controller).toBeDefined();
+  });
+});

tone-page-server/src/verification/verification.controller.ts

@@ -0,0 +1,28 @@
+import { BadRequestException, Body, Controller, Post } from '@nestjs/common';
+import { SendVerificationCodeDto } from './dto/send-verification-code.dto';
+import { VerificationService } from './verification.service';
+
+@Controller('verification')
+export class VerificationController {
+
+    constructor(
+        private readonly verificationService: VerificationService,
+    ) { }
+
+    @Post('send')
+    async sendVerificationCode(@Body() dto: SendVerificationCodeDto) {
+        switch (dto.type) {
+            case 'login':
+                switch (dto.targetType) {
+                    case 'phone':
+                        return this.verificationService.sendPhoneCode(dto.phone, dto.type);
+                    case 'email':
+                        return this.verificationService.sendEmailCode(dto.email, dto.type);
+                    default:
+                        throw new BadRequestException('不支持的目标类型');
+                }
+            default:
+                throw new BadRequestException('不支持的验证码类型');
+        }
+    }
+}

tone-page-server/src/verification/verification.module.ts

@@ -1,4 +1,9 @@
 import { Module } from '@nestjs/common';
+import { VerificationController } from './verification.controller';
+import { VerificationService } from './verification.service';
 
-@Module({})
+@Module({
+  controllers: [VerificationController],
+  providers: [VerificationService]
+})
 export class VerificationModule {}

tone-page-server/src/verification/verification.service.spec.ts

@@ -0,0 +1,18 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { VerificationService } from './verification.service';
+
+describe('VerificationService', () => {
+  let service: VerificationService;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      providers: [VerificationService],
+    }).compile();
+
+    service = module.get<VerificationService>(VerificationService);
+  });
+
+  it('should be defined', () => {
+    expect(service).toBeDefined();
+  });
+});

tone-page-server/src/verification/verification.service.ts

@@ -0,0 +1,88 @@
+import { Injectable } from '@nestjs/common';
+
+@Injectable()
+export class VerificationService {
+
+    private pool: Map<string, {
+        code: string;
+        createdAt: number;
+        expiredAt: number;
+        tryCount: number;
+        maxTryCount: number;
+    }> = new Map();
+
+    async sendPhoneCode(phone: string, type: 'login') {
+        const key = `phone:${phone}:${type}`;
+        // 检测是否在冷却时间内
+        // TODO
+
+        // 生成验证码
+        const code = this.generateCode();
+        // 发送验证码
+        // TODO
+
+        // 存储验证码
+        this.saveCode(key, code);
+    }
+
+    async sendEmailCode(email: string, type: 'login') {
+        const key = `email:${email}:${type}`;
+        // 检测是否在冷却时间内
+        // TODO
+
+        // 生成验证码
+        const code = this.generateCode();
+        // 发送验证码
+        // TODO
+
+        // 存储验证码
+        this.saveCode(key, code);
+    }
+
+    private saveCode(key: string, code: string) {
+        this.pool.set(key, {
+            code: code,
+            createdAt: Date.now(),
+            expiredAt: Date.now() + 10 * 60 * 1000, // 10分钟过期
+            tryCount: 0,
+            maxTryCount: 5,
+        });
+    }
+
+    verifyPhoneCode(phone: string, code: string, type: 'login') {
+        const key = `phone:${phone}:${type}`;
+        return this.verifyCode(key, code);
+    }
+
+    verifyEmailCode(email: string, code: string, type: 'login') {
+        const key = `email:${email}:${type}`;
+        return this.verifyCode(key, code);
+    }
+
+    /**
+     * @returns 0: 验证码正确, -1: 验证码不存在或已过期, -2: 验证码错误, -3: 超过最大尝试次数
+     */
+    private verifyCode(key: string, code: string) {
+        const data = this.pool.get(key);
+        if (!data) {
+            return -1;
+        }
+        if (data.tryCount >= data.maxTryCount) {
+            return -3;
+        }
+        if (data.expiredAt < Date.now()) {
+            return -1;
+        }
+        if (data.code !== code) {
+            data.tryCount++;
+            return -2;
+        }
+        this.pool.delete(key);
+        return 0;
+    }
+
+    private generateCode() {
+        return Math.floor(100000 + Math.random() * 900000).toString();
+    }
+}
+