@qwickapps/qwickbrain-proxy 1.0.0

package/dist/lib/__tests__/cache-manager.test.js

@@ -0,0 +1,202 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { mkdtempSync, rmSync } from 'fs';
+import { join } from 'path';
+import { tmpdir } from 'os';
+import { createDatabase, runMigrations } from '../../db/client.js';
+import { CacheManager } from '../cache-manager.js';
+describe('CacheManager', () => {
+    let tmpDir;
+    let cacheManager;
+    let db;
+    beforeEach(() => {
+        // Create temporary directory for test database
+        tmpDir = mkdtempSync(join(tmpdir(), 'cache-test-'));
+        const dbResult = createDatabase(tmpDir);
+        db = dbResult.db;
+        // Run migrations to create tables
+        runMigrations(db);
+        const config = {
+            dir: tmpDir,
+            ttl: {
+                workflows: 3600,
+                rules: 3600,
+                documents: 1800,
+                memories: 900,
+            },
+            preload: [],
+        };
+        cacheManager = new CacheManager(db, config);
+    });
+    afterEach(() => {
+        // Clean up temporary directory
+        rmSync(tmpDir, { recursive: true, force: true });
+    });
+    describe('setDocument and getDocument', () => {
+        it('should store and retrieve a document', async () => {
+            await cacheManager.setDocument('workflow', 'test-workflow', 'content here');
+            const cached = await cacheManager.getDocument('workflow', 'test-workflow');
+            expect(cached).not.toBeNull();
+            expect(cached?.data.content).toBe('content here');
+            expect(cached?.data.doc_type).toBe('workflow');
+            expect(cached?.data.name).toBe('test-workflow');
+            expect(cached?.isExpired).toBe(false);
+        });
+        it('should store document with metadata', async () => {
+            const metadata = { author: 'test', version: 1 };
+            await cacheManager.setDocument('frd', 'test-frd', 'frd content', undefined, metadata);
+            const cached = await cacheManager.getDocument('frd', 'test-frd');
+            expect(cached?.data.metadata).toEqual(metadata);
+        });
+        it('should store document with project scope', async () => {
+            await cacheManager.setDocument('design', 'test-design', 'design content', 'my-project');
+            const cached = await cacheManager.getDocument('design', 'test-design', 'my-project');
+            expect(cached).not.toBeNull();
+            expect(cached?.data.project).toBe('my-project');
+        });
+        it('should distinguish between global and project-scoped documents', async () => {
+            await cacheManager.setDocument('rule', 'test-rule', 'global rule');
+            await cacheManager.setDocument('rule', 'test-rule', 'project rule', 'my-project');
+            const globalDoc = await cacheManager.getDocument('rule', 'test-rule');
+            const projectDoc = await cacheManager.getDocument('rule', 'test-rule', 'my-project');
+            expect(globalDoc?.data.content).toBe('global rule');
+            expect(globalDoc?.data.project).toBe('');
+            expect(projectDoc?.data.content).toBe('project rule');
+            expect(projectDoc?.data.project).toBe('my-project');
+        });
+        it('should return null for non-existent document', async () => {
+            const cached = await cacheManager.getDocument('workflow', 'non-existent');
+            expect(cached).toBeNull();
+        });
+        it('should update existing document on conflict', async () => {
+            await cacheManager.setDocument('workflow', 'test', 'version 1');
+            await cacheManager.setDocument('workflow', 'test', 'version 2');
+            const cached = await cacheManager.getDocument('workflow', 'test');
+            expect(cached?.data.content).toBe('version 2');
+        });
+        it('should mark document as expired after TTL', async () => {
+            // Override config to have very short TTL for testing
+            const shortTTLConfig = {
+                dir: tmpDir,
+                ttl: {
+                    workflows: 0, // Expire immediately
+                    rules: 0,
+                    documents: 0,
+                    memories: 0,
+                },
+                preload: [],
+            };
+            const shortCacheManager = new CacheManager(db, shortTTLConfig);
+            await shortCacheManager.setDocument('workflow', 'test', 'content');
+            // Wait to ensure timestamp difference
+            await new Promise(resolve => setTimeout(resolve, 100));
+            const cached = await shortCacheManager.getDocument('workflow', 'test');
+            expect(cached).not.toBeNull();
+            expect(cached?.isExpired).toBe(true);
+        });
+    });
+    describe('setMemory and getMemory', () => {
+        it('should store and retrieve a memory', async () => {
+            await cacheManager.setMemory('test-memory', 'memory content');
+            const cached = await cacheManager.getMemory('test-memory');
+            expect(cached).not.toBeNull();
+            expect(cached?.data.content).toBe('memory content');
+            expect(cached?.data.name).toBe('test-memory');
+            expect(cached?.isExpired).toBe(false);
+        });
+        it('should store memory with metadata', async () => {
+            const metadata = { lastUpdated: Date.now() };
+            await cacheManager.setMemory('test-memory', 'content', undefined, metadata);
+            const cached = await cacheManager.getMemory('test-memory');
+            expect(cached?.data.metadata).toEqual(metadata);
+        });
+        it('should store memory with project scope', async () => {
+            await cacheManager.setMemory('patterns', 'project patterns', 'my-project');
+            const cached = await cacheManager.getMemory('patterns', 'my-project');
+            expect(cached).not.toBeNull();
+            expect(cached?.data.project).toBe('my-project');
+        });
+        it('should distinguish between global and project-scoped memories', async () => {
+            await cacheManager.setMemory('context', 'global context');
+            await cacheManager.setMemory('context', 'project context', 'my-project');
+            const globalMem = await cacheManager.getMemory('context');
+            const projectMem = await cacheManager.getMemory('context', 'my-project');
+            expect(globalMem?.data.content).toBe('global context');
+            expect(projectMem?.data.content).toBe('project context');
+        });
+        it('should return null for non-existent memory', async () => {
+            const cached = await cacheManager.getMemory('non-existent');
+            expect(cached).toBeNull();
+        });
+        it('should mark memory as expired after TTL', async () => {
+            const shortTTLConfig = {
+                dir: tmpDir,
+                ttl: {
+                    workflows: 0,
+                    rules: 0,
+                    documents: 0,
+                    memories: 0,
+                },
+                preload: [],
+            };
+            const shortCacheManager = new CacheManager(db, shortTTLConfig);
+            await shortCacheManager.setMemory('test', 'content');
+            await new Promise(resolve => setTimeout(resolve, 100));
+            const cached = await shortCacheManager.getMemory('test');
+            expect(cached).not.toBeNull();
+            expect(cached?.isExpired).toBe(true);
+        });
+    });
+    describe('cleanupExpiredEntries', () => {
+        it('should delete expired documents and memories', async () => {
+            const shortTTLConfig = {
+                dir: tmpDir,
+                ttl: {
+                    workflows: 0,
+                    rules: 0,
+                    documents: 0,
+                    memories: 0,
+                },
+                preload: [],
+            };
+            const shortCacheManager = new CacheManager(db, shortTTLConfig);
+            // Add some items that will immediately expire
+            await shortCacheManager.setDocument('workflow', 'test1', 'content1');
+            await shortCacheManager.setDocument('rule', 'test2', 'content2');
+            await shortCacheManager.setMemory('memory1', 'content3');
+            // Wait to ensure they're expired (need sufficient time for clock to advance)
+            await new Promise(resolve => setTimeout(resolve, 100));
+            // Clean up
+            const result = await shortCacheManager.cleanupExpiredEntries();
+            expect(result.documentsDeleted).toBe(2);
+            expect(result.memoriesDeleted).toBe(1);
+            // Verify they're gone
+            const doc1 = await shortCacheManager.getDocument('workflow', 'test1');
+            const mem1 = await shortCacheManager.getMemory('memory1');
+            expect(doc1).toBeNull();
+            expect(mem1).toBeNull();
+        });
+        it('should not delete non-expired items', async () => {
+            await cacheManager.setDocument('workflow', 'test', 'content');
+            await cacheManager.setMemory('memory', 'content');
+            const result = await cacheManager.cleanupExpiredEntries();
+            expect(result.documentsDeleted).toBe(0);
+            expect(result.memoriesDeleted).toBe(0);
+            // Verify they're still there
+            const doc = await cacheManager.getDocument('workflow', 'test');
+            const mem = await cacheManager.getMemory('memory');
+            expect(doc).not.toBeNull();
+            expect(mem).not.toBeNull();
+        });
+    });
+    describe('cache age calculation', () => {
+        it('should calculate age correctly', async () => {
+            await cacheManager.setDocument('workflow', 'test', 'content');
+            // Wait 100ms
+            await new Promise(resolve => setTimeout(resolve, 100));
+            const cached = await cacheManager.getDocument('workflow', 'test');
+            expect(cached?.age).toBeGreaterThanOrEqual(0);
+            expect(cached?.age).toBeLessThan(1); // Less than 1 second
+        });
+    });
+});
+//# sourceMappingURL=cache-manager.test.js.map

