ack PR comments

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/components/action-bar/action-bar.tsx

@@ -53,8 +53,11 @@ export function ActionBar() {
   }
 
   const handleHide = async () => {
-    setContextMenu(null)
-    await updateSetting.mutateAsync({ key: 'showActionBar', value: false })
+    try {
+      await updateSetting.mutateAsync({ key: 'showActionBar', value: false })
+    } finally {
+      setContextMenu(null)
+    }
   }
 
   if (!showActionBar) {

apps/sim/app/workspace/[workspaceId]/w/[workflowId]/workflow.tsx

@@ -212,7 +212,6 @@ const WorkflowContent = React.memo(() => {
   const [potentialParentId, setPotentialParentId] = useState<string | null>(null)
   const [selectedEdges, setSelectedEdges] = useState<SelectedEdgesMap>(new Map())
   const [isShiftPressed, setIsShiftPressed] = useState(false)
-  const [isSelectionDragActive, setIsSelectionDragActive] = useState(false)
   const [isErrorConnectionDrag, setIsErrorConnectionDrag] = useState(false)
   const canvasMode = useCanvasModeStore((state) => state.mode)
   const isHandMode = canvasMode === 'hand'
@@ -1928,7 +1927,6 @@ const WorkflowContent = React.memo(() => {
     }
     const handleFocusLoss = () => {
       setIsShiftPressed(false)
-      setIsSelectionDragActive(false)
     }
     const handleVisibilityChange = () => {
       if (document.hidden) {
@@ -2843,17 +2841,6 @@ const WorkflowContent = React.memo(() => {
     ]
   )
 
-  // Lock selection mode when selection drag starts (captures Shift state at drag start)
-  const onSelectionStart = useCallback(() => {
-    if (isShiftPressed) {
-      setIsSelectionDragActive(true)
-    }
-  }, [isShiftPressed])
-
-  const onSelectionEnd = useCallback(() => {
-    requestAnimationFrame(() => setIsSelectionDragActive(false))
-  }, [])
-
   /** Captures initial positions when selection drag starts (for marquee-selected nodes). */
   const onSelectionDragStart = useCallback(
     (_event: React.MouseEvent, nodes: Node[]) => {
@@ -3003,7 +2990,6 @@ const WorkflowContent = React.memo(() => {
 
   const onSelectionDragStop = useCallback(
     (_event: React.MouseEvent, nodes: any[]) => {
-      requestAnimationFrame(() => setIsSelectionDragActive(false))
       clearDragHighlights()
       if (nodes.length === 0) return
 
@@ -3311,9 +3297,9 @@ const WorkflowContent = React.memo(() => {
               edgeTypes={edgeTypes}
               onDrop={effectivePermissions.canEdit ? onDrop : undefined}
               onDragOver={effectivePermissions.canEdit ? onDragOver : undefined}
-              onInit={() => {
+              onInit={(instance) => {
                 requestAnimationFrame(() => {
-                  fitViewToBounds({ padding: 0.1, maxZoom: 1.0 })
+                  instance.fitView(reactFlowFitViewOptions)
                   setIsCanvasReady(true)
                 })
               }}
@@ -3334,11 +3320,9 @@ const WorkflowContent = React.memo(() => {
               onPointerMove={handleCanvasPointerMove}
               onPointerLeave={handleCanvasPointerLeave}
               elementsSelectable={true}
-              selectionOnDrag={!isHandMode || isSelectionDragActive}
+              selectionOnDrag={!isHandMode}
               selectionMode={SelectionMode.Partial}
               panOnDrag={isHandMode ? [0, 1] : false}
-              onSelectionStart={onSelectionStart}
-              onSelectionEnd={onSelectionEnd}
               multiSelectionKeyCode={['Meta', 'Control', 'Shift']}
               nodesConnectable={effectivePermissions.canEdit}
               nodesDraggable={effectivePermissions.canEdit}