@pocketcoder/host 0.0.2

package/dist/gateway/__tests__/websocket-client.test.js

@@ -0,0 +1,249 @@
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import { WebSocketServer } from "ws";
+import { GatewayClient } from "../websocket-client.js";
+/**
+ * Helper: create a local WebSocket server on a random port for testing.
+ * Returns the server and its URL.
+ */
+async function createTestServer() {
+    const wss = new WebSocketServer({ port: 0, host: "127.0.0.1" });
+    await new Promise((resolve) => {
+        wss.on("listening", () => resolve());
+    });
+    const { port } = wss.address();
+    return { wss, url: `ws://127.0.0.1:${port}`, port };
+}
+/** Helper: wait for the client to reach a specific state */
+function waitForState(client, state, timeoutMs = 5000) {
+    return new Promise((resolve, reject) => {
+        const timer = setTimeout(() => reject(new Error(`Timed out waiting for state: ${state}`)), timeoutMs);
+        if (client.state === state) {
+            clearTimeout(timer);
+            resolve();
+            return;
+        }
+        const event = state === "connected" ? "connected" : "disconnected";
+        const handler = () => {
+            clearTimeout(timer);
+            client.off(event, handler);
+            resolve();
+        };
+        client.on(event, handler);
+    });
+}
+describe("GatewayClient", () => {
+    let wss;
+    let url;
+    let client;
+    beforeEach(async () => {
+        const server = await createTestServer();
+        wss = server.wss;
+        url = server.url;
+        client = new GatewayClient({ url });
+    });
+    afterEach(async () => {
+        client.disconnect();
+        await new Promise((resolve) => {
+            wss.close(() => resolve());
+        });
+    });
+    it("should connect to WebSocket server", async () => {
+        client.connect();
+        await waitForState(client, "connected");
+        expect(client.state).toBe("connected");
+        expect(client.connected).toBe(true);
+    });
+    it("should send JSON messages", async () => {
+        const received = new Promise((resolve) => {
+            wss.on("connection", (ws) => {
+                ws.on("message", (data) => {
+                    resolve(data.toString());
+                });
+            });
+        });
+        client.connect();
+        await waitForState(client, "connected");
+        const message = {
+            type: "daemon_register",
+            payload: {
+                machineKey: "test-key",
+                token: "test-token",
+                machine: {
+                    os: "linux",
+                    osVersion: "Ubuntu 22.04",
+                    hostname: "test-host",
+                    username: "test-user",
+                    label: null,
+                },
+                projects: [],
+            },
+        };
+        client.send(message);
+        const data = await received;
+        expect(JSON.parse(data)).toEqual(message);
+    });
+    it("should receive and parse messages", async () => {
+        const messageReceived = new Promise((resolve) => {
+            client.on("message", (msg) => {
+                if (msg.type !== "ping") {
+                    resolve(msg);
+                }
+            });
+        });
+        wss.on("connection", (ws) => {
+            const response = {
+                type: "machines_list_response",
+                requestId: "req-1",
+                payload: { machines: [] },
+            };
+            ws.send(JSON.stringify(response));
+        });
+        client.connect();
+        await waitForState(client, "connected");
+        const msg = await messageReceived;
+        expect(msg.type).toBe("machines_list_response");
+        expect(msg.requestId).toBe("req-1");
+    });
+    it("should handle disconnection", async () => {
+        const disconnected = new Promise((resolve) => {
+            client.on("disconnected", (code, reason) => {
+                resolve({ code, reason });
+            });
+        });
+        wss.on("connection", (ws) => {
+            // Close from server side after brief delay
+            setTimeout(() => ws.close(1000, "server closing"), 50);
+        });
+        client.connect();
+        await waitForState(client, "connected");
+        const result = await disconnected;
+        expect(result.code).toBe(1000);
+        expect(client.state).toBe("disconnected");
+        expect(client.connected).toBe(false);
+    });
+    it("should handle connection errors", async () => {
+        // Close the server before connecting so the connection fails
+        await new Promise((resolve) => {
+            wss.close(() => resolve());
+        });
+        const errorReceived = new Promise((resolve) => {
+            client.on("error", (err) => {
+                resolve(err);
+            });
+        });
+        client.connect();
+        const err = await errorReceived;
+        expect(err).toBeInstanceOf(Error);
+    });
+    it("should auto-respond to ping with pong", async () => {
+        const pongReceived = new Promise((resolve) => {
+            wss.on("connection", (ws) => {
+                ws.on("message", (data) => {
+                    const msg = JSON.parse(data.toString());
+                    if (msg.type === "pong") {
+                        resolve(msg.type);
+                    }
+                });
+                // Send ping after connection
+                setTimeout(() => ws.send(JSON.stringify({ type: "ping" })), 50);
+            });
+        });
+        client.connect();
+        await waitForState(client, "connected");
+        const type = await pongReceived;
+        expect(type).toBe("pong");
+    });
+    it("should emit message event for ping", async () => {
+        const pingReceived = new Promise((resolve) => {
+            client.on("message", (msg) => {
+                if (msg.type === "ping") {
+                    resolve(msg);
+                }
+            });
+        });
+        wss.on("connection", (ws) => {
+            setTimeout(() => ws.send(JSON.stringify({ type: "ping" })), 50);
+        });
+        client.connect();
+        await waitForState(client, "connected");
+        const msg = await pingReceived;
+        expect(msg.type).toBe("ping");
+    });
+    it("should not connect if already connected", async () => {
+        client.connect();
+        await waitForState(client, "connected");
+        // Second connect should be a no-op
+        client.connect();
+        expect(client.state).toBe("connected");
+    });
+    it("should throw when sending on disconnected client", () => {
+        const message = {
+            type: "daemon_register",
+            payload: {
+                machineKey: "k",
+                token: "t",
+                machine: {
+                    os: "linux",
+                    osVersion: "",
+                    hostname: "",
+                    username: "",
+                    label: null,
+                },
+                projects: [],
+            },
+        };
+        expect(() => client.send(message)).toThrow("WebSocket is not connected");
+    });
+    it("should disconnect gracefully", async () => {
+        client.connect();
+        await waitForState(client, "connected");
+        client.disconnect();
+        expect(client.state).toBe("disconnected");
+        expect(client.connected).toBe(false);
+    });
+    it("should handle disconnect when already disconnected", () => {
+        // Should not throw
+        client.disconnect();
+        expect(client.state).toBe("disconnected");
+    });
+    describe("ping()", () => {
+        it("should return true when server responds to ping", async () => {
+            wss.on("connection", (ws) => {
+                // ws library auto-responds to WebSocket protocol pings with pongs
+                ws.on("ping", () => {
+                    ws.pong();
+                });
+            });
+            client.connect();
+            await waitForState(client, "connected");
+            const result = await client.ping(3000);
+            expect(result).toBe(true);
+        });
+        it("should return false when not connected", async () => {
+            const result = await client.ping(1000);
+            expect(result).toBe(false);
+        });
+        it("should return false when ping times out", async () => {
+            let serverWs = null;
+            wss.on("connection", (ws) => {
+                serverWs = ws;
+            });
+            client.connect();
+            await waitForState(client, "connected");
+            // Simulate network partition - terminate the server socket without close handshake
+            // but before the ping can complete
+            serverWs.terminate();
+            // Give the client a moment to not know about the termination
+            const result = await client.ping(100); // Short timeout
+            expect(result).toBe(false);
+        });
+        it("should return false after disconnect", async () => {
+            client.connect();
+            await waitForState(client, "connected");
+            client.disconnect();
+            const result = await client.ping(1000);
+            expect(result).toBe(false);
+        });
+    });
+});
+//# sourceMappingURL=websocket-client.test.js.map

