forked from Ynov/ptitspas-ynov-back
27 lines
789 B
TypeScript
27 lines
789 B
TypeScript
import { CanActivate, ExecutionContext, Injectable } from "@nestjs/common";
|
|
import { Reflector } from "@nestjs/core";
|
|
import { Observable } from "rxjs";
|
|
|
|
@Injectable()
|
|
export class RolesGuard implements CanActivate {
|
|
constructor(private readonly reflector: Reflector) {}
|
|
|
|
canActivate(context: ExecutionContext): boolean | Promise<boolean> | Observable<boolean> {
|
|
const requiredRoles = this.reflector.getAllAndOverride<string[]>('roles', [
|
|
context.getHandler(),
|
|
context.getClass(),
|
|
]);
|
|
|
|
if (!requiredRoles || requiredRoles.length === 0) {
|
|
return true;
|
|
}
|
|
|
|
const request = context.switchToHttp().getRequest();
|
|
const user = request.user;
|
|
if (!user || !user.role) {
|
|
return false;
|
|
}
|
|
return requiredRoles.includes(user.role);
|
|
}
|
|
}
|