mm_session 1.5.0 → 1.5.2

package/test.js

@@ -0,0 +1,190 @@
+const { Session, Store } = require('./index.js');
+
+String.prototype.aes_encode = function(key) {
+  // 简单的AES编码模拟
+  return Buffer.from(this).toString('base64') + '_' + key.substring(0, 8);
+};
+
+// 模拟Koa应用
+class MockKoaApp {
+  constructor() {
+    this.middlewares = [];
+  }
+
+  use(middleware) {
+    this.middlewares.push(middleware);
+  }
+
+  async handleRequest(ctx) {
+    // 执行中间件链
+    const executeMiddleware = async (index) => {
+      if (index < this.middlewares.length) {
+        const middleware = this.middlewares[index];
+        await middleware(ctx, () => executeMiddleware(index + 1));
+      }
+    };
+    
+    await executeMiddleware(0);
+  }
+}
+
+// 模拟Koa的上下文对象
+class MockContext {
+  constructor() {
+    this.cookies = new MockCookies();
+    this.headers = {};
+    this.ip = '127.0.0.1';
+    this.request = {
+      header: {
+        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+      }
+    };
+    this.session = null;
+  }
+}
+
+// 模拟Koa的cookies对象
+class MockCookies {
+  constructor() {
+    this._cookies = new Map();
+  }
+
+  get(key, options) {
+    return this._cookies.get(key);
+  }
+
+  set(key, value, options) {
+    if (value === null) {
+      this._cookies.delete(key);
+    } else {
+      this._cookies.set(key, value);
+    }
+  }
+}
+
+// 测试session模块
+async function testSession() {
+  console.log('开始测试session模块...\n');
+
+  try {
+    // 测试1: 创建Session实例
+    console.log('测试1: 创建Session实例');
+    const session = new Session({
+      key: 'test_session',
+      max_age: 3600
+    });
+    console.log('✅ Session实例创建成功\n');
+
+    // 测试2: 创建Koa应用并注册中间件
+    console.log('测试2: 创建Koa应用并注册中间件');
+    const app = new MockKoaApp();
+    
+    // 使用Koa标准方式注册session中间件
+    app.use(session.middleware());
+    
+    // 添加一个业务中间件来测试session功能
+    app.use(async (ctx, next) => {
+      // 测试session数据操作
+      if (!ctx.session.user_id) {
+        ctx.session.user_id = 123;
+        ctx.session.username = 'test_user';
+      }
+      
+      await next();
+    });
+    
+    console.log('✅ Koa应用创建成功，中间件注册完成\n');
+
+    // 测试3: 测试新session创建
+    console.log('测试3: 测试新session创建');
+    const ctx1 = new MockContext();
+    
+    await app.handleRequest(ctx1);
+    
+    // 等待session保存完成（模拟异步保存过程）
+    await new Promise(resolve => setTimeout(resolve, 50));
+    
+    if (ctx1.session) {
+      console.log('✅ 新session创建成功');
+      console.log('   session对象:', typeof ctx1.session);
+      console.log('   user_id:', ctx1.session.user_id);
+      console.log('   username:', ctx1.session.username);
+    } else {
+      throw new Error('session创建失败');
+    }
+
+    // 测试4: 测试session保存和读取
+    console.log('\n测试4: 测试session保存和读取');
+    
+    // 检查cookie是否设置
+    const sessionCookie = ctx1.cookies.get('test_session');
+    if (sessionCookie) {
+      console.log('✅ session保存成功，cookie已设置');
+      console.log('   session ID:', sessionCookie);
+    } else {
+      console.log('⚠️  session保存但cookie未设置');
+    }
+
+    // 测试5: 测试session读取（模拟有cookie的情况）
+    console.log('\n测试5: 测试session读取');
+    const ctx2 = new MockContext();
+    
+    // 设置cookie模拟已有session
+    if (sessionCookie) {
+      ctx2.cookies.set('test_session', sessionCookie);
+    }
+    
+    await app.handleRequest(ctx2);
+    
+    if (ctx2.session) {
+      console.log('✅ session读取成功');
+      console.log('   session对象存在');
+      console.log('   user_id:', ctx2.session.user_id);
+      console.log('   username:', ctx2.session.username);
+    } else {
+      console.log('⚠️  session读取失败（可能是新session）');
+    }
+
+    // 测试6: 测试Store类
+    console.log('\n测试6: 测试Store类');
+    const store = new Store('test');
+    
+    // 模拟获取session ID
+    const sessionId = await store.getID(ctx1);
+    console.log('✅ Store类测试成功');
+    console.log('   生成的session ID:', sessionId);
+
+    // 测试7: 测试session销毁
+    console.log('\n测试7: 测试session销毁');
+    const ctx3 = new MockContext();
+    
+    // 添加一个中间件来测试session销毁
+    const destroyApp = new MockKoaApp();
+    destroyApp.use(session.middleware());
+    destroyApp.use(async (ctx, next) => {
+      // 设置session为null触发销毁
+      ctx.session = null;
+      await next();
+    });
+    
+    await destroyApp.handleRequest(ctx3);
+    console.log('✅ session销毁逻辑测试完成');
+
+    console.log('\n🎉 所有测试通过！session模块功能正常。');
+
+  } catch (error) {
+    console.error('❌ 测试失败:', error.message);
+    console.error(error.stack);
+  }
+}
+
+// 运行测试
+if (require.main === module) {
+  testSession().catch(console.error);
+}
+
+module.exports = {
+  MockContext,
+  MockCookies,
+  testSession
+};

