Function collectExecutableBlocks

packages/cli/src/commands/run.ts:362–399

* Collect executable blocks from a DeepnoteFile. * Handles notebook and block filtering, and validates that requested blocks exist.

(
  file: DeepnoteFile,
  options: { notebook?: string; block?: string; blockIds?: string[] }
)

 * Handles notebook and block filtering, and validates that requested blocks exist.
 */
function collectExecutableBlocks(
  file: DeepnoteFile,
  options: { notebook?: string; block?: string; blockIds?: string[] }
): DryRunBlockInfo[] {
  const notebooks = getNotebooksForExecutionScope(file, options)
  const idsToValidate = options.blockIds ?? (options.block ? [options.block] : [])
  const blockIdFilter = options.blockIds ? new Set(options.blockIds) : options.block ? new Set([options.block]) : null

  // Collect all executable blocks
  const executableBlocks: DryRunBlockInfo[] = []
  for (const notebook of notebooks) {
    const sortedBlocks = [...notebook.blocks].sort((a, b) => a.sortingKey.localeCompare(b.sortingKey))
    for (const block of sortedBlocks) {
      if (!executableBlockTypeSet.has(block.type)) {
        continue
      }
      // Skip if filtering by block IDs and this isn't in the set
      if (blockIdFilter && !blockIdFilter.has(block.id)) {
        continue
      }
      executableBlocks.push({
        id: block.id,
        type: block.type,
        label: getBlockLabel(block),
        notebook: notebook.name,
      })
    }
  }

  // Validate requested IDs when filtering yields no executable blocks.
  if (executableBlocks.length === 0) {
    for (const blockId of idsToValidate) {
      assertExecutableBlockExists(blockId, notebooks)
    }
  }

  return executableBlocks
}

function getNotebooksForExecutionScope(
  file: DeepnoteFile,