koa 使用 jwt

参考文章

JSON Web Token 入门教程

koa-jwt 使用详解

npm koa-jwt

npm jsonwebtoken

后端

老样子, 先新建一个空的koa项目.然后安装依赖

koa
koa-router
koa-bodyparser
koa2-cors
jsonwebtoken
koa-jwt

直接看完整的代码

const Koa = require("koa");
const KoaRouter = require("koa-router");
const bodyParser = require("koa-bodyparser");
const cors = require("koa2-cors");

const jwt = require("jsonwebtoken");
const koajwt = require("koa-jwt");

const app = new Koa();
const SECRET = "your-jwt-secret"; // 秘钥

app.use(cors());
app.use(bodyParser());

const router = new KoaRouter();

router.post("/api/user/login", async (ctx) => {
  const { username } = ctx.request.body;
  // 根据用户名生成 token (其实应该还有 password, 这里忽略了)
  const token = jwt.sign({ username }, SECRET);
  // 然后返回给前端
  ctx.body = { code: 20000, msg: "login success", token };
});

router.post("/api/user/info", async (ctx) => {
  // 从 'ctx.user' 中获取内容
  ctx.body = { code: 20000, info: ctx.state.user };
});

// 自定义的权限错误处理, 当然这是特殊的业务需求
async function customAuthorizationCatcher(ctx, next) {
  try {
    await next();
  } catch (err) {
    // 由 koa-jwt 抛出的错误
    if (err.status === 401) {
      // 强制修改网络状态, 在接口中返回业务类型状态码(根据需求)
      ctx.status = 200;
      ctx.body = { code: 40100, msg: "无效 token" };
    } else {
      throw err;
    }
  }
}

app.use(customAuthorizationCatcher); // 这个中间件要放在'koa-jwt'的前面
// koa-jwt 中间件会获取前端请求中的token,进行检验
app.use(
  koajwt({
    secret: SECRET,
    // key: "user", 默认把token解析的内容保存到 'ctx.user' 中
  }).unless({ path: ["/api/user/login"] })
);
app.use(router.routes());

app.listen(8000);

前端

依然是使用react + axios

直接看组件

import React, { useCallback } from "react";
import axios from "axios";

// 后端接口地址
const SERVER_ORIGIN = "http://localhost:8000";

const App = () => {
  const login = useCallback(() => {
    const f = async () => {
      try {
        const res = await axios.post(`${SERVER_ORIGIN}/api/user/login`, {
          username: "my-name",
        });

        console.log("res", res.data);
        // 把 token 存起来
        window.localStorage.setItem("token", res.data.token);
      } catch (err) {
        console.error(err);
      }
    };

    f();
  }, []);

  const getInfo = useCallback(() => {
    const f = async () => {
      try {
        const token = window.localStorage.getItem("token");
        const res = await axios.post(`${SERVER_ORIGIN}/api/user/info`, null, {
          // 请求时, 在 header 中添加 token
          headers: { Authorization: `Bearer ${token}` },
        });
        console.log("info", res.data);
      } catch (err) {
        console.error(err);
      }
    };
    f();
  }, []);

  return (
    <>
      <h1>koa jwt</h1>
      <button onClick={login}>login</button>
      <button onClick={getInfo}>get user info</button>
    </>
  );
};

export default App;

坑

这里需要说明一下前端请求的坑, 为什么 header添加的是Authorization, 而且还要加Bearer这个单词???

因为后端使用的是koa-jwt插件, 我们来看看插件的源码

// /lib/resolvers/auth-header.js

module.exports = function resolveAuthorizationHeader(ctx, opts) {
  if (!ctx.header || !ctx.header.authorization) {
    return;
  }

  const parts = ctx.header.authorization.split(" ");

  if (parts.length === 2) {
    const scheme = parts[0];
    const credentials = parts[1];

    if (/^Bearer$/i.test(scheme)) {
      return credentials;
    }
  }
  if (!opts.passthrough) {
    ctx.throw(
      401,
      'Bad Authorization header format. Format is "Authorization: Bearer <token>"'
    );
  }
};

ctx.header.authorization 规定了header使用的字段
/^Bearer$/i这个正则规定了token前面要添加的单词
split(" ") 规定了 Bearer这个单词和token中还必须有一个空格