Function handleAddBlock

packages/mcp/src/tools/writing.ts:501–607

(args: Record<string, unknown>)

}

async function handleAddBlock(args: Record<string, unknown>) {
  const parsedArgs = addBlockArgsSchema.safeParse(args)
  if (!parsedArgs.success) {
    return writingError(`Invalid args in handleAddBlock: ${formatFirstIssue(parsedArgs.error)}`)
  }
  const filePath = parsedArgs.data.path
  const notebookFilter = parsedArgs.data.notebook
  const blockSpec = parsedArgs.data.block
  const position = parsedArgs.data.position
  const dryRun = parsedArgs.data.dryRun === true

  const file = await loadDeepnoteFile(filePath)
  if (file.project.notebooks.length === 0) {
    return {
      content: [{ type: 'text', text: `No notebooks found for file "${file.project.name}" (${file.project.id})` }],
      isError: true,
    }
  }

  // Find the target notebook
  let notebook = file.project.notebooks[0]
  if (notebookFilter) {
    const found = file.project.notebooks.find(n => n.name === notebookFilter || n.id === notebookFilter)
    if (!found) {
      return {
        content: [{ type: 'text', text: `Notebook not found: ${notebookFilter}` }],
        isError: true,
      }
    }
    notebook = found
  }

  // Determine insertion index
  let insertIndex = notebook.blocks.length
  if (position?.after) {
    const afterIdx = findBlockIndex(notebook.blocks, position.after)
    if (afterIdx < 0) {
      throw new Error(`Invalid position.after: block not found (${position.after})`)
    }
    insertIndex = afterIdx + 1
  } else if (position?.before) {
    const beforeIdx = findBlockIndex(notebook.blocks, position.before)
    if (beforeIdx < 0) {
      throw new Error(`Invalid position.before: block not found (${position.before})`)
    }
    insertIndex = beforeIdx
  } else if (position?.index !== undefined) {
    if (!Number.isInteger(position.index) || position.index < 0 || position.index > notebook.blocks.length) {
      throw new Error(`Invalid position.index: expected integer in range 0..${notebook.blocks.length}`)
    }
    insertIndex = position.index
  }

  // Create the new block
  const newBlock = createBlock(blockSpec, insertIndex)

  // Insert the block
  notebook.blocks.splice(insertIndex, 0, newBlock)