ホームページ >ウェブフロントエンド >jsチュートリアル >NestJS でデコレータとミドルウェアを作成して使用する方法

NestJS でデコレータとミドルウェアを作成して使用する方法

Mary-Kate Olsenオリジナル: 2025-01-13 06:05:43205ブラウズ

デコレータとミドルウェアは NestJS の基礎であり、アプローチの仕方に応じて、作業を驚くほど簡単にすることも、少し圧倒することもできるツールです。

今日は、作業を軽く簡単にしながら、ユーザー認証用のカスタムデコレーターとミドルウェアを作成する手順を見てみましょう。コーヒーを飲んで、さあ始めましょう!

1. インターフェース

まず、ユーザーオブジェクトのインターフェイスを定義しましょう。

これにより、型の安全性が確保され、IDE が快適に保たれます (そして、幸せな IDE を好まない人はいないでしょうか?)。

export interface IUser {
  id: string;
  name: string;
  primaryEmail: string;
  phoneNumber: string | null;
  countryCode: string | null;
  dob: Date | null;
  createdAt: Date;
  updatedAt?: Date;
  deletedAt?: Date | null;
}

2. カスタムデコレータの作成

カスタムデコレータは、NestJS アプリケーションのクールな子供のようなものです。

ここでは、リクエストオブジェクトからユーザーメタデータを取得するものを作成しています。

import { createParamDecorator, ExecutionContext } from '@nestjs/common';
import { IUser } from '../interface/user.interface';

export const UserMetadata = createParamDecorator(
  (_data: unknown, ctx: ExecutionContext) => {
    const request = ctx.switchToHttp().getRequest();
    return request.user as IUser;
  },
);

それだけです！このデコレータを使用して、コントローラメソッドでユーザー情報を直接取得できるようになりました。

How to Create and Use a Decorator and Middleware in NestJS

3. 認証ミドルウェアの作成

次に、仮想バウンサーのようにエンドポイントを保護する AuthGuard を作成しましょう。

import {
  CanActivate,
  ExecutionContext,
  ForbiddenException,
  Injectable
} from '@nestjs/common';
import { Reflector } from '@nestjs/core';
import { verify } from 'jsonwebtoken';
import { Observable } from 'rxjs';
import { IS_PUBLIC_KEY } from '../constant/core';
import { IUser } from '../interface/user.interface';

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

  canActivate(
    context: ExecutionContext,
  ): boolean | Promise<boolean> | Observable<boolean> {
    const isPublic = this.reflector.getAllAndOverride<boolean>(IS_PUBLIC_KEY, [
      context.getHandler(),
      context.getClass(),
    ]);
    if (isPublic) {
      return true;
    }

    const request = context.switchToHttp().getRequest();
    const headers = request.headers;

    const token = (headers['authorization'] || '').split(' ')[1];
    if (!token) {
      throw new ForbiddenException('Not Authenticated');
    }

    const jwtOpts = {
      expiresIn: '1h', // Replace with env vars in real use
      audience: 'your-audience',
      algorithm: 'HS256',
      issuer: 'your-issuer',
    };

    try {
      const decoded = verify(token, "my-jwt-secret-token", {
        audience: jwtOpts.audience,
        issuer: jwtOpts.issuer,
      }) as { user: IUser };

      request.user = decoded.user;
      return true;
    } catch (err) {
      throw new ForbiddenException('Session Expired or Invalid');
    }
  }
}

4. パブリックルートと内部ルートのメタデータの追加

一部のルートはパブリックである必要があり (ログインなど)、一部は内部である可能性があります。

そのために 2 つの単純なデコレータを追加しましょう。

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

export const IS_PUBLIC_KEY = 'isPublic';
export const IS_INTERNAL = 'isInternal';

export const Public = () => SetMetadata(IS_PUBLIC_KEY, true);
export const Internal = () => SetMetadata(IS_INTERNAL, true);

5. コントローラーでの使用

最後に、これらすべてをコントローラーで使用する方法を示します。

import { Controller, Get, UseGuards } from '@nestjs/common';
import { UserMetadata } from '../decorators/user.decorator';
import { AuthGuard } from '../guards/auth.guard';
import { Public } from '../decorators/public.decorator';

@Controller('users')
export class UserController {

  @Public()
  @Get('login')
  login() {
    return { message: 'Login endpoint (public)' };
  }

  @UseGuards(AuthGuard)
  @Get('profile')
  getProfile(@UserMetadata() user: IUser) {
    return {
      message: 'User Profile',
      user,
    };
  }
}