orca/src/renderer/src/runtime/sync-runtime-graph/agent-status-projection.ts

import type { AppState } from '@/store/types'
import { AGENT_STATUS_SYNC_UPDATED_AT_BUCKET_MS, graphState } from './graph-state'
import type { AgentStatusProjectionCacheEntry } from './types'

function serializeAgentStatusEntry(
  paneKey: string,
  entry: AppState['agentStatusByPaneKey'][string]
): string {
  return JSON.stringify({
    paneKey,
    entryPaneKey: entry.paneKey,
    state: entry.state,
    workingMode: entry.workingMode ?? null,
    prompt: entry.prompt,
    updatedAtBucket: Math.floor(entry.updatedAt / AGENT_STATUS_SYNC_UPDATED_AT_BUCKET_MS),
    stateStartedAt: entry.stateStartedAt,
    agentType: entry.agentType ?? null,
    terminalTitle: entry.terminalTitle ?? null,
    stateHistory: entry.stateHistory.map((history) => ({
      state: history.state,
      prompt: history.prompt,
      startedAt: history.startedAt,
      interrupted: history.interrupted ?? null
    })),
    toolName: entry.toolName ?? null,
    toolInput: entry.toolInput ?? null,
    // Include the prompt so a newly captured question re-fires mobile publication.
    interactivePrompt: entry.interactivePrompt ?? null,
    lastAssistantMessage: entry.lastAssistantMessage ?? null,
    interrupted: entry.interrupted ?? null
  })
}

export function buildRuntimeMobileAgentStatusProjection(
  agentStatusByPaneKey: AppState['agentStatusByPaneKey']
): string {
  const cached = graphState.cachedAgentStatusProjection
  if (cached?.source === agentStatusByPaneKey) {
    return cached.projection
  }

  // A status ping replaces one entry and re-spreads the map; reuse every other entry.
  const entries = new Map<string, AgentStatusProjectionCacheEntry>()
  const parts: string[] = []
  for (const [paneKey, entry] of Object.entries(agentStatusByPaneKey).sort(([a], [b]) =>
    a.localeCompare(b)
  )) {
    const previous = cached?.entries.get(paneKey)
    const entryCache =
      previous?.entry === entry
        ? previous
        : { entry, projection: serializeAgentStatusEntry(paneKey, entry) }
    entries.set(paneKey, entryCache)
    parts.push(entryCache.projection)
  }
  const projection = `[${parts.join(',')}]`
  graphState.cachedAgentStatusProjection = {
    source: agentStatusByPaneKey,
    entries,
    projection
  }
  return projection
}

export function buildRuntimeMobileAgentStatusProjectionForTests(
  agentStatusByPaneKey: AppState['agentStatusByPaneKey']
): string {
  return buildRuntimeMobileAgentStatusProjection(agentStatusByPaneKey)
}

export const AGENT_STATUS_SYNC_UPDATED_AT_BUCKET_MS_FOR_TESTS =
  AGENT_STATUS_SYNC_UPDATED_AT_BUCKET_MS

export function resetRuntimeMobileAgentStatusProjectionCacheForTests(): void {
  graphState.cachedAgentStatusProjection = null
}