fix(tables): restore frozenColumns on delete-column undo/redo

Author: waleedlatif1

apps/sim/app/workspace/[workspaceId]/tables/[tableId]/components/table-grid/table-grid.tsx

@@ -490,6 +490,11 @@ export function TableGrid({
     setColumnWidths(widths)
   }
 
+  function handleFrozenColumnsChange(frozen: string[]) {
+    setFrozenColumns(frozen)
+    frozenColumnsRef.current = frozen
+  }
+
   const handleFreezeToggle = useCallback((columnName: string) => {
     const col = columnsRef.current.find((c) => c.name === columnName)
     const siblings: string[] = col?.workflowGroupId
@@ -533,6 +538,7 @@ export function TableGrid({
     onColumnOrderChange: handleColumnOrderChange,
     onColumnRename: handleColumnRename,
     onColumnWidthsChange: handleColumnWidthsChange,
+    onFrozenColumnsChange: handleFrozenColumnsChange,
     getColumnWidths,
   })
   const undoRef = useRef(undo)
@@ -2820,6 +2826,7 @@ export function TableGrid({
         .map((r) => ({ rowId: r.id, value: r.data[columnToDelete] }))
       const previousWidth = columnWidthsRef.current[columnToDelete] ?? null
       const orderSnapshot = currentOrder ? [...currentOrder] : null
+      const frozenSnapshot = [...frozenColumnsRef.current]
 
       const onDeleted = () => {
         deletedOriginalPositions.push(entry.position)
@@ -2833,6 +2840,7 @@ export function TableGrid({
           cellData,
           previousOrder: orderSnapshot,
           previousWidth,
+          previousFrozenColumns: frozenSnapshot,
         })
 
         const { [columnToDelete]: _removedWidth, ...cleanedWidths } = columnWidthsRef.current

apps/sim/hooks/use-table-undo.ts

@@ -31,6 +31,7 @@ interface UseTableUndoProps {
   onColumnOrderChange?: (order: string[]) => void
   onColumnRename?: (oldName: string, newName: string) => void
   onColumnWidthsChange?: (widths: Record<string, number>) => void
+  onFrozenColumnsChange?: (frozen: string[]) => void
   getColumnWidths?: () => Record<string, number>
 }
 
@@ -40,6 +41,7 @@ export function useTableUndo({
   onColumnOrderChange,
   onColumnRename,
   onColumnWidthsChange,
+  onFrozenColumnsChange,
   getColumnWidths,
 }: UseTableUndoProps) {
   const push = useTableUndoStore((s) => s.push)
@@ -69,6 +71,8 @@ export function useTableUndo({
   onColumnRenameRef.current = onColumnRename
   const onColumnWidthsChangeRef = useRef(onColumnWidthsChange)
   onColumnWidthsChangeRef.current = onColumnWidthsChange
+  const onFrozenColumnsChangeRef = useRef(onFrozenColumnsChange)
+  onFrozenColumnsChangeRef.current = onFrozenColumnsChange
   const getColumnWidthsRef = useRef(getColumnWidths)
   getColumnWidthsRef.current = getColumnWidths
 
@@ -273,6 +277,10 @@ export function useTableUndo({
                       metadata.columnWidths = merged
                       onColumnWidthsChangeRef.current?.(merged)
                     }
+                    if (action.previousFrozenColumns !== null) {
+                      onFrozenColumnsChangeRef.current?.(action.previousFrozenColumns)
+                      metadata.frozenColumns = action.previousFrozenColumns
+                    }
                     if (Object.keys(metadata).length > 0) {
                       updateMetadataMutation.mutate(metadata)
                     }
@@ -294,6 +302,13 @@ export function useTableUndo({
                     metadata.columnWidths = rest
                     onColumnWidthsChangeRef.current?.(rest)
                   }
+                  if (action.previousFrozenColumns !== null) {
+                    const newFrozen = action.previousFrozenColumns.filter(
+                      (n) => n !== action.columnName
+                    )
+                    onFrozenColumnsChangeRef.current?.(newFrozen)
+                    metadata.frozenColumns = newFrozen
+                  }
                   if (Object.keys(metadata).length > 0) {
                     updateMetadataMutation.mutate(metadata)
                   }

apps/sim/stores/table/types.ts

@@ -44,6 +44,7 @@ export type TableUndoAction =
       cellData: Array<{ rowId: string; value: unknown }>
       previousOrder: string[] | null
       previousWidth: number | null
+      previousFrozenColumns: string[] | null
     }
   | { type: 'rename-column'; oldName: string; newName: string }
   | {