package/dist/gateway/__tests__/websocket-client.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket-client.test.js","sourceRoot":"","sources":["../../../src/gateway/__tests__/websocket-client.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAErE,OAAO,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AAIrC,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAEvD;;;GAGG;AACH,KAAK,UAAU,gBAAgB;IAK7B,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;IAChE,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;QAClC,GAAG,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IACH,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,CAAC,OAAO,EAAiB,CAAC;IAC9C,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,kBAAkB,IAAI,EAAE,EAAE,IAAI,EAAE,CAAC;AACtD,CAAC;AAED,4DAA4D;AAC5D,SAAS,YAAY,CACnB,MAAqB,EACrB,KAAmC,EACnC,SAAS,GAAG,IAAI;IAEhB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,KAAK,GAAG,UAAU,CACtB,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,gCAAgC,KAAK,EAAE,CAAC,CAAC,EAChE,SAAS,CACV,CAAC;QAEF,IAAI,MAAM,CAAC,KAAK,KAAK,KAAK,EAAE,CAAC;YAC3B,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,KAAK,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,cAAc,CAAC;QACnE,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC3B,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,GAAoB,CAAC;IACzB,IAAI,GAAW,CAAC;IAChB,IAAI,MAAqB,CAAC;IAE1B,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,MAAM,GAAG,MAAM,gBAAgB,EAAE,CAAC;QACxC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACjB,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACjB,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,CAAC,UAAU,EAAE,CAAC;QACpB,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,QAAQ,GAAG,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,EAAE;YAC/C,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;gBAC1B,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;oBACxB,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC3B,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE;gBACP,UAAU,EAAE,UAAU;gBACtB,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE;oBACP,EAAE,EAAE,OAAO;oBACX,SAAS,EAAE,cAAc;oBACzB,QAAQ,EAAE,WAAW;oBACrB,QAAQ,EAAE,WAAW;oBACrB,KAAK,EAAE,IAAI;iBACZ;gBACD,QAAQ,EAAE,EAAE;aACb;SACF,CAAC;QAEF,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,eAAe,GAAG,IAAI,OAAO,CAAU,CAAC,OAAO,EAAE,EAAE;YACvD,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,OAAO,CAAC,GAAc,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;YAC1B,MAAM,QAAQ,GAAY;gBACxB,IAAI,EAAE,wBAAwB;gBAC9B,SAAS,EAAE,OAAO;gBAClB,OAAO,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;aAC1B,CAAC;YACF,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,GAAG,GAAG,MAAM,eAAe,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAChD,MAAM,CAAE,GAA6B,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,YAAY,GAAG,IAAI,OAAO,CAAmC,CAAC,OAAO,EAAE,EAAE;YAC7E,MAAM,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;gBACzC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5B,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;YAC1B,2CAA2C;YAC3C,UAAU,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,gBAAgB,CAAC,EAAE,EAAE,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,6DAA6D;QAC7D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,OAAO,CAAQ,CAAC,OAAO,EAAE,EAAE;YACnD,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACzB,OAAO,CAAC,GAAG,CAAC,CAAC;YACf,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,EAAE,CAAC;QAEjB,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC;QAChC,MAAM,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,YAAY,GAAG,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,EAAE;YACnD,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;gBAC1B,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;oBACxB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;oBACxC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACxB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBACpB,CAAC;gBACH,CAAC,CAAC,CAAC;gBACH,6BAA6B;gBAC7B,UAAU,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,YAAY,GAAG,IAAI,OAAO,CAAmB,CAAC,OAAO,EAAE,EAAE;YAC7D,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,OAAO,CAAC,GAAG,CAAC,CAAC;gBACf,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;YAC1B,UAAU,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC;QAC/B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,mCAAmC;QACnC,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,OAAO,GAAY;YACvB,IAAI,EAAE,iBAAiB;YACvB,OAAO,EAAE;gBACP,UAAU,EAAE,GAAG;gBACf,KAAK,EAAE,GAAG;gBACV,OAAO,EAAE;oBACP,EAAE,EAAE,OAAO;oBACX,SAAS,EAAE,EAAE;oBACb,QAAQ,EAAE,EAAE;oBACZ,QAAQ,EAAE,EAAE;oBACZ,KAAK,EAAE,IAAI;iBACZ;gBACD,QAAQ,EAAE,EAAE;aACb;SACF,CAAC;QAEF,MAAM,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,4BAA4B,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,CAAC,OAAO,EAAE,CAAC;QACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QAExC,MAAM,CAAC,UAAU,EAAE,CAAC;QACpB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,mBAAmB;QACnB,MAAM,CAAC,UAAU,EAAE,CAAC;QACpB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;gBAC1B,kEAAkE;gBAClE,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;oBACjB,EAAE,CAAC,IAAI,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAExC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,IAAI,QAAQ,GAAqB,IAAI,CAAC;YACtC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;gBAC1B,QAAQ,GAAG,EAAE,CAAC;YAChB,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAExC,mFAAmF;YACnF,mCAAmC;YACnC,QAAS,CAAC,SAAS,EAAE,CAAC;YAEtB,6DAA6D;YAC7D,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,gBAAgB;YACvD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACxC,MAAM,CAAC,UAAU,EAAE,CAAC;YAEpB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/gateway/gateway-request.d.ts

