41 lines
1.3 KiB
TypeScript
41 lines
1.3 KiB
TypeScript
|
import { Injectable, CanActivate, ExecutionContext, UnauthorizedException } from '@nestjs/common';
|
||
|
|
import { JwtService } from '@nestjs/jwt';
|
||
|
|
import { Request } from 'express';
|
||
|
|
import { AuthService } from '../services/AuthService';
|
||
|
|
|
||
|
|
@Injectable()
|
||
|
|
export class JwtAuthGuard implements CanActivate {
|
||
|
|
constructor(
|
||
|
|
private readonly jwtService: JwtService,
|
||
|
|
private readonly authService: AuthService,
|
||
|
|
) {}
|
||
|
|
|
||
|
|
async canActivate(context: ExecutionContext): Promise<boolean> {
|
||
|
|
const request = context.switchToHttp().getRequest();
|
||
|
|
const token = this.extractTokenFromHeader(request);
|
||
|
|
|
||
|
|
if (!token) {
|
||
|
|
throw new UnauthorizedException('未提供访问令牌');
|
||
|
|
}
|
||
|
|
|
||
|
|
try {
|
||
|
|
// 验证Token
|
||
|
|
const payload = await this.authService.validateToken(token);
|
||
|
|
|
||
|
|
if (!payload) {
|
||
|
|
throw new UnauthorizedException('访问令牌无效或已过期');
|
||
|
|
}
|
||
|
|
|
||
|
|
// 将用户信息添加到请求对象中
|
||
|
|
request.user = payload;
|
||
|
|
return true;
|
||
|
|
} catch (error) {
|
||
|
|
throw new UnauthorizedException('访问令牌验证失败');
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private extractTokenFromHeader(request: Request): string | undefined {
|
||
|
|
const [type, token] = request.headers.authorization?.split(' ') ?? [];
|
||
|
|
return type === 'Bearer' ? token : undefined;
|
||
|
|
}
|
||
|
|
}
|