커스텀 데코레이터 만들기

import { createParamDecorator, ExecutionContext } from '@nestjs/common';

export const User = createParamDecorator(
  (data: unknown, ctx: ExecutionContext) => {
    // ExecutionContext는 Http, RPC, Websockets 컨텍스트를 추상화합니다.
    // 여기서는 HTTP 요청을 다루므로 Http 컨텍스트로 전환합니다.
    const request = ctx.switchToHttp().getRequest();

    // data 인자는 데코레이터에 전달된 인자입니다 (예: @User('firstName')).
    // 만약 data가 있다면 해당 속성만 반환하고, 없다면 전체 user 객체를 반환합니다.
    return data ? request.user?.[data] : request.user;
  },
);

import { Injectable, NestMiddleware } from '@nestjs/common';
import { Request, Response, NextFunction } from 'express';

@Injectable()
export class AuthMiddleware implements NestMiddleware {
  use(req: Request, res: Response, next: NextFunction) {
    // 실제 인증 로직 (DB 조회, JWT 검증 등)
    // 여기서는 임시로 가짜 사용자 정보를 req.user에 할당합니다.
    (req as any).user = { userId: 1, username: 'nest_user', roles: ['admin'] };
    console.log('User mocked by AuthMiddleware:', (req as any).user);
    next();
  }
}

import { Module, NestModule, MiddlewareConsumer } from '@nestjs/common';
import { AppController } from './app.controller';
import { AppService } from './app.service';
import { AuthMiddleware } from './auth/auth.middleware'; // 미들웨어 임포트

@Module({
  imports: [],
  controllers: [AppController],
  providers: [AppService],
})
export class AppModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer
      .apply(AuthMiddleware)
      .forRoutes('*'); // 모든 경로에 AuthMiddleware 적용
  }
}

import { Controller, Get } from '@nestjs/common';
import { AppService } from './app.service';
import { User } from './common/decorators/user.decorator'; // 커스텀 데코레이터 임포트

// 사용자 정보의 타입을 정의해두면 타입 안정성을 높일 수 있습니다.
interface CurrentUser {
  userId: number;
  username: string;
  roles: string[];
}

@Controller()
export class AppController {
  constructor(private readonly appService: AppService) {}

  @Get('profile')
  // @User() 데코레이터를 사용하여 요청 객체에서 user 정보를 가져옵니다.
  getProfile(@User() user: CurrentUser) {
    console.log('Current User:', user);
    return `Hello, ${user.username}! Your ID is ${user.userId} and roles are ${user.roles.join(', ')}.`;
  }

  @Get('username')
  // @User('username') 데코레이터를 사용하여 user 객체에서 특정 속성만 가져옵니다.
  getUsername(@User('username') username: string) {
    console.log('Current Username:', username);
    return `Your username is: ${username}`;
  }
}

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

export const Roles = (...roles: string[]) => SetMetadata('roles', roles);

import { Injectable, CanActivate, ExecutionContext } from '@nestjs/common';
import { Reflector } from '@nestjs/core'; // Reflector를 사용하여 메타데이터를 읽습니다.

@Injectable()
export class RolesGuard implements CanActivate {
  constructor(private reflector: Reflector) {}

  canActivate(context: ExecutionContext): boolean {
    const requiredRoles = this.reflector.getAllAndOverride<string[]>('roles', [
      context.getHandler(),
      context.getClass(),
    ]);
    if (!requiredRoles) {
      return true; // 역할 제한이 없으면 접근 허용
    }
    const { user } = context.switchToHttp().getRequest();
    // 실제 로직: user의 roles가 requiredRoles에 포함되는지 확인
    return user.roles.some((role: string) => requiredRoles.includes(role));
  }
}

import { Controller, Get, UseGuards } from '@nestjs/common';
import { AuthGuard } from '../auth/guards/auth.guard';
import { RolesGuard } from '../auth/guards/roles.guard';
import { Roles } from '../common/decorators/roles.decorator';

@Controller('admin')
@UseGuards(AuthGuard, RolesGuard) // AuthGuard 먼저, RolesGuard 나중에 실행
export class AdminController {
  @Get('dashboard')
  @Roles('admin') // 이 메서드는 'admin' 역할만 접근 가능
  getAdminDashboard() {
    return 'Welcome to the admin dashboard!';
  }
}