自定义 jwt 策略验证失败时的响应 NestJs

Question

我使用 NestJs 成功实现了用于身份验证的 jwt 策略。

下面是jwt策略的代码

import { ServerResponse } from './../helpers/serverResponse.helper';
import { Injectable, UnauthorizedException, HttpStatus } from '@nestjs/common';
import { PassportStrategy } from '@nestjs/passport';
import { ExtractJwt, Strategy } from 'passport-jwt';
import { config as env } from 'dotenv';
import { Bugsnag } from '../helpers/bugsnag.helper';

env();

@Injectable()
export class JwtStrategy extends PassportStrategy(Strategy, 'jwt') {
    constructor(
    private readonly logger: Bugsnag,
    ) {
    super({
        jwtFromRequest: ExtractJwt.fromAuthHeaderAsBearerToken(),
        secretOrKey: process.env.JWT_SECRET_KEY,
        passReqToCallback: true,
    });

    }

    async validate(payload, done: Function) {
    try {
        const validClaims = await this.authService.verifyTokenClaims(payload);

        if (!validClaims)
            return done(new UnauthorizedException('invalid token claims'), false);
        done(null, payload);
    } catch (err) {
        this.logger.notify(err);
        return ServerResponse.throwError({
        success: false,
        status: HttpStatus.INTERNAL_SERVER_ERROR,
        message: 'JwtStrategy class, validate function',
        errors: [err],
        });
    }
    }
}

我在这里看到，仅当请求标头中提供了有效令牌时才会调用验证函数，我对此表示同意。但是，我想知道是否可以自定义在这种情况下发送的响应对象（提供了无效的令牌）。

如果是，我该怎么做？

Answer 1

如果您愿意，可以使用异常过滤器来捕获UnauthorizedException并修改响应。另一种选择是扩展mixin 类并在forAuthGuard('jwt')周围添加一些逻辑，然后在错误中读取原因并抛出特定的自定义消息try/catchsuper.canActivate(context)UnauthorizedException

Answer 2

您可以使用AuthGuard('jwt')的handleRequest方法在 JWT 验证失败时引发任何异常。

@Injectable()
export class JwtAuthGuard extends AuthGuard('jwt') {
  handleRequest(err: any, user: any, info: any, context: any, status: any) {
    if (info instanceof JsonWebTokenError) {
      throw new UnauthorizedException('Invalid Token!');
    }

    return super.handleRequest(err, user, info, context, status);
  }
}

JsonWebTokenError来自jsonwebtoken图书馆，护照内部使用。