package/dist/lib/__tests__/cache-manager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache-manager.test.js","sourceRoot":"","sources":["../../../src/lib/__tests__/cache-manager.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AAC5B,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGnD,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,MAAc,CAAC;IACnB,IAAI,YAA0B,CAAC;IAC/B,IAAI,EAA2C,CAAC;IAEhD,UAAU,CAAC,GAAG,EAAE;QACd,+CAA+C;QAC/C,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QACpD,MAAM,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QACxC,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;QAEjB,kCAAkC;QAClC,aAAa,CAAC,EAAE,CAAC,CAAC;QAElB,MAAM,MAAM,GAAoB;YAC9B,GAAG,EAAE,MAAM;YACX,GAAG,EAAE;gBACH,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,IAAI;gBACX,SAAS,EAAE,IAAI;gBACf,QAAQ,EAAE,GAAG;aACd;YACD,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,YAAY,GAAG,IAAI,YAAY,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,+BAA+B;QAC/B,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;QAC3C,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC;YAE5E,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;YAE3E,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,QAAQ,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;YAChD,MAAM,YAAY,CAAC,WAAW,CAAC,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;YAEtF,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAEjE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,YAAY,CAAC,WAAW,CAAC,QAAQ,EAAE,aAAa,EAAE,gBAAgB,EAAE,YAAY,CAAC,CAAC;YAExF,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,QAAQ,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;YAErF,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,YAAY,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;YACnE,MAAM,YAAY,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC;YAElF,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACtE,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,MAAM,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;YAErF,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACpD,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACzC,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACtD,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAChE,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAElE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,qDAAqD;YACrD,MAAM,cAAc,GAAoB;gBACtC,GAAG,EAAE,MAAM;gBACX,GAAG,EAAE;oBACH,SAAS,EAAE,CAAC,EAAE,qBAAqB;oBACnC,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE,CAAC;oBACZ,QAAQ,EAAE,CAAC;iBACZ;gBACD,OAAO,EAAE,EAAE;aACZ,CAAC;YAEF,MAAM,iBAAiB,GAAG,IAAI,YAAY,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAC/D,MAAM,iBAAiB,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAEnE,sCAAsC;YACtC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAEvE,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,YAAY,CAAC,SAAS,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;YAE9D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,QAAQ,GAAG,EAAE,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAC7C,MAAM,YAAY,CAAC,SAAS,CAAC,aAAa,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;YAE5E,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAE3D,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,YAAY,CAAC,SAAS,CAAC,UAAU,EAAE,kBAAkB,EAAE,YAAY,CAAC,CAAC;YAE3E,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,YAAY,CAAC,SAAS,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;YAC1D,MAAM,YAAY,CAAC,SAAS,CAAC,SAAS,EAAE,iBAAiB,EAAE,YAAY,CAAC,CAAC;YAEzE,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC1D,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YAEzE,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACvD,MAAM,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,cAAc,GAAoB;gBACtC,GAAG,EAAE,MAAM;gBACX,GAAG,EAAE;oBACH,SAAS,EAAE,CAAC;oBACZ,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE,CAAC;oBACZ,QAAQ,EAAE,CAAC;iBACZ;gBACD,OAAO,EAAE,EAAE;aACZ,CAAC;YAEF,MAAM,iBAAiB,GAAG,IAAI,YAAY,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAC/D,MAAM,iBAAiB,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAErD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,cAAc,GAAoB;gBACtC,GAAG,EAAE,MAAM;gBACX,GAAG,EAAE;oBACH,SAAS,EAAE,CAAC;oBACZ,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE,CAAC;oBACZ,QAAQ,EAAE,CAAC;iBACZ;gBACD,OAAO,EAAE,EAAE;aACZ,CAAC;YAEF,MAAM,iBAAiB,GAAG,IAAI,YAAY,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAE/D,8CAA8C;YAC9C,MAAM,iBAAiB,CAAC,WAAW,CAAC,UAAU,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;YACrE,MAAM,iBAAiB,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;YACjE,MAAM,iBAAiB,CAAC,SAAS,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YAEzD,6EAA6E;YAC7E,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,WAAW;YACX,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,qBAAqB,EAAE,CAAC;YAE/D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEvC,sBAAsB;YACtB,MAAM,IAAI,GAAG,MAAM,iBAAiB,CAAC,WAAW,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACtE,MAAM,IAAI,GAAG,MAAM,iBAAiB,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAE1D,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAC9D,MAAM,YAAY,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YAElD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,qBAAqB,EAAE,CAAC;YAE1D,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEvC,6BAA6B;YAC7B,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAC/D,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAEnD,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;YAE9D,aAAa;YACb,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAElE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,qBAAqB;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/lib/__tests__/connection-manager.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=connection-manager.test.d.ts.map