package/store.js

@@ -1,96 +0,0 @@
-const crypto = require('crypto');
-const CacheBase = require('mm_cachebase');
-
-if (!$.cache) {
-	$.cache = new CacheBase();
-}
-
-/**
- * @class 缓存
- */
-class Store {
-	/**
-	 * @description 构造函数
-	 * @param {String} key 键
-	 * @constructor
-	 */
-	constructor(key) {
-		if (key) {
-			this.key = key + "_";
-		} else {
-			this.key = $.dict.session_id + "_";
-		}
-	}
-}
-
-/**
- * @description 获取session ID
- * @param {Object} ctx HTTP上下文
- */
-Store.prototype.getID = async function(ctx) {
-	var hd = ctx.request.header;
-	var agent = hd['user-agent'];
-	if (!agent) {
-		agent = "mm";
-	}
-	var start = agent.md5().substring(0, 32);
-	var stamp = Date.parse(new Date()) / 1000;
-	var uuid = (ctx.ip + '_' + stamp).aes_encode(start);
-	return uuid;
-};
-
-/**
- * @description 获取session缓存
- * @param {String} uuid 客户端唯一ID
- */
-Store.prototype.get = async function(uuid) {
-	if (!$.cache.has(this.key + uuid)) return undefined;
-	// 我们正在解码数据来自我们的srote, 我们假定它是在储存之前
-	var val = await $.cache.get(this.key + uuid);
-	if (val) {
-		if (typeof(val) == "string") {
-			return JSON.parse(val);
-		}
-		return val;
-	} else {
-		return null;
-	}
-};
-
-/**
- * @description 设置session到缓存
- * @param {Object} obj 
- * @property {String} obj.uuid 客户端唯一表示ID
- * @property {Number} obj.maxAge 最大寿命
- * @param {Object} ctx HTTP请求上下文
- */
-Store.prototype.set = async function(session, {
-	uuid,
-	maxAge
-} = {}, ctx) {
-	if (!uuid) {
-		uuid = await this.getID(ctx);
-	}
-	if (!maxAge) {
-		maxAge = 7200;
-	}
-	try {
-		if (typeof(session) == "object") {
-			session = JSON.stringify(session);
-		}
-		await $.cache.set(this.key + uuid, session, maxAge);
-	} catch (err) {
-		console.log('Set session error:', err);
-	}
-	return uuid;
-};
-
-/**
- * @description 销毁缓存
- * @param {String} uuid
- */
-Store.prototype.destroy = function(uuid) {
-	$.cache.del(this.key + uuid);
-};
-
-module.exports = Store;