fix

Shopify · Mar 21, 2024 · f73ab99 · f73ab99
1 parent 82f1811
commit f73ab99
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/polaris-react/src/utilities/use-index-resource-state.ts b/polaris-react/src/utilities/use-index-resource-state.ts
@@ -110,20 +110,26 @@ export function useIndexResourceState<T extends {[key: string]: unknown}>(
             return newSelectedResources;
           });
           break;
+
         case SelectionType.All:
         case SelectionType.Page: {
           const resourceList = resourceFilter
             ? resources.filter(resourceFilter)
             : resources;
+          const mappedResources = new Set(resourceList.map(resourceIDResolver));
+
           const hasRoomForMoreSelection =
             isSelecting && selectedResources.size < resourceList.length;
-          const resourceIDs = new Set(resourceList.map(resourceIDResolver));
 
           setSelectedResources(
-            hasRoomForMoreSelection ? resourceIDs : new Set(),
+            hasRoomForMoreSelection || isSelecting
+              ? mappedResources
+              : new Set(),
           );
           setUnselectedResources(
-            hasRoomForMoreSelection ? new Set() : resourceIDs,
+            hasRoomForMoreSelection || isSelecting
+              ? new Set()
+              : mappedResources,
           );
           break;
         }