@@ -0,0 +1,17 @@
+/**
+ * Utility for making request-response calls through the gateway.
+ *
+ * Some handlers (e.g., machines_list) need to forward a request to the gateway
+ * and wait for the corresponding response.  This module provides a simple
+ * promise-based mechanism backed by a one-time message listener.
+ */
+import type { Message } from "@pocketcoder/shared";
+import type { GatewayClient } from "./websocket-client.js";
+/**
+ * Send a request message through the gateway and wait for a response whose
+ * `type` matches `expectedResponseType` and `requestId` matches.
+ *
+ * Rejects if no response arrives within `timeoutMs` (default 15 s).
+ */
+export declare function gatewayRequest(client: GatewayClient, request: Message, expectedResponseType: string, timeoutMs?: number): Promise<Message>;
+//# sourceMappingURL=gateway-request.d.ts.map

package/dist/gateway/gateway-request.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gateway-request.d.ts","sourceRoot":"","sources":["../../src/gateway/gateway-request.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAEnD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAK3D;;;;;GAKG;AACH,wBAAgB,cAAc,CAC5B,MAAM,EAAE,aAAa,EACrB,OAAO,EAAE,OAAO,EAChB,oBAAoB,EAAE,MAAM,EAC5B,SAAS,SAAqB,GAC7B,OAAO,CAAC,OAAO,CAAC,CA0BlB"}

