Function getUpstreamBlocks

packages/reactivity/src/dag-analyzer.ts:63–81

(dag: BlockDependencyDag, blocksIds: string[], visited: Set<string> = new Set())

}

function getUpstreamBlocks(dag: BlockDependencyDag, blocksIds: string[], visited: Set<string> = new Set()): string[] {
  const inputVariables = dag.nodes.filter(node => blocksIds.includes(node.id)).flatMap(node => node.inputVariables)

  const blocksThatDefineVariables = dag.nodes
    .filter(node => node.outputVariables.find(outputVariable => inputVariables.includes(outputVariable)))
    .map(node => node.id)

  if (blocksThatDefineVariables.length === 0) {
    return []
  }

  // This prevents the function from making recursive calls for blocks that have already been visited.
  const nextBlocks = blocksThatDefineVariables.filter(id => !visited.has(id))
  nextBlocks.forEach(id => {
    visited.add(id)
  })

  return [...nextBlocks, ...getUpstreamBlocks(dag, nextBlocks, visited)]
}