66 lines
1.4 KiB
JavaScript
66 lines
1.4 KiB
JavaScript
|
/**
|
|||
|
|
*
|
|||
|
|
* @authors yutent (yutent@doui.cc)
|
|||
|
|
* @date 2018-07-26 15:50:25
|
|||
|
|
* @version $Id$
|
|||
|
|
*/
|
|||
|
|
const redisStore = require('../module/redis-store')
|
|||
|
|
const nativeStore = require('../module/native-store')
|
|||
|
|
|
|||
|
|
module.exports = function(req, res, next) {
|
|||
|
|
var opt = this.get('session')
|
|||
|
|
var jwt = this.get('jwt')
|
|||
|
|
var cookie = this.ins('cookie')
|
|||
|
|
var session = null
|
|||
|
|
var uuid = Util.sec.uuid()
|
|||
|
|
var ssid = ''
|
|||
|
|
|
|||
|
|
opt.jwt = jwt
|
|||
|
|
|
|||
|
|
if (req.method === 'OPTIONS') {
|
|||
|
|
return next()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (jwt) {
|
|||
|
|
var auth = req.header('authorization')
|
|||
|
|
if (auth) {
|
|||
|
|
ssid = auth.split('.').pop()
|
|||
|
|
uuid = auth
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
ssid = cookie('NODESSID')
|
|||
|
|
// 校验级别为1, 则混入ua
|
|||
|
|
if (opt.level > 0) {
|
|||
|
|
uuid += req.header('user-agent')
|
|||
|
|
}
|
|||
|
|
// 校验级别为2, 则混入ip
|
|||
|
|
if (opt.level > 1) {
|
|||
|
|
uuid += req.ip()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
uuid = Util.sec.sha1(uuid)
|
|||
|
|
|
|||
|
|
if (opt.type === 'redis') {
|
|||
|
|
session = new redisStore(this.__SESSION_STORE__, opt, uuid)
|
|||
|
|
} else {
|
|||
|
|
session = new nativeStore(this.__SESSION_STORE__, opt, uuid)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 启用SESSION
|
|||
|
|
// ssid非法或过期时，需要重写
|
|||
|
|
if (!ssid || ssid !== session.start(ssid)) {
|
|||
|
|
ssid = session.start(ssid)
|
|||
|
|
if (!jwt) {
|
|||
|
|
cookie('NODESSID', ssid, {
|
|||
|
|
httpOnly: true,
|
|||
|
|
expires: opt.ttl,
|
|||
|
|
domain: opt.domain
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.__INSTANCE__.session = session
|
|||
|
|
|
|||
|
|
next()
|
|||
|
|
}
|