如何检查Nestjs Guard中的角色

Question

我有一个提供JWT令牌的外部服务。在Nestjs中，我首先有JWTGuard类：

@Injectable()
export class JwtGuard extends AuthGuard('JWT_STRATEGY') {
  constructor() {
    super();
  }

  getRequest(context: ExecutionContext) {
    console.log('JwtGuard');
    const ctx = GqlExecutionContext.create(context);
    return ctx.getContext().req;
  }
}

然后是护照策略：

@Injectable()
export class JwtStrategy extends PassportStrategy(Strategy, 'JWT_STRATEGY') {
  constructor(private configService: ConfigService) {
    super({
      secretOrKeyProvider: passportJwtSecret({
        cache: true,
        rateLimit: true,
        jwksRequestsPerMinute: 5,
        jwksUri: configService.get<string>('ADFS_KEYS_URL'),
      }),
      ignoreExpiration: false,
      jwtFromRequest: ExtractJwt.fromAuthHeaderAsBearerToken(),
      audience: configService.get<string>('ADFS_AUDIENCE'),
      issuer: configService.get<string>('ADFS_ISSUER'),
      algorithms: ['RS256'],
    });
  }

  validate(payload: unknown): unknown {
    console.log('jwt strategy');
    console.log(payload);
    return payload;
  }
}

JWTGuard似乎首先运行，然后是策略。但是，如果我想做额外的警卫和检查，请说角色。一个人在哪里做？我需要在护照策略之后的另一个后卫吗？我有两个角色“用户”和“ admin”。

Answer 1

首先，在appModule中定义全局守卫（称为rolesguard）如下：

providers: [
  AppService,
  {
    provide: APP_GUARD,
    useClass: JwtAuthGuard,
  },
  {
    provide: APP_GUARD,
    useClass: RolesGuard,
  },
]

然后在RolesGuard中我们有以下内容：

export enum FamilyRole {
  Admin = 'Admin',
  User = 'User',
}

...

export class FamilyRolesGuard implements CanActivate {

  async canActivate(context: ExecutionContext): Promise<boolean> {
    const requiredRoles = this.reflector.getAllAndOverride<FamilyRole>(
      ROLES_KEY,
      [context.getHandler(), context.getClass()],
    );

    if (!requiredRoles) {
      return true;
    }

    const { user } = context.switchToHttp().getRequest();
    // do the rest and return either true or false
  }
}

然后创建您自己的装饰器，您可以装饰您的装饰器API如果您需要该API来根据警卫保护您的应用程序。

import { SetMetadata } from '@nestjs/common';

export const ROLES_KEY = 'FamilyRoles';
export const FamilyRoles = (...roles: FamilyRole[]) =>
  SetMetadata(ROLES_KEY, roles);

然后，您可以在API中使用Decorator：

@Post('user')
@FamilyRoles(FamilyRole.Admin)
...

因此，在API中，如果您没有family，则在警卫中，您将不会有sircessuriperroles， 如果 block 将返回true。

更多信息： https://docs.nestjs.com/security/authorization