@ash-ai/server 0.0.2 → 0.0.4

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,45 @@
+import { describe, it, expect } from 'vitest';
+import { encrypt, decrypt } from '../crypto.js';
+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 } = encrypt(plain, masterKey);
+        const decrypted = decrypt(encrypted, iv, authTag, masterKey);
+        expect(decrypted).toBe(plain);
+    });
+    it('produces different ciphertext for same plaintext (unique IV)', () => {
+        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);
+    });
+    it('fails to decrypt with wrong master key', () => {
+        const plain = 'sk-ant-abc123xyz';
+        const { encrypted, iv, authTag } = encrypt(plain, masterKey);
+        expect(() => decrypt(encrypted, iv, authTag, 'wrong-key')).toThrow();
+    });
+    it('fails to decrypt with tampered authTag', () => {
+        const plain = 'sk-ant-abc123xyz';
+        const { encrypted, iv, authTag } = encrypt(plain, masterKey);
+        // Flip bits in the auth tag — guaranteed to fail GCM authentication
+        const buf = Buffer.from(authTag, 'base64');
+        buf[0] ^= 0xff;
+        const tampered = buf.toString('base64');
+        expect(() => decrypt(encrypted, iv, tampered, masterKey)).toThrow();
+    });
+    it('handles empty string', () => {
+        const plain = '';
+        const { encrypted, iv, authTag } = encrypt(plain, masterKey);
+        const decrypted = decrypt(encrypted, iv, authTag, masterKey);
+        expect(decrypted).toBe('');
+    });
+    it('handles long keys', () => {
+        const plain = 'sk-' + 'a'.repeat(1000);
+        const { encrypted, iv, authTag } = encrypt(plain, masterKey);
+        const decrypted = decrypt(encrypted, iv, authTag, 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;AAEhD,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,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAC7D,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,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;IAC9B,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,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC7D,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IACvE,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,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC7D,oEAAoE;QACpE,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,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;IACtE,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,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAC7D,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,GAAG,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QAC7D,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":""}

package/dist/__tests__/file-store.test.js

@@ -0,0 +1,105 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { tmpdir } from 'node:os';
+import { mkdtempSync, rmSync } from 'node:fs';
+import { join } from 'node:path';
+import { LocalFileStore } from '../../../sandbox/src/file-store-local.js';
+import { createFileStore } from '../../../sandbox/src/file-store.js';
+describe('LocalFileStore', () => {
+    let dir;
+    let store;
+    beforeEach(() => {
+        dir = mkdtempSync(join(tmpdir(), 'ash-filestore-'));
+        store = new LocalFileStore(dir);
+    });
+    afterEach(() => {
+        rmSync(dir, { recursive: true, force: true });
+    });
+    it('put and get round-trip', async () => {
+        const content = Buffer.from('hello world');
+        await store.put('test.txt', content);
+        const got = await store.get('test.txt');
+        expect(got).not.toBeNull();
+        expect(got.toString()).toBe('hello world');
+    });
+    it('get returns null for missing key', async () => {
+        const got = await store.get('nonexistent.txt');
+        expect(got).toBeNull();
+    });
+    it('exists returns true for existing file', async () => {
+        await store.put('exists.txt', Buffer.from('data'));
+        expect(await store.exists('exists.txt')).toBe(true);
+        expect(await store.exists('nope.txt')).toBe(false);
+    });
+    it('delete removes a file', async () => {
+        await store.put('to-delete.txt', Buffer.from('data'));
+        expect(await store.exists('to-delete.txt')).toBe(true);
+        await store.delete('to-delete.txt');
+        expect(await store.exists('to-delete.txt')).toBe(false);
+    });
+    it('delete is a no-op for missing file', async () => {
+        // Should not throw
+        await store.delete('nonexistent.txt');
+    });
+    it('put creates nested directories', async () => {
+        await store.put('a/b/c/deep.txt', Buffer.from('deep'));
+        const got = await store.get('a/b/c/deep.txt');
+        expect(got.toString()).toBe('deep');
+    });
+    it('list returns all files', async () => {
+        await store.put('file1.txt', Buffer.from('1'));
+        await store.put('dir/file2.txt', Buffer.from('2'));
+        await store.put('dir/sub/file3.txt', Buffer.from('3'));
+        const files = await store.list();
+        const keys = files.map(f => f.key).sort();
+        expect(keys).toEqual(['dir/file2.txt', 'dir/sub/file3.txt', 'file1.txt']);
+    });
+    it('list with prefix filters to subdirectory', async () => {
+        await store.put('a/1.txt', Buffer.from('1'));
+        await store.put('a/2.txt', Buffer.from('2'));
+        await store.put('b/3.txt', Buffer.from('3'));
+        const files = await store.list('a');
+        const keys = files.map(f => f.key).sort();
+        expect(keys).toEqual(['a/1.txt', 'a/2.txt']);
+    });
+    it('list returns empty array for missing prefix', async () => {
+        const files = await store.list('nonexistent');
+        expect(files).toEqual([]);
+    });
+    it('rejects path traversal', async () => {
+        await expect(store.put('../escape.txt', Buffer.from('bad'))).rejects.toThrow('path traversal');
+    });
+    it('list includes file metadata', async () => {
+        await store.put('meta.txt', Buffer.from('some content'));
+        const files = await store.list();
+        expect(files).toHaveLength(1);
+        expect(files[0].key).toBe('meta.txt');
+        expect(files[0].size).toBe(12); // 'some content'.length
+        expect(files[0].lastModified).toBeTruthy();
+    });
+});
+describe('createFileStore', () => {
+    it('creates LocalFileStore for file:// URL', async () => {
+        const dir = mkdtempSync(join(tmpdir(), 'ash-fs-factory-'));
+        const store = await createFileStore(`file://${dir}`);
+        await store.put('hello.txt', Buffer.from('world'));
+        const got = await store.get('hello.txt');
+        expect(got.toString()).toBe('world');
+        rmSync(dir, { recursive: true, force: true });
+    });
+    it('throws for unsupported scheme', async () => {
+        await expect(createFileStore('ftp://bad')).rejects.toThrow('Unsupported file store URL scheme');
+    });
+    it('throws for s3:// without AWS SDK installed', async () => {
+        // S3FileStore requires @aws-sdk/client-s3 — this test verifies error handling
+        // In CI without the SDK, it should give a helpful error
+        // (If SDK IS installed, this would succeed, so we just verify it doesn't crash unexpectedly)
+        try {
+            await createFileStore('s3://test-bucket/prefix/');
+            // If it doesn't throw, the SDK is available — that's fine
+        }
+        catch (err) {
+            expect(err.message).toMatch(/aws-sdk|S3/i);
+        }
+    });
+});
+//# sourceMappingURL=file-store.test.js.map