session_auth.js 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. 'use strict';
  2. // 加密类
  3. const crypto = require('crypto');
  4. module.exports = options => {
  5. /**
  6. * session判断中间件
  7. *
  8. * @param {function} next - 中间件继续执行的方法
  9. * @return {void}
  10. */
  11. return function* sessionAuth(next) {
  12. try {
  13. // 判断session
  14. const sessionUser = this.session.sessionUser;
  15. if (sessionUser === undefined) {
  16. throw '不存在session';
  17. }
  18. // 校验session
  19. if (sessionUser.account === undefined || sessionUser.loginTime === undefined) {
  20. throw '用户数据不完整';
  21. }
  22. // 校验session
  23. const sessionToken = crypto.createHmac('sha1', sessionUser.loginTime + '')
  24. .update(sessionUser.account).digest().toString('base64');
  25. if (sessionToken !== sessionUser.sessionToken) {
  26. throw 'session数据错误';
  27. }
  28. // 同步消息
  29. yield this.service.notify.syncNotifyData();
  30. } catch (error) {
  31. return this.redirect('/');
  32. }
  33. yield next;
  34. };
  35. };