@ash-ai/server 0.0.3 → 0.0.5

package/dist/__tests__/attachments.test.d.ts

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

package/dist/__tests__/attachments.test.d.ts.map

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

package/dist/__tests__/attachments.test.js

@@ -0,0 +1,57 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { tmpdir } from 'node:os';
+import { mkdtempSync } from 'node:fs';
+import { join } from 'node:path';
+import { initDb, closeDb, insertAttachment, getAttachment, listAttachmentsByMessage, listAttachmentsBySession, deleteAttachment } from '../db/index.js';
+const tenant = 'attach-test';
+let n = 0;
+const uid = () => `a-${Date.now()}-${++n}`;
+describe('attachments', () => {
+    beforeEach(async () => {
+        const dir = mkdtempSync(join(tmpdir(), 'ash-attach-'));
+        await initDb({ dataDir: dir });
+    });
+    afterEach(async () => {
+        await closeDb();
+    });
+    it('inserts and retrieves an attachment', async () => {
+        const id = uid();
+        const att = await insertAttachment(id, tenant, 'msg-1', 'sess-1', 'test.txt', 'text/plain', 42, '/path/to/test.txt');
+        expect(att.id).toBe(id);
+        expect(att.filename).toBe('test.txt');
+        expect(att.mimeType).toBe('text/plain');
+        expect(att.size).toBe(42);
+        const got = await getAttachment(id);
+        expect(got).not.toBeNull();
+        expect(got.id).toBe(id);
+    });
+    it('returns null for nonexistent attachment', async () => {
+        const got = await getAttachment('nope');
+        expect(got).toBeNull();
+    });
+    it('lists attachments by message', async () => {
+        const a = uid(), b = uid();
+        await insertAttachment(a, tenant, 'msg-1', 'sess-1', 'file1.txt', 'text/plain', 10, '/p1');
+        await insertAttachment(b, tenant, 'msg-2', 'sess-1', 'file2.txt', 'text/plain', 20, '/p2');
+        const forMsg1 = await listAttachmentsByMessage('msg-1', tenant);
+        expect(forMsg1.length).toBe(1);
+        expect(forMsg1[0].id).toBe(a);
+    });
+    it('lists attachments by session', async () => {
+        const a = uid(), b = uid(), c = uid();
+        await insertAttachment(a, tenant, 'msg-1', 'sess-1', 'f1.txt', 'text/plain', 10, '/p1');
+        await insertAttachment(b, tenant, 'msg-2', 'sess-1', 'f2.txt', 'text/plain', 20, '/p2');
+        await insertAttachment(c, tenant, 'msg-3', 'sess-2', 'f3.txt', 'text/plain', 30, '/p3');
+        const forSess1 = await listAttachmentsBySession('sess-1', tenant);
+        expect(forSess1.length).toBe(2);
+    });
+    it('deletes an attachment', async () => {
+        const id = uid();
+        await insertAttachment(id, tenant, 'msg-1', 'sess-1', 'del.txt', 'text/plain', 5, '/del');
+        const deleted = await deleteAttachment(id);
+        expect(deleted).toBe(true);
+        const got = await getAttachment(id);
+        expect(got).toBeNull();
+    });
+});
+//# sourceMappingURL=attachments.test.js.map