package/dist/gateway/gateway-request.js

@@ -0,0 +1,39 @@
+/**
+ * Utility for making request-response calls through the gateway.
+ *
+ * Some handlers (e.g., machines_list) need to forward a request to the gateway
+ * and wait for the corresponding response.  This module provides a simple
+ * promise-based mechanism backed by a one-time message listener.
+ */
+const DEFAULT_TIMEOUT_MS = 15_000;
+/**
+ * Send a request message through the gateway and wait for a response whose
+ * `type` matches `expectedResponseType` and `requestId` matches.
+ *
+ * Rejects if no response arrives within `timeoutMs` (default 15 s).
+ */
+export function gatewayRequest(client, request, expectedResponseType, timeoutMs = DEFAULT_TIMEOUT_MS) {
+    return new Promise((resolve, reject) => {
+        const requestId = request.requestId;
+        const cleanup = () => {
+            client.off("message", listener);
+            clearTimeout(timer);
+        };
+        const listener = (msg) => {
+            if (msg.type === "ping")
+                return;
+            const m = msg;
+            if (m.type === expectedResponseType && m.requestId === requestId) {
+                cleanup();
+                resolve(m);
+            }
+        };
+        const timer = setTimeout(() => {
+            cleanup();
+            reject(new Error(`Gateway request timed out after ${timeoutMs}ms`));
+        }, timeoutMs);
+        client.on("message", listener);
+        client.send(request);
+    });
+}
+//# sourceMappingURL=gateway-request.js.map

package/dist/gateway/gateway-request.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gateway-request.js","sourceRoot":"","sources":["../../src/gateway/gateway-request.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAOH,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAElC;;;;;GAKG;AACH,MAAM,UAAU,cAAc,CAC5B,MAAqB,EACrB,OAAgB,EAChB,oBAA4B,EAC5B,SAAS,GAAG,kBAAkB;IAE9B,OAAO,IAAI,OAAO,CAAU,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC9C,MAAM,SAAS,GAAI,OAA8C,CAAC,SAAmB,CAAC;QAEtF,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAChC,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,CAAC,GAA0B,EAAQ,EAAE;YACpD,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;gBAAE,OAAO;YAChC,MAAM,CAAC,GAAG,GAAuC,CAAC;YAClD,IAAI,CAAC,CAAC,IAAI,KAAK,oBAAoB,IAAI,CAAC,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;gBACjE,OAAO,EAAE,CAAC;gBACV,OAAO,CAAC,CAAC,CAAC,CAAC;YACb,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,OAAO,EAAE,CAAC;YACV,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,SAAS,IAAI,CAAC,CAAC,CAAC;QACtE,CAAC,EAAE,SAAS,CAAC,CAAC;QAEd,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/gateway/message-router.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Message router for incoming gateway messages.
+ *
+ * Dispatches incoming messages to appropriate handlers based on message type.
+ * Errors in handlers are caught and logged, never crashing the router.
+ */
+import type { Message, MessageType, RequestMessage } from "@pocketcoder/shared";
+import type { GatewayClient } from "./websocket-client.js";
+/**
+ * Handler function signature.
+ * Receives the parsed request message and a send function for responses.
+ * The handler can send one or more messages back (e.g., streaming events + final response).
+ */
+export type MessageHandler = (message: RequestMessage, send: (response: Message) => void) => Promise<void>;
+export declare class MessageRouter {
+    private handlers;
+    private client;
+    /** Register a handler for a specific message type */
+    registerHandler(type: MessageType, handler: MessageHandler): void;
+    /** Remove a handler for a specific message type */
+    removeHandler(type: MessageType): void;
+    /** Attach to a GatewayClient and start routing messages */
+    attach(client: GatewayClient): void;
+    /** Detach from the current GatewayClient */
+    detach(): void;
+    /** The bound message handler for the gateway client */
+    private handleMessage;
+    /** Send a generic error response */
+    private sendErrorResponse;
+}
+//# sourceMappingURL=message-router.d.ts.map

