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

package/dist/index.d.mts

@@ -478,7 +478,7 @@ declare function createCliChildPriorityPolicy(deps?: PriorityDependencies): <T e
  * registering listeners/doing other work. There is a spawn-to-setPriority race:
  * a child can run (and fork) before this call. Later descendants inherit nice
  * on POSIX; early descendants are not retroactively changed. Windows creation
- * and ConPTY inheritance require verification on the MainPC.
+ * and ConPTY inheritance require verification on the DesktopA.
  * Optional darwin-bg is applied asynchronously by taskpolicy after nice; the
  * extra helper startup adds a further race before the background class takes
  * effect. It also throttles I/O, so nice alone remains the default.
@@ -662,6 +662,7 @@ declare function createLineParser<TEnvelope = SessionHostWireEnvelope>(onEnvelop
 interface SessionHostClientOptions {
     endpoint?: SessionHostEndpoint;
     appName?: string;
+    requestTimeoutMs?: number;
 }
 /**
  * Why an established session-host connection went away.
@@ -670,6 +671,7 @@ interface SessionHostClientOptions {
  * - `closed` — socket fully closed ('close') without a preceding error/end
  */
 type SessionHostDisconnectReason = 'error' | 'ended' | 'closed';
+declare function disconnectSessionHostClients(endpoint?: SessionHostEndpoint, error?: Error): void;
 interface SessionHostDisconnectInfo {
     reason: SessionHostDisconnectReason;
     endpointPath: string;
@@ -690,12 +692,17 @@ interface SessionHostDisconnectInfo {
 declare class SessionHostClient {
     readonly endpoint: SessionHostEndpoint;
     private socket;
+    private connectPromise;
+    private requestTimeoutMs;
     private requestWaiters;
     private eventListeners;
     private disconnectListeners;
     /** Guards against emitting both an 'error'-path and a 'close'-path disconnect for one socket. */
     private disconnectedSockets;
+    /** Session subscriptions established via create_session/attach_session, replayed automatically on reconnect. */
+    private attachedSessions;
     constructor(options?: SessionHostClientOptions);
+    static disconnectClients(endpoint?: SessionHostEndpoint, error?: Error): void;
     /**
      * Fires when an established connection to the session host goes away.
      *
@@ -710,9 +717,11 @@ declare class SessionHostClient {
      */
     onDisconnect(listener: (info: SessionHostDisconnectInfo) => void): () => void;
     private handleDisconnect;
+    private sendRawRequestOnSocket;
     connect(): Promise<void>;
     onEvent(listener: (event: SessionHostEvent) => void): () => void;
     request<T = unknown>(request: SessionHostRequest): Promise<SessionHostResponse<T>>;
+    disconnect(error?: Error): void;
     close(): Promise<void>;
 }
 declare function createResponseEnvelope(requestId: string, response: SessionHostResponse): SessionHostResponseEnvelope;
@@ -871,4 +880,4 @@ interface SessionHostControlTransport {
  */
 declare function createSessionHostControlPlane(transport: SessionHostControlTransport): SessionHostControlPlane;
 
-export { type AcquireWritePayload, type AttachSessionPayload, CLI_CHILD_NICE_ENV, CLI_CHILD_NICE_INCREMENT, CLI_CHILD_SCHEDULING_ENV, type ClearSessionBufferPayload, type CreateSessionPayload, DEFAULT_CONFIG_DIR_NAME, DEFAULT_SESSION_RING_BUFFER_MAX_BYTES, type DetachSessionPayload, type ForceDetachClientPayload, type GetHostDiagnosticsPayload, type GetSnapshotPayload, type GetTerminalSnapshotPayload, type HostUpgradeLeasePayload, type PruneDuplicateSessionsPayload, type ReleaseWritePayload, type ResizeSessionPayload, type RestartSessionPayload, type ResumeSessionPayload, SESSION_HOST_SUPPORTED_REQUEST_TYPES, type SendInputPayload, type SendSignalPayload, type SessionAttachedClient, type SessionBufferSnapshot, type SessionClientType, type SessionHostCategory, SessionHostClient, type SessionHostClientOptions, type SessionHostControlPlane, type SessionHostControlTransport, type SessionHostDiagnostics, type SessionHostDisconnectInfo, type SessionHostDisconnectReason, type SessionHostDuplicateSessionGroup, type SessionHostEndpoint, type SessionHostEndpointOptions, type SessionHostEvent, type SessionHostEventEnvelope, type SessionHostLogEntry, type SessionHostPruneDuplicatesResult, type SessionHostRecord, SessionHostRegistry, type SessionHostRequest, type SessionHostRequestEnvelope, type SessionHostRequestTrace, type SessionHostRequestType, type SessionHostResponse, type SessionHostResponseEnvelope, type SessionHostRuntimeTransition, type SessionHostSnapshot, type SessionHostSurfaceKind, type SessionHostSurfaceRecordLike, type SessionHostWireEnvelope, type SessionLaunchCommand, type SessionLifecycle, type SessionOwnerType, SessionRingBuffer, type SessionRingBufferOptions, type SessionTerminalSnapshot, type SessionTerminalState, type SessionTermination, type SessionTransport, type SessionWriteOwner, type StopSessionPayload, type UpdateSessionMetaPayload, applyTerminalColorEnv, buildRuntimeDisplayName, buildRuntimeKey, canonicalizeInstancePath, classifyTermination, createCliChildPriorityPolicy, createLineParser, createResponseEnvelope, createSessionHostControlPlane, ensureNodePtySpawnHelperPermissions, formatRuntimeOwner, getDefaultSessionHostEndpoint, getSessionHostRecoveryLabel, getSessionHostSurfaceKind, getWorkspaceLabel, hiddenSpawn as hiddenSpawnProcess, isDefaultInstanceConfigDir, isSessionHostLiveRuntime, isSessionHostRecoverySnapshot, isTerminalRecord, partitionSessionHostDiagnosticsSessions, partitionSessionHostRecords, replayLaunchEnv, resolveAttachableRuntimeRecord, resolveInstanceConfigDir, resolveRuntimeRecord, resolveSessionHostIpcKey, sanitizeSpawnEnv, spawnCliChild, writeEnvelope };
+export { type AcquireWritePayload, type AttachSessionPayload, CLI_CHILD_NICE_ENV, CLI_CHILD_NICE_INCREMENT, CLI_CHILD_SCHEDULING_ENV, type ClearSessionBufferPayload, type CreateSessionPayload, DEFAULT_CONFIG_DIR_NAME, DEFAULT_SESSION_RING_BUFFER_MAX_BYTES, type DetachSessionPayload, type ForceDetachClientPayload, type GetHostDiagnosticsPayload, type GetSnapshotPayload, type GetTerminalSnapshotPayload, type HostUpgradeLeasePayload, type PruneDuplicateSessionsPayload, type ReleaseWritePayload, type ResizeSessionPayload, type RestartSessionPayload, type ResumeSessionPayload, SESSION_HOST_SUPPORTED_REQUEST_TYPES, type SendInputPayload, type SendSignalPayload, type SessionAttachedClient, type SessionBufferSnapshot, type SessionClientType, type SessionHostCategory, SessionHostClient, type SessionHostClientOptions, type SessionHostControlPlane, type SessionHostControlTransport, type SessionHostDiagnostics, type SessionHostDisconnectInfo, type SessionHostDisconnectReason, type SessionHostDuplicateSessionGroup, type SessionHostEndpoint, type SessionHostEndpointOptions, type SessionHostEvent, type SessionHostEventEnvelope, type SessionHostLogEntry, type SessionHostPruneDuplicatesResult, type SessionHostRecord, SessionHostRegistry, type SessionHostRequest, type SessionHostRequestEnvelope, type SessionHostRequestTrace, type SessionHostRequestType, type SessionHostResponse, type SessionHostResponseEnvelope, type SessionHostRuntimeTransition, type SessionHostSnapshot, type SessionHostSurfaceKind, type SessionHostSurfaceRecordLike, type SessionHostWireEnvelope, type SessionLaunchCommand, type SessionLifecycle, type SessionOwnerType, SessionRingBuffer, type SessionRingBufferOptions, type SessionTerminalSnapshot, type SessionTerminalState, type SessionTermination, type SessionTransport, type SessionWriteOwner, type StopSessionPayload, type UpdateSessionMetaPayload, applyTerminalColorEnv, buildRuntimeDisplayName, buildRuntimeKey, canonicalizeInstancePath, classifyTermination, createCliChildPriorityPolicy, createLineParser, createResponseEnvelope, createSessionHostControlPlane, disconnectSessionHostClients, ensureNodePtySpawnHelperPermissions, formatRuntimeOwner, getDefaultSessionHostEndpoint, getSessionHostRecoveryLabel, getSessionHostSurfaceKind, getWorkspaceLabel, hiddenSpawn as hiddenSpawnProcess, isDefaultInstanceConfigDir, isSessionHostLiveRuntime, isSessionHostRecoverySnapshot, isTerminalRecord, partitionSessionHostDiagnosticsSessions, partitionSessionHostRecords, replayLaunchEnv, resolveAttachableRuntimeRecord, resolveInstanceConfigDir, resolveRuntimeRecord, resolveSessionHostIpcKey, sanitizeSpawnEnv, spawnCliChild, writeEnvelope };

package/dist/index.d.ts

@@ -478,7 +478,7 @@ declare function createCliChildPriorityPolicy(deps?: PriorityDependencies): <T e
  * registering listeners/doing other work. There is a spawn-to-setPriority race:
  * a child can run (and fork) before this call. Later descendants inherit nice
  * on POSIX; early descendants are not retroactively changed. Windows creation
- * and ConPTY inheritance require verification on the MainPC.
+ * and ConPTY inheritance require verification on the DesktopA.
  * Optional darwin-bg is applied asynchronously by taskpolicy after nice; the
  * extra helper startup adds a further race before the background class takes
  * effect. It also throttles I/O, so nice alone remains the default.
@@ -662,6 +662,7 @@ declare function createLineParser<TEnvelope = SessionHostWireEnvelope>(onEnvelop
 interface SessionHostClientOptions {
     endpoint?: SessionHostEndpoint;
     appName?: string;
+    requestTimeoutMs?: number;
 }
 /**
  * Why an established session-host connection went away.
@@ -670,6 +671,7 @@ interface SessionHostClientOptions {
  * - `closed` — socket fully closed ('close') without a preceding error/end
  */
 type SessionHostDisconnectReason = 'error' | 'ended' | 'closed';
+declare function disconnectSessionHostClients(endpoint?: SessionHostEndpoint, error?: Error): void;
 interface SessionHostDisconnectInfo {
     reason: SessionHostDisconnectReason;
     endpointPath: string;
@@ -690,12 +692,17 @@ interface SessionHostDisconnectInfo {
 declare class SessionHostClient {
     readonly endpoint: SessionHostEndpoint;
     private socket;
+    private connectPromise;
+    private requestTimeoutMs;
     private requestWaiters;
     private eventListeners;
     private disconnectListeners;
     /** Guards against emitting both an 'error'-path and a 'close'-path disconnect for one socket. */
     private disconnectedSockets;
+    /** Session subscriptions established via create_session/attach_session, replayed automatically on reconnect. */
+    private attachedSessions;
     constructor(options?: SessionHostClientOptions);
+    static disconnectClients(endpoint?: SessionHostEndpoint, error?: Error): void;
     /**
      * Fires when an established connection to the session host goes away.
      *
@@ -710,9 +717,11 @@ declare class SessionHostClient {
      */
     onDisconnect(listener: (info: SessionHostDisconnectInfo) => void): () => void;
     private handleDisconnect;
+    private sendRawRequestOnSocket;
     connect(): Promise<void>;
     onEvent(listener: (event: SessionHostEvent) => void): () => void;
     request<T = unknown>(request: SessionHostRequest): Promise<SessionHostResponse<T>>;
+    disconnect(error?: Error): void;
     close(): Promise<void>;
 }
 declare function createResponseEnvelope(requestId: string, response: SessionHostResponse): SessionHostResponseEnvelope;
@@ -871,4 +880,4 @@ interface SessionHostControlTransport {
  */
 declare function createSessionHostControlPlane(transport: SessionHostControlTransport): SessionHostControlPlane;
 
-export { type AcquireWritePayload, type AttachSessionPayload, CLI_CHILD_NICE_ENV, CLI_CHILD_NICE_INCREMENT, CLI_CHILD_SCHEDULING_ENV, type ClearSessionBufferPayload, type CreateSessionPayload, DEFAULT_CONFIG_DIR_NAME, DEFAULT_SESSION_RING_BUFFER_MAX_BYTES, type DetachSessionPayload, type ForceDetachClientPayload, type GetHostDiagnosticsPayload, type GetSnapshotPayload, type GetTerminalSnapshotPayload, type HostUpgradeLeasePayload, type PruneDuplicateSessionsPayload, type ReleaseWritePayload, type ResizeSessionPayload, type RestartSessionPayload, type ResumeSessionPayload, SESSION_HOST_SUPPORTED_REQUEST_TYPES, type SendInputPayload, type SendSignalPayload, type SessionAttachedClient, type SessionBufferSnapshot, type SessionClientType, type SessionHostCategory, SessionHostClient, type SessionHostClientOptions, type SessionHostControlPlane, type SessionHostControlTransport, type SessionHostDiagnostics, type SessionHostDisconnectInfo, type SessionHostDisconnectReason, type SessionHostDuplicateSessionGroup, type SessionHostEndpoint, type SessionHostEndpointOptions, type SessionHostEvent, type SessionHostEventEnvelope, type SessionHostLogEntry, type SessionHostPruneDuplicatesResult, type SessionHostRecord, SessionHostRegistry, type SessionHostRequest, type SessionHostRequestEnvelope, type SessionHostRequestTrace, type SessionHostRequestType, type SessionHostResponse, type SessionHostResponseEnvelope, type SessionHostRuntimeTransition, type SessionHostSnapshot, type SessionHostSurfaceKind, type SessionHostSurfaceRecordLike, type SessionHostWireEnvelope, type SessionLaunchCommand, type SessionLifecycle, type SessionOwnerType, SessionRingBuffer, type SessionRingBufferOptions, type SessionTerminalSnapshot, type SessionTerminalState, type SessionTermination, type SessionTransport, type SessionWriteOwner, type StopSessionPayload, type UpdateSessionMetaPayload, applyTerminalColorEnv, buildRuntimeDisplayName, buildRuntimeKey, canonicalizeInstancePath, classifyTermination, createCliChildPriorityPolicy, createLineParser, createResponseEnvelope, createSessionHostControlPlane, ensureNodePtySpawnHelperPermissions, formatRuntimeOwner, getDefaultSessionHostEndpoint, getSessionHostRecoveryLabel, getSessionHostSurfaceKind, getWorkspaceLabel, hiddenSpawn as hiddenSpawnProcess, isDefaultInstanceConfigDir, isSessionHostLiveRuntime, isSessionHostRecoverySnapshot, isTerminalRecord, partitionSessionHostDiagnosticsSessions, partitionSessionHostRecords, replayLaunchEnv, resolveAttachableRuntimeRecord, resolveInstanceConfigDir, resolveRuntimeRecord, resolveSessionHostIpcKey, sanitizeSpawnEnv, spawnCliChild, writeEnvelope };
+export { type AcquireWritePayload, type AttachSessionPayload, CLI_CHILD_NICE_ENV, CLI_CHILD_NICE_INCREMENT, CLI_CHILD_SCHEDULING_ENV, type ClearSessionBufferPayload, type CreateSessionPayload, DEFAULT_CONFIG_DIR_NAME, DEFAULT_SESSION_RING_BUFFER_MAX_BYTES, type DetachSessionPayload, type ForceDetachClientPayload, type GetHostDiagnosticsPayload, type GetSnapshotPayload, type GetTerminalSnapshotPayload, type HostUpgradeLeasePayload, type PruneDuplicateSessionsPayload, type ReleaseWritePayload, type ResizeSessionPayload, type RestartSessionPayload, type ResumeSessionPayload, SESSION_HOST_SUPPORTED_REQUEST_TYPES, type SendInputPayload, type SendSignalPayload, type SessionAttachedClient, type SessionBufferSnapshot, type SessionClientType, type SessionHostCategory, SessionHostClient, type SessionHostClientOptions, type SessionHostControlPlane, type SessionHostControlTransport, type SessionHostDiagnostics, type SessionHostDisconnectInfo, type SessionHostDisconnectReason, type SessionHostDuplicateSessionGroup, type SessionHostEndpoint, type SessionHostEndpointOptions, type SessionHostEvent, type SessionHostEventEnvelope, type SessionHostLogEntry, type SessionHostPruneDuplicatesResult, type SessionHostRecord, SessionHostRegistry, type SessionHostRequest, type SessionHostRequestEnvelope, type SessionHostRequestTrace, type SessionHostRequestType, type SessionHostResponse, type SessionHostResponseEnvelope, type SessionHostRuntimeTransition, type SessionHostSnapshot, type SessionHostSurfaceKind, type SessionHostSurfaceRecordLike, type SessionHostWireEnvelope, type SessionLaunchCommand, type SessionLifecycle, type SessionOwnerType, SessionRingBuffer, type SessionRingBufferOptions, type SessionTerminalSnapshot, type SessionTerminalState, type SessionTermination, type SessionTransport, type SessionWriteOwner, type StopSessionPayload, type UpdateSessionMetaPayload, applyTerminalColorEnv, buildRuntimeDisplayName, buildRuntimeKey, canonicalizeInstancePath, classifyTermination, createCliChildPriorityPolicy, createLineParser, createResponseEnvelope, createSessionHostControlPlane, disconnectSessionHostClients, ensureNodePtySpawnHelperPermissions, formatRuntimeOwner, getDefaultSessionHostEndpoint, getSessionHostRecoveryLabel, getSessionHostSurfaceKind, getWorkspaceLabel, hiddenSpawn as hiddenSpawnProcess, isDefaultInstanceConfigDir, isSessionHostLiveRuntime, isSessionHostRecoverySnapshot, isTerminalRecord, partitionSessionHostDiagnosticsSessions, partitionSessionHostRecords, replayLaunchEnv, resolveAttachableRuntimeRecord, resolveInstanceConfigDir, resolveRuntimeRecord, resolveSessionHostIpcKey, sanitizeSpawnEnv, spawnCliChild, writeEnvelope };

package/dist/index.js

@@ -50,6 +50,7 @@ __export(index_exports, {
   createLineParser: () => createLineParser,
   createResponseEnvelope: () => createResponseEnvelope,
   createSessionHostControlPlane: () => createSessionHostControlPlane,
+  disconnectSessionHostClients: () => disconnectSessionHostClients,
   ensureNodePtySpawnHelperPermissions: () => ensureNodePtySpawnHelperPermissions,
   formatRuntimeOwner: () => formatRuntimeOwner,
   getDefaultSessionHostEndpoint: () => getDefaultSessionHostEndpoint,
@@ -804,16 +805,40 @@ function createLineParser(onEnvelope) {
   return parser;
 }
 var establishedSockets = /* @__PURE__ */ new WeakSet();
+var activeClientRefs = /* @__PURE__ */ new Set();
+function disconnectSessionHostClients(endpoint, error) {
+  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);
+    }
+  }
+}
 var SessionHostClient = class {
   endpoint;
   socket = null;
+  connectPromise = null;
+  requestTimeoutMs = 3e4;
   requestWaiters = /* @__PURE__ */ new Map();
   eventListeners = /* @__PURE__ */ new Set();
   disconnectListeners = /* @__PURE__ */ new Set();
   /** Guards against emitting both an 'error'-path and a 'close'-path disconnect for one socket. */
   disconnectedSockets = /* @__PURE__ */ new WeakSet();
+  /** Session subscriptions established via create_session/attach_session, replayed automatically on reconnect. */
+  attachedSessions = /* @__PURE__ */ new Map();
   constructor(options = {}) {
     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, error) {
+    disconnectSessionHostClients(endpoint, error);
   }
   /**
    * Fires when an established connection to the session host goes away.
@@ -856,6 +881,7 @@ var SessionHostClient = class {
     this.requestWaiters.clear();
     if (this.socket === socket) {
       this.socket = null;
+      this.connectPromise = null;
     }
     try {
       socket.destroy();
@@ -875,8 +901,48 @@ var SessionHostClient = class {
       }
     }
   }
+  sendRawRequestOnSocket(socket, request) {
+    const requestId = generateUUID();
+    const envelope = {
+      kind: "request",
+      requestId,
+      request
+    };
+    const timeoutMs = this.requestTimeoutMs > 0 ? this.requestTimeoutMs : 3e4;
+    const timeoutLabel = timeoutMs % 1e3 === 0 ? `${timeoutMs / 1e3}s` : `${timeoutMs}ms`;
+    return new Promise((resolve3, 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 {
+          }
+          reject(timeoutError);
+        }
+      }, timeoutMs);
+      this.requestWaiters.set(requestId, {
+        resolve: (value) => {
+          clearTimeout(timeout);
+          resolve3(value);
+        },
+        reject: (error) => {
+          clearTimeout(timeout);
+          reject(error);
+        }
+      });
+      socket.write(serializeEnvelope(envelope));
+    });
+  }
   async connect() {
-    if (this.socket && !this.socket.destroyed) return;
+    if (this.socket && !this.socket.destroyed) {
+      if (this.connectPromise) await this.connectPromise;
+      return;
+    }
     if (this.socket) {
       try {
         this.socket.destroy();
@@ -884,6 +950,7 @@ var SessionHostClient = class {
       }
       this.socket = null;
     }
+    this.connectPromise = null;
     const socket = net.createConnection(this.endpoint.path);
     this.socket = socket;
     const parser = createLineParser((envelope) => {
@@ -917,13 +984,35 @@ var SessionHostClient = class {
     socket.on("end", () => {
       this.handleDisconnect(socket, "ended", void 0, flushParser().length);
     });
-    await new Promise((resolve3, reject) => {
-      socket.once("connect", () => {
-        establishedSockets.add(socket);
-        resolve3();
+    const connectTask = (async () => {
+      await new Promise((resolve3, reject) => {
+        socket.once("connect", () => {
+          establishedSockets.add(socket);
+          resolve3();
+        });
+        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) {
     this.eventListeners.add(listener);
@@ -933,33 +1022,50 @@ var SessionHostClient = class {
   }
   async request(request) {
     await this.connect();
-    if (!this.socket) throw new Error("Session host socket unavailable");
-    const requestId = generateUUID();
-    const envelope = {
-      kind: "request",
-      requestId,
-      request
-    };
-    const response = await new Promise((resolve3, reject) => {
-      const timeout = setTimeout(() => {
-        this.requestWaiters.delete(requestId);
-        reject(new Error(`Session host request timed out after 30s (${request.type})`));
-      }, 3e4);
-      this.requestWaiters.set(requestId, {
-        resolve: (value) => {
-          clearTimeout(timeout);
-          resolve3(value);
-        },
-        reject: (error) => {
-          clearTimeout(timeout);
-          reject(error);
+    const requestSocket = this.socket;
+    if (!requestSocket) throw new Error("Session host socket unavailable");
+    const response = await this.sendRawRequestOnSocket(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?.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
+          });
         }
-      });
-      this.socket?.write(serializeEnvelope(envelope));
-    });
+      } else if (request.type === "detach_session" || request.type === "delete_session") {
+        this.attachedSessions.delete(request.payload.sessionId);
+      }
+    }
     return response;
   }
+  disconnect(error) {
+    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() {
+    this.attachedSessions.clear();
+    this.connectPromise = null;
     if (!this.socket) return;
     const socket = this.socket;
     this.socket = null;
@@ -1177,6 +1283,7 @@ function createSessionHostControlPlane(transport) {
   createLineParser,
   createResponseEnvelope,
   createSessionHostControlPlane,
+  disconnectSessionHostClients,
   ensureNodePtySpawnHelperPermissions,
   formatRuntimeOwner,
   getDefaultSessionHostEndpoint,