fix(tables): drain past the cap so exclusions can't shrink a select-all chip

Cursor Bugbot: for a gutter select-all the menu count comes from
selectedRowCount (capped), but the chip was built by loading exactly
MAX_TABLE_SELECTION_ROWS and filtering exclusions AFTER. Any excluded row inside
that prefix left the chip short of the advertised count.

Third appearance of the same invariant — label vs payload — this time in the
select-all path specifically, which the earlier fixes did not touch.

Loads the cap plus the exclusion count, which covers the worst case where every
exclusion falls inside the prefix, so the filtered result still reaches the cap
whenever the table has the rows. Extracted as drainTargetForChip so the
compensation is stated and tested rather than an inline arithmetic detail.

Author: waleedlatif1

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

@@ -12,7 +12,6 @@ import { useParams } from 'next/navigation'
 import { usePostHog } from 'posthog-js/react'
 import type { RunLimit, RunMode, TableFindMatch } from '@/lib/api/contracts/tables'
 import { attachSelectionContextToClipboard } from '@/lib/copilot/chat/selection-clipboard'
-import { MAX_TABLE_SELECTION_ROWS } from '@/lib/copilot/chat/selection-context'
 import { captureEvent } from '@/lib/posthog/client'
 import type {
   ColumnDefinition,
@@ -73,6 +72,7 @@ import {
   classifyExecStatusMix,
   collectRowSnapshots,
   computeNormalizedSelection,
+  drainTargetForChip,
   type ExecStatusMix,
   expandToDisplayColumns,
   isCellInSelection,
@@ -3840,7 +3840,11 @@ export function TableGrid({
     let sourceRowIds = addToChatRowIds
     if (contextMenuIsSelectAll || isColumnSelectionRef.current) {
       try {
-        const { rows: loaded } = await ensureRowsLoadedUpToRef.current(MAX_TABLE_SELECTION_ROWS)
+        const excludedCount =
+          rowSelectionRef.current.kind === 'all' ? (rowSelectionRef.current.excluded?.size ?? 0) : 0
+        const { rows: loaded } = await ensureRowsLoadedUpToRef.current(
+          drainTargetForChip(excludedCount)
+        )
         // A column selection spans all rows; a gutter select-all filters by the
         // (exclusion-aware) row selection.
         const drained = (

apps/sim/app/workspace/[workspaceId]/tables/[tableId]/components/table-grid/utils.test.ts

@@ -12,6 +12,7 @@ import {
   buildTableSelectionContext,
   canWriteRowsWithChip,
   chipRowCount,
+  drainTargetForChip,
   selectedColumnIds,
 } from './utils'
 
@@ -131,3 +132,18 @@ describe('chipRowCount', () => {
     }
   )
 })
+
+describe('drainTargetForChip', () => {
+  it('still yields a full cap when every exclusion lands in the loaded prefix', () => {
+    // The worst case for a gutter select-all: exclusions are filtered out AFTER
+    // loading, so loading only the cap would leave the chip short of the count
+    // the menu already advertised.
+    const excluded = 30
+
+    expect(drainTargetForChip(excluded) - excluded).toBe(MAX_TABLE_SELECTION_ROWS)
+  })
+
+  it('loads exactly the cap when nothing is excluded', () => {
+    expect(drainTargetForChip(0)).toBe(MAX_TABLE_SELECTION_ROWS)
+  })
+})

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

@@ -404,6 +404,18 @@ export function buildTableSelectionContext(opts: {
   }
 }
 
+/**
+ * How many rows to load before building a select-all chip. A gutter select-all
+ * can carry exclusions anywhere in the table, and they are filtered out AFTER
+ * loading — so loading only {@link MAX_TABLE_SELECTION_ROWS} yields fewer than
+ * the cap whenever an excluded row sits in that prefix, leaving the chip short
+ * of the count the menu advertised. Loading the cap plus the exclusion count
+ * covers the worst case, where every exclusion falls inside the prefix.
+ */
+export function drainTargetForChip(excludedCount: number): number {
+  return MAX_TABLE_SELECTION_ROWS + excludedCount
+}
+
 /**
  * Rows a chip will actually reference for a selection of `requested` rows —
  * {@link buildTableSelectionContext} caps its `rowIds`, so any count shown to