package/dist/gateway/message-router.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-router.d.ts","sourceRoot":"","sources":["../../src/gateway/message-router.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,EAAiB,MAAM,qBAAqB,CAAC;AAK/F,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAG3D;;;;GAIG;AACH,MAAM,MAAM,cAAc,GAAG,CAC3B,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,CAAC,QAAQ,EAAE,OAAO,KAAK,IAAI,KAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;AAEnB,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAA0C;IAC1D,OAAO,CAAC,MAAM,CAA8B;IAE5C,qDAAqD;IACrD,eAAe,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc,GAAG,IAAI;IAIjE,mDAAmD;IACnD,aAAa,CAAC,IAAI,EAAE,WAAW,GAAG,IAAI;IAItC,2DAA2D;IAC3D,MAAM,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI;IAKnC,4CAA4C;IAC5C,MAAM,IAAI,IAAI;IAOd,uDAAuD;IACvD,OAAO,CAAC,aAAa,CAyDnB;IAEF,oCAAoC;IACpC,OAAO,CAAC,iBAAiB;CAU1B"}

package/dist/gateway/message-router.js

@@ -0,0 +1,91 @@
+/**
+ * Message router for incoming gateway messages.
+ *
+ * Dispatches incoming messages to appropriate handlers based on message type.
+ * Errors in handlers are caught and logged, never crashing the router.
+ */
+import { ErrorCode } from "@pocketcoder/shared/constants";
+import { logger } from "../utils/logger.js";
+import { isShutdownInProgress, trackInflight } from "../services/shutdown.js";
+export class MessageRouter {
+    handlers = new Map();
+    client = null;
+    /** Register a handler for a specific message type */
+    registerHandler(type, handler) {
+        this.handlers.set(type, handler);
+    }
+    /** Remove a handler for a specific message type */
+    removeHandler(type) {
+        this.handlers.delete(type);
+    }
+    /** Attach to a GatewayClient and start routing messages */
+    attach(client) {
+        this.client = client;
+        client.on("message", this.handleMessage);
+    }
+    /** Detach from the current GatewayClient */
+    detach() {
+        if (this.client) {
+            this.client.off("message", this.handleMessage);
+            this.client = null;
+        }
+    }
+    /** The bound message handler for the gateway client */
+    handleMessage = (message) => {
+        // Ignore ping messages (handled by GatewayClient itself)
+        if (message.type === "ping") {
+            return;
+        }
+        const msg = message;
+        // Skip response and event messages — the daemon only handles requests
+        if (msg.type.endsWith("_response") || msg.type === "error") {
+            return;
+        }
+        // Reject new requests during shutdown
+        if (isShutdownInProgress()) {
+            if ("requestId" in msg && typeof msg.requestId === "string") {
+                this.sendErrorResponse(msg.requestId, ErrorCode.DAEMON_BUSY, "Daemon is shutting down");
+            }
+            return;
+        }
+        const handler = this.handlers.get(msg.type);
+        if (!handler) {
+            logger.warn(`Unknown message type: ${msg.type}`);
+            // Send error response if the message has a requestId
+            if ("requestId" in msg && typeof msg.requestId === "string") {
+                this.sendErrorResponse(msg.requestId, ErrorCode.INVALID_REQUEST, `Unknown message type: ${msg.type}`);
+            }
+            return;
+        }
+        const send = (response) => {
+            if (this.client) {
+                this.client.send(response);
+            }
+        };
+        // Track in-flight request for graceful shutdown draining
+        const release = trackInflight();
+        // Execute handler, catching any errors
+        handler(msg, send)
+            .catch((err) => {
+            const errorMessage = err instanceof Error ? err.message : String(err);
+            logger.error(`Handler error for ${msg.type}: ${errorMessage}`);
+            // Send error response if the message has a requestId
+            if ("requestId" in msg && typeof msg.requestId === "string") {
+                this.sendErrorResponse(msg.requestId, ErrorCode.UNKNOWN_ERROR, errorMessage);
+            }
+        })
+            .finally(release);
+    };
+    /** Send a generic error response */
+    sendErrorResponse(requestId, code, message) {
+        if (!this.client)
+            return;
+        const errorResponse = {
+            type: "error",
+            requestId,
+            error: { code, message },
+        };
+        this.client.send(errorResponse);
+    }
+}
+//# sourceMappingURL=message-router.js.map

