[Nestjs] 实现生成图形验证码以及校验过程-阿里云开发者社区

[Nestjs] 实现生成图形验证码以及校验过程

2023-08-24 532

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 要在 NestJS 中实现图形验证码的校验过程，你可以按照以下步骤进行操作：1. 安装依赖：首先，使用 npm 或 yarn 安装 svg-captcha 和 cookie-parser，其中 svg-captcha 是用于生成 SVG 格式的验证码图片的库，cookie-parser 是用于处理和解析 Cookie 的库。

要在 NestJS 中实现图形验证码的校验过程，你可以按照以下步骤进行操作：

安装依赖：首先，使用 npm 或 yarn 安装 svg-captcha和 cookie-parser，其中 svg-captcha 是用于生成 SVG 格式的验证码图片的库，cookie-parser 是用于处理和解析 Cookie 的库。

生成和发送验证码：在验证码控制器中，编写生成和发送验证码图片的接口。同时将验证码文本存储在会话或数据库中，以便后续验证。

import { Controller, Get, Res, Req } from '@nestjs/common';
import * as svgCaptcha from 'svg-captcha';
@Controller('captcha')
export class CaptchaController {
  @Get()
  generateCaptcha(@Res() res, @Req() req) {
    const captcha = svgCaptcha.create();
    // 将验证码文本存储在会话或数据库中，以便后续验证
    req.session.captcha = captcha.text;
    res.set('Content-Type', 'image/svg+xml');
    res.send(captcha.data);
  }
}

校验验证码：在验证码控制器中，编写用于校验验证码的接口。

import { Controller, Post, Body, Req } from '@nestjs/common';
@Controller('captcha')
export class CaptchaController {
  // ...
  @Post('verify')
  verifyCaptcha(@Body() body, @Req() req) {
    const { captchaText } = body;
    const storedCaptcha = req.session.captcha;
    if (captchaText && storedCaptcha && captchaText.toLowerCase() === storedCaptcha.toLowerCase()) {
      // 验证码校验成功
      return { success: true };
    } else {
      // 验证码校验失败
      return { success: false };
    }
  }
}

配置模块和路由：在 app.module.ts 文件中导入 CookieParserMiddleware 和 SessionMiddleware，配置 CaptchaModule，并添加中间件到应用程序的路由配置中。

import { Module, NestModule, MiddlewareConsumer } from '@nestjs/common';
import { CaptchaController } from './captcha/captcha.controller';
import * as cookieParser from 'cookie-parser';
import * as session from 'express-session';
@Module({
  controllers: [CaptchaController],
})
export class CaptchaModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer.apply(cookieParser(), session({ secret: 'your_secret_key' })).forRoutes(CaptchaController);
  }
}