Function analyzeDag

packages/cli/src/commands/dag.ts:99–144

(
  path: string | undefined,
  options: DagOptions
)

type BlockMap = Map<string, BlockInfo>

async function analyzeDag(
  path: string | undefined,
  options: DagOptions
): Promise<{ dag: BlockDependencyDag; blocks: DeepnoteBlock[]; blockMap: BlockMap }> {
  const { absolutePath } = await resolvePathToDeepnoteFile(path)

  const { file: deepnoteFile, warnings } = await loadAndResolveDeepnoteFile(absolutePath)
  emitInitResolverWarnings(warnings, options.output !== undefined)

  // Collect all blocks, optionally filtering by notebook
  const allBlocks: DeepnoteBlock[] = []
  const blockMap: BlockMap = new Map()

  for (const notebook of deepnoteFile.project.notebooks) {
    if (options.notebook && notebook.name !== options.notebook) {
      continue
    }

    for (const block of notebook.blocks) {
      allBlocks.push(block)
      blockMap.set(block.id, {
        id: block.id,
        label: getBlockLabel(block),
        type: block.type,
        notebookName: notebook.name,
      })
    }
  }

  if (allBlocks.length === 0) {
    if (options.notebook) {
      throw new Error(`No blocks found in notebook "${options.notebook}"`)
    }
    throw new Error('No blocks found in the project')
  }

  debug(`Analyzing ${allBlocks.length} blocks...`)

  const pythonInterpreter = options.python ? await resolvePythonExecutable(options.python) : undefined
  const { dag } = await getDagForBlocks(allBlocks, {
    acceptPartialDAG: true,
    pythonInterpreter,
  })

  return { dag, blocks: allBlocks, blockMap }
}

function outputDagShow(
  dag: BlockDependencyDag,