feat:添加统一异常的token刷新机制

Author: pedro

lib/exception.ts

@@ -261,3 +261,24 @@ export class MethodNotAllowed extends HttpException {
     }
   }
 }
+
+export class RefreshException extends HttpException {
+  public code = 401;
+  public msg = "refresh token 获取失败";
+  public errorCode = 10100;
+
+  constructor(ex?: Exception) {
+    super();
+    if (ex && ex.code) {
+      assert(isInteger(ex.code));
+      this.code = ex.code;
+    }
+    if (ex && ex.msg) {
+      this.msg = ex.msg;
+    }
+    if (ex && ex.errorCode) {
+      assert(isInteger(ex.errorCode));
+      this.errorCode = ex.errorCode;
+    }
+  }
+}

lib/jwt.ts

@@ -3,7 +3,8 @@ import {
   ExpiredTokenException,
   InvalidTokenException,
   AuthFailed,
-  NotFound
+  NotFound,
+  RefreshException
 } from "./exception";
 import Application from "koa";
 import { RouterContext } from "koa-router";
@@ -180,6 +181,26 @@ async function refreshTokenRequired(
   }
 }
 
+/**
+ * 守卫函数，用户刷新令牌，统一异常
+ */
+async function refreshTokenRequiredWithUnifyException(
+  ctx: RouterContext,
+  next: () => Promise<any>
+) {
+  // 添加access 和 refresh 的标识位
+  if (ctx.request.method !== "OPTIONS") {
+    try {
+      await parseHeader(ctx, TokenType.REFRESH);
+    } catch (error) {
+      throw new RefreshException();
+    }
+    await next();
+  } else {
+    await next();
+  }
+}
+
 /**
  * 守卫函数，用于权限组鉴权
  */

package.json

@@ -1,6 +1,6 @@
 {
   "name": "lin-mizar",
-  "version": "0.0.1-alpha.9",
+  "version": "0.0.1-alpha.10",
   "description": "The core library of Lin CMS",
   "main": "lin/index.js",
   "scripts": {