Fix: controlled focussed cell not kept in focus

packages/table/src/tableHotkeys.ts

@@ -52,8 +52,16 @@ export class TableHotkeys {
         this.props = props;
     }
 
-    public setState(state: TableState) {
-        this.state = state;
+    public setState(newState: TableState) {
+        if (
+            newState.focusedCell != null &&
+            (this.state.focusedCell == null || this.state.focusedCell !== newState.focusedCell)
+        ) {
+            // Ensure controlled focused cell is scrolled into view
+            this.scrollBodyToFocusedCell(newState.focusedCell);
+        }
+
+        this.state = newState;
     }
 
     // Selection
@@ -295,6 +303,8 @@ export class TableHotkeys {
             return;
         }
 
+        // The `viewportRect`'s origin does _not_ include frozen cells but the cumulative cell heights / widths _do_.
+        // So, there is a mismatch in origins that we fix this way (for now).
         const frozenRowsHeight = this.grid.getCumulativeHeightBefore(this.state.numFrozenRowsClamped);
         const frozenColumnsWidth = this.grid.getCumulativeWidthBefore(this.state.numFrozenColumnsClamped);