package/dist/__tests__/attachments.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"attachments.test.js","sourceRoot":"","sources":["../../src/__tests__/attachments.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAExJ,MAAM,MAAM,GAAG,aAAa,CAAC;AAC7B,IAAI,CAAC,GAAG,CAAC,CAAC;AACV,MAAM,GAAG,GAAG,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;AAE3C,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QACvD,MAAM,MAAM,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,OAAO,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,EAAE,GAAG,GAAG,EAAE,CAAC;QACjB,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,mBAAmB,CAAC,CAAC;QACrH,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxB,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACxC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAE1B,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,CAAC,GAAG,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC;QAC3B,MAAM,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QAC3F,MAAM,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QAE3F,MAAM,OAAO,GAAG,MAAM,wBAAwB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAChE,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,CAAC,GAAG,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC;QACtC,MAAM,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QACxF,MAAM,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QACxF,MAAM,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;QAExF,MAAM,QAAQ,GAAG,MAAM,wBAAwB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClE,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;QACrC,MAAM,EAAE,GAAG,GAAG,EAAE,CAAC;QACjB,MAAM,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1F,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,EAAE,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/bundle.test.d.ts

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

package/dist/__tests__/bundle.test.d.ts.map

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

package/dist/__tests__/bundle.test.js

@@ -0,0 +1,55 @@
+import { describe, it, expect, afterEach } from 'vitest';
+import { tmpdir } from 'node:os';
+import { mkdtempSync, writeFileSync, readFileSync, rmSync, mkdirSync, existsSync } from 'node:fs';
+import { join } from 'node:path';
+import { createBundle, extractBundle } from '@ash-ai/sandbox';
+describe('workspace bundles', () => {
+    const dirs = [];
+    function tmpDir() {
+        const d = mkdtempSync(join(tmpdir(), 'ash-bundle-'));
+        dirs.push(d);
+        return d;
+    }
+    afterEach(() => {
+        for (const d of dirs) {
+            rmSync(d, { recursive: true, force: true });
+        }
+        dirs.length = 0;
+    });
+    it('round-trip: createBundle → extractBundle preserves files', () => {
+        const src = tmpDir();
+        writeFileSync(join(src, 'hello.txt'), 'world');
+        mkdirSync(join(src, 'sub'));
+        writeFileSync(join(src, 'sub', 'nested.txt'), 'deep');
+        const bundle = createBundle(src);
+        expect(bundle.length).toBeGreaterThan(0);
+        const dest = tmpDir();
+        extractBundle(bundle, dest);
+        expect(readFileSync(join(dest, 'hello.txt'), 'utf-8')).toBe('world');
+        expect(readFileSync(join(dest, 'sub', 'nested.txt'), 'utf-8')).toBe('deep');
+    });
+    it('createBundle excludes node_modules', () => {
+        const src = tmpDir();
+        writeFileSync(join(src, 'index.js'), 'console.log("hi")');
+        mkdirSync(join(src, 'node_modules', 'dep'), { recursive: true });
+        writeFileSync(join(src, 'node_modules', 'dep', 'index.js'), 'big module');
+        const bundle = createBundle(src);
+        const dest = tmpDir();
+        extractBundle(bundle, dest);
+        expect(readFileSync(join(dest, 'index.js'), 'utf-8')).toBe('console.log("hi")');
+        // node_modules should be excluded
+        expect(existsSync(join(dest, 'node_modules'))).toBe(false);
+    });
+    it('throws for nonexistent workspace', () => {
+        expect(() => createBundle('/nonexistent/path')).toThrow('does not exist');
+    });
+    it('extractBundle creates target dir if needed', () => {
+        const src = tmpDir();
+        writeFileSync(join(src, 'file.txt'), 'data');
+        const bundle = createBundle(src);
+        const dest = join(tmpDir(), 'new', 'nested', 'dir');
+        extractBundle(bundle, dest);
+        expect(readFileSync(join(dest, 'file.txt'), 'utf-8')).toBe('data');
+    });
+});
+//# sourceMappingURL=bundle.test.js.map

package/dist/__tests__/bundle.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bundle.test.js","sourceRoot":"","sources":["../../src/__tests__/bundle.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAClG,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAE9D,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,MAAM,IAAI,GAAa,EAAE,CAAC;IAE1B,SAAS,MAAM;QACb,MAAM,CAAC,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC;QACrD,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACb,OAAO,CAAC,CAAC;IACX,CAAC;IAED,SAAS,CAAC,GAAG,EAAE;QACb,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrB,MAAM,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,CAAC;QAC/C,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;QAC5B,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,YAAY,CAAC,EAAE,MAAM,CAAC,CAAC;QAEtD,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAEzC,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC;QACtB,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAE5B,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrE,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,EAAE,mBAAmB,CAAC,CAAC;QAC1D,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,EAAE,KAAK,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,EAAE,YAAY,CAAC,CAAC;QAE1E,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC;QACtB,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAE5B,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAChF,kCAAkC;QAClC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,aAAa,CAAC,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QAEjC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;QACpD,aAAa,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/coordinator.test.d.ts

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

package/dist/__tests__/coordinator.test.d.ts.map

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

package/dist/__tests__/coordinator.test.js

@@ -0,0 +1,283 @@
+import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
+import { mkdirSync, rmSync } from 'node:fs';
+import { join } from 'node:path';
+import { randomUUID } from 'node:crypto';
+import { initDb, closeDb, insertSession, updateSessionStatus, updateSessionRunner } from '../db/index.js';
+import { RunnerCoordinator } from '../runner/coordinator.js';
+const TEST_DIR = join(import.meta.dirname ?? '.', '..', '..', '.test-coordinator-' + process.pid);
+describe('RunnerCoordinator (DB-backed)', () => {
+    let coordinator;
+    beforeEach(async () => {
+        mkdirSync(TEST_DIR, { recursive: true });
+        await initDb({ dataDir: TEST_DIR });
+        coordinator = new RunnerCoordinator({});
+    });
+    afterEach(async () => {
+        coordinator.stopLivenessSweep();
+        await closeDb();
+        rmSync(TEST_DIR, { recursive: true, force: true });
+    });
+    describe('registerRunner', () => {
+        it('registers a new runner and persists to DB', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-1',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            const info = await coordinator.getRunnerInfoFromDb();
+            expect(info).toHaveLength(1);
+            expect(info[0].runnerId).toBe('runner-1');
+            expect(info[0].host).toBe('host-1');
+            expect(info[0].port).toBe(4200);
+            expect(info[0].max).toBe(50);
+        });
+        it('re-registers an existing runner with updated connection info', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-1',
+                host: 'old-host',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            await coordinator.registerRunner({
+                runnerId: 'runner-1',
+                host: 'new-host',
+                port: 4201,
+                maxSandboxes: 100,
+            });
+            const info = await coordinator.getRunnerInfoFromDb();
+            expect(info).toHaveLength(1);
+            expect(info[0].host).toBe('new-host');
+            expect(info[0].port).toBe(4201);
+            expect(info[0].max).toBe(100);
+        });
+    });
+    describe('heartbeat', () => {
+        it('updates runner stats in DB', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-1',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            await coordinator.heartbeat('runner-1', {
+                total: 10, cold: 0, warming: 2, warm: 3, waiting: 3, running: 5,
+                maxCapacity: 50, resumeWarmHits: 0, resumeColdHits: 0,
+            });
+            const info = await coordinator.getRunnerInfoFromDb();
+            expect(info[0].active).toBe(5); // running count
+        });
+    });
+    describe('selectBackend', () => {
+        it('throws when no runners available and no local backend', async () => {
+            await expect(coordinator.selectBackend()).rejects.toThrow('No runners available');
+        });
+        it('falls back to local backend when no remote runners', async () => {
+            const localBackend = {
+                createSandbox: vi.fn(),
+                destroySandbox: vi.fn(),
+                destroyAll: vi.fn(),
+                sendCommand: vi.fn(),
+                interrupt: vi.fn(),
+                getSandbox: vi.fn(),
+                isSandboxAlive: vi.fn(),
+                markRunning: vi.fn(),
+                markWaiting: vi.fn(),
+                recordWarmHit: vi.fn(),
+                recordColdHit: vi.fn(),
+                persistState: vi.fn(),
+                getStats: vi.fn(),
+                activeCount: 0,
+            };
+            const coordWithLocal = new RunnerCoordinator({ localBackend });
+            const result = await coordWithLocal.selectBackend();
+            expect(result.runnerId).toBe('__local__');
+            expect(result.backend).toBe(localBackend);
+        });
+        it('selects runner with most available capacity', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-1',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            await coordinator.registerRunner({
+                runnerId: 'runner-2',
+                host: 'host-2',
+                port: 4201,
+                maxSandboxes: 100,
+            });
+            // runner-1 has 40 active sandboxes (10 available)
+            await coordinator.heartbeat('runner-1', {
+                total: 50, cold: 0, warming: 5, warm: 0, waiting: 5, running: 40,
+                maxCapacity: 50, resumeWarmHits: 0, resumeColdHits: 0,
+            });
+            // runner-2 has 10 active sandboxes (90 available)
+            await coordinator.heartbeat('runner-2', {
+                total: 100, cold: 0, warming: 0, warm: 0, waiting: 0, running: 10,
+                maxCapacity: 100, resumeWarmHits: 0, resumeColdHits: 0,
+            });
+            const result = await coordinator.selectBackend();
+            expect(result.runnerId).toBe('runner-2');
+        });
+    });
+    describe('handleDeadRunner', () => {
+        it('marks active sessions on dead runner as paused and removes runner', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-dead',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            // Create a session on this runner
+            const sessionId = randomUUID();
+            await insertSession(sessionId, 'test-agent', sessionId);
+            await updateSessionRunner(sessionId, 'runner-dead');
+            await updateSessionStatus(sessionId, 'active');
+            await coordinator.handleDeadRunner('runner-dead');
+            // Session should be paused
+            const { getSession } = await import('../db/index.js');
+            const session = await getSession(sessionId);
+            expect(session?.status).toBe('paused');
+            // Runner should be removed from DB
+            const info = await coordinator.getRunnerInfoFromDb();
+            expect(info).toHaveLength(0);
+        });
+        it('does not re-pause already paused sessions', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-dead',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            const sessionId = randomUUID();
+            await insertSession(sessionId, 'test-agent', sessionId);
+            await updateSessionRunner(sessionId, 'runner-dead');
+            await updateSessionStatus(sessionId, 'paused');
+            // Should not throw
+            await coordinator.handleDeadRunner('runner-dead');
+            const { getSession } = await import('../db/index.js');
+            const session = await getSession(sessionId);
+            expect(session?.status).toBe('paused');
+        });
+    });
+    describe('getRunnerInfoFromDb', () => {
+        it('returns all runners visible to any coordinator', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-1',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            await coordinator.registerRunner({
+                runnerId: 'runner-2',
+                host: 'host-2',
+                port: 4201,
+                maxSandboxes: 100,
+            });
+            // A second coordinator can see both runners (same DB)
+            const coordinator2 = new RunnerCoordinator({});
+            const info = await coordinator2.getRunnerInfoFromDb();
+            expect(info).toHaveLength(2);
+            const runnerIds = info.map((r) => r.runnerId).sort();
+            expect(runnerIds).toEqual(['runner-1', 'runner-2']);
+        });
+    });
+    describe('deregisterRunner', () => {
+        it('immediately pauses sessions and removes runner from DB', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-shutting-down',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            // Create two sessions: one active, one ended
+            const activeId = randomUUID();
+            const endedId = randomUUID();
+            await insertSession(activeId, 'test-agent', activeId);
+            await updateSessionRunner(activeId, 'runner-shutting-down');
+            await updateSessionStatus(activeId, 'active');
+            await insertSession(endedId, 'test-agent', endedId);
+            await updateSessionRunner(endedId, 'runner-shutting-down');
+            await updateSessionStatus(endedId, 'ended');
+            await coordinator.deregisterRunner('runner-shutting-down');
+            const { getSession } = await import('../db/index.js');
+            // Active session should be paused
+            const active = await getSession(activeId);
+            expect(active?.status).toBe('paused');
+            // Ended session should remain ended (not re-paused)
+            const ended = await getSession(endedId);
+            expect(ended?.status).toBe('ended');
+            // Runner should be gone
+            const info = await coordinator.getRunnerInfoFromDb();
+            expect(info).toHaveLength(0);
+        });
+    });
+    describe('bulkPauseSessionsByRunner', () => {
+        it('pauses multiple active sessions in a single operation', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-bulk',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            // Create 5 sessions: 3 active, 1 paused, 1 ended
+            const ids = Array.from({ length: 5 }, () => randomUUID());
+            for (const id of ids) {
+                await insertSession(id, 'test-agent', id);
+                await updateSessionRunner(id, 'runner-bulk');
+            }
+            await updateSessionStatus(ids[0], 'active');
+            await updateSessionStatus(ids[1], 'active');
+            await updateSessionStatus(ids[2], 'active');
+            await updateSessionStatus(ids[3], 'paused');
+            await updateSessionStatus(ids[4], 'ended');
+            await coordinator.handleDeadRunner('runner-bulk');
+            const { getSession } = await import('../db/index.js');
+            // All 3 active sessions should be paused
+            for (const id of ids.slice(0, 3)) {
+                const session = await getSession(id);
+                expect(session?.status).toBe('paused');
+            }
+            // Already paused stays paused
+            expect((await getSession(ids[3]))?.status).toBe('paused');
+            // Ended stays ended
+            expect((await getSession(ids[4]))?.status).toBe('ended');
+        });
+    });
+    describe('multi-coordinator consistency', () => {
+        it('second coordinator can select runner registered by first', async () => {
+            // First coordinator registers a runner
+            await coordinator.registerRunner({
+                runnerId: 'runner-shared',
+                host: 'shared-host',
+                port: 4200,
+                maxSandboxes: 100,
+            });
+            // Second coordinator should be able to select it
+            const coordinator2 = new RunnerCoordinator({});
+            const result = await coordinator2.selectBackend();
+            expect(result.runnerId).toBe('runner-shared');
+        });
+        it('dead runner handling is idempotent across coordinators', async () => {
+            await coordinator.registerRunner({
+                runnerId: 'runner-dying',
+                host: 'host-1',
+                port: 4200,
+                maxSandboxes: 50,
+            });
+            const sessionId = randomUUID();
+            await insertSession(sessionId, 'test-agent', sessionId);
+            await updateSessionRunner(sessionId, 'runner-dying');
+            await updateSessionStatus(sessionId, 'active');
+            const coordinator2 = new RunnerCoordinator({});
+            // Both coordinators handle the dead runner — should be idempotent
+            await coordinator.handleDeadRunner('runner-dying');
+            await coordinator2.handleDeadRunner('runner-dying');
+            const { getSession } = await import('../db/index.js');
+            const session = await getSession(sessionId);
+            expect(session?.status).toBe('paused');
+        });
+    });
+});
+//# sourceMappingURL=coordinator.test.js.map