package/dist/gateway/message-router.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"message-router.js","sourceRoot":"","sources":["../../src/gateway/message-router.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,SAAS,EAAE,MAAM,+BAA+B,CAAC;AAE1D,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,oBAAoB,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAc9E,MAAM,OAAO,aAAa;IAChB,QAAQ,GAAG,IAAI,GAAG,EAA+B,CAAC;IAClD,MAAM,GAAyB,IAAI,CAAC;IAE5C,qDAAqD;IACrD,eAAe,CAAC,IAAiB,EAAE,OAAuB;QACxD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,mDAAmD;IACnD,aAAa,CAAC,IAAiB;QAC7B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,2DAA2D;IAC3D,MAAM,CAAC,MAAqB;QAC1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IAC3C,CAAC;IAED,4CAA4C;IAC5C,MAAM;QACJ,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;YAC/C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;IACH,CAAC;IAED,uDAAuD;IAC/C,aAAa,GAAG,CAAC,OAA8B,EAAQ,EAAE;QAC/D,yDAAyD;QACzD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,GAAG,GAAG,OAAkB,CAAC;QAE/B,sEAAsE;QACtE,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC3D,OAAO;QACT,CAAC;QAED,sCAAsC;QACtC,IAAI,oBAAoB,EAAE,EAAE,CAAC;YAC3B,IAAI,WAAW,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC5D,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE,yBAAyB,CAAC,CAAC;YAC1F,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAmB,CAAC,CAAC;QAE3D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,yBAAyB,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;YACjD,qDAAqD;YACrD,IAAI,WAAW,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC5D,IAAI,CAAC,iBAAiB,CACpB,GAAG,CAAC,SAAS,EACb,SAAS,CAAC,eAAe,EACzB,yBAAyB,GAAG,CAAC,IAAI,EAAE,CACpC,CAAC;YACJ,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,QAAiB,EAAQ,EAAE;YACvC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC,CAAC;QAEF,yDAAyD;QACzD,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;QAEhC,uCAAuC;QACvC,OAAO,CAAC,GAAqB,EAAE,IAAI,CAAC;aACjC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;YACtB,MAAM,YAAY,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACtE,MAAM,CAAC,KAAK,CAAC,qBAAqB,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC,CAAC;YAE/D,qDAAqD;YACrD,IAAI,WAAW,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC5D,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YAC/E,CAAC;QACH,CAAC,CAAC;aACD,OAAO,CAAC,OAAO,CAAC,CAAC;IACtB,CAAC,CAAC;IAEF,oCAAoC;IAC5B,iBAAiB,CAAC,SAAiB,EAAE,IAAY,EAAE,OAAe;QACxE,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QAEzB,MAAM,aAAa,GAAkB;YACnC,IAAI,EAAE,OAAO;YACb,SAAS;YACT,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;SACzB,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAClC,CAAC;CACF"}

package/dist/gateway/messages/__tests__/daemon-register.test.d.ts

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

package/dist/gateway/messages/__tests__/daemon-register.test.d.ts.map

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

package/dist/gateway/messages/__tests__/daemon-register.test.js

