• Index
• Search by APIs
• Search by Words
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• pknote-backend-master
  • .eslintignore
  • src
    • entity
      • MsgReply.ts
      • OrderDetail.ts
      • FundsCreditRelated.ts
      • MyColumn.ts
      • BaseOrg.ts
      • StudyInfo.ts
      • AuthorCertified.ts
      • OrderPayInfo.ts
      • VipCategory.ts
      • MyChapter.ts
      • BaseFile.ts
      • MyComment.ts
      • BaseTenant.ts
      • UserTags.ts
      • BaseCompetence.ts
      • VipGroup.ts
      • FundsWithdraw.ts
      • FundsAccount.ts
      • BaseResource.ts
      • OrderInfo.ts
      • MsgInfo.ts
      • BaseRole.ts
      • WorkflowInfo.ts
      • MyContent.ts
      • FundsCredit.ts
      • UserShipping.ts
      • VipRelatedContent.ts
      • Chapter.ts
      • UserInfo.ts
    • app.module.ts
    • common
      • providers
        • redis-client.service.ts
        • sms.service.ts
      • decorators
        • get-user.decorator.ts
      • exceptions
        • code-expire.exception.ts
      • interfaces
        • sms-response.interface.ts
        • http-response.interface.ts
    • app.service.ts
    • config
      • typeorm.config.ts
      • redis.config.ts
    • app.controller.spec.ts
    • main.ts
    • system
      • content
        • dto
          • create-content.dto.ts
          • update-content.dto.ts
        • content.module.ts
        • content.service.ts
        • content.controller.ts
    • app.controller.ts
    • auth
      • providers
        • dto
          • login-by-pwd.dto.ts
          • auth-credentials.dto.ts
        • jwt.strategy.ts
        • auth.service.ts
        • user.repository.ts
      • controllers
        • auth.controller.ts
      • interfaces
        • jwt-payload.interface.ts
      • auth.module.ts
  • .husky
    • common.sh
    • pre-commit
    • lintstagedrc.js
  • .prettierrc
  • LICENSE
  • test
    • app.e2e-spec.ts
    • jest-e2e.json
  • commitlint.config.js
  • config
    • default.yml
    • production.yml
  • tsconfig.build.json
  • README.md
  • package.json
  • .cz-config.js
  • Dockerfile
  • .eslintrc.js
  • tsconfig.json
  • .gitignore
  • .prettierignore
  • .dockerignore
  • .vscode
    • launch.json
  • nest-cli.json

import {
  Injectable,
  UnauthorizedException,
  Logger,
  // HttpException,
} from '@nestjs/common';
import { UserRepository } from './user.repository';
import { InjectRepository } from '@nestjs/typeorm';
import { AuthCredentialsDto } from './dto/auth-credentials.dto';
import { JwtService } from '@nestjs/jwt';
import { JwtPayload } from '../interfaces/jwt-payload.interface';
import { SmsService } from '@/common/providers/sms.service';
import { RedisClientService } from '@/common/providers/redis-client.service';
import { HttpResponse } from '@/common/interfaces/http-response.interface';
import { CodeExpireException } from '@/common/exceptions/code-expire.exception';
import { LoginByPwdDto } from './dto/login-by-pwd.dto';

@Injectable()
export class AuthService {
  private logger = new Logger('AuthService');
  constructor(
    @InjectRepository(UserRepository)
    private userRepository: UserRepository,
    private jwtService: JwtService,
    private smsService: SmsService,
    private redisClientService: RedisClientService,
  ) {}

  async register(authCredentialsDto: AuthCredentialsDto): Promise<void> {
    const { phone, code } = authCredentialsDto;
    if (process.env.NODE_ENV === 'development' && code === '666666') {
      return this.userRepository.register(authCredentialsDto);
    }
    const result = await this.redisClientService.getValue(phone);
    if (!result || result != code) {
      throw new CodeExpireException();
    } else {
      // 否则使用过后，删除该key
      await this.redisClientService.delKey(phone);
    }
    return this.userRepository.register(authCredentialsDto);
  }

  async loginByPwd(
    loginData: LoginByPwdDto,
  ): Promise<{ accessToken: string; user: unknown }> {
    const user = await this.userRepository.validateuserPassword(loginData);

    if (!user) {
      throw new UnauthorizedException('Invalid credentials');
    }

    const { userId } = user;
    const payload: JwtPayload = { userId };
    const accessToken: string = await this.jwtService.sign(payload);

    this.logger.debug(
      `Generator JWT token with payload ${JSON.stringify(payload)}`,
    );
    return { accessToken, user };
  }

  // 发送短信
  async sendCode(phone: string): Promise<HttpResponse> {
    const code = String(Math.random()).slice(-6);
    console.log('---> 验证码: ', code);
    await this.redisClientService.setValue(phone, code, 'EX', 60 * 10);
    const result = await this.smsService.sendCode(phone, code);
    // const result = 1;
    return {
      statusCode: result ? 0 : 1,
    };
  }
}