@pocketcoder/host 0.0.2

package/dist/services/__tests__/message-queue.test.js

@@ -0,0 +1,203 @@
+import { afterEach, describe, expect, it, vi } from "vitest";
+import { initQueue, enqueueMessage, startDrain, isProcessing, getQueueSize, clearQueue, removeSession, clearAllQueues, } from "../message-queue.js";
+vi.mock("../../utils/logger.js", () => ({
+    logger: { debug: vi.fn(), info: vi.fn(), warn: vi.fn(), error: vi.fn() },
+}));
+function makeMessage(agentSessionId, content = "test", requestId = "req-1") {
+    return {
+        projectPath: "/tmp/project",
+        content,
+        sessionId: "session-1",
+        requestId,
+        agentSessionId,
+    };
+}
+describe("message-queue", () => {
+    afterEach(() => {
+        clearAllQueues();
+    });
+    describe("initQueue", () => {
+        it("should initialize a queue for a session", () => {
+            const processFn = vi.fn();
+            initQueue("session-a", processFn);
+            expect(getQueueSize("session-a")).toBe(0);
+            expect(isProcessing("session-a")).toBe(false);
+        });
+        it("should not overwrite existing queue", () => {
+            const processFn1 = vi.fn();
+            const processFn2 = vi.fn();
+            initQueue("session-a", processFn1);
+            enqueueMessage(makeMessage("session-a", "msg1"));
+            initQueue("session-a", processFn2);
+            // Queue should still have the message
+            expect(getQueueSize("session-a")).toBe(1);
+        });
+    });
+    describe("enqueueMessage", () => {
+        it("should return 'started' for first message when not processing", () => {
+            initQueue("session-a", vi.fn());
+            const result = enqueueMessage(makeMessage("session-a"));
+            expect(result).toBe("started");
+        });
+        it("should queue requests per project/session", () => {
+            initQueue("session-a", vi.fn());
+            initQueue("session-b", vi.fn());
+            enqueueMessage(makeMessage("session-a", "msg1", "r1"));
+            enqueueMessage(makeMessage("session-a", "msg2", "r2"));
+            enqueueMessage(makeMessage("session-b", "msg3", "r3"));
+            expect(getQueueSize("session-a")).toBe(2);
+            expect(getQueueSize("session-b")).toBe(1);
+        });
+        it("should reject when queue full (50 messages)", () => {
+            initQueue("session-a", vi.fn());
+            // Fill the queue to max
+            for (let i = 0; i < 50; i++) {
+                enqueueMessage(makeMessage("session-a", `msg-${i}`, `req-${i}`));
+            }
+            expect(getQueueSize("session-a")).toBe(50);
+            // 51st message should be rejected
+            const result = enqueueMessage(makeMessage("session-a", "overflow", "req-overflow"));
+            expect(result).toBe("full");
+            expect(getQueueSize("session-a")).toBe(50);
+        });
+        it("should return 'full' for uninitialized session", () => {
+            const result = enqueueMessage(makeMessage("unknown-session"));
+            expect(result).toBe("full");
+        });
+    });
+    describe("startDrain / sequential processing", () => {
+        it("should process messages in FIFO order", async () => {
+            const processed = [];
+            const processFn = async (msg) => {
+                processed.push(msg.content);
+            };
+            initQueue("session-a", processFn);
+            enqueueMessage(makeMessage("session-a", "first", "r1"));
+            enqueueMessage(makeMessage("session-a", "second", "r2"));
+            enqueueMessage(makeMessage("session-a", "third", "r3"));
+            startDrain("session-a");
+            // Wait for async drain to complete
+            await vi.waitFor(() => {
+                expect(processed).toEqual(["first", "second", "third"]);
+            });
+        });
+        it("should process sequentially (one at a time)", async () => {
+            let concurrency = 0;
+            let maxConcurrency = 0;
+            const processFn = async () => {
+                concurrency++;
+                maxConcurrency = Math.max(maxConcurrency, concurrency);
+                // Simulate async work
+                await new Promise((resolve) => setTimeout(resolve, 10));
+                concurrency--;
+            };
+            initQueue("session-a", processFn);
+            enqueueMessage(makeMessage("session-a", "msg1", "r1"));
+            enqueueMessage(makeMessage("session-a", "msg2", "r2"));
+            enqueueMessage(makeMessage("session-a", "msg3", "r3"));
+            startDrain("session-a");
+            await vi.waitFor(() => {
+                expect(isProcessing("session-a")).toBe(false);
+            }, { timeout: 1000 });
+            expect(maxConcurrency).toBe(1);
+            expect(getQueueSize("session-a")).toBe(0);
+        });
+        it("should mark isProcessing during drain", async () => {
+            let resolveProcess;
+            const processFn = async () => {
+                await new Promise((resolve) => {
+                    resolveProcess = resolve;
+                });
+            };
+            initQueue("session-a", processFn);
+            enqueueMessage(makeMessage("session-a"));
+            expect(isProcessing("session-a")).toBe(false);
+            startDrain("session-a");
+            // Give the drain a tick to start
+            await new Promise((resolve) => setTimeout(resolve, 0));
+            expect(isProcessing("session-a")).toBe(true);
+            // Complete the processing
+            resolveProcess();
+            await vi.waitFor(() => {
+                expect(isProcessing("session-a")).toBe(false);
+            });
+        });
+        it("should return 'queued' when already processing", async () => {
+            const resolvers = [];
+            const processFn = async () => {
+                await new Promise((resolve) => {
+                    resolvers.push(resolve);
+                });
+            };
+            initQueue("session-a", processFn);
+            enqueueMessage(makeMessage("session-a", "first", "r1"));
+            startDrain("session-a");
+            // Wait for processing to start
+            await new Promise((resolve) => setTimeout(resolve, 0));
+            // Second message should be queued
+            const result = enqueueMessage(makeMessage("session-a", "second", "r2"));
+            expect(result).toBe("queued");
+            // Resolve all pending processors to cleanup
+            for (const resolve of resolvers)
+                resolve();
+            // Give drain a moment to pick up the second message
+            await new Promise((resolve) => setTimeout(resolve, 0));
+            for (const resolve of resolvers)
+                resolve();
+            await vi.waitFor(() => {
+                expect(isProcessing("session-a")).toBe(false);
+            });
+        });
+        it("should handle processing errors without stopping drain", async () => {
+            const processed = [];
+            const processFn = async (msg) => {
+                if (msg.content === "error") {
+                    throw new Error("processing error");
+                }
+                processed.push(msg.content);
+            };
+            initQueue("session-a", processFn);
+            enqueueMessage(makeMessage("session-a", "first", "r1"));
+            enqueueMessage(makeMessage("session-a", "error", "r2"));
+            enqueueMessage(makeMessage("session-a", "third", "r3"));
+            startDrain("session-a");
+            await vi.waitFor(() => {
+                expect(isProcessing("session-a")).toBe(false);
+            });
+            // first and third processed, error skipped
+            expect(processed).toEqual(["first", "third"]);
+        });
+    });
+    describe("clearQueue", () => {
+        it("should clear pending messages", () => {
+            initQueue("session-a", vi.fn());
+            enqueueMessage(makeMessage("session-a", "msg1", "r1"));
+            enqueueMessage(makeMessage("session-a", "msg2", "r2"));
+            expect(getQueueSize("session-a")).toBe(2);
+            clearQueue("session-a");
+            expect(getQueueSize("session-a")).toBe(0);
+        });
+    });
+    describe("removeSession", () => {
+        it("should remove all state for a session", () => {
+            initQueue("session-a", vi.fn());
+            enqueueMessage(makeMessage("session-a"));
+            expect(getQueueSize("session-a")).toBe(1);
+            removeSession("session-a");
+            expect(getQueueSize("session-a")).toBe(0);
+            expect(isProcessing("session-a")).toBe(false);
+        });
+    });
+    describe("clearAllQueues", () => {
+        it("should clear all session queues", () => {
+            initQueue("session-a", vi.fn());
+            initQueue("session-b", vi.fn());
+            enqueueMessage(makeMessage("session-a"));
+            enqueueMessage(makeMessage("session-b"));
+            clearAllQueues();
+            expect(getQueueSize("session-a")).toBe(0);
+            expect(getQueueSize("session-b")).toBe(0);
+        });
+    });
+});
+//# sourceMappingURL=message-queue.test.js.map

