Class DaemonEventSocket

apps/kimi-web/src/api/daemon/ws.ts:69–478

}

export class DaemonEventSocket {
  private ws: WebSocket | null = null;
  private connected = false;
  private closed = false;

  /** subscriptions we manage: sessionId → last known cursor {seq, epoch} */
  private readonly subscriptions = new Map<string, SessionCursor>();

  /** subscriptions queued while not yet connected */
  private readonly pendingSubscriptions: PendingSubscription[] = [];
  private readonly terminalAttachments = new Map<string, TerminalAttachment>();

  private msgSeq = 0;

  /** Automatic reconnect (exponential backoff, reset on a successful hello). */
  private reconnectAttempts = 0;
  private reconnectTimer: ReturnType<typeof setTimeout> | null = null;

  constructor(
    private readonly wsUrl: string,
    private readonly clientId: string,
    private readonly handlers: DaemonEventSocketHandlers,
  ) {}

  /** Open the WebSocket connection. No-op while one is open or after close(). */
  connect(): void {
    if (this.ws !== null || this.closed) return;

    traceWsLifecycle('connect', { url: this.wsUrl, attempt: this.reconnectAttempts });
    const credential = getCredential();
    const protocols =
      credential !== undefined ? [`${WS_BEARER_PROTOCOL_PREFIX}${credential}`] : undefined;
    const ws = new WebSocket(this.wsUrl, protocols);
    this.ws = ws;

    ws.onopen = () => {
      // Don't mark as connected yet — wait for server_hello
      traceWsLifecycle('open');
    };

    ws.onmessage = (ev: MessageEvent) => {
      try {
        const frame = JSON.parse(String(ev.data)) as WireServerFrame;
        traceWsIn(frame);
        this.handleFrame(frame);
      } catch (error) {
        traceWsLifecycle('parse-error', { error: String(error) });
        this.handlers.onError(0, `Failed to parse WS frame: ${String(error)}`, false);
      }
    };

    ws.onerror = () => {
      // The error details are not exposed by the browser WS API; the close
      // event with a reason code follows immediately.
      traceWsLifecycle('error');
      this.handlers.onError(0, 'WebSocket error', false);
    };