feat: getDate 逻辑

Author: lucifer

config/index.js

@@ -0,0 +1,21 @@
+const startTime = new Date();
+
+startTime.setMonth(5);
+startTime.setDate(1);
+startTime.setHours(0);
+startTime.setMinutes(0);
+startTime.setSeconds(0);
+
+module.exports = {
+  startTime: startTime.getTime(),
+  secre: process.env.secret,
+  clientId: "c16b80e7b58a5a007157",
+  db: [
+    {
+      login: "azl397985856123",
+    },
+    {
+      login: "Yueqi-19",
+    },
+  ],
+};

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: false,
+    pay: true,
   };
 
   await next();

middleware/passport.js

@@ -1,18 +1,7 @@
 const fetch = require("node-fetch");
 const { encrypt, decrypt } = require("../utils/crypto");
 const { fail } = require("../utils/request");
-
-const secret = process.env.secret;
-
-const clientId = "c16b80e7b58a5a007157";
-const db = [
-  {
-    login: "azl397985856123",
-  },
-  {
-    login: "Yueqi-19",
-  },
-];
+const { secret, db, clientId } = require("../config/index");
 
 module.exports = async function checkAuth(ctx, next) {
   if (!ctx.session) {
@@ -21,6 +10,7 @@ module.exports = async function checkAuth(ctx, next) {
   if (ctx.session.user) {
     await next();
   } else {
+    // 1. 如果有 token ，则说明是之前种植过的，直接解析（如果是别人伪造的则会解析失败）
     const token = ctx.cookies.get("token");
 
     if (token) {
@@ -37,6 +27,7 @@ module.exports = async function checkAuth(ctx, next) {
         }
       }
     }
+    // 2. 如果没有 token，就必须有 code，因此这个时候需要拿 code 去 github 登录，取用户的信息。
     const code = ctx.query.code;
     if (!code) {
       ctx.body = fail({ message: "请先登录~", code: 91 });
@@ -46,6 +37,7 @@ module.exports = async function checkAuth(ctx, next) {
       return;
     }
     try {
+      // 3. 根据 code  获取用户信息
       const { access_token } = await fetch(
         `https://github.com/login/oauth/access_token?code=${code}&client_id=${clientId}&client_secret=${secret}`,
         {
@@ -66,59 +58,28 @@ module.exports = async function checkAuth(ctx, next) {
       // user.login 存在表示登录成功
       if (user.login) {
         // 付费用户
-        if (db.find((q) => q.login === user.login)) {
-          // TODO: 如果不在组织中，自动邀请进 Github 组织
-          // see #1 https://octokit.github.io/rest.js/v18#orgs-check-membership
-          // see #2 https://github.com/octokit/octokit.js
-          // see #3 https://github.com/thundergolfer/automated-github-organization-invites/blob/bb1bb3d42a330716f4dd5c49256245e4bde27489/web_app.rb
-          ctx.session.user = {
-            ...user,
-            pay: true,
-          };
-
-          ctx.cookies.set(
-            "token",
-            encrypt(
-              Buffer.from(
-                JSON.stringify({
-                  ...user,
-                  pay: true,
-                }),
-                "utf8"
-              )
-            ),
-            {
-              httpOnly: false,
-              expires: new Date(24 * 60 * 60 * 1000 + Date.now()),
-            }
-          );
-        } else {
-          ctx.session.user = {
-            ...user,
-            pay: false,
-          };
-
-          ctx.cookies.set(
-            "token",
-            encrypt(
-              Buffer.from(
-                JSON.stringify({
-                  ...user,
-                  pay: false,
-                }),
-                "utf8"
-              )
-            ),
-            {
-              httpOnly: false,
-              expires: new Date(24 * 60 * 60 * 1000 + Date.now()),
-            }
-          );
-        }
+        const u = {
+          ...user,
+          pay: !!db.find((q) => q.login === user.login),
+        };
+        // TODO: 如果不在组织中，自动邀请进 Github 组织
+        // see #1 https://octokit.github.io/rest.js/v18#orgs-check-membership
+        // see #2 https://github.com/octokit/octokit.js
+        // see #3 https://github.com/thundergolfer/automated-github-organization-invites/blob/bb1bb3d42a330716f4dd5c49256245e4bde27489/web_app.rb
+        ctx.session.user = u;
+        ctx.cookies.set(
+          "token",
+          encrypt(Buffer.from(JSON.stringify(u), "utf8")),
+          {
+            httpOnly: false,
+            expires: new Date(24 * 60 * 60 * 1000 + Date.now()), // 一天后过期，后期考虑延长时间
+          }
+        );
       }
 
       await next();
     } catch (err) {
+      // 4. 登录过程中出错，会跳转至此
       ctx.body = fail({ message: "登录失败， code 码已失效~", code: 93 });
     }
   }

routes/dailyProblem.js

@@ -3,9 +3,12 @@ const solutions = require("../static/solution/solutions.json");
 const { decrypt } = require("../utils/crypto");
 
 const { success, fail } = require("../utils/request");
+const { startTime } = require("../config/index");
 
-function getCurrentDay(date) {
-  return 1;
+const MS_PER_DAY = 24 * 60 * 60 * 1000;
+
+function getDay(date) {
+  return ((date - startTime + MS_PER_DAY - 1) / MS_PER_DAY) >> 0;
 }
 
 router.get("/api/v1/daily-problem", async (ctx) => {
@@ -14,9 +17,8 @@ router.get("/api/v1/daily-problem", async (ctx) => {
   // 3. 根据 Day 几 计算出具体返回哪一个题目
   // ！！注意： 如果用户指定的时间大于今天，则返回”题目不存在，仅支持查询历史每日一题“
 
-  const date = ctx.query.date; // 用户指定的实际
-
-  if (date) {
+  const date = getDay(ctx.query.date || new Date().getTime()); // 用户指定的实际
+  if (date === 2) {
     ctx.body = success({
       day: 2,
       title: "821. 字符的最短距离",
@@ -38,7 +40,7 @@ router.get("/api/v1/daily-problem", async (ctx) => {
       
       `,
     });
-  } else {
+  } else if (date <= 1) {
     ctx.body = success({
       day: 1,
       title: "66. 加一",
@@ -67,6 +69,10 @@ router.get("/api/v1/daily-problem", async (ctx) => {
 解释: 输入数组表示数字 4321。
     `,
     });
+  } else {
+    ctx.body = fail({
+      message: "当前暂时没有每日一题，请联系当前讲师进行处理~",
+    });
   }
 });
 
@@ -75,7 +81,7 @@ router.get("/api/v1/daily-problem/solution", async (ctx) => {
 
   // ！！注意： 如果用户指定的 day 大于今天对应的day，则返回”题解不存在，仅支持查询历史官方题解“
 
-  const day = ctx.query.day || getCurrentDay(); // 用户指定的实际第几天（注意这里的 day 是数字，含义为第 day 天）
+  const day = ctx.query.day || getDay(new Date().getTime()); // 用户指定的实际第几天（注意这里的 day 是数字，含义为第 day 天）
   if (day in solutions) {
     ctx.body = success({
       content: decrypt(solutions[day].content),