package/dist/services/__tests__/message-queue.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-queue.test.js","sourceRoot":"","sources":["../../../src/services/__tests__/message-queue.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE7D,OAAO,EACL,SAAS,EACT,cAAc,EACd,UAAU,EACV,YAAY,EACZ,YAAY,EACZ,UAAU,EACV,aAAa,EACb,cAAc,GACf,MAAM,qBAAqB,CAAC;AAG7B,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;CACzE,CAAC,CAAC,CAAC;AAEJ,SAAS,WAAW,CAAC,cAAsB,EAAE,OAAO,GAAG,MAAM,EAAE,SAAS,GAAG,OAAO;IAChF,OAAO;QACL,WAAW,EAAE,cAAc;QAC3B,OAAO;QACP,SAAS,EAAE,WAAW;QACtB,SAAS;QACT,cAAc;KACf,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,SAAS,CAAC,GAAG,EAAE;QACb,cAAc,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1B,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAClC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,SAAS,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;YACnC,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC;YACjD,SAAS,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;YACnC,sCAAsC;YACtC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAEhC,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YACvD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YACvD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YAEvD,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAEhC,wBAAwB;YACxB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC5B,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YACnE,CAAC;YACD,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAE3C,kCAAkC;YAClC,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC;YACpF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,SAAS,GAAa,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAc,KAAK,EAAE,GAAG,EAAE,EAAE;gBACzC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC9B,CAAC,CAAC;YAEF,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAClC,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YACxD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;YACzD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YAExD,UAAU,CAAC,WAAW,CAAC,CAAC;YAExB,mCAAmC;YACnC,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YAC1D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,cAAc,GAAG,CAAC,CAAC;YAEvB,MAAM,SAAS,GAAc,KAAK,IAAI,EAAE;gBACtC,WAAW,EAAE,CAAC;gBACd,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;gBACvD,sBAAsB;gBACtB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;gBACxD,WAAW,EAAE,CAAC;YAChB,CAAC,CAAC;YAEF,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAClC,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YACvD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YACvD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YAEvD,UAAU,CAAC,WAAW,CAAC,CAAC;YAExB,MAAM,EAAE,CAAC,OAAO,CACd,GAAG,EAAE;gBACH,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC,EACD,EAAE,OAAO,EAAE,IAAI,EAAE,CAClB,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,IAAI,cAAwC,CAAC;YAC7C,MAAM,SAAS,GAAc,KAAK,IAAI,EAAE;gBACtC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;oBAClC,cAAc,GAAG,OAAO,CAAC;gBAC3B,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YAEF,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAClC,cAAc,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YAEzC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9C,UAAU,CAAC,WAAW,CAAC,CAAC;YAExB,iCAAiC;YACjC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACvD,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE7C,0BAA0B;YAC1B,cAAe,EAAE,CAAC;YAClB,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,SAAS,GAAsB,EAAE,CAAC;YACxC,MAAM,SAAS,GAAc,KAAK,IAAI,EAAE;gBACtC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;oBAClC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC1B,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YAEF,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAClC,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YACxD,UAAU,CAAC,WAAW,CAAC,CAAC;YAExB,+BAA+B;YAC/B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YAEvD,kCAAkC;YAClC,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAE9B,4CAA4C;YAC5C,KAAK,MAAM,OAAO,IAAI,SAAS;gBAAE,OAAO,EAAE,CAAC;YAC3C,oDAAoD;YACpD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACvD,KAAK,MAAM,OAAO,IAAI,SAAS;gBAAE,OAAO,EAAE,CAAC;YAC3C,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,SAAS,GAAa,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAc,KAAK,EAAE,GAAG,EAAE,EAAE;gBACzC,IAAI,GAAG,CAAC,OAAO,KAAK,OAAO,EAAE,CAAC;oBAC5B,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;gBACtC,CAAC;gBACD,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC9B,CAAC,CAAC;YAEF,SAAS,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;YAClC,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YACxD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YACxD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;YAExD,UAAU,CAAC,WAAW,CAAC,CAAC;YAExB,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE;gBACpB,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;YAEH,2CAA2C;YAC3C,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YACvD,cAAc,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;YACvD,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE1C,UAAU,CAAC,WAAW,CAAC,CAAC;YACxB,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,cAAc,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAE1C,aAAa,CAAC,WAAW,CAAC,CAAC;YAC3B,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,SAAS,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAChC,cAAc,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YACzC,cAAc,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,CAAC;YAEzC,cAAc,EAAE,CAAC;YACjB,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/__tests__/queue-overflow.test.d.ts

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

