feat: 图形验证码

Author: kuangshp

package-lock.json

@@ -34,12 +34,14 @@
     "class-transformer": "^0.5.1",
     "class-validator": "^0.14.0",
     "cross-env": "^7.0.3",
+    "express-session": "^1.17.3",
     "ioredis": "^5.3.2",
     "lodash": "^4.17.21",
     "moment": "^2.29.4",
     "mysql2": "^3.6.1",
     "reflect-metadata": "^0.1.13",
     "rxjs": "^7.8.1",
+    "svg-captcha": "^1.4.0",
     "typeorm": "^0.3.17",
     "uuid": "^9.0.1",
     "winston": "^3.10.0",
@@ -53,6 +55,7 @@
     "@nestjs/schematics": "^10.0.0",
     "@nestjs/testing": "^10.0.0",
     "@types/express": "^4.17.17",
+    "@types/express-session": "^1.17.8",
     "@types/jest": "^29.5.2",
     "@types/lodash": "^4.14.199",
     "@types/node": "^20.3.1",

package.json

@@ -9,4 +9,7 @@ export class LoginDto {
   @Length(6, 12, { message: '密码长度必须是6-12位的' })
   @IsNotEmpty({ message: '密码不能为空' })
   readonly password!: string;
+
+  @IsNotEmpty({ message: '验证码不能空' })
+  readonly captcha!: string;
 }

src/api/login/dto/login.dto.ts

@@ -1,19 +1,41 @@
-import { Body, Controller, Get, Post, Query } from '@nestjs/common';
+import { Body, Controller, Get, Post, Query, Res, Session } from '@nestjs/common';
 import { LoginDto } from './dto/login.dto';
 import { LoginService } from './login.service';
 import { LoginVo } from './vo/login.vo';
-
+import * as svgCaptcha from 'svg-captcha';
+import { HttpException } from '@nestjs/common/exceptions';
+import { HttpStatus } from '@nestjs/common/enums';
 @Controller()
 export class LoginController {
   constructor(private readonly loginService: LoginService) {}
 
   @Post('login')
-  async loginApi(@Body() req: LoginDto): Promise<LoginVo> {
-    return await this.loginService.loginApi(req);
+  async loginApi(@Body() req: LoginDto, @Session() session: any): Promise<LoginVo> {
+    const storedCaptcha = session.code;
+    if (req.captcha && storedCaptcha && req.captcha.toLowerCase() === storedCaptcha.toLowerCase()) {
+      return await this.loginService.loginApi(req);
+    } else {
+      throw new HttpException('验证码错误', HttpStatus.OK);
+    }
   }
 
   @Get('refresh')
   async refreshTokenApi(@Query('token') token: string): Promise<LoginVo> {
     return await this.loginService.refreshTokenApi(token);
   }
+
+  @Get('captcha')
+  getCaptchaApi(@Res() res: any, @Session() session: any) {
+    const captcha = svgCaptcha.create({
+      size: 4, //生成几个验证码
+      fontSize: 50, //文字大小
+      width: 100, //宽度
+      height: 34, //高度
+      background: '#cc9966', //背景颜色
+    });
+    console.log(captcha, '??');
+    session['code'] = captcha['text']; //存储验证码记录到session
+    res.set('Content-Type', 'image/svg+xml');
+    res.send(captcha['data']);
+  }
 }