package/dist/lib/__tests__/connection-manager.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection-manager.test.d.ts","sourceRoot":"","sources":["../../../src/lib/__tests__/connection-manager.test.ts"],"names":[],"mappings":""}

package/dist/lib/__tests__/connection-manager.test.js

@@ -0,0 +1,188 @@
+import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
+import { ConnectionManager } from '../connection-manager.js';
+describe('ConnectionManager', () => {
+    let mockClient;
+    let connectionManager;
+    let config;
+    beforeEach(() => {
+        config = {
+            healthCheckInterval: 100, // Short interval for testing
+            timeout: 5000,
+            maxReconnectAttempts: 3,
+            reconnectBackoff: {
+                initial: 50,
+                multiplier: 2,
+                max: 500,
+            },
+        };
+        mockClient = {
+            healthCheck: vi.fn().mockResolvedValue(true),
+        };
+        connectionManager = new ConnectionManager(mockClient, config);
+    });
+    afterEach(() => {
+        connectionManager.stop();
+    });
+    describe('getState', () => {
+        it('should return initial state as disconnected', () => {
+            expect(connectionManager.getState()).toBe('disconnected');
+        });
+    });
+    describe('start and healthCheck', () => {
+        it('should transition to connected on successful health check', async () => {
+            await connectionManager.start();
+            expect(connectionManager.getState()).toBe('connected');
+            expect(mockClient.healthCheck).toHaveBeenCalled();
+        });
+        it('should transition to reconnecting after failed health check', async () => {
+            mockClient.healthCheck = vi.fn().mockRejectedValue(new Error('Connection failed'));
+            await connectionManager.start();
+            // After health check fails, it immediately schedules reconnect
+            expect(connectionManager.getState()).toBe('reconnecting');
+        });
+        it('should emit stateChange event on state transition', async () => {
+            const stateChangeListener = vi.fn();
+            connectionManager.on('stateChange', stateChangeListener);
+            await connectionManager.start();
+            expect(stateChangeListener).toHaveBeenCalledWith({
+                from: 'disconnected',
+                to: 'connected',
+            });
+        });
+        it('should emit connected event with latency', async () => {
+            const connectedListener = vi.fn();
+            connectionManager.on('connected', connectedListener);
+            await connectionManager.start();
+            expect(connectedListener).toHaveBeenCalled();
+            const call = connectedListener.mock.calls[0][0];
+            expect(call).toHaveProperty('latencyMs');
+            expect(typeof call.latencyMs).toBe('number');
+        });
+        it('should emit disconnected event on failure', async () => {
+            mockClient.healthCheck = vi.fn().mockRejectedValue(new Error('Failed'));
+            const disconnectedListener = vi.fn();
+            connectionManager.on('disconnected', disconnectedListener);
+            await connectionManager.start();
+            expect(disconnectedListener).toHaveBeenCalled();
+            expect(disconnectedListener.mock.calls[0][0]).toHaveProperty('error');
+        });
+    });
+    describe('stop', () => {
+        it('should transition to offline state', () => {
+            connectionManager.stop();
+            expect(connectionManager.getState()).toBe('offline');
+        });
+        it('should stop health check timer', async () => {
+            await connectionManager.start();
+            const healthCheckCount = mockClient.healthCheck.mock.calls.length;
+            connectionManager.stop();
+            // Wait longer than health check interval
+            await new Promise(resolve => setTimeout(resolve, 300));
+            // Health check should not have been called again
+            expect(mockClient.healthCheck.mock.calls.length).toBe(healthCheckCount);
+        });
+    });
+    describe('reconnection logic', () => {
+        it('should attempt to reconnect on health check failure', async () => {
+            let callCount = 0;
+            mockClient.healthCheck = vi.fn().mockImplementation(() => {
+                callCount++;
+                if (callCount === 1) {
+                    return Promise.reject(new Error('Failed'));
+                }
+                return Promise.resolve(true);
+            });
+            const reconnectingListener = vi.fn();
+            connectionManager.on('reconnecting', reconnectingListener);
+            await connectionManager.start();
+            // Wait for reconnect attempt
+            await new Promise(resolve => setTimeout(resolve, 100));
+            expect(reconnectingListener).toHaveBeenCalled();
+            expect(reconnectingListener.mock.calls[0][0]).toHaveProperty('attempt');
+            expect(reconnectingListener.mock.calls[0][0]).toHaveProperty('delay');
+        });
+        it('should use exponential backoff for reconnection', async () => {
+            mockClient.healthCheck = vi.fn().mockRejectedValue(new Error('Failed'));
+            const reconnectingListener = vi.fn();
+            connectionManager.on('reconnecting', reconnectingListener);
+            await connectionManager.start();
+            // Wait for multiple reconnect attempts
+            await new Promise(resolve => setTimeout(resolve, 500));
+            const delays = reconnectingListener.mock.calls.map(call => call[0].delay);
+            // Delays should increase exponentially
+            for (let i = 1; i < delays.length; i++) {
+                expect(delays[i]).toBeGreaterThan(delays[i - 1]);
+            }
+        });
+        it('should emit maxReconnectAttemptsReached after max attempts', async () => {
+            mockClient.healthCheck = vi.fn().mockRejectedValue(new Error('Failed'));
+            const maxAttemptsListener = vi.fn();
+            connectionManager.on('maxReconnectAttemptsReached', maxAttemptsListener);
+            await connectionManager.start();
+            // Wait for all reconnect attempts
+            await new Promise(resolve => setTimeout(resolve, 2000));
+            expect(maxAttemptsListener).toHaveBeenCalled();
+            expect(connectionManager.getState()).toBe('offline');
+        });
+        it('should reset reconnect attempts on successful connection', async () => {
+            let callCount = 0;
+            mockClient.healthCheck = vi.fn().mockImplementation(() => {
+                callCount++;
+                // Fail first two, then succeed
+                if (callCount <= 2) {
+                    return Promise.reject(new Error('Failed'));
+                }
+                return Promise.resolve(true);
+            });
+            await connectionManager.start();
+            // Wait for reconnection
+            await new Promise(resolve => setTimeout(resolve, 300));
+            expect(connectionManager.getState()).toBe('connected');
+        });
+    });
+    describe('execute', () => {
+        it('should execute function when connected', async () => {
+            await connectionManager.start();
+            const testFn = vi.fn().mockResolvedValue('result');
+            const result = await connectionManager.execute(testFn);
+            expect(result).toBe('result');
+            expect(testFn).toHaveBeenCalled();
+        });
+        it('should throw error when not connected', async () => {
+            const testFn = vi.fn();
+            await expect(connectionManager.execute(testFn)).rejects.toThrow('QwickBrain unavailable');
+            expect(testFn).not.toHaveBeenCalled();
+        });
+        it('should record failure if execution throws', async () => {
+            await connectionManager.start();
+            const testFn = vi.fn().mockRejectedValue(new Error('Execution failed'));
+            await expect(connectionManager.execute(testFn)).rejects.toThrow('Execution failed');
+            // Should transition to reconnecting after failure
+            expect(connectionManager.getState()).toBe('reconnecting');
+        });
+        it('should wait for state transitions to complete (race condition test)', async () => {
+            await connectionManager.start();
+            // Simulate rapid state changes
+            connectionManager.setState('disconnected');
+            connectionManager.setState('connected');
+            const testFn = vi.fn().mockResolvedValue('result');
+            // This should wait for state transitions to settle
+            const result = await connectionManager.execute(testFn);
+            expect(result).toBe('result');
+        });
+    });
+    describe('recordFailure', () => {
+        it('should transition to reconnecting when connected', () => {
+            connectionManager['state'] = 'connected';
+            connectionManager.recordFailure();
+            // After failure, it immediately schedules reconnect
+            expect(connectionManager.getState()).toBe('reconnecting');
+        });
+        it('should not affect offline state', () => {
+            connectionManager.stop();
+            connectionManager.recordFailure();
+            expect(connectionManager.getState()).toBe('offline');
+        });
+    });
+});
+//# sourceMappingURL=connection-manager.test.js.map

