feat: 多环境支持

Author: lucifer

Diff for: app.js

@@ -11,6 +11,7 @@ const passport = require("./middleware/passport");
 const index = require("./routes/index");
 const users = require("./routes/users");
 const lectures = require("./routes/lectures");
+const mockUserInfo = require("./middleware/mockUserInfo");
 
 // error handler
 onerror(app);
@@ -24,7 +25,11 @@ app.use(
 app.use(cors({ credentials: true }));
 app.use(json());
 app.use(logger());
-app.use(passport);
+if (process.env.NODE_ENV === "development") {
+  app.use(mockUserInfo);
+} else {
+  app.use(passport);
+}
 app.use(require("koa-static")(__dirname + "/public"));
 
 app.use(

Diff for: middleware/mockUserInfo.js

@@ -0,0 +1,13 @@
+module.exports = async function checkAuth(ctx, next) {
+  if (!ctx.session) {
+    ctx.session = {};
+  }
+  ctx.session.user = {
+    login: "azl397985856",
+    avatar_url: "https://avatars.githubusercontent.com/u/12479470?v=4",
+    name: "lucifer",
+    pay: true,
+  };
+
+  await next();
+};

Diff for: middleware/passport.js

@@ -14,7 +14,10 @@ module.exports = async function checkAuth(ctx, next) {
   if (!ctx.session) {
     ctx.session = {};
   }
-  if (ctx.session.user) next();
+  if (ctx.session.user) {
+    await next();
+    return;
+  }
   const token = ctx.cookies.get("token");
 
   if (token) {
@@ -79,5 +82,5 @@ module.exports = async function checkAuth(ctx, next) {
     };
   }
 
-  next();
+  await next();
 };

Diff for: package.json

@@ -3,8 +3,8 @@
   "version": "0.1.0",
   "private": true,
   "scripts": {
-    "start": "node bin/www",
-    "dev": "./node_modules/.bin/nodemon bin/www",
+    "start": "NODE_ENV=production node bin/www",
+    "dev": "NODE_ENV=development ./node_modules/.bin/nodemon bin/www",
     "prd": "pm2 start bin/www",
     "test": "echo \"Error: no test specified\" && exit 1"
   },

Diff for: routes/lectures.js

@@ -1,6 +1,7 @@
 const router = require("koa-router")();
 const lectures = require("../static/lectures/lectures.json");
 const { decrypt } = require("../utils/crypto");
+const { success, fail } = require("../utils/request");
 
 router.get("/api/v1/lectures/basic", async (ctx) => {
   ctx.body = {
@@ -24,11 +25,14 @@ router.get("/api/v1/lectures/basic", async (ctx) => {
 router.get("/api/v1/lectures/basic/:id", async (ctx) => {
   const id = ctx.params.id;
   const lecture = lectures[id];
-
-  ctx.body = {
-    ...lecture,
-    content: decrypt(lectures[id].content),
-  };
+  if (!lecture) {
+    ctx.body = fail({ message: "讲义不存在" });
+  } else {
+    ctx.body = success({
+      ...lecture,
+      content: decrypt(lecture.content),
+    });
+  }
 });
 
 module.exports = router;

Diff for: routes/users.js

@@ -1,16 +1,24 @@
 const router = require("koa-router")();
 
+const { success, fail } = require("../utils/request");
+
 router.get("/api/v1/user", async (ctx) => {
   if (ctx.session && ctx.session.user) {
-    ctx.body = ctx.session.body;
+    ctx.body = success(ctx.session.user);
   } else {
-    ctx.body = {
-      success: false,
+    ctx.body = fail({
       code: 91,
-      data: null,
       message: "您还没有登录，请先登录~",
-    };
+    });
+  }
+});
+
+router.get("/api/v1/user/logout", async (ctx) => {
+  if (ctx.session && ctx.session.user) {
+    ctx.session = {};
   }
+  ctx.cookies.set("token", "");
+  ctx.body = success("ok");
 });
 
 module.exports = router;

Diff for: utils/request.js

@@ -0,0 +1,16 @@
+module.exports = {
+  success(data) {
+    return {
+      success: true,
+      data,
+    };
+  },
+  fail({ message, code = 10001 }) {
+    return {
+      success: false,
+      data: null,
+      message,
+      code,
+    };
+  },
+};