调整session的调用方式;增加index.d.ts
parent
ebccbf270f
commit
4673b541a6
|
@ -0,0 +1,38 @@
|
||||||
|
/**
|
||||||
|
* {}
|
||||||
|
* @author yutent<yutent.io@gmail.com>
|
||||||
|
* @date 2025/01/03 10:39:13
|
||||||
|
*/
|
||||||
|
|
||||||
|
declare interface RedisConfig {
|
||||||
|
host: string
|
||||||
|
port: number
|
||||||
|
db: number
|
||||||
|
}
|
||||||
|
|
||||||
|
declare interface SessionConfig {
|
||||||
|
ttl?: number
|
||||||
|
domain?: string
|
||||||
|
level?: 0 | 2 | 4 | 6
|
||||||
|
secret: string
|
||||||
|
db: RedisConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
declare class Store {
|
||||||
|
update(ssid: string): void
|
||||||
|
|
||||||
|
get(ssid: string, key?: string): Promise<string | object>
|
||||||
|
|
||||||
|
set(ssid: string, key: string, val: string): void
|
||||||
|
|
||||||
|
unset(ssid: string, key: string): void
|
||||||
|
|
||||||
|
clear(ssid: string): void
|
||||||
|
}
|
||||||
|
|
||||||
|
declare interface Session {
|
||||||
|
name: 'session'
|
||||||
|
install(conf?: SessionConfig): Store
|
||||||
|
}
|
||||||
|
|
||||||
|
export function createSession(): Session
|
125
index.js
125
index.js
|
@ -6,11 +6,11 @@
|
||||||
|
|
||||||
import { uuid, sha1 } from 'crypto.js'
|
import { uuid, sha1 } from 'crypto.js'
|
||||||
|
|
||||||
import RedisStore from './lib/redis-store.js'
|
import Store from './lib/redis-store.js'
|
||||||
|
|
||||||
const DEFAULT_CONFIG = {
|
const DEFAULT_CONFIG = {
|
||||||
ttl: 3600 * 24 * 7,
|
ttl: 3600 * 24 * 7,
|
||||||
domain: '', // NODESSID域, 默认等于domain
|
domain: '', // NODESSID域
|
||||||
level: 0, // 校验级别, 0: 不校验客户端, 2: ua, 4: ip, 6: ua + ip
|
level: 0, // 校验级别, 0: 不校验客户端, 2: ua, 4: ip, 6: ua + ip
|
||||||
secret: 'it_is_secret_key', // jwt密钥, 使用时请修改
|
secret: 'it_is_secret_key', // jwt密钥, 使用时请修改
|
||||||
db: {
|
db: {
|
||||||
|
@ -20,71 +20,74 @@ const DEFAULT_CONFIG = {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 会话安装包
|
function sessionMiddleware(req, res, next) {
|
||||||
export const SessionModule = {
|
let opt = this.get('session')
|
||||||
name: 'session',
|
let cache = req.cookie('NODESSID')
|
||||||
install(conf = {}) {
|
let deviceID = ''
|
||||||
if (!conf.secret) {
|
let ssid
|
||||||
console.warn(
|
|
||||||
new Error(
|
// options请求不处理会话
|
||||||
'You must set a `secret` key for session, or it will use the default key.'
|
if (req.method === 'OPTIONS') {
|
||||||
)
|
return next()
|
||||||
)
|
|
||||||
}
|
|
||||||
let session = Object.assign({}, DEFAULT_CONFIG, conf)
|
|
||||||
this.set({ session })
|
|
||||||
// 这里只创建session的存储器, 而初始化操作在中间件中进行
|
|
||||||
return new RedisStore(session)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 校验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, {
|
||||||
|
maxAge: opt.ttl,
|
||||||
|
httpOnly: true,
|
||||||
|
domain: opt.domain
|
||||||
|
})
|
||||||
|
// 缓存ssid到req上
|
||||||
|
req.ssid = ssid
|
||||||
|
this.$$session.update(ssid)
|
||||||
|
|
||||||
|
next()
|
||||||
}
|
}
|
||||||
|
|
||||||
// 会话中间件
|
// 会话中间件
|
||||||
export function createSession() {
|
export function createSession() {
|
||||||
return function (req, res, next) {
|
return {
|
||||||
let opt = this.get('session')
|
name: 'session',
|
||||||
let cache = req.cookie('NODESSID')
|
install(conf = {}) {
|
||||||
let deviceID = ''
|
if (!conf.secret) {
|
||||||
let ssid
|
console.warn(
|
||||||
|
new Error(
|
||||||
// options请求不处理会话
|
'You must set a `secret` key for session, or it will use the default key.'
|
||||||
if (req.method === 'OPTIONS') {
|
)
|
||||||
return next()
|
)
|
||||||
}
|
|
||||||
|
|
||||||
// 校验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 {
|
let session = Object.assign({}, DEFAULT_CONFIG, conf)
|
||||||
ssid = cache || sha1(uuid())
|
this.set({ session })
|
||||||
|
|
||||||
|
// 注册中间件
|
||||||
|
this.use(sessionMiddleware)
|
||||||
|
|
||||||
|
// 这里只创建session的存储器, 而初始化操作在中间件中进行
|
||||||
|
return new Store(session)
|
||||||
}
|
}
|
||||||
|
|
||||||
res.cookie('NODESSID', ssid, {
|
|
||||||
maxAge: opt.ttl,
|
|
||||||
httpOnly: true,
|
|
||||||
domain: opt.domain
|
|
||||||
})
|
|
||||||
// 缓存ssid到req上
|
|
||||||
req.ssid = ssid
|
|
||||||
this.$$session.update(ssid)
|
|
||||||
|
|
||||||
next()
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,8 +6,7 @@
|
||||||
import 'es.shim'
|
import 'es.shim'
|
||||||
import Ioredis from 'ioredis'
|
import Ioredis from 'ioredis'
|
||||||
|
|
||||||
export default class Session {
|
export default class Store {
|
||||||
|
|
||||||
#store = null
|
#store = null
|
||||||
#ttl = 60
|
#ttl = 60
|
||||||
|
|
||||||
|
|
|
@ -1,10 +1,11 @@
|
||||||
{
|
{
|
||||||
"name": "@gm5/session",
|
"name": "@gm5/session",
|
||||||
"version": "2.0.0",
|
"version": "3.0.0",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"description": "会话中间件。",
|
"description": "会话中间件。",
|
||||||
"main": "index.js",
|
|
||||||
"author": "yutent <yutent.io@gmail.com>",
|
"author": "yutent <yutent.io@gmail.com>",
|
||||||
|
"main": "index.js",
|
||||||
|
"types": "index.d.ts",
|
||||||
"keywords": [
|
"keywords": [
|
||||||
"fivejs",
|
"fivejs",
|
||||||
"gm5",
|
"gm5",
|
||||||
|
|
Loading…
Reference in New Issue