Node.js에서 JWT를 이용한 토큰 발급 및 검증 로직 구현하기

import jwt from "jsonwebtoken"; /* jwt를 사용하기 위해 jsonwebtoken import */

const expirationTime = Math.floor(Date.now() / 1000) + 60*60; /* 현재시간 + @, e.g. 60*60 = 1시간 후 만료되도록 설정 */
const payload = { 
    user_id : "sample_id",
    role : "USER",
    exp : expirationTime,
}; /* payload에 담을 객체를 선언합니다. */

const secretKey = "jwt-secret-key"; /* 사용할 secret 키를 선언합니다. */
const token = jwt.sign(payload, secretKey); /* jwt의 sign 메소드를 사용해서 token을 생성합니다. */

import jwt from "jsonwebtoken"; 
import { Router } from 'express';
const router = Router();

// 서버를 구동하기 위한 코드는 생략되었습니다.

router.post("/signin", async (req, res) => {
    // ... 유저와 비밀번호를 검증
    const expirationTime = Math.floor(Date.now() / 1000) + 60*60;
    const payload = { 
        user_id : "sample_id",
        role : "USER",
        exp : expirationTime,
    };

    const secretKey = "jwt-secret-key";
    const token = jwt.sign(payload, secretKey);

    res.setHeader('Authorization', `Bearer ${token}`);
    res.status(200)
        .json({message: "로그인에 성공했습니다."});
});

import { Router } from 'express';
const router = Router();
import login-middleware from "../middlewares/login-middleware.js";

/* 라우터에서 실행 코드가 실행되기 전, 아래 처럼 미들웨어를 선언하여 줍니다. 이렇게 미들웨어로 작성하게 되면 다양한 라우터에서 동일하게 사용할 수 있다는 장점이 있습니다. */
router.get("/users/:email",
    login-middleware,
    (req, res, next) => {
        // ...
    }
);

async function login_required (req, res, next) {
   const { shortId } = req.params || {};

   const userToken = req.headers["authorization"]?.split(" ")[1];

   if (!userToken) {
      res.status(400).json({ message: "토큰이 없습니다."});
      return;
   }

   try {
      const secretKey = process.env.JWT_SECRET_KEY || "jwt-secret-key";
      const jwtDecoded = jwt.verify(userToken, secretKey);
      const currentUser = await User.findOne(
         {shortId: shortId},
         { _id: 1}
      );

      // params에 shortId가 없고 역할이 ADMIN이 아닌 경우,
      // ADMIN이 아니고, params의 shortId는 존재하지만 
      // 토큰의 id를 가진 사용자의 shortId와 일치하지 않는 경우 (=본인 토큰이 아닌 경우)
      // 에러를 반환
      if( jwtDecoded.role !== "ADMIN" && !shortId ) {
         res.status(400).json({ message : "권한이 없습니다."});
         return;
      } else if ( jwtDecoded.role !== "ADMIN" && currentUser._id != jwtDecoded.user_id ) {
         res.status(400).json({ message : "잘못된 토큰입니다."});
         return;
      }
      
      next();
         
   } catch (error) {
      if (error.name === "TokenExpiredError") {
         res.status(400).json({ message: "토큰이 만료되었습니다." });
      } else {
         res.status(400).json({ message: "정상적인 토큰이 아닙니다." });
      }
      return;
   }
}
  
export { login_required };