certd/packages/ui/certd-server/src/controller/basic/code-controller.ts

import { Rule, RuleType } from '@midwayjs/validate';
import { ALL, Body, Controller, Get, Inject, Post, Provide, Query } from '@midwayjs/core';
import { BaseController, Constants } from '@certd/lib-server';
import { CodeService } from '../../modules/basic/service/code-service.js';
import { EmailService } from '../../modules/basic/service/email-service.js';

export class SmsCodeReq {
  @Rule(RuleType.string().required())
  phoneCode: string;

  @Rule(RuleType.string().required())
  mobile: string;

  @Rule(RuleType.string().required().max(10))
  randomStr: string;

  @Rule(RuleType.string().required().max(4))
  imgCode: string;

  @Rule(RuleType.string())
  verificationType: string;
}

export class EmailCodeReq {
  @Rule(RuleType.string().required())
  email: string;

  @Rule(RuleType.string().required().max(10))
  randomStr: string;

  @Rule(RuleType.string().required().max(4))
  imgCode: string;

  @Rule(RuleType.string())
  verificationType: string;
}

// 找回密码的验证码有效期
const FORGOT_PASSWORD_CODE_DURATION = 3

/**
 */
@Provide()
@Controller('/api/basic/code')
export class BasicController extends BaseController {
  @Inject()
  codeService: CodeService;

  @Inject()
  emailService: EmailService;

  @Post('/sendSmsCode', { summary: Constants.per.guest })
  public async sendSmsCode(
    @Body(ALL)
    body: SmsCodeReq
  ) {
    const opts = {
      verificationType: body.verificationType,
      verificationCodeLength: undefined,
      duration: undefined,
    };
    if(body?.verificationType === 'forgotPassword') {
      opts.duration = FORGOT_PASSWORD_CODE_DURATION;
      // opts.verificationCodeLength = 6; //部分厂商这里会设置参数长度这里就不改了
    }

    await this.codeService.checkCaptcha(body.randomStr, body.imgCode);
    await this.codeService.sendSmsCode(body.phoneCode, body.mobile, body.randomStr, opts);
    return this.ok(null);
  }

  @Post('/sendEmailCode', { summary: Constants.per.guest })
  public async sendEmailCode(
    @Body(ALL)
    body: EmailCodeReq
  ) {
    const opts = {
      verificationType: body.verificationType,
      verificationCodeLength: undefined,
      title: undefined,
      content: undefined,
      duration: undefined,
    };
    if(body?.verificationType === 'forgotPassword') {
      opts.title = '找回密码';
      opts.content = '验证码：${code}。您正在找回密码，请输入验证码并完成操作。如非本人操作请忽略';
      opts.duration = FORGOT_PASSWORD_CODE_DURATION;
      opts.verificationCodeLength = 6;
    }

    await this.codeService.checkCaptcha(body.randomStr, body.imgCode);
    await this.codeService.sendEmailCode(body.email, body.randomStr, opts);
    // 设置缓存内容
    return this.ok(null);
  }

  @Get('/captcha', { summary: Constants.per.guest })
  public async getCaptcha(@Query('randomStr') randomStr: any) {
    const captcha = await this.codeService.generateCaptcha(randomStr);
    this.ctx.res.setHeader('Content-Type', 'image/svg+xml');
    return captcha.data;
  }
}