certd/packages/ui/certd-server/src/controller/basic/login/register-controller.ts

import { ALL, Body, Controller, Inject, Post, Provide, RequestIP } from "@midwayjs/core";
import { BaseController, Constants, SysSettingsService } from "@certd/lib-server";
import { RegisterType, UserService } from "../../../modules/sys/authority/service/user-service.js";
import { CodeService } from "../../../modules/basic/service/code-service.js";
import { checkComm, checkPlus } from "@certd/plus-core";
import { InviteService } from "@certd/commercial-core";

export type RegisterReq = {
  type: RegisterType;
  username: string;
  password: string;
  mobile: string;
  email: string;
  phoneCode?: string;

  validateCode: string;
  captcha: any;
  inviteCode?: string;
};

/**
 */
@Provide()
@Controller("/api/")
export class RegisterController extends BaseController {
  @Inject()
  userService: UserService;
  @Inject()
  codeService: CodeService;

  @Inject()
  sysSettingsService: SysSettingsService;

  @Inject()
  inviteService: InviteService;

  @Post("/register", { description: Constants.per.guest })
  public async register(
    @Body(ALL)
    body: RegisterReq,
    @RequestIP() remoteIp: string
  ) {
    const sysPublicSettings = await this.sysSettingsService.getPublicSettings();
    if (sysPublicSettings.registerEnabled === false) {
      throw new Error("当前站点已禁止自助注册功能");
    }

    if (body.username && ["admin", "certd"].includes(body.username)) {
      throw new Error("用户名不能为保留字");
    }

    if (body.type === "username") {
      if (sysPublicSettings.usernameRegisterEnabled === false) {
        throw new Error("当前站点已禁止用户名注册功能");
      }
      if (!body.username) {
        throw new Error("用户名不能为空");
      }

      await this.codeService.checkCaptcha(body.captcha, { remoteIp });
      const registerUser = {
        username: body.username,
        password: body.password,
      } as any;
      const newUser = await this.userService.register(body.type, registerUser, async txManager => {
        await this.inviteService.bindInvitee({ manager: txManager }, { inviteeUserId: registerUser.id, inviteCode: body.inviteCode });
      });
      return this.ok(newUser);
    } else if (body.type === "mobile") {
      if (sysPublicSettings.mobileRegisterEnabled === false) {
        throw new Error("当前站点已禁止手机号注册功能");
      }
      checkComm();
      //验证短信验证码
      await this.codeService.checkSmsCode({
        mobile: body.mobile,
        phoneCode: body.phoneCode,
        smsCode: body.validateCode,
        throwError: true,
      });
      const registerUser = {
        username: body.username,
        phoneCode: body.phoneCode,
        mobile: body.mobile,
        password: body.password,
      } as any;
      const newUser = await this.userService.register(body.type, registerUser, async txManager => {
        await this.inviteService.bindInvitee({ manager: txManager }, { inviteeUserId: registerUser.id, inviteCode: body.inviteCode });
      });
      return this.ok(newUser);
    } else if (body.type === "email") {
      if (sysPublicSettings.emailRegisterEnabled === false) {
        throw new Error("当前站点已禁止Email注册功能");
      }
      checkPlus();
      this.codeService.checkEmailCode({
        email: body.email,
        validateCode: body.validateCode,
        throwError: true,
      });
      const registerUser = {
        username: body.username,
        email: body.email,
        password: body.password,
      } as any;
      const newUser = await this.userService.register(body.type, registerUser, async txManager => {
        await this.inviteService.bindInvitee({ manager: txManager }, { inviteeUserId: registerUser.id, inviteCode: body.inviteCode });
      });
      return this.ok(newUser);
    }
  }
}