package/dist/services/__tests__/queue-overflow.test.d.ts.map

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

package/dist/services/__tests__/queue-overflow.test.js

@@ -0,0 +1,115 @@
+import { mkdtemp, rm, mkdir } from "node:fs/promises";
+import { join } from "node:path";
+import { tmpdir } from "node:os";
+import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
+import { ErrorCode, MAX_QUEUE_PER_PROJECT } from "@pocketcoder/shared/constants";
+import { openDatabase, closeDatabase } from "../../db/database.js";
+import { createProject } from "../../db/repositories/projects.js";
+import { handleAgentMessage } from "../../handlers/agent-message.js";
+import { clearAllQueues } from "../message-queue.js";
+vi.mock("../../utils/logger.js", () => ({
+    logger: { debug: vi.fn(), info: vi.fn(), warn: vi.fn(), error: vi.fn() },
+}));
+vi.mock("../../opencode/proxy.js", () => ({
+    proxyToOpenCode: vi.fn(),
+}));
+import { proxyToOpenCode } from "../../opencode/proxy.js";
+const mockProxyToOpenCode = vi.mocked(proxyToOpenCode);
+describe("queue overflow handling", () => {
+    let tempDir;
+    let projectDir;
+    let dbPath;
+    let projectId;
+    const agentSessionId = "overflow-test-session";
+    beforeEach(async () => {
+        tempDir = await mkdtemp(join(tmpdir(), "pocketcoder-queue-overflow-test-"));
+        projectDir = join(tempDir, "project");
+        dbPath = join(tempDir, "daemon.db");
+        await mkdir(projectDir);
+        closeDatabase();
+        openDatabase(dbPath);
+        const project = createProject("test-project", projectDir);
+        projectId = project.project_id;
+        mockProxyToOpenCode.mockReset();
+        clearAllQueues();
+    });
+    afterEach(async () => {
+        closeDatabase();
+        clearAllQueues();
+        await rm(tempDir, { recursive: true, force: true });
+    });
+    function createSendCollector() {
+        const messages = [];
+        return { messages, send: (msg) => messages.push(msg) };
+    }
+    async function fillQueue(send) {
+        // Block first call so all subsequent messages queue
+        mockProxyToOpenCode.mockImplementation(async () => new Promise(() => {
+            /* never resolves */
+        }));
+        // Send first message — starts processing (occupies the processor)
+        const first = {
+            type: "agent_message",
+            requestId: "req-first",
+            sessionId: projectId,
+            payload: { agentSessionId, content: "First" },
+        };
+        await handleAgentMessage(first, send);
+        // Fill queue to the limit
+        for (let i = 0; i < MAX_QUEUE_PER_PROJECT; i++) {
+            const msg = {
+                type: "agent_message",
+                requestId: `req-fill-${i}`,
+                sessionId: projectId,
+                payload: { agentSessionId, content: `Fill ${i}` },
+            };
+            await handleAgentMessage(msg, send);
+        }
+    }
+    it("should reject when queue full", async () => {
+        const { messages, send } = createSendCollector();
+        await fillQueue(send);
+        // Try to add one more — should be rejected
+        const overflow = {
+            type: "agent_message",
+            requestId: "req-overflow",
+            sessionId: projectId,
+            payload: { agentSessionId, content: "Overflow" },
+        };
+        await handleAgentMessage(overflow, send);
+        const resp = messages.find((m) => m.requestId === "req-overflow");
+        expect(resp).toBeDefined();
+        expect(resp.error).toBeDefined();
+        expect(resp.payload).toBeUndefined();
+    });
+    it("should return DAEMON_BUSY code", async () => {
+        const { messages, send } = createSendCollector();
+        await fillQueue(send);
+        const overflow = {
+            type: "agent_message",
+            requestId: "req-overflow-code",
+            sessionId: projectId,
+            payload: { agentSessionId, content: "Overflow" },
+        };
+        await handleAgentMessage(overflow, send);
+        const resp = messages.find((m) => m.requestId === "req-overflow-code");
+        expect(resp.error.code).toBe(ErrorCode.DAEMON_BUSY);
+        expect(resp.error.message).toBe("Message queue is full");
+    });
+    it("should include retry hint", async () => {
+        const { messages, send } = createSendCollector();
+        await fillQueue(send);
+        const overflow = {
+            type: "agent_message",
+            requestId: "req-overflow-hint",
+            sessionId: projectId,
+            payload: { agentSessionId, content: "Overflow" },
+        };
+        await handleAgentMessage(overflow, send);
+        const resp = messages.find((m) => m.requestId === "req-overflow-hint");
+        expect(resp.error.details).toBeDefined();
+        expect(resp.error.details.queueSize).toBe(MAX_QUEUE_PER_PROJECT);
+        expect(resp.error.details.retryAfterMs).toBe(5000);
+    });
+});
+//# sourceMappingURL=queue-overflow.test.js.map

