public/npm-package
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

temp: 临时

Browse Source
This commit is contained in:
lik 2020-12-04 14:31:54 +08:00
parent 4e922091e6
commit 2e2d2b6c3e
1 changed files with 23 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
auth/index.js
View File

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