fix: clear selected rows when a refresh is done (#25)

Fixes: gisce/webclient#1289
gisce · Oct 15, 2024 · 925332d · 925332d
1 parent d437219
commit 925332d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/components/InfiniteTable/InfiniteTable.tsx b/src/components/InfiniteTable/InfiniteTable.tsx
@@ -106,7 +106,7 @@ const InfiniteTableComp = forwardRef<InfiniteTableRef, InfiniteTableProps>(
     const notifyColumnChanges = useRef(false);
     const firstTimeResized = useRef(false);
 
-    useDeepCompareEffect(() => {
+    const updateSelectedRowKeys = useCallback(() => {
       gridRef.current?.api?.forEachNode((node) => {
         if (node?.data?.id && selectedRowKeys.includes(node.data.id)) {
           node.setSelected(true);
@@ -116,6 +116,10 @@ const InfiniteTableComp = forwardRef<InfiniteTableRef, InfiniteTableProps>(
       });
     }, [selectedRowKeys]);
 
+    useDeepCompareEffect(() => {
+      updateSelectedRowKeys();
+    }, [selectedRowKeys]);
+
     useImperativeHandle(ref, () => ({
       setSelectedRows: (keys: number[]) => {
         gridRef.current?.api?.forEachNode((node) => {
@@ -130,6 +134,7 @@ const InfiniteTableComp = forwardRef<InfiniteTableRef, InfiniteTableProps>(
         gridRef.current?.api?.deselectAll();
       },
       refresh: () => {
+        gridRef.current?.api?.deselectAll();
         gridRef.current?.api?.purgeInfiniteCache();
       },
     }));
@@ -383,6 +388,7 @@ const InfiniteTableComp = forwardRef<InfiniteTableRef, InfiniteTableProps>(
         selectedRowKeys,
         columnsPersistedStateRef,
         memoizedOnRowStatus,
+        updateSelectedRowKeys,
         applyAutoFitState,
         scrollToSavedPosition,
       ],