80 lines
1.6 KiB
JavaScript
80 lines
1.6 KiB
JavaScript
|
/**
|
||
|
|
* 会话模块
|
||
|
|
* @author yutent<yutent.io@gmail.com>
|
||
|
|
* @date 2020/09/24 11:18:23
|
||
|
|
*/
|
||
|
|
|
||
|
|
import { uuid, sha1 } from 'crypto.js'
|
||
|
|
|
||
|
|
import RedisStore from './lib/redis-store.js'
|
||
|
|
import MemStore from './lib/mem-store.js'
|
||
|
|
|
||
|
|
// 会话储存器
|
||
|
|
export const sessionStore = {
|
||
|
|
name: 'session',
|
||
|
|
install() {
|
||
|
|
var session = this.get('session')
|
||
|
|
|
||
|
|
// 这里只创建session的存储器, 而初始化操作在中间件中进行
|
||
|
|
if (session.type === 'redis') {
|
||
|
|
return new RedisStore(session)
|
||
|
|
} else {
|
||
|
|
return new MemStore(session)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// 会话中间件
|
||
|
|
export function sessionWare(req, res, next) {
|
||
|
|
var opt = this.get('session')
|
||
|
|
var jwt = this.get('jwt')
|
||
|
|
var ssid
|
||
|
|
|
||
|
|
// options请求不处理会话
|
||
|
|
if (req.method === 'OPTIONS') {
|
||
|
|
return next()
|
||
|
|
}
|
||
|
|
|
||
|
|
// jwt模式的校验不在这里处理
|
||
|
|
if (jwt) {
|
||
|
|
var auth = req.header('authorization')
|
||
|
|
if (auth) {
|
||
|
|
ssid = auth.split('.').pop()
|
||
|
|
this.$$session.start(ssid)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
var cache = req.cookie('NODESSID')
|
||
|
|
var deviceID = ''
|
||
|
|
|
||
|
|
// 校验UA
|
||
|
|
if (opt.level & 2) {
|
||
|
|
deviceID += req.header('user-agent')
|
||
|
|
}
|
||
|
|
|
||
|
|
// 校验IP
|
||
|
|
if (opt.level & 4) {
|
||
|
|
deviceID += req.ip()
|
||
|
|
}
|
||
|
|
|
||
|
|
if (deviceID) {
|
||
|
|
deviceID = sha1(deviceID)
|
||
|
|
|
||
|
|
// ssid 最后16位是指纹
|
||
|
|
if (cache) {
|
||
|
|
if (cache.slice(-16) === deviceID.slice(-16)) {
|
||
|
|
ssid = cache
|
||
|
|
} else {
|
||
|
|
ssid = uuid('') + deviceID.slice(-16)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
ssid = cache || sha1(uuid())
|
||
|
|
}
|
||
|
|
|
||
|
|
res.cookie('NODESSID', ssid)
|
||
|
|
this.$$session.start(ssid)
|
||
|
|
}
|
||
|
|
|
||
|
|
next()
|
||
|
|
}
|