package/dist/lib/__tests__/connection-manager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection-manager.test.js","sourceRoot":"","sources":["../../../src/lib/__tests__/connection-manager.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAI7D,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,UAA4B,CAAC;IACjC,IAAI,iBAAoC,CAAC;IACzC,IAAI,MAA4B,CAAC;IAEjC,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG;YACP,mBAAmB,EAAE,GAAG,EAAE,6BAA6B;YACvD,OAAO,EAAE,IAAI;YACb,oBAAoB,EAAE,CAAC;YACvB,gBAAgB,EAAE;gBAChB,OAAO,EAAE,EAAE;gBACX,UAAU,EAAE,CAAC;gBACb,GAAG,EAAE,GAAG;aACT;SACF,CAAC;QAEF,UAAU,GAAG;YACX,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC;SACd,CAAC;QAEjC,iBAAiB,GAAG,IAAI,iBAAiB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,iBAAiB,CAAC,IAAI,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACvD,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,UAAU,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;YAEnF,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,+DAA+D;YAC/D,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,iBAAiB,CAAC,EAAE,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;YAEzD,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,MAAM,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC;gBAC/C,IAAI,EAAE,cAAc;gBACpB,EAAE,EAAE,WAAW;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,iBAAiB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,iBAAiB,CAAC,EAAE,CAAC,WAAW,EAAE,iBAAiB,CAAC,CAAC;YAErD,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,MAAM,CAAC,iBAAiB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChD,MAAM,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YACzC,MAAM,CAAC,OAAO,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,UAAU,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;YACxE,MAAM,oBAAoB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,iBAAiB,CAAC,EAAE,CAAC,cAAc,EAAE,oBAAoB,CAAC,CAAC;YAE3D,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,MAAM,CAAC,oBAAoB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAChD,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,iBAAiB,CAAC,IAAI,EAAE,CAAC;YAEzB,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAChC,MAAM,gBAAgB,GAAI,UAAU,CAAC,WAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;YAE3E,iBAAiB,CAAC,IAAI,EAAE,CAAC;YAEzB,yCAAyC;YACzC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,iDAAiD;YACjD,MAAM,CAAE,UAAU,CAAC,WAAmB,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,UAAU,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACvD,SAAS,EAAE,CAAC;gBACZ,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;oBACpB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAC7C,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC,CAAC,CAAC;YAEH,MAAM,oBAAoB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,iBAAiB,CAAC,EAAE,CAAC,cAAc,EAAE,oBAAoB,CAAC,CAAC;YAE3D,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,6BAA6B;YAC7B,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,MAAM,CAAC,oBAAoB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAChD,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;YACxE,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,UAAU,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;YAExE,MAAM,oBAAoB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,iBAAiB,CAAC,EAAE,CAAC,cAAc,EAAE,oBAAoB,CAAC,CAAC;YAE3D,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,uCAAuC;YACvC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YAE1E,uCAAuC;YACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YACnD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,UAAU,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;YAExE,MAAM,mBAAmB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,iBAAiB,CAAC,EAAE,CAAC,6BAA6B,EAAE,mBAAmB,CAAC,CAAC;YAEzE,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,kCAAkC;YAClC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YAExD,MAAM,CAAC,mBAAmB,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC/C,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,UAAU,CAAC,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACvD,SAAS,EAAE,CAAC;gBACZ,+BAA+B;gBAC/B,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;oBACnB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAC7C,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC,CAAC,CAAC;YAEH,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,wBAAwB;YACxB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;YAEvD,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEvB,MAAM,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;YAC1F,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC;YAExE,MAAM,MAAM,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YAEpF,kDAAkD;YAClD,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,MAAM,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAEhC,+BAA+B;YAC/B,iBAAiB,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;YAC3C,iBAAiB,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YAExC,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEnD,mDAAmD;YACnD,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,iBAAiB,CAAC,OAAO,CAAC,GAAG,WAAW,CAAC;YAEzC,iBAAiB,CAAC,aAAa,EAAE,CAAC;YAElC,oDAAoD;YACpD,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,iBAAiB,CAAC,IAAI,EAAE,CAAC;YAEzB,iBAAiB,CAAC,aAAa,EAAE,CAAC;YAElC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/lib/__tests__/proxy-server.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=proxy-server.test.d.ts.map

