feat: 付费功能完成

Author: lucifer

app.js

@@ -8,6 +8,7 @@ const bodyparser = require("koa-bodyparser");
 const logger = require("koa-logger");
 
 const passport = require("./middleware/passport");
+const pay = require("./middleware/pay");
 const index = require("./routes/index");
 const users = require("./routes/users");
 const dailyProblem = require("./routes/dailyProblem");
@@ -31,6 +32,7 @@ if (process.env.NODE_ENV === "development") {
 } else {
   app.use(passport);
 }
+app.use(pay({ whitelist: ["/api/v1/user", "/api/v1/logout"] }));
 app.use(require("koa-static")(__dirname + "/public"));
 
 app.use(

middleware/mockUserInfo.js

@@ -6,7 +6,7 @@ module.exports = async function checkAuth(ctx, next) {
     login: "azl397985856",
     avatar_url: "https://avatars.githubusercontent.com/u/12479470?v=4",
     name: "lucifer",
-    pay: true,
+    pay: false,
   };
 
   await next();

middleware/passport.js

@@ -33,13 +33,7 @@ module.exports = async function checkAuth(ctx, next) {
             ctx.session.user = duser;
             await next();
             return;
-          } else {
-            ctx.body = fail({
-              message: "本页内容需付费才能查看，请和 lucifer 进行联系~",
-              code: 93,
-            });
           }
-          return;
         } catch (err) {
           console.log("token 解析失败:", err);
           return;

middleware/pay.js

@@ -0,0 +1,20 @@
+const { fail } = require("../utils/request");
+//
+module.exports = function ({ whitelist }) {
+  return async function checkAuth(ctx, next) {
+    if (whitelist.includes(ctx.path)) await next();
+    else {
+      if (ctx?.session?.user) {
+        if (ctx.session.user.pay) {
+          await next();
+        } else {
+          ctx.body = fail({
+            message: "本页内容需付费才能查看，请和 lucifer 进行联系~",
+            code: 93,
+          });
+          return;
+        }
+      }
+    }
+  };
+};