Function setAllBlocksCollapsedState

cli/src/utils/collapse-helpers.ts:214–256

(
  messages: ChatMessage[],
  collapsed: boolean,
)

 * @returns Updated messages array with all collapsible blocks set to the specified state
 */
export function setAllBlocksCollapsedState(
  messages: ChatMessage[],
  collapsed: boolean,
): ChatMessage[] {
  return messages.map((message) => {
    let updatedMessage = message
    let messageChanged = false

    // Handle agent variant messages (message-level isCollapsed)
    if (message.variant === 'agent') {
      // Treat undefined as collapsed (true) to match hasAnyExpandedBlocks semantics
      const currentCollapsed = message.metadata?.isCollapsed ?? true
      if (currentCollapsed !== collapsed) {
        messageChanged = true
        updatedMessage = {
          ...updatedMessage,
          metadata: {
            ...updatedMessage.metadata,
            isCollapsed: collapsed,
            userOpened: !collapsed ? true : updatedMessage.metadata?.userOpened,
          },
        }
      }
    }

    // Handle blocks within messages (applies to all message variants)
    if (message.blocks) {
      const { blocks: updatedBlocks, changed } = updateBlocksRecursively(
        message.blocks,
        collapsed,
      )
      if (changed) {
        messageChanged = true
        updatedMessage = {
          ...updatedMessage,
          blocks: updatedBlocks,
        }
      }
    }

    return messageChanged ? updatedMessage : message
  })
}