package/dist/lib/__tests__/proxy-server.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proxy-server.test.d.ts","sourceRoot":"","sources":["../../../src/lib/__tests__/proxy-server.test.ts"],"names":[],"mappings":""}

package/dist/lib/__tests__/proxy-server.test.js

@@ -0,0 +1,205 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
+import { mkdtempSync, rmSync } from 'fs';
+import { join } from 'path';
+import { tmpdir } from 'os';
+import { createDatabase, runMigrations } from '../../db/client.js';
+import { ProxyServer } from '../proxy-server.js';
+// Mock MCP SDK
+vi.mock('@modelcontextprotocol/sdk/client/index.js', () => ({
+    Client: vi.fn().mockImplementation(() => ({
+        connect: vi.fn().mockResolvedValue(undefined),
+        close: vi.fn().mockResolvedValue(undefined),
+        callTool: vi.fn(),
+        listTools: vi.fn().mockResolvedValue({ tools: [] }),
+    })),
+}));
+vi.mock('@modelcontextprotocol/sdk/server/index.js', () => ({
+    Server: vi.fn().mockImplementation(() => ({
+        setRequestHandler: vi.fn(),
+        connect: vi.fn().mockResolvedValue(undefined),
+        close: vi.fn().mockResolvedValue(undefined),
+    })),
+}));
+vi.mock('@modelcontextprotocol/sdk/server/stdio.js', () => ({
+    StdioServerTransport: vi.fn(),
+}));
+vi.mock('@modelcontextprotocol/sdk/client/sse.js', () => ({
+    SSEClientTransport: vi.fn(),
+}));
+describe('ProxyServer', () => {
+    let tmpDir;
+    let config;
+    let db;
+    let server;
+    beforeEach(() => {
+        tmpDir = mkdtempSync(join(tmpdir(), 'proxy-test-'));
+        const dbResult = createDatabase(tmpDir);
+        db = dbResult.db;
+        // Run migrations to create tables
+        runMigrations(db);
+        config = {
+            qwickbrain: {
+                mode: 'sse',
+                url: 'http://test.local:3000',
+            },
+            cache: {
+                dir: tmpDir,
+                ttl: {
+                    workflows: 3600,
+                    rules: 3600,
+                    documents: 1800,
+                    memories: 900,
+                },
+                preload: [],
+            },
+            connection: {
+                healthCheckInterval: 30000,
+                timeout: 5000,
+                maxReconnectAttempts: 5,
+                reconnectBackoff: {
+                    initial: 1000,
+                    multiplier: 2,
+                    max: 30000,
+                },
+            },
+            sync: {
+                interval: 60000,
+                batchSize: 10,
+            },
+        };
+        server = new ProxyServer(db, config);
+    });
+    afterEach(async () => {
+        await server.stop();
+        rmSync(tmpDir, { recursive: true, force: true });
+    });
+    describe('initialization', () => {
+        it('should initialize with all components', () => {
+            expect(server['server']).toBeDefined();
+            expect(server['connectionManager']).toBeDefined();
+            expect(server['cacheManager']).toBeDefined();
+            expect(server['qwickbrainClient']).toBeDefined();
+        });
+        it('should set up MCP request handlers', () => {
+            const setRequestHandlerMock = server['server'].setRequestHandler;
+            // Should have set up handlers for ListTools and CallTool
+            expect(setRequestHandlerMock).toHaveBeenCalled();
+            expect(setRequestHandlerMock.mock.calls.length).toBeGreaterThanOrEqual(2);
+        });
+    });
+    describe('cache cleanup on startup', () => {
+        it('should clean up expired cache entries on start', async () => {
+            // Add some expired entries
+            const shortTTLConfig = { ...config };
+            shortTTLConfig.cache.ttl = {
+                workflows: 0,
+                rules: 0,
+                documents: 0,
+                memories: 0,
+            };
+            const tempServer = new ProxyServer(db, shortTTLConfig);
+            await tempServer['cacheManager'].setDocument('workflow', 'test', 'content');
+            await new Promise(resolve => setTimeout(resolve, 100));
+            // Verify entry exists before cleanup
+            const beforeCleanup = await tempServer['cacheManager'].getDocument('workflow', 'test');
+            expect(beforeCleanup).not.toBeNull();
+            await tempServer.start();
+            // Entry should be removed after cleanup
+            const afterCleanup = await tempServer['cacheManager'].getDocument('workflow', 'test');
+            expect(afterCleanup).toBeNull();
+            await tempServer.stop();
+        });
+    });
+    describe('graceful degradation', () => {
+        it('should serve stale cache when disconnected', async () => {
+            // Create server with short TTL to make cache expire
+            const shortTTLConfig = { ...config };
+            shortTTLConfig.cache.ttl = {
+                workflows: 0,
+                rules: 0,
+                documents: 0,
+                memories: 0,
+            };
+            const tempServer = new ProxyServer(db, shortTTLConfig);
+            // Add item to cache
+            await tempServer['cacheManager'].setDocument('workflow', 'test', 'cached content');
+            // Wait for cache to expire
+            await new Promise(resolve => setTimeout(resolve, 100));
+            // Simulate disconnected state
+            tempServer['connectionManager']['state'] = 'disconnected';
+            const result = await tempServer['handleGetDocument']('workflow', 'test');
+            expect(result.data.content).toBe('cached content');
+            expect(result._metadata.source).toBe('stale_cache');
+            await tempServer.stop();
+        });
+        it('should return error when no cache available and disconnected', async () => {
+            // Simulate disconnected state with no cache
+            server['connectionManager']['state'] = 'disconnected';
+            const result = await server['handleGetDocument']('workflow', 'non-existent');
+            expect(result.error).toBeDefined();
+            expect(result.error?.code).toBe('UNAVAILABLE');
+            expect(result.error?.message).toContain('QwickBrain unavailable');
+        });
+    });
+    describe('metadata', () => {
+        it('should include correct metadata for cache hit', async () => {
+            await server['cacheManager'].setDocument('workflow', 'test', 'content');
+            // Simulate connected state
+            server['connectionManager']['state'] = 'connected';
+            const result = await server['handleGetDocument']('workflow', 'test');
+            expect(result._metadata.source).toBe('cache');
+            expect(result._metadata.age_seconds).toBeDefined();
+            expect(typeof result._metadata.age_seconds).toBe('number');
+        });
+        it('should create metadata with correct source', () => {
+            const cacheMeta = server['createMetadata']('cache', 100);
+            expect(cacheMeta).toEqual({
+                source: 'cache',
+                age_seconds: 100,
+                status: server['connectionManager'].getState(),
+            });
+            const liveMeta = server['createMetadata']('live');
+            expect(liveMeta).toEqual({
+                source: 'live',
+                status: server['connectionManager'].getState(),
+            });
+            const staleMeta = server['createMetadata']('stale_cache', 500);
+            expect(staleMeta).toEqual({
+                source: 'stale_cache',
+                age_seconds: 500,
+                status: server['connectionManager'].getState(),
+            });
+        });
+    });
+    describe('error handling', () => {
+        it('should handle QwickBrain errors gracefully', async () => {
+            // Mock client to throw error
+            server['qwickbrainClient'].getDocument = vi.fn().mockRejectedValue(new Error('Network error'));
+            // Simulate connected state
+            server['connectionManager']['state'] = 'connected';
+            const result = await server['handleGetDocument']('workflow', 'test');
+            // Should fall back to error response when remote fails and no cache
+            expect(result.error).toBeDefined();
+            expect(result.error?.code).toBe('UNAVAILABLE');
+        });
+    });
+    describe('project scoping', () => {
+        it('should handle project-scoped documents', async () => {
+            await server['cacheManager'].setDocument('design', 'test-design', 'design content', 'my-project');
+            server['connectionManager']['state'] = 'connected';
+            const result = await server['handleGetDocument']('design', 'test-design', 'my-project');
+            expect(result.data.content).toBe('design content');
+            expect(result.data.project).toBe('my-project');
+        });
+        it('should distinguish global vs project-scoped items', async () => {
+            await server['cacheManager'].setDocument('rule', 'test', 'global rule');
+            await server['cacheManager'].setDocument('rule', 'test', 'project rule', 'proj1');
+            server['connectionManager']['state'] = 'connected';
+            const globalResult = await server['handleGetDocument']('rule', 'test');
+            const projectResult = await server['handleGetDocument']('rule', 'test', 'proj1');
+            expect(globalResult.data.content).toBe('global rule');
+            expect(projectResult.data.content).toBe('project rule');
+        });
+    });
+});
+//# sourceMappingURL=proxy-server.test.js.map