package/dist/services/__tests__/queue-overflow.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"queue-overflow.test.js","sourceRoot":"","sources":["../../../src/services/__tests__/queue-overflow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAGzE,OAAO,EAAE,SAAS,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAEjF,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;CACzE,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;CACzB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,MAAM,mBAAmB,GAAG,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;AAEvD,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,OAAe,CAAC;IACpB,IAAI,UAAkB,CAAC;IACvB,IAAI,MAAc,CAAC;IACnB,IAAI,SAAiB,CAAC;IAEtB,MAAM,cAAc,GAAG,uBAAuB,CAAC;IAE/C,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,kCAAkC,CAAC,CAAC,CAAC;QAC5E,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QACtC,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAEpC,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC;QACxB,aAAa,EAAE,CAAC;QAChB,YAAY,CAAC,MAAM,CAAC,CAAC;QAErB,MAAM,OAAO,GAAG,aAAa,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;QAC1D,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC;QAE/B,mBAAmB,CAAC,SAAS,EAAE,CAAC;QAChC,cAAc,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,aAAa,EAAE,CAAC;QAChB,cAAc,EAAE,CAAC;QACjB,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,SAAS,mBAAmB;QAI1B,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,GAAY,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;IAClE,CAAC;IAED,KAAK,UAAU,SAAS,CAAC,IAA4B;QACnD,oDAAoD;QACpD,mBAAmB,CAAC,kBAAkB,CACpC,KAAK,IAAI,EAAE,CACT,IAAI,OAAO,CAAC,GAAG,EAAE;YACf,oBAAoB;QACtB,CAAC,CAAC,CACL,CAAC;QAEF,kEAAkE;QAClE,MAAM,KAAK,GAAwB;YACjC,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,WAAW;YACtB,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE;SAC9C,CAAC;QACF,MAAM,kBAAkB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAEtC,0BAA0B;QAC1B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,qBAAqB,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/C,MAAM,GAAG,GAAwB;gBAC/B,IAAI,EAAE,eAAe;gBACrB,SAAS,EAAE,YAAY,CAAC,EAAE;gBAC1B,SAAS,EAAE,SAAS;gBACpB,OAAO,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,EAAE;aAClD,CAAC;YACF,MAAM,kBAAkB,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,mBAAmB,EAAE,CAAC;QAEjD,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC;QAEtB,2CAA2C;QAC3C,MAAM,QAAQ,GAAwB;YACpC,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,cAAc;YACzB,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE;SACjD,CAAC;QACF,MAAM,kBAAkB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAEzC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CACxB,CAAC,CAAC,EAAE,EAAE,CAAE,CAA0B,CAAC,SAAS,KAAK,cAAc,CACxC,CAAC;QAE1B,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QACjC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,mBAAmB,EAAE,CAAC;QAEjD,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC;QAEtB,MAAM,QAAQ,GAAwB;YACpC,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,mBAAmB;YAC9B,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE;SACjD,CAAC;QACF,MAAM,kBAAkB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAEzC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CACxB,CAAC,CAAC,EAAE,EAAE,CAAE,CAA0B,CAAC,SAAS,KAAK,mBAAmB,CAC7C,CAAC;QAE1B,MAAM,CAAC,IAAI,CAAC,KAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,CAAC,IAAI,CAAC,KAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,mBAAmB,EAAE,CAAC;QAEjD,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC;QAEtB,MAAM,QAAQ,GAAwB;YACpC,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,mBAAmB;YAC9B,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE,UAAU,EAAE;SACjD,CAAC;QACF,MAAM,kBAAkB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAEzC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CACxB,CAAC,CAAC,EAAE,EAAE,CAAE,CAA0B,CAAC,SAAS,KAAK,mBAAmB,CAC7C,CAAC;QAE1B,MAAM,CAAC,IAAI,CAAC,KAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC,KAAM,CAAC,OAAQ,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnE,MAAM,CAAC,IAAI,CAAC,KAAM,CAAC,OAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/__tests__/shutdown.test.d.ts

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

package/dist/services/__tests__/shutdown.test.d.ts.map

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

package/dist/services/__tests__/shutdown.test.js

@@ -0,0 +1,181 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+vi.mock("../../utils/logger.js", () => ({
+    logger: { debug: vi.fn(), info: vi.fn(), warn: vi.fn(), error: vi.fn() },
+}));
+vi.mock("../../db/database.js", () => ({
+    closeDatabase: vi.fn(),
+}));
+vi.mock("../../utils/pid.js", () => ({
+    removePidFile: vi.fn().mockResolvedValue(undefined),
+}));
+import { registerShutdownHandler, clearShutdownHandlers, gracefulShutdown, isShutdownInProgress, trackInflight, getInflightCount, } from "../shutdown.js";
+import { closeDatabase } from "../../db/database.js";
+import { removePidFile } from "../../utils/pid.js";
+describe("shutdown orchestrator", () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        clearShutdownHandlers();
+        vi.useRealTimers();
+    });
+    it("should stop accepting new requests by setting shutdown flag", async () => {
+        expect(isShutdownInProgress()).toBe(false);
+        const p = gracefulShutdown();
+        expect(isShutdownInProgress()).toBe(true);
+        await vi.runAllTimersAsync();
+        await p;
+    });
+    it("should run registered handlers in reverse order (LIFO)", async () => {
+        const order = [];
+        registerShutdownHandler("first", () => {
+            order.push("first");
+        });
+        registerShutdownHandler("second", () => {
+            order.push("second");
+        });
+        registerShutdownHandler("third", () => {
+            order.push("third");
+        });
+        const p = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p;
+        expect(order).toEqual(["third", "second", "first"]);
+    });
+    it("should close database and remove PID file", async () => {
+        const p = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p;
+        expect(closeDatabase).toHaveBeenCalled();
+        expect(removePidFile).toHaveBeenCalled();
+    });
+    it("should wait for in-flight requests before running handlers", async () => {
+        const events = [];
+        const release = trackInflight();
+        expect(getInflightCount()).toBe(1);
+        registerShutdownHandler("handler", () => {
+            events.push("handler-ran");
+        });
+        const p = gracefulShutdown();
+        // Advance time a little — handler should NOT have run yet
+        await vi.advanceTimersByTimeAsync(100);
+        expect(events).toEqual([]);
+        // Release the in-flight request
+        release();
+        expect(getInflightCount()).toBe(0);
+        // Now advance past all timers to complete shutdown
+        await vi.advanceTimersByTimeAsync(100);
+        await vi.runAllTimersAsync();
+        await p;
+        expect(events).toEqual(["handler-ran"]);
+    });
+    it("should drain multiple in-flight requests", async () => {
+        const release1 = trackInflight();
+        const release2 = trackInflight();
+        const release3 = trackInflight();
+        expect(getInflightCount()).toBe(3);
+        const p = gracefulShutdown();
+        release1();
+        release2();
+        expect(getInflightCount()).toBe(1);
+        release3();
+        expect(getInflightCount()).toBe(0);
+        await vi.runAllTimersAsync();
+        await p;
+    });
+    it("should timeout in-flight drain and proceed with shutdown", async () => {
+        // Start an in-flight request that never completes
+        trackInflight();
+        expect(getInflightCount()).toBe(1);
+        registerShutdownHandler("handler", vi.fn());
+        const p = gracefulShutdown();
+        // Advance past the in-flight drain timeout (3 seconds)
+        await vi.advanceTimersByTimeAsync(3_000);
+        // Advance past the overall shutdown timeout to fully complete
+        await vi.advanceTimersByTimeAsync(5_000);
+        await p;
+        // Shutdown should have completed despite the stuck in-flight request
+        expect(closeDatabase).toHaveBeenCalled();
+        expect(removePidFile).toHaveBeenCalled();
+    });
+    it("should close terminal sessions via registered handler", async () => {
+        const killAllTerminals = vi.fn();
+        registerShutdownHandler("terminal sessions", killAllTerminals);
+        const p = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p;
+        expect(killAllTerminals).toHaveBeenCalled();
+    });
+    it("should terminate OpenCode gracefully via registered handler", async () => {
+        const stopOpenCode = vi.fn().mockResolvedValue(undefined);
+        registerShutdownHandler("opencode process", stopOpenCode);
+        const p = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p;
+        expect(stopOpenCode).toHaveBeenCalled();
+    });
+    it("should remove PID file during shutdown", async () => {
+        const p = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p;
+        expect(removePidFile).toHaveBeenCalled();
+    });
+    it("should force complete after overall timeout", async () => {
+        // Register a handler that takes too long
+        registerShutdownHandler("slow handler", async () => {
+            await new Promise((resolve) => setTimeout(resolve, 10_000));
+        });
+        const p = gracefulShutdown();
+        // Advance past the overall shutdown timeout (5 seconds)
+        await vi.advanceTimersByTimeAsync(5_000);
+        await p;
+        // Should still clean up final resources even after timeout
+        expect(closeDatabase).toHaveBeenCalled();
+        expect(removePidFile).toHaveBeenCalled();
+    });
+    it("should be idempotent — second call is a no-op", async () => {
+        const handler = vi.fn();
+        registerShutdownHandler("handler", handler);
+        const p1 = gracefulShutdown();
+        const p2 = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p1;
+        await p2;
+        expect(handler).toHaveBeenCalledTimes(1);
+    });
+    it("should continue shutdown even if a handler throws", async () => {
+        const events = [];
+        registerShutdownHandler("first", () => {
+            events.push("first");
+        });
+        registerShutdownHandler("failing", () => {
+            throw new Error("handler failure");
+        });
+        registerShutdownHandler("third", () => {
+            events.push("third");
+        });
+        const p = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p;
+        // Third runs first (LIFO), then failing throws, then first runs
+        expect(events).toEqual(["third", "first"]);
+        expect(closeDatabase).toHaveBeenCalled();
+    });
+    it("should handle trackInflight release being called multiple times", () => {
+        const release = trackInflight();
+        expect(getInflightCount()).toBe(1);
+        release();
+        expect(getInflightCount()).toBe(0);
+        release(); // Double release should be safe
+        expect(getInflightCount()).toBe(0);
+    });
+    it("should exit with code 0 on success (handlers complete normally)", async () => {
+        registerShutdownHandler("quick", vi.fn());
+        const p = gracefulShutdown();
+        await vi.runAllTimersAsync();
+        await p;
+        // If gracefulShutdown resolves without throwing, it's a success (code 0 path)
+        expect(isShutdownInProgress()).toBe(true);
+    });
+});
+//# sourceMappingURL=shutdown.test.js.map

