@addai/node 0.30.1 → 0.30.2

package/dist/desktop/relay-client.js

@@ -83,6 +83,30 @@ let missedPongs = 0;
  *  people watching the same screen each need their own connection and their
  *  own handshake. x11vnc runs with -shared precisely so it will serve them. */
 const bridges = new Map();
+/**
+ * Bytes that arrived while their bridge was still being built.
+ *
+ * ⚠️ There is an await between 'attach' and the socket existing — the row has
+ * to be looked up to know which port to dial — and a viewer can send in that
+ * window. RFB never did: x11vnc speaks first, so noVNC has nothing to say
+ * until the server has already said hello, and the gap was invisible for as
+ * long as the screen was the only channel.
+ *
+ * xpra is the other way round. Its client sends its own hello the moment the
+ * socket opens, so on the windows channel the first packet ALWAYS raced the
+ * lookup — and losing it was not a dropped packet but a dead session: the data
+ * branch found no bridge, reported the viewer gone, and the connection that
+ * finished opening a moment later sat there with nothing on it until xpra
+ * timed out 20 seconds on:
+ *
+ *     New tcp connection received from '172.17.0.1:58222'
+ *     0 packets received
+ *     1 packets sent (25 bytes)
+ *
+ * A viewerId present here means "attaching" — distinct from present in
+ * `bridges` (attached) and from neither (gone, and the sender should be told).
+ */
+const pendingWrites = new Map();
 const wakeHooks = new Map();
 function onWake(kind, fn) {
     wakeHooks.set(kind, fn);
@@ -95,6 +119,9 @@ function dropBridges() {
         catch { /* gone */ }
     }
     bridges.clear();
+    // The link went down mid-attach. Nothing will ever flush these, and keeping
+    // them would hand a reconnecting viewer the last session's first packet.
+    pendingWrites.clear();
 }
 /**
  * Start a connection attempt that CANNOT take the daemon down with it.
@@ -175,11 +202,17 @@ async function connect() {
         const gone = (viewerId, reason) => {
             bridges.get(viewerId)?.destroy();
             bridges.delete(viewerId);
+            // Anything still waiting for a bridge that is never coming.
+            pendingWrites.delete(viewerId);
             if (sock.readyState === ws_1.default.OPEN) {
                 sock.send(JSON.stringify({ type: 'gone', viewerId, reason }));
             }
         };
         if (msg.type === 'attach') {
+            // Claim the viewer BEFORE the await, so a packet arriving during the
+            // lookup is held rather than treated as arriving after a lost bridge.
+            if (!pendingWrites.has(msg.viewerId))
+                pendingWrites.set(msg.viewerId, []);
             const row = (await (0, manager_1.listDesktops)()).find(d => d.id === msg.desktopId);
             if (!row?.vnc_port) {
                 gone(msg.viewerId, 'desktop is not running');
@@ -223,14 +256,28 @@ async function connect() {
             tcp.on('close', () => gone(msg.viewerId, 'the desktop closed the connection'));
             bridges.get(msg.viewerId)?.destroy();
             bridges.set(msg.viewerId, tcp);
+            // Whatever arrived while the row was being looked up, in order and
+            // before anything newer. net.connect returns a socket that is still
+            // connecting, and node queues writes on it until it is — so this needs
+            // no wait of its own.
+            const queued = pendingWrites.get(msg.viewerId) ?? [];
+            pendingWrites.delete(msg.viewerId);
+            for (const chunk of queued)
+                tcp.write(chunk);
             return;
         }
         if (msg.type === 'data' && msg.b64) {
             const bridge = bridges.get(msg.viewerId);
-            // Input for a bridge that no longer exists. Dropping it quietly is how
-            // "take control does nothing" happens: the pointer never moves and the
-            // viewer has no idea why.
             if (!bridge) {
+                // Still attaching: hold it. The first xpra packet always lands here.
+                const queue = pendingWrites.get(msg.viewerId);
+                if (queue) {
+                    queue.push(Buffer.from(msg.b64, 'base64'));
+                    return;
+                }
+                // Input for a bridge that no longer exists. Dropping it quietly is how
+                // "take control does nothing" happens: the pointer never moves and the
+                // viewer has no idea why.
                 gone(msg.viewerId, 'no longer connected to the desktop');
                 return;
             }
@@ -260,6 +307,8 @@ async function connect() {
         if (msg.type === 'detach') {
             bridges.get(msg.viewerId)?.destroy();
             bridges.delete(msg.viewerId);
+            // A viewer that left while it was still attaching.
+            pendingWrites.delete(msg.viewerId);
         }
     });
     const retry = () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@addai/node",
-  "version": "0.30.1",
+  "version": "0.30.2",
   "description": "Daemon that pairs a machine with your +Ai account and runs Claude / Codex / Kimi / Gemini agents on its behalf. Reachable via Supabase from Vault, Entity Studio, or any other +Ai surface.",
   "license": "MIT",
   "keywords": [