@@ -0,0 +1,113 @@
+import { describe, expect, it, vi, beforeEach } from "vitest";
+// Mock dependencies
+vi.mock("../../../auth/token-storage.js");
+vi.mock("../../../db/repositories/projects.js");
+vi.mock("../../../utils/machine-info.js");
+const { loadTokens } = await import("../../../auth/token-storage.js");
+const { listProjects } = await import("../../../db/repositories/projects.js");
+const { getMachineMetadata } = await import("../../../utils/machine-info.js");
+const { buildDaemonRegisterMessage, sendDaemonRegister } = await import("../daemon-register.js");
+const mockLoadTokens = vi.mocked(loadTokens);
+const mockListProjects = vi.mocked(listProjects);
+const mockGetMachineMetadata = vi.mocked(getMachineMetadata);
+const sampleTokens = {
+    machine_key: "550e8400-e29b-41d4-a716-446655440000",
+    access_token: "eyJhbGciOiJIUzI1NiJ9.test",
+    refresh_token: "refresh-token-123",
+    expires_at: Date.now() / 1000 + 3600,
+    user_id: "user-uuid-123",
+};
+const sampleMachine = {
+    os: "linux",
+    osVersion: "Linux 6.1.0",
+    hostname: "dev-machine",
+    username: "testuser",
+    label: null,
+};
+const sampleProjects = [
+    {
+        project_id: "proj-1",
+        name: "my-app",
+        path: "/home/user/my-app",
+        created_at: "2026-01-01T00:00:00.000Z",
+        last_opened_at: null,
+    },
+    {
+        project_id: "proj-2",
+        name: "api-server",
+        path: "/home/user/api-server",
+        created_at: "2026-01-02T00:00:00.000Z",
+        last_opened_at: "2026-01-10T12:00:00.000Z",
+    },
+];
+function createMockClient() {
+    return { send: vi.fn() };
+}
+describe("daemon-register", () => {
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockGetMachineMetadata.mockReturnValue(sampleMachine);
+        mockListProjects.mockReturnValue(sampleProjects);
+        mockLoadTokens.mockResolvedValue(sampleTokens);
+    });
+    describe("buildDaemonRegisterMessage", () => {
+        it("should build correct registration message", async () => {
+            const message = await buildDaemonRegisterMessage();
+            expect(message).not.toBeNull();
+            expect(message.type).toBe("daemon_register");
+            expect(message.payload.machineKey).toBe(sampleTokens.machine_key);
+            expect(message.payload.token).toBe(sampleTokens.access_token);
+        });
+        it("should include machine metadata", async () => {
+            const message = await buildDaemonRegisterMessage();
+            expect(message.payload.machine).toEqual(sampleMachine);
+            expect(mockGetMachineMetadata).toHaveBeenCalledWith(null);
+        });
+        it("should include project list", async () => {
+            const message = await buildDaemonRegisterMessage();
+            expect(message.payload.projects).toEqual([
+                { projectId: "proj-1", name: "my-app" },
+                { projectId: "proj-2", name: "api-server" },
+            ]);
+        });
+        it("should return null when not authenticated", async () => {
+            mockLoadTokens.mockResolvedValue(null);
+            const message = await buildDaemonRegisterMessage();
+            expect(message).toBeNull();
+        });
+        it("should pass label to getMachineMetadata", async () => {
+            await buildDaemonRegisterMessage("Work Laptop");
+            expect(mockGetMachineMetadata).toHaveBeenCalledWith("Work Laptop");
+        });
+        it("should handle empty project list", async () => {
+            mockListProjects.mockReturnValue([]);
+            const message = await buildDaemonRegisterMessage();
+            expect(message.payload.projects).toEqual([]);
+        });
+    });
+    describe("sendDaemonRegister", () => {
+        it("should send registration message via client", async () => {
+            const mockClient = createMockClient();
+            const result = await sendDaemonRegister(mockClient);
+            expect(result).toBe(true);
+            expect(vi.mocked(mockClient.send)).toHaveBeenCalledOnce();
+            const sent = vi.mocked(mockClient.send).mock.calls[0][0];
+            expect(sent.type).toBe("daemon_register");
+            expect(sent.payload.machineKey).toBe(sampleTokens.machine_key);
+        });
+        it("should return false when not authenticated", async () => {
+            mockLoadTokens.mockResolvedValue(null);
+            const mockClient = createMockClient();
+            const result = await sendDaemonRegister(mockClient);
+            expect(result).toBe(false);
+            expect(vi.mocked(mockClient.send)).not.toHaveBeenCalled();
+        });
+        it("should re-register on reconnect (callable multiple times)", async () => {
+            const mockClient = createMockClient();
+            await sendDaemonRegister(mockClient);
+            await sendDaemonRegister(mockClient);
+            expect(vi.mocked(mockClient.send)).toHaveBeenCalledTimes(2);
+        });
+    });
+});
+//# sourceMappingURL=daemon-register.test.js.map

