sim/apps/sim/executor/utils/block-data.ts

import { normalizeName } from '@/executor/constants'
import type { ExecutionContext } from '@/executor/types'

export interface BlockDataCollection {
  blockData: Record<string, any>
  blockNameMapping: Record<string, string>
}

export function collectBlockData(ctx: ExecutionContext): BlockDataCollection {
  const blockData: Record<string, any> = {}
  const blockNameMapping: Record<string, string> = {}

  for (const [id, state] of ctx.blockStates.entries()) {
    if (state.output !== undefined) {
      blockData[id] = state.output
      const workflowBlock = ctx.workflow?.blocks?.find((b) => b.id === id)
      if (workflowBlock?.metadata?.name) {
        blockNameMapping[normalizeName(workflowBlock.metadata.name)] = id
      }
    }
  }

  return { blockData, blockNameMapping }
}