From 35fe3cfbe530fe8ca70873793790c532514ff3b7 Mon Sep 17 00:00:00 2001
From: Kevin Van Cott <kevinvandy656@gmail.com>
Date: Fri, 9 Feb 2024 10:58:07 -0600
Subject: [PATCH] fix clear selection button to clear all pages

---
 .../src/components/inputs/MRT_SelectCheckbox.tsx              | 2 +-
 .../src/components/toolbar/MRT_ToolbarAlertBanner.tsx         | 4 +++-
 packages/material-react-table/src/utils/row.utils.ts          | 3 ++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/packages/material-react-table/src/components/inputs/MRT_SelectCheckbox.tsx b/packages/material-react-table/src/components/inputs/MRT_SelectCheckbox.tsx
index 1683abd7c..d45481bb8 100644
--- a/packages/material-react-table/src/components/inputs/MRT_SelectCheckbox.tsx
+++ b/packages/material-react-table/src/components/inputs/MRT_SelectCheckbox.tsx
@@ -87,7 +87,7 @@ export const MRT_SelectCheckbox = <TData extends MRT_RowData>({
     },
     onChange: (event) => {
       event.stopPropagation();
-      row ? onSelectionChange!(event) : onSelectAllChange(event);
+      selectAll ? onSelectAllChange(event) : onSelectionChange!(event);
     },
     size: (density === 'compact' ? 'small' : 'medium') as 'medium' | 'small',
     ...checkboxProps,
diff --git a/packages/material-react-table/src/components/toolbar/MRT_ToolbarAlertBanner.tsx b/packages/material-react-table/src/components/toolbar/MRT_ToolbarAlertBanner.tsx
index 806d523de..76f5e5917 100644
--- a/packages/material-react-table/src/components/toolbar/MRT_ToolbarAlertBanner.tsx
+++ b/packages/material-react-table/src/components/toolbar/MRT_ToolbarAlertBanner.tsx
@@ -68,7 +68,9 @@ export const MRT_ToolbarAlertBanner = <TData extends MRT_RowData>({
           ?.replace('{selectedCount}', selectedRowCount.toLocaleString())
           ?.replace('{rowCount}', totalRowCount.toString())}
         <Button
-          onClick={(event) => getMRT_SelectAllHandler({ table })(event, false)}
+          onClick={(event) =>
+            getMRT_SelectAllHandler({ table })(event, false, true)
+          }
           size="small"
           sx={{ p: '2px' }}
         >
diff --git a/packages/material-react-table/src/utils/row.utils.ts b/packages/material-react-table/src/utils/row.utils.ts
index f0295c4f4..576bc0060 100644
--- a/packages/material-react-table/src/utils/row.utils.ts
+++ b/packages/material-react-table/src/utils/row.utils.ts
@@ -243,13 +243,14 @@ export const getMRT_SelectAllHandler =
   (
     event: ChangeEvent<HTMLInputElement> | MouseEvent<HTMLButtonElement>,
     value?: boolean,
+    forceAll?: boolean,
   ) => {
     const {
       options: { enableRowPinning, rowPinningDisplayMode, selectAllMode },
       refs: { lastSelectedRowId },
     } = table;
 
-    selectAllMode === 'all'
+    selectAllMode === 'all' || forceAll
       ? table.toggleAllRowsSelected(value ?? (event as any).target.checked)
       : table.toggleAllPageRowsSelected(value ?? (event as any).target.checked);
     if (enableRowPinning && rowPinningDisplayMode?.includes('select')) {