TypeORM与NestJS应用中密码自动哈希的实现指南

npm install bcrypt
npm install -D @types/bcrypt

// src/user/user.entity.ts
import { Entity, PrimaryGeneratedColumn, Column, BeforeInsert, BeforeUpdate } from 'typeorm';
import * as bcrypt from 'bcrypt';

@Entity()
export class User {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({ unique: true })
  username: string;

  @Column()
  password: string;

  // ... 其他字段
}

// src/user/user.entity.ts (添加在 User 类内部)
// ...
export class User {
  // ... 其他字段和属性

  @BeforeInsert()
  async hashPasswordOnInsert() {
    if (this.password) {
      this.password = await bcrypt.hash(this.password, 10); // 10 是 saltRounds
    }
  }
}

// src/user/user.entity.ts (添加在 User 类内部)
// ...
export class User {
  // ... 其他字段和属性

  @BeforeInsert()
  async hashPasswordOnInsert() {
    if (this.password) {
      this.password = await bcrypt.hash(this.password, 10);
    }
  }

  @BeforeUpdate()
  async hashPasswordOnUpdate() {
    // 只有当密码字段被修改时才重新哈希
    // 注意：TypeORM的@BeforeUpdate钩子不会自动告诉你哪些字段被修改。
    // 如果你在服务层手动设置了新的明文密码，这个钩子会捕获到。
    // 如果你只更新了其他字段，而没有触碰password字段，则不会重新哈希。
    // 为了更精确地判断密码是否改变，可能需要额外逻辑（如在服务层比较旧密码或设置一个标志）。
    if (this.password && this.password.length < 60) { // 简单判断是否为明文密码（哈希密码通常更长）
      this.password = await bcrypt.hash(this.password, 10);
    }
  }
}

// src/user/user.entity.ts
import { Entity, PrimaryGeneratedColumn, Column, BeforeInsert, BeforeUpdate } from 'typeorm';
import * as bcrypt from 'bcrypt';

@Entity()
export class User {
  @PrimaryGeneratedColumn()
  id: number;

  @Column({ unique: true })
  username: string;

  @Column()
  password: string;

  @BeforeInsert()
  async hashPasswordOnInsert() {
    if (this.password) {
      this.password = await bcrypt.hash(this.password, 10);
    }
  }

  @BeforeUpdate()
  async hashPasswordOnUpdate() {
    // 只有当密码字段被修改且看起来是明文时才重新哈希
    // 这里使用长度判断作为一种简单机制，实际应用中可能需要更复杂的逻辑
    if (this.password && this.password.length < 60) {
      this.password = await bcrypt.hash(this.password, 10);
    }
  }

  // 辅助方法：用于验证密码，不直接存储在数据库中
  async validatePassword(password: string): Promise<boolean> {
    return bcrypt.compare(password, this.password);
  }
}

// src/user/user.service.ts
import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { User } from './user.entity';
import { CreateUserDto } from './dto/create-user.dto';
import { UpdateUserDto } from './dto/update-user.dto';

@Injectable()
export class UserService {
  constructor(
    @InjectRepository(User)
    private usersRepository: Repository<User>,
  ) {}

  async create(createUserDto: CreateUserDto): Promise<User> {
    const newUser = this.usersRepository.create(createUserDto);
    // 在这里设置明文密码，@BeforeInsert 会自动哈希
    // newUser.password = createUserDto.password; // create方法通常会直接映射dto，所以这行可能不需要显式写
    return this.usersRepository.save(newUser);
  }

  async update(id: number, updateUserDto: UpdateUserDto): Promise<User> {
    const user = await this.usersRepository.findOneBy({ id });
    if (!user) {
      throw new Error('User not found');
    }
    // 如果 updateUserDto 包含新密码，则会更新 user.password
    // @BeforeUpdate 会自动处理新密码的哈希
    Object.assign(user, updateUserDto);
    return this.usersRepository.save(user);
  }

  async findOneByUsername(username: string): Promise<User | undefined> {
    return this.usersRepository.findOne({ where: { username } });
  }
}

// src/auth/auth.service.ts (示例)
import { Injectable, UnauthorizedException } from '@nestjs/common';
import { UserService } from '../user/user.service';
import { LoginDto } from './dto/login.dto';

@Injectable()
export class AuthService {
  constructor(private userService: UserService) {}

  async validateUser(username: string, pass: string): Promise<any> {
    const user = await this.userService.findOneByUsername(username);
    if (user && await user.validatePassword(pass)) { // 使用实体中的辅助方法验证密码
      const { password, ...result } = user;
      return result;
    }
    return null;
  }

  async login(loginDto: LoginDto) {
    const user = await this.validateUser(loginDto.username, loginDto.password);
    if (!user) {
      throw new UnauthorizedException('Invalid credentials');
    }
    // ... 生成 JWT token 等后续操作
    return user;
  }
}