@adhdev/session-host-core 1.0.82-rc.24 → 1.0.82-rc.26

package/src/ipc.ts

@@ -4,6 +4,7 @@ import * as net from 'net';
 import { generateUUID } from './uuid.js';
 import { StringDecoder } from 'string_decoder';
 import type {
+  AttachSessionPayload,
   SessionHostEvent,
   SessionHostRequest,
   SessionHostRequestEnvelope,
@@ -122,6 +123,7 @@ function createLineParser<TEnvelope = SessionHostWireEnvelope>(
 export interface SessionHostClientOptions {
   endpoint?: SessionHostEndpoint;
   appName?: string;
+  requestTimeoutMs?: number;
 }
 
 /**
@@ -138,6 +140,20 @@ export type SessionHostDisconnectReason = 'error' | 'ended' | 'closed';
  * (host not up yet) — only the former is a session-host death signal.
  */
 const establishedSockets = new WeakSet<net.Socket>();
+const activeClientRefs = new Set<WeakRef<SessionHostClient>>();
+
+export function disconnectSessionHostClients(endpoint?: SessionHostEndpoint, error?: Error): void {
+  for (const ref of activeClientRefs) {
+    const client = ref.deref();
+    if (!client) {
+      activeClientRefs.delete(ref);
+      continue;
+    }
+    if (!endpoint || client.endpoint.path === endpoint.path) {
+      client.disconnect(error);
+    }
+  }
+}
 
 export interface SessionHostDisconnectInfo {
   reason: SessionHostDisconnectReason;
@@ -161,14 +177,26 @@ export class SessionHostClient {
   readonly endpoint: SessionHostEndpoint;
 
   private socket: net.Socket | null = null;
+  private connectPromise: Promise<void> | null = null;
+  private requestTimeoutMs = 30_000;
   private requestWaiters = new Map<string, { resolve: (value: SessionHostResponse) => void; reject: (error: Error) => void }>();
   private eventListeners = new Set<(event: SessionHostEvent) => void>();
   private disconnectListeners = new Set<(info: SessionHostDisconnectInfo) => void>();
   /** Guards against emitting both an 'error'-path and a 'close'-path disconnect for one socket. */
   private disconnectedSockets = new WeakSet<net.Socket>();
+  /** Session subscriptions established via create_session/attach_session, replayed automatically on reconnect. */
+  private attachedSessions = new Map<string, AttachSessionPayload>();
 
   constructor(options: SessionHostClientOptions = {}) {
     this.endpoint = options.endpoint || getDefaultSessionHostEndpoint(options.appName || 'adhdev');
+    if (typeof options.requestTimeoutMs === 'number' && options.requestTimeoutMs > 0) {
+      this.requestTimeoutMs = options.requestTimeoutMs;
+    }
+    activeClientRefs.add(new WeakRef(this));
+  }
+
+  static disconnectClients(endpoint?: SessionHostEndpoint, error?: Error): void {
+    disconnectSessionHostClients(endpoint, error);
   }
 
   /**
@@ -228,6 +256,7 @@ export class SessionHostClient {
     // guard at the top of connect().
     if (this.socket === socket) {
       this.socket = null;
+      this.connectPromise = null;
     }
     try { socket.destroy(); } catch { /* noop */ }
 
@@ -246,13 +275,49 @@ export class SessionHostClient {
     }
   }
 
+  private sendRawRequestOnSocket<T = unknown>(
+    socket: net.Socket,
+    request: SessionHostRequest,
+  ): Promise<SessionHostResponse<T>> {
+    const requestId = generateUUID();
+    const envelope: SessionHostRequestEnvelope = {
+      kind: 'request',
+      requestId,
+      request,
+    };
+    const timeoutMs = this.requestTimeoutMs > 0 ? this.requestTimeoutMs : 30_000;
+    const timeoutLabel = timeoutMs % 1000 === 0 ? `${timeoutMs / 1000}s` : `${timeoutMs}ms`;
+    return new Promise<SessionHostResponse<T>>((resolve, reject) => {
+      const timeout = setTimeout(() => {
+        const timeoutError = new Error(`Session host request timed out after ${timeoutLabel} (${request.type})`);
+        if (this.socket === socket) {
+          this.disconnect(timeoutError);
+        } else {
+          this.requestWaiters.delete(requestId);
+          this.disconnectedSockets.add(socket);
+          try { socket.destroy(); } catch { /* noop */ }
+          reject(timeoutError);
+        }
+      }, timeoutMs);
+      this.requestWaiters.set(requestId, {
+        resolve: (value) => { clearTimeout(timeout); resolve(value as SessionHostResponse<T>); },
+        reject: (error) => { clearTimeout(timeout); reject(error); },
+      });
+      socket.write(serializeEnvelope(envelope));
+    });
+  }
+
   async connect(): Promise<void> {
-    if (this.socket && !this.socket.destroyed) return;
+    if (this.socket && !this.socket.destroyed) {
+      if (this.connectPromise) await this.connectPromise;
+      return;
+    }
     // Cleanup stale socket reference left after error/disconnect
     if (this.socket) {
       try { this.socket.destroy(); } catch { /* noop */ }
       this.socket = null;
     }
+    this.connectPromise = null;
 
     const socket = net.createConnection(this.endpoint.path);
     this.socket = socket;
@@ -314,13 +379,36 @@ export class SessionHostClient {
       this.handleDisconnect(socket, 'ended', undefined, flushParser().length);
     });
 
-    await new Promise<void>((resolve, reject) => {
-      socket.once('connect', () => {
-        establishedSockets.add(socket);
-        resolve();
+    const connectTask = (async () => {
+      await new Promise<void>((resolve, reject) => {
+        socket.once('connect', () => {
+          establishedSockets.add(socket);
+          resolve();
+        });
+        socket.once('error', reject);
       });
-      socket.once('error', reject);
-    });
+      if (this.attachedSessions.size > 0 && this.socket === socket && !socket.destroyed) {
+        for (const [sessionId, sub] of [...this.attachedSessions.entries()]) {
+          if (this.socket !== socket || socket.destroyed) break;
+          const replayResponse = await this.sendRawRequestOnSocket(socket, {
+            type: 'attach_session',
+            payload: sub,
+          }).catch(() => null);
+          if (replayResponse && !replayResponse.success) {
+            this.attachedSessions.delete(sessionId);
+          }
+        }
+      }
+    })();
+
+    this.connectPromise = connectTask;
+    try {
+      await connectTask;
+    } finally {
+      if (this.connectPromise === connectTask) {
+        this.connectPromise = null;
+      }
+    }
   }
 
   onEvent(listener: (event: SessionHostEvent) => void): () => void {
@@ -332,31 +420,53 @@ export class SessionHostClient {
 
   async request<T = unknown>(request: SessionHostRequest): Promise<SessionHostResponse<T>> {
     await this.connect();
-    if (!this.socket) throw new Error('Session host socket unavailable');
-
-    const requestId = generateUUID();
-    const envelope: SessionHostRequestEnvelope = {
-      kind: 'request',
-      requestId,
-      request,
-    };
-
-    const response = await new Promise<SessionHostResponse>((resolve, reject) => {
-      const timeout = setTimeout(() => {
-        this.requestWaiters.delete(requestId);
-        reject(new Error(`Session host request timed out after 30s (${request.type})`));
-      }, 30_000);
-      this.requestWaiters.set(requestId, {
-        resolve: (value) => { clearTimeout(timeout); resolve(value); },
-        reject: (error) => { clearTimeout(timeout); reject(error); },
-      });
-      this.socket?.write(serializeEnvelope(envelope));
-    });
+    const requestSocket = this.socket;
+    if (!requestSocket) throw new Error('Session host socket unavailable');
+
+    const response = await this.sendRawRequestOnSocket<T>(requestSocket, request);
+    if (response.success) {
+      if (request.type === 'attach_session') {
+        this.attachedSessions.set(request.payload.sessionId, {
+          sessionId: request.payload.sessionId,
+          clientId: request.payload.clientId,
+          clientType: request.payload.clientType,
+          readOnly: request.payload.readOnly === true,
+        });
+      } else if (request.type === 'create_session') {
+        const createdId = (response.result as { sessionId?: string } | undefined)?.sessionId || request.payload.sessionId;
+        const clientId = typeof request.payload.clientId === 'string' ? request.payload.clientId.trim() : '';
+        if (createdId && clientId) {
+          this.attachedSessions.set(createdId, {
+            sessionId: createdId,
+            clientId,
+            clientType: request.payload.clientType || 'daemon',
+            readOnly: false,
+          });
+        }
+      } else if (request.type === 'detach_session' || request.type === 'delete_session') {
+        this.attachedSessions.delete(request.payload.sessionId);
+      }
+    }
+    return response;
+  }
 
-    return response as SessionHostResponse<T>;
+  disconnect(error?: Error): void {
+    const socket = this.socket;
+    if (socket) {
+      this.handleDisconnect(socket, error ? 'error' : 'closed', error, 0);
+      return;
+    }
+    this.connectPromise = null;
+    const failure = error || new Error(`Session host client disconnected (${this.endpoint.path})`);
+    for (const waiter of this.requestWaiters.values()) {
+      waiter.reject(failure);
+    }
+    this.requestWaiters.clear();
   }
 
   async close(): Promise<void> {
+    this.attachedSessions.clear();
+    this.connectPromise = null;
     if (!this.socket) return;
     const socket = this.socket;
     this.socket = null;