feat: cookie

Author: lucifer

routes/index.js

@@ -10,41 +10,12 @@ const db = [
   },
 ];
 
+const loginedUsers = Set();
+
 router.get("/", async (ctx) => {
   await ctx.render("index", {
     title: "欢迎来到 91 天学算法~",
   });
 });
 
-router.get("/api/v1/user", async (ctx) => {
-  const code = ctx.query.code;
-  const { access_token } = await fetch(
-    `https://github.com/login/oauth/access_token?code=${code}&client_id=${clientId}&client_secret=${clientSecret}`,
-    {
-      method: "POST",
-      headers: {
-        Accept: "application/json",
-      },
-    }
-  ).then((res) => res.json());
-
-  const user = await fetch("https://api.github.com/user", {
-    headers: {
-      Accept: "application/json",
-      Authorization: `token ${access_token}`,
-    },
-  }).then((res) => res.json());
-  if (db.find((q) => q.login === user.login)) {
-    ctx.body = {
-      ...user,
-      pay: true,
-    };
-  } else {
-    ctx.body = {
-      ...user,
-      pay: false,
-    };
-  }
-});
-
 module.exports = router;

routes/users.js

@@ -1,13 +1,63 @@
-const router = require('koa-router')()
+const router = require("koa-router")();
+const crypto = require("crypto");
 
-router.prefix('/users')
+const secret = process.env.secret;
 
-router.get('/', function (ctx, next) {
-  ctx.body = 'this is a users response!'
-})
+router.prefix("/api/v1/user");
 
-router.get('/bar', function (ctx, next) {
-  ctx.body = 'this is a users/bar response'
-})
+function encrypt(str) {
+  const cipher = crypto.createCipher("aes192", secret);
+  let enc = cipher.update(str, "utf8", "hex"); //编码方式从utf-8转为hex;
+  enc += cipher.final("hex"); //编码方式从转为hex;
+  return enc;
+}
 
-module.exports = router
+function decrypt(str) {
+  const decipher = crypto.createDecipher("aes192", secret);
+  let dec = decipher.update(str, "hex", "utf8"); //编码方式从hex转为utf-8;
+  dec += decipher.final("utf8"); //编码方式从utf-8;
+  return dec;
+}
+
+router.get("/", async (ctx) => {
+  const token = ctx.cookies.get("token");
+
+  if (token) {
+    const duser = decrypt(token);
+    if (db.find((q) => q.login === duser.login)) return duser;
+  }
+  const code = ctx.query.code;
+  const { access_token } = await fetch(
+    `https://github.com/login/oauth/access_token?code=${code}&client_id=${clientId}&client_secret=${clientSecret}`,
+    {
+      method: "POST",
+      headers: {
+        Accept: "application/json",
+      },
+    }
+  ).then((res) => res.json());
+
+  const user = await fetch("https://api.github.com/user", {
+    headers: {
+      Accept: "application/json",
+      Authorization: `token ${access_token}`,
+    },
+  }).then((res) => res.json());
+  if (user.login) {
+    loginedUsers.add(user.login);
+  }
+  if (db.find((q) => q.login === user.login)) {
+    ctx.cookies.set("token", encrypt(JSON.stringify(user)));
+    ctx.body = {
+      ...user,
+      pay: true,
+    };
+  } else {
+    ctx.body = {
+      ...user,
+      pay: false,
+    };
+  }
+});
+
+module.exports = router;