瀏覽代碼

验证码用于注册

枫叶秋林 2 年之前
父節點
當前提交
94532297b7
共有 2 個文件被更改,包括 15 次插入8 次删除
  1. 9 4
      src/auth/auth.controller.ts
  2. 6 4
      src/cod/cod.module.ts

+ 9 - 4
src/auth/auth.controller.ts

@@ -1,4 +1,5 @@
-import { Body, Controller, Get, Post, Req, UseGuards } from '@nestjs/common'
+import { CodService } from '@/cod/cod.service'
+import { Body, Controller, Get, Headers, Post, Req, UseGuards } from '@nestjs/common'
 import { AuthGuard } from '@nestjs/passport'
 import { Request } from 'express'
 import { AuthService } from './auth.service'
@@ -7,10 +8,14 @@ import registerDto from './dto/register.dto'
 
 @Controller('auth')
 export class AuthController {
-  constructor(private auto: AuthService) {}
+  constructor(private auto: AuthService, private readonly codService: CodService) {}
   @Post('register')
-  register(@Body() dto: registerDto) {
-    return this.auto.register(dto)
+  async register(@Body() dto: registerDto, @Req() req: Request, @Headers() headers: any) {
+    const rescod = await this.codService.verifycod(req.ip, headers.cod)
+    if (!rescod.status) {
+      return rescod
+    }
+    return await this.auto.register(dto)
   }
   @Post('login')
   login(@Body() dto: LoginDto) {

+ 6 - 4
src/cod/cod.module.ts

@@ -1,9 +1,11 @@
-import { Module } from '@nestjs/common';
-import { CodService } from './cod.service';
-import { CodController } from './cod.controller';
+import { Global, Module } from '@nestjs/common'
+import { CodService } from './cod.service'
+import { CodController } from './cod.controller'
 
+@Global()
 @Module({
   providers: [CodService],
-  controllers: [CodController]
+  controllers: [CodController],
+  exports: [CodService],
 })
 export class CodModule {}