Class BranchingRpc

test/bindingRuntime.misc.test.ts:102–175

}

class BranchingRpc implements StdioProcess {
  private handlers: Array<(m: JsonRpcMessage) => void> = [];
  written: JsonRpcMessage[] = [];
  promptRequests: JsonRpcRequest[] = [];

  private readonly updates: any[];
  private readonly loadSession: boolean;
  private sessionId = 'sess-branch';

  constructor(params: { updates: any[]; loadSession: boolean }) {
    this.updates = params.updates;
    this.loadSession = params.loadSession;
  }

  write(message: JsonRpcMessage): void {
    this.written.push(message);
    if (!('method' in message)) return;

    const req = message as JsonRpcRequest;
    if (req.method === 'initialize') {
      queueMicrotask(() => {
        this.emit({
          jsonrpc: '2.0',
          id: req.id,
          result: {
            protocolVersion: 1,
            agentCapabilities: { loadSession: this.loadSession },
          },
        } as JsonRpcResponse);
      });
      return;
    }

    if (req.method === 'session/new') {
      queueMicrotask(() => {
        this.emit({
          jsonrpc: '2.0',
          id: req.id,
          result: { sessionId: this.sessionId },
        } as JsonRpcResponse);
      });
      return;
    }

    if (req.method === 'session/prompt') {
      this.promptRequests.push(req);
      queueMicrotask(() => {
        this.updates.forEach((update) => {
          this.emit({
            jsonrpc: '2.0',
            method: 'session/update',
            params: { sessionId: this.sessionId, update },
          } as any);
        });
        this.emit({
          jsonrpc: '2.0',
          id: req.id,
          result: { stopReason: 'end' },