@threadbase-sh/streamer 1.69.5 → 1.69.6

package/dist/index.js

@@ -9962,6 +9962,13 @@ var SessionHandlers = class {
   get sessionSubscribers() {
     return this.deps.sessionSubscribers;
   }
+  // Prompt lifecycle events (question / permission and their cancellations)
+  // carry prompt content and go ONLY to the session's subscribers — never to
+  // every connected socket. A client that subscribes after a prompt opened
+  // gets it from the subscribe replay (server-wiring), not from a broadcast.
+  broadcastToSession(sessionId, message) {
+    this.wsHub.broadcastToClients(this.sessionSubscribers.get(sessionId) ?? [], message);
+  }
   get agentConfig() {
     return this.deps.agentConfig;
   }
@@ -10516,7 +10523,7 @@ var SessionHandlers = class {
     const toolUseId = `screen:${sessionId}:${key.length}`;
     this.pendingQuestions.set(sessionId, { toolUseId, questions, origin: "pty" });
     this.pendingQuestionKey.set(sessionId, key);
-    this.wsHub.broadcast({ type: "question", sessionId, toolUseId, questions });
+    this.broadcastToSession(sessionId, { type: "question", sessionId, toolUseId, questions });
   }
   // Permission gate opened/closed (OSC 777 + scraped options). Broadcasts the
   // additive `permission` / `permission_cancelled` events. Mobile answers by
@@ -10526,7 +10533,7 @@ var SessionHandlers = class {
       if (!this.pendingPermission.has(sessionId)) return;
       this.pendingPermission.delete(sessionId);
       this.pendingPermissionKey.delete(sessionId);
-      this.wsHub.broadcast({ type: "permission_cancelled", sessionId });
+      this.broadcastToSession(sessionId, { type: "permission_cancelled", sessionId });
       return;
     }
     const key = permissionContentKey(gate);
@@ -10540,7 +10547,7 @@ var SessionHandlers = class {
       `[ws.broadcast_permission] ${sessionId.slice(0, 8)} subscribers=${subscriberCount}`,
       { event: "ws.broadcast_permission", sessionId, subscriberCount }
     );
-    this.wsHub.broadcast({
+    this.broadcastToSession(sessionId, {
       type: "permission",
       sessionId,
       ...gate.prompt ? { prompt: gate.prompt } : {},
@@ -10588,7 +10595,7 @@ var SessionHandlers = class {
     const gateClosed = () => {
       this.pendingPermission.delete(sessionId);
       this.pendingPermissionKey.delete(sessionId);
-      this.wsHub.broadcast({ type: "permission_cancelled", sessionId });
+      this.broadcastToSession(sessionId, { type: "permission_cancelled", sessionId });
       json(res, 409, { ok: false, reason: "gate_closed" });
     };
     const gate = this.pendingPermission.get(sessionId);
@@ -10674,7 +10681,7 @@ var SessionHandlers = class {
     if (!await this.questionMenuStillOpen(sessionId)) {
       this.pendingQuestions.delete(sessionId);
       this.pendingQuestionKey.delete(sessionId);
-      this.wsHub.broadcast({ type: "question_cancelled", sessionId, toolUseId });
+      this.broadcastToSession(sessionId, { type: "question_cancelled", sessionId, toolUseId });
       json(res, 409, { ok: false, reason: "question_gone" });
       return;
     }
@@ -10686,7 +10693,7 @@ var SessionHandlers = class {
       return;
     }
     this.pendingQuestions.delete(sessionId);
-    this.wsHub.broadcast({ type: "question_cancelled", sessionId, toolUseId });
+    this.broadcastToSession(sessionId, { type: "question_cancelled", sessionId, toolUseId });
     json(res, 200, { ok: true });
   }
   // Best-effort: a session we don't own a PTY for, or one that raced away
@@ -17807,7 +17814,7 @@ var StreamerServer = class {
         priorToolUseId
       });
       this.pendingQuestionKey.set(sessionId, key);
-      if (broadcast) this.wsHub.broadcast(m);
+      if (broadcast) this.wsHub.broadcastToClients(this.sessionSubscribers.get(sessionId) ?? [], m);
     }
   }
   cancelPendingQuestion(sessionId) {
@@ -17815,7 +17822,11 @@ var StreamerServer = class {
     if (!pq) return;
     this.pendingQuestions.delete(sessionId);
     this.pendingQuestionKey.delete(sessionId);
-    this.wsHub.broadcast({ type: "question_cancelled", sessionId, toolUseId: pq.toolUseId });
+    this.wsHub.broadcastToClients(this.sessionSubscribers.get(sessionId) ?? [], {
+      type: "question_cancelled",
+      sessionId,
+      toolUseId: pq.toolUseId
+    });
   }
   async handleBrowse(url, res) {
     if (!this.browseRoot) {