gemini-cli/packages/core/src/context/sidecar/inbox.ts

import type { InboxMessage, InboxSnapshot } from '../pipeline.js';

export class LiveInbox {
  private messages: InboxMessage[] = [];

  publish<T>(topic: string, payload: T, idGenerator: { generateId(): string }): void {
    this.messages.push({
      id: idGenerator.generateId(),
      topic,
      payload,
      timestamp: Date.now(),
    });
  }

  getMessages(): ReadonlyArray<InboxMessage> {
    return [...this.messages];
  }

  drainConsumed(consumedIds: Set<string>): void {
    this.messages = this.messages.filter((m) => !consumedIds.has(m.id));
  }
}

export class InboxSnapshotImpl implements InboxSnapshot {
  private messages: ReadonlyArray<InboxMessage>;
  private consumedIds = new Set<string>();

  constructor(messages: ReadonlyArray<InboxMessage>) {
    this.messages = messages;
  }

  getMessages<T = unknown>(topic: string): ReadonlyArray<InboxMessage<T>> {
    return this.messages.filter((m) => m.topic === topic) as unknown as ReadonlyArray<InboxMessage<T>>;
  }

  consume(messageId: string): void {
    this.consumedIds.add(messageId);
  }

  getConsumedIds(): Set<string> {
    return this.consumedIds;
  }
}