add skip-auth.decorator

This commit is contained in:
秦秋旭 2023-02-16 16:19:24 +08:00
parent 205689e130
commit 93127d2bd6
5 changed files with 23 additions and 6 deletions

View File

@ -1,8 +1,10 @@
import { APP_GUARD } from '@nestjs/core'
import { Logger, Module } from '@nestjs/common' import { Logger, Module } from '@nestjs/common'
import { ConfigModule } from '@nestjs/config' import { ConfigModule } from '@nestjs/config'
import { PrismaModule, loggingMiddleware } from 'nestjs-prisma' import { PrismaModule, loggingMiddleware } from 'nestjs-prisma'
import { UsersModule } from './users/users.module' import { UsersModule } from './users/users.module'
import { AuthModule } from './auth/auth.module' import { AuthModule } from './auth/auth.module'
import { JwtAuthGuard } from './common/guards/jwt-auth.guard'
@Module({ @Module({
imports: [ imports: [
@ -22,6 +24,6 @@ import { AuthModule } from './auth/auth.module'
AuthModule, AuthModule,
], ],
controllers: [], controllers: [],
providers: [], providers: [{ provide: APP_GUARD, useClass: JwtAuthGuard }],
}) })
export class AppModule {} export class AppModule {}

View File

@ -5,17 +5,20 @@ import { ApiTags } from '@nestjs/swagger'
import { LoginInputDto } from './dto/login-input.dto' import { LoginInputDto } from './dto/login-input.dto'
import { UserEntity } from 'src/users/entities/user.entity' import { UserEntity } from 'src/users/entities/user.entity'
import { User } from 'src/common/decorators/user.decorator' import { User } from 'src/common/decorators/user.decorator'
import { SkipAuth } from 'src/common/decorators/skip-auth.decorator'
@ApiTags('auth') @ApiTags('auth')
@Controller() @Controller()
export class AuthController { export class AuthController {
constructor(private readonly authService: AuthService) {} constructor(private readonly authService: AuthService) {}
@SkipAuth()
@Post('api/register') @Post('api/register')
async register(@Body() userData: CreateUserDto) { async register(@Body() userData: CreateUserDto) {
return this.authService.register(userData) return this.authService.register(userData)
} }
@SkipAuth()
@Post('api/login') @Post('api/login')
async login(@Body() user: LoginInputDto) { async login(@Body() user: LoginInputDto) {
return this.authService.login(user.email, user.password) return this.authService.login(user.email, user.password)

View File

@ -0,0 +1,4 @@
import { SetMetadata } from '@nestjs/common'
export const SKIP_AUTH_KEY = 'skip-auth'
export const SkipAuth = () => SetMetadata(SKIP_AUTH_KEY, true)

View File

@ -1,3 +1,4 @@
import { Reflector } from '@nestjs/core'
import { import {
ExecutionContext, ExecutionContext,
Injectable, Injectable,
@ -5,12 +6,22 @@ import {
} from '@nestjs/common' } from '@nestjs/common'
import { AuthGuard } from '@nestjs/passport' import { AuthGuard } from '@nestjs/passport'
import 'rxjs' import 'rxjs'
import { SKIP_AUTH_KEY } from '../decorators/skip-auth.decorator'
@Injectable() @Injectable()
export class JwtAuthGuard extends AuthGuard('jwt') { export class JwtAuthGuard extends AuthGuard('jwt') {
constructor(private reflector: Reflector) {
super()
}
canActivate(context: ExecutionContext) { canActivate(context: ExecutionContext) {
// Add your custom authentication logic here const isPublic = this.reflector.getAllAndOverride<boolean>(SKIP_AUTH_KEY, [
// for example, call super.logIn(request) to establish a session. context.getHandler(),
context.getClass(),
])
if (isPublic) {
return true
}
return super.canActivate(context) return super.canActivate(context)
} }

View File

@ -4,15 +4,12 @@ import { ConfigService } from '@nestjs/config'
import { DocumentBuilder, SwaggerModule } from '@nestjs/swagger' import { DocumentBuilder, SwaggerModule } from '@nestjs/swagger'
import { PrismaClientExceptionFilter, PrismaService } from 'nestjs-prisma' import { PrismaClientExceptionFilter, PrismaService } from 'nestjs-prisma'
import { AppModule } from './app.module' import { AppModule } from './app.module'
import { JwtAuthGuard } from './common/guards/jwt-auth.guard'
async function bootstrap() { async function bootstrap() {
const app = await NestFactory.create(AppModule) const app = await NestFactory.create(AppModule)
// Validation // Validation
app.useGlobalPipes(new ValidationPipe({ whitelist: true })) app.useGlobalPipes(new ValidationPipe({ whitelist: true }))
// Global Guard
app.useGlobalGuards(new JwtAuthGuard())
// enable shutdown hook // enable shutdown hook
const prismaService = app.get(PrismaService) const prismaService = app.get(PrismaService)