package/dist/__tests__/coordinator.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"coordinator.test.js","sourceRoot":"","sources":["../../src/__tests__/coordinator.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAC1G,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAE7D,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;AAElG,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC7C,IAAI,WAA8B,CAAC;IAEnC,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzC,MAAM,MAAM,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpC,WAAW,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,WAAW,CAAC,iBAAiB,EAAE,CAAC;QAChC,MAAM,OAAO,EAAE,CAAC;QAChB,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACrD,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,UAAU;gBAChB,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,GAAG;aAClB,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACrD,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,WAAW,CAAC,SAAS,CAAC,UAAU,EAAE;gBACtC,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC;gBAC/D,WAAW,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC;aACtD,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACrD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAE,gBAAgB;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,MAAM,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QACpF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,YAAY,GAAG;gBACnB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;gBACtB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;gBACvB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;gBACnB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;gBACpB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;gBAClB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;gBACnB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;gBACvB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;gBACpB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;gBACpB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;gBACtB,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;gBACtB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;gBACrB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;gBACjB,WAAW,EAAE,CAAC;aACf,CAAC;YAEF,MAAM,cAAc,GAAG,IAAI,iBAAiB,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,aAAa,EAAE,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,GAAG;aAClB,CAAC,CAAC;YAEH,kDAAkD;YAClD,MAAM,WAAW,CAAC,SAAS,CAAC,UAAU,EAAE;gBACtC,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE;gBAChE,WAAW,EAAE,EAAE,EAAE,cAAc,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC;aACtD,CAAC,CAAC;YAEH,kDAAkD;YAClD,MAAM,WAAW,CAAC,SAAS,CAAC,UAAU,EAAE;gBACtC,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE;gBACjE,WAAW,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC;aACvD,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,aAAa,EAAE,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,aAAa;gBACvB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,kCAAkC;YAClC,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;YAC/B,MAAM,aAAa,CAAC,SAAS,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;YACxD,MAAM,mBAAmB,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;YACpD,MAAM,mBAAmB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAE/C,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;YAElD,2BAA2B;YAC3B,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAEvC,mCAAmC;YACnC,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACrD,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,aAAa;gBACvB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;YAC/B,MAAM,aAAa,CAAC,SAAS,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;YACxD,MAAM,mBAAmB,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;YACpD,MAAM,mBAAmB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAE/C,mBAAmB;YACnB,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;YAElD,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,UAAU;gBACpB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,GAAG;aAClB,CAAC,CAAC;YAEH,sDAAsD;YACtD,MAAM,YAAY,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,mBAAmB,EAAE,CAAC;YACtD,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAE7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;YACrD,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,sBAAsB;gBAChC,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,6CAA6C;YAC7C,MAAM,QAAQ,GAAG,UAAU,EAAE,CAAC;YAC9B,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;YAC7B,MAAM,aAAa,CAAC,QAAQ,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;YACtD,MAAM,mBAAmB,CAAC,QAAQ,EAAE,sBAAsB,CAAC,CAAC;YAC5D,MAAM,mBAAmB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAE9C,MAAM,aAAa,CAAC,OAAO,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;YACpD,MAAM,mBAAmB,CAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;YAC3D,MAAM,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE5C,MAAM,WAAW,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,CAAC;YAE3D,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACtD,kCAAkC;YAClC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAEtC,oDAAoD;YACpD,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,CAAC;YACxC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEpC,wBAAwB;YACxB,MAAM,IAAI,GAAG,MAAM,WAAW,CAAC,mBAAmB,EAAE,CAAC;YACrD,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,aAAa;gBACvB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,iDAAiD;YACjD,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC;YAC1D,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;gBACrB,MAAM,aAAa,CAAC,EAAE,EAAE,YAAY,EAAE,EAAE,CAAC,CAAC;gBAC1C,MAAM,mBAAmB,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;YAC/C,CAAC;YACD,MAAM,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC5C,MAAM,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC5C,MAAM,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC5C,MAAM,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAC5C,MAAM,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YAE3C,MAAM,WAAW,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;YAElD,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACtD,yCAAyC;YACzC,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;gBACjC,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,EAAE,CAAC,CAAC;gBACrC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACzC,CAAC;YACD,8BAA8B;YAC9B,MAAM,CAAC,CAAC,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1D,oBAAoB;YACpB,MAAM,CAAC,CAAC,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,uCAAuC;YACvC,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,eAAe;gBACzB,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,GAAG;aAClB,CAAC,CAAC;YAEH,iDAAiD;YACjD,MAAM,YAAY,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,aAAa,EAAE,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,WAAW,CAAC,cAAc,CAAC;gBAC/B,QAAQ,EAAE,cAAc;gBACxB,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,IAAI;gBACV,YAAY,EAAE,EAAE;aACjB,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;YAC/B,MAAM,aAAa,CAAC,SAAS,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;YACxD,MAAM,mBAAmB,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YACrD,MAAM,mBAAmB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAE/C,MAAM,YAAY,GAAG,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE/C,kEAAkE;YAClE,MAAM,WAAW,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;YACnD,MAAM,YAAY,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;YAEpD,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,CAAC;YAC5C,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/crypto.test.d.ts

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

package/dist/__tests__/crypto.test.d.ts.map

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

package/dist/__tests__/crypto.test.js

@@ -0,0 +1,72 @@
+import { describe, it, expect } from 'vitest';
+import { encrypt, decrypt } from '../crypto.js';
+import { createCipheriv, randomBytes, createHash } from 'node:crypto';
+describe('credential encryption', () => {
+    const masterKey = 'test-master-key-for-unit-tests';
+    it('round-trips a plaintext key', () => {
+        const plain = 'sk-ant-abc123xyz';
+        const { encrypted, iv, authTag, salt } = encrypt(plain, masterKey);
+        expect(salt).toBeDefined();
+        const decrypted = decrypt(encrypted, iv, authTag, masterKey, salt);
+        expect(decrypted).toBe(plain);
+    });
+    it('produces different ciphertext for same plaintext (unique IV and salt)', () => {
+        const plain = 'sk-ant-abc123xyz';
+        const a = encrypt(plain, masterKey);
+        const b = encrypt(plain, masterKey);
+        expect(a.encrypted).not.toBe(b.encrypted);
+        expect(a.iv).not.toBe(b.iv);
+        expect(a.salt).not.toBe(b.salt);
+    });
+    it('fails to decrypt with wrong master key', () => {
+        const plain = 'sk-ant-abc123xyz';
+        const { encrypted, iv, authTag, salt } = encrypt(plain, masterKey);
+        expect(() => decrypt(encrypted, iv, authTag, 'wrong-key', salt)).toThrow();
+    });
+    it('fails to decrypt with tampered authTag', () => {
+        const plain = 'sk-ant-abc123xyz';
+        const { encrypted, iv, authTag, salt } = encrypt(plain, masterKey);
+        const buf = Buffer.from(authTag, 'base64');
+        buf[0] ^= 0xff;
+        const tampered = buf.toString('base64');
+        expect(() => decrypt(encrypted, iv, tampered, masterKey, salt)).toThrow();
+    });
+    it('handles empty string', () => {
+        const plain = '';
+        const { encrypted, iv, authTag, salt } = encrypt(plain, masterKey);
+        const decrypted = decrypt(encrypted, iv, authTag, masterKey, salt);
+        expect(decrypted).toBe('');
+    });
+    it('handles long keys', () => {
+        const plain = 'sk-' + 'a'.repeat(1000);
+        const { encrypted, iv, authTag, salt } = encrypt(plain, masterKey);
+        const decrypted = decrypt(encrypted, iv, authTag, masterKey, salt);
+        expect(decrypted).toBe(plain);
+    });
+    it('decrypts legacy credentials (no salt, SHA-256 derived key)', () => {
+        // Simulate legacy encryption: unsalted SHA-256 key derivation
+        const plain = 'sk-ant-legacy-key';
+        const legacyKey = createHash('sha256').update(masterKey).digest();
+        const iv = randomBytes(12);
+        const cipher = createCipheriv('aes-256-gcm', legacyKey, iv);
+        let encrypted = cipher.update(plain, 'utf8', 'base64');
+        encrypted += cipher.final('base64');
+        const authTag = cipher.getAuthTag();
+        // Decrypt with salt=null triggers legacy path
+        const decrypted = decrypt(encrypted, iv.toString('base64'), authTag.toString('base64'), masterKey, null);
+        expect(decrypted).toBe(plain);
+    });
+    it('decrypt with salt=undefined also triggers legacy path', () => {
+        const plain = 'sk-ant-legacy-key-2';
+        const legacyKey = createHash('sha256').update(masterKey).digest();
+        const iv = randomBytes(12);
+        const cipher = createCipheriv('aes-256-gcm', legacyKey, iv);
+        let encrypted = cipher.update(plain, 'utf8', 'base64');
+        encrypted += cipher.final('base64');
+        const authTag = cipher.getAuthTag();
+        // Omit salt entirely
+        const decrypted = decrypt(encrypted, iv.toString('base64'), authTag.toString('base64'), masterKey);
+        expect(decrypted).toBe(plain);
+    });
+});
+//# sourceMappingURL=crypto.test.js.map

package/dist/__tests__/crypto.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"crypto.test.js","sourceRoot":"","sources":["../../src/__tests__/crypto.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEtE,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,MAAM,SAAS,GAAG,gCAAgC,CAAC;IAEnD,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,KAAK,GAAG,kBAAkB,CAAC;QACjC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACnE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAC3B,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QACnE,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,KAAK,GAAG,kBAAkB,CAAC;QACjC,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACpC,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACpC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC5B,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,KAAK,GAAG,kBAAkB,CAAC;QACjC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACnE,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,KAAK,GAAG,kBAAkB,CAAC;QACjC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACnE,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC3C,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;QACf,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACxC,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,KAAK,GAAG,EAAE,CAAC;QACjB,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACnE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QACnE,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC3B,MAAM,KAAK,GAAG,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACnE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QACnE,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,8DAA8D;QAC9D,MAAM,KAAK,GAAG,mBAAmB,CAAC;QAClC,MAAM,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,CAAC;QAClE,MAAM,EAAE,GAAG,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,MAAM,GAAG,cAAc,CAAC,aAAa,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QAC5D,IAAI,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACvD,SAAS,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAEpC,8CAA8C;QAC9C,MAAM,SAAS,GAAG,OAAO,CACvB,SAAS,EACT,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACrB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAC1B,SAAS,EACT,IAAI,CACL,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,KAAK,GAAG,qBAAqB,CAAC;QACpC,MAAM,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,EAAE,CAAC;QAClE,MAAM,EAAE,GAAG,WAAW,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,MAAM,GAAG,cAAc,CAAC,aAAa,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC;QAC5D,IAAI,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACvD,SAAS,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QAEpC,qBAAqB;QACrB,MAAM,SAAS,GAAG,OAAO,CACvB,SAAS,EACT,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EACrB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAC1B,SAAS,CACV,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/file-store.test.d.ts

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

package/dist/__tests__/file-store.test.d.ts.map

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