fix(autolayout): recalculate dimensions after grid snapping

Author: waleedlatif1

apps/sim/lib/workflows/autolayout/core.ts

@@ -420,10 +420,13 @@ export function layoutBlocksCore(
   calculatePositions(layers, edges, layoutOptions)
 
   // 5. Normalize positions
-  const dimensions = normalizePositions(nodes, { isContainer: options.isContainer })
+  let dimensions = normalizePositions(nodes, { isContainer: options.isContainer })
 
-  // 6. Snap to grid if gridSize is specified
-  snapNodesToGrid(nodes, layoutOptions.gridSize)
+  // 6. Snap to grid if gridSize is specified (recalculates dimensions)
+  const snappedDimensions = snapNodesToGrid(nodes, layoutOptions.gridSize)
+  if (snappedDimensions) {
+    dimensions = snappedDimensions
+  }
 
   return { nodes, dimensions }
 }

apps/sim/lib/workflows/autolayout/targeted.ts

@@ -1,4 +1,3 @@
-import { createLogger } from '@sim/logger'
 import {
   CONTAINER_PADDING,
   DEFAULT_HORIZONTAL_SPACING,
@@ -14,12 +13,11 @@ import {
   isContainerType,
   prepareContainerDimensions,
   shouldSkipAutoLayout,
+  snapPositionToGrid,
 } from '@/lib/workflows/autolayout/utils'
 import { CONTAINER_DIMENSIONS } from '@/lib/workflows/blocks/block-dimensions'
 import type { BlockState } from '@/stores/workflows/workflow/types'
 
-const logger = createLogger('AutoLayout:Targeted')
-
 export interface TargetedLayoutOptions extends LayoutOptions {
   changedBlockIds: string[]
   verticalSpacing?: number
@@ -186,10 +184,7 @@ function layoutGroup(
     const block = blocks[id]
     const newPos = layoutPositions.get(id)
     if (!block || !newPos) continue
-    block.position = {
-      x: newPos.x + offsetX,
-      y: newPos.y + offsetY,
-    }
+    block.position = snapPositionToGrid({ x: newPos.x + offsetX, y: newPos.y + offsetY }, gridSize)
   }
 }
 

apps/sim/lib/workflows/autolayout/utils.ts

@@ -43,15 +43,29 @@ export function snapPositionToGrid(
 }
 
 /**
- * Snaps all node positions in a graph to grid positions.
- * Only applies if gridSize > 0.
+ * Snaps all node positions in a graph to grid positions and returns updated dimensions.
+ * Returns null if gridSize is not set or no snapping was needed.
  */
-export function snapNodesToGrid(nodes: Map<string, GraphNode>, gridSize: number | undefined): void {
-  if (!gridSize || gridSize <= 0) {
-    return
+export function snapNodesToGrid(
+  nodes: Map<string, GraphNode>,
+  gridSize: number | undefined
+): { width: number; height: number } | null {
+  if (!gridSize || gridSize <= 0 || nodes.size === 0) {
+    return null
   }
+
+  let maxX = Number.NEGATIVE_INFINITY
+  let maxY = Number.NEGATIVE_INFINITY
+
   for (const node of nodes.values()) {
     node.position = snapPositionToGrid(node.position, gridSize)
+    maxX = Math.max(maxX, node.position.x + node.metrics.width)
+    maxY = Math.max(maxY, node.position.y + node.metrics.height)
+  }
+
+  return {
+    width: maxX + CONTAINER_PADDING,
+    height: maxY + CONTAINER_PADDING,
   }
 }