package/dist/services/__tests__/shutdown.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdown.test.js","sourceRoot":"","sources":["../../../src/services/__tests__/shutdown.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEzE,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE;CACzE,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;CACvB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;CACpD,CAAC,CAAC,CAAC;AAEJ,OAAO,EACL,uBAAuB,EACvB,qBAAqB,EACrB,gBAAgB,EAChB,oBAAoB,EACpB,aAAa,EACb,gBAAgB,GACjB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,qBAAqB,EAAE,CAAC;QACxB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;IACV,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,uBAAuB,CAAC,OAAO,EAAE,GAAG,EAAE;YACpC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC,CAAC,CAAC;QACH,uBAAuB,CAAC,QAAQ,EAAE,GAAG,EAAE;YACrC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;QACH,uBAAuB,CAAC,OAAO,EAAE,GAAG,EAAE;YACpC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACzC,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,MAAM,GAAa,EAAE,CAAC;QAE5B,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;QAChC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEnC,uBAAuB,CAAC,SAAS,EAAE,GAAG,EAAE;YACtC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAE7B,0DAA0D;QAC1D,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAE3B,gCAAgC;QAChC,OAAO,EAAE,CAAC;QACV,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEnC,mDAAmD;QACnD,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;QACvC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,QAAQ,GAAG,aAAa,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAG,aAAa,EAAE,CAAC;QACjC,MAAM,QAAQ,GAAG,aAAa,EAAE,CAAC;QACjC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEnC,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAE7B,QAAQ,EAAE,CAAC;QACX,QAAQ,EAAE,CAAC;QACX,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEnC,QAAQ,EAAE,CAAC;QACX,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEnC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;IACV,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,kDAAkD;QAClD,aAAa,EAAE,CAAC;QAChB,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAEnC,uBAAuB,CAAC,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAE5C,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAE7B,uDAAuD;QACvD,MAAM,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QAEzC,8DAA8D;QAC9D,MAAM,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;QAER,qEAAqE;QACrE,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACzC,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACjC,uBAAuB,CAAC,mBAAmB,EAAE,gBAAgB,CAAC,CAAC;QAE/D,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,MAAM,CAAC,gBAAgB,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC1D,uBAAuB,CAAC,kBAAkB,EAAE,YAAY,CAAC,CAAC;QAE1D,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,MAAM,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,yCAAyC;QACzC,uBAAuB,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAE7B,wDAAwD;QACxD,MAAM,EAAE,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;QAER,2DAA2D;QAC3D,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACzC,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,uBAAuB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAE5C,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QAC9B,MAAM,EAAE,GAAG,gBAAgB,EAAE,CAAC;QAC9B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC;QACT,MAAM,EAAE,CAAC;QAET,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,uBAAuB,CAAC,OAAO,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;QACH,uBAAuB,CAAC,SAAS,EAAE,GAAG,EAAE;YACtC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QACH,uBAAuB,CAAC,OAAO,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,gEAAgE;QAChE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACzE,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;QAChC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,OAAO,EAAE,CAAC;QACV,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,OAAO,EAAE,CAAC,CAAC,gCAAgC;QAC3C,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,uBAAuB,CAAC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAE1C,MAAM,CAAC,GAAG,gBAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC7B,MAAM,CAAC,CAAC;QAER,8EAA8E;QAC9E,MAAM,CAAC,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/__tests__/terminal-exit.test.d.ts

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

package/dist/services/__tests__/terminal-exit.test.d.ts.map

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