From 2e2d2b6c3ebee1b41a89d7ae6e28a0d5a7a83504 Mon Sep 17 00:00:00 2001
From: lik <likeagle@163.com>
Date: Fri, 4 Dec 2020 14:31:54 +0800
Subject: [PATCH] =?UTF-8?q?temp:=20=E4=B8=B4=E6=97=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 auth/index.js | 34 +++++++++++++++++++++++-----------
 1 file changed, 23 insertions(+), 11 deletions(-)

diff --git a/auth/index.js b/auth/index.js
index e2cf101..a8ce697 100644
--- a/auth/index.js
+++ b/auth/index.js
@@ -3,11 +3,12 @@
 const crypto = require('crypto');
 const Redis = require('ioredis');
 
-function AuthToken(redisdb) {
+function AuthToken(redisdb, expiresSeconds) {
   this.tokenDB = redisdb;
+  this.defaultExpiresSeconds = expiresSeconds;
 }
 
-AuthToken.prototype.genToken = async function (userData, userkey, expiresSeconds) {
+AuthToken.prototype.genToken = async function ({ userData, userapp }) {
   // 生成系统内部的user token
   let hash = crypto.createHash('md5');
   hash.update(JSON.stringify(userData) + Date() + Math.random());
@@ -16,14 +17,29 @@ AuthToken.prototype.genToken = async function (userData, userkey, expiresSeconds
   // 缓存到redis
   let tokenData = {
     userData: userData,
-    userkey: userkey,
-    expires: { ttl: expiresSeconds, ts: Math.floor(Date.now() / 1000) }
+    apps: {},
+    createts: Math.floor(Date.now() / 1000),
+    ttl: this.defaultExpiresSeconds
   };
-  await this.tokenDB.set(userToken, JSON.stringify(tokenData), 'EX', expiresSeconds);
+  tokenData.apps[userapp.appid] = userapp;
+
+  await this.tokenDB.set(userToken, JSON.stringify(tokenData), 'EX', this.defaultExpiresSeconds);
 
   return userToken;
 };
 
+AuthToken.prototype.update = async function (userToken, { userData, userapp }) {
+  let tokenData = await this.tokenDB.get(userToken).then(function (data) {
+    return JSON.parse(data);
+  });
+
+  if (tokenData) {
+    if (userData) tokenData.userData = userData;
+    if (userapp) tokenData.apps[userapp.appid] = userapp;
+    await this.tokenDB.set(userToken, JSON.stringify(tokenData), 'EX', this.defaultExpiresSeconds);
+  }
+};
+
 AuthToken.prototype.delToken = async function (userToken) {
   let tokenData = await this.tokenDB.get(userToken).then(function (data) {
     return JSON.parse(data);
@@ -34,7 +50,7 @@ AuthToken.prototype.delToken = async function (userToken) {
   }
 };
 
-AuthToken.prototype.checkToken = async function (userToken, userKey, checkKey = false, updateExpire = true) {
+AuthToken.prototype.checkToken = async function (userToken, updateExpire = true) {
   let tokenData = await this.tokenDB.get(userToken).then(function (data) {
     return JSON.parse(data);
   });
@@ -44,10 +60,6 @@ AuthToken.prototype.checkToken = async function (userToken, userKey, checkKey =
     return null;
   }
 
-  if (checkKey && userKey != tokenData.userkey) {
-    return null;
-  }
-
   if (updateExpire) {
     tokenData.expires.ts = Math.floor(Date.now() / 1000);
     this.tokenDB.set(userToken, JSON.stringify(tokenData), 'EX', tokenData.expires.ttl);
@@ -88,7 +100,7 @@ AuthToken.prototype.checkTokenKoaRequest = async function (ctx, next) {
     };
     return;
   }
-  
+
   ctx.userData = tokenData.userData;
 
   return next();