package/dist/gateway/messages/__tests__/daemon-register.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"daemon-register.test.js","sourceRoot":"","sources":["../../../../src/gateway/messages/__tests__/daemon-register.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAO9D,oBAAoB;AACpB,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;AAC1C,EAAE,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;AAChD,EAAE,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;AAE1C,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,gCAAgC,CAAC,CAAC;AACtE,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,sCAAsC,CAAC,CAAC;AAC9E,MAAM,EAAE,kBAAkB,EAAE,GAAG,MAAM,MAAM,CAAC,gCAAgC,CAAC,CAAC;AAE9E,MAAM,EAAE,0BAA0B,EAAE,kBAAkB,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;AAEjG,MAAM,cAAc,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;AAC7C,MAAM,gBAAgB,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;AACjD,MAAM,sBAAsB,GAAG,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;AAE7D,MAAM,YAAY,GAAe;IAC/B,WAAW,EAAE,sCAAsC;IACnD,YAAY,EAAE,2BAA2B;IACzC,aAAa,EAAE,mBAAmB;IAClC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI;IACpC,OAAO,EAAE,eAAe;CACzB,CAAC;AAEF,MAAM,aAAa,GAAG;IACpB,EAAE,EAAE,OAAgB;IACpB,SAAS,EAAE,aAAa;IACxB,QAAQ,EAAE,aAAa;IACvB,QAAQ,EAAE,UAAU;IACpB,KAAK,EAAE,IAAqB;CAC7B,CAAC;AAEF,MAAM,cAAc,GAAG;IACrB;QACE,UAAU,EAAE,QAAQ;QACpB,IAAI,EAAE,QAAQ;QACd,IAAI,EAAE,mBAAmB;QACzB,UAAU,EAAE,0BAA0B;QACtC,cAAc,EAAE,IAAI;KACrB;IACD;QACE,UAAU,EAAE,QAAQ;QACpB,IAAI,EAAE,YAAY;QAClB,IAAI,EAAE,uBAAuB;QAC7B,UAAU,EAAE,0BAA0B;QACtC,cAAc,EAAE,0BAA0B;KAC3C;CACF,CAAC;AAEF,SAAS,gBAAgB;IACvB,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAA8B,CAAC;AACvD,CAAC;AAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,sBAAsB,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC;QACtD,gBAAgB,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QACjD,cAAc,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,OAAO,GAAG,MAAM,0BAA0B,EAAE,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAC/B,MAAM,CAAC,OAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC9C,MAAM,CAAC,OAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;YACnE,MAAM,CAAC,OAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,OAAO,GAAG,MAAM,0BAA0B,EAAE,CAAC;YAEnD,MAAM,CAAC,OAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YACxD,MAAM,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,OAAO,GAAG,MAAM,0BAA0B,EAAE,CAAC;YAEnD,MAAM,CAAC,OAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;gBACxC,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACvC,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE;aAC5C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,cAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEvC,MAAM,OAAO,GAAG,MAAM,0BAA0B,EAAE,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,0BAA0B,CAAC,aAAa,CAAC,CAAC;YAEhD,MAAM,CAAC,sBAAsB,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,gBAAgB,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YAErC,MAAM,OAAO,GAAG,MAAM,0BAA0B,EAAE,CAAC;YAEnD,MAAM,CAAC,OAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YAEtC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAEpD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1B,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,oBAAoB,EAAE,CAAC;YAC1D,MAAM,IAAI,GAAG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAA0B,CAAC;YAClF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC1C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,cAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YAEtC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAEpD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC3B,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;YAEtC,MAAM,kBAAkB,CAAC,UAAU,CAAC,CAAC;YACrC,MAAM,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAErC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/gateway/messages/__tests__/project-sync.test.d.ts

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

package/dist/gateway/messages/__tests__/project-sync.test.d.ts.map

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