From 26dd94f07286c906395d8c6e2ef05571cc18a5e1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20B=C3=BCchele?= <danielbuechele@fb.com>
Date: Fri, 21 Sep 2018 10:30:07 -0700
Subject: [PATCH] selecting rows from outside

Summary: Allows to set the selected rows of a table using the `highlightedRows` property.

Reviewed By: passy

Differential Revision: D9850597

fbshipit-source-id: 62239192113f43ac257d7328b21b386b584a7cfb
---
 src/ui/components/table/ManagedTable.js | 26 +++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/src/ui/components/table/ManagedTable.js b/src/ui/components/table/ManagedTable.js
index 582945af9..a12d4f5cc 100644
--- a/src/ui/components/table/ManagedTable.js
+++ b/src/ui/components/table/ManagedTable.js
@@ -102,6 +102,10 @@ export type ManagedTableProps = {|
    * Whether to hide the column names at the top of the table.
    */
   hideHeader?: boolean,
+  /**
+   * Rows that are highlighted initially.
+   */
+  highlightedRows?: Set<string>,
 |};
 
 type ManagedTableState = {|
@@ -141,7 +145,7 @@ class ManagedTable extends React.Component<
       this.props.columnOrder ||
       Object.keys(this.props.columns).map(key => ({key, visible: true})),
     columnSizes: this.props.columnSizes || {},
-    highlightedRows: new Set(),
+    highlightedRows: this.props.highlightedRows || new Set(),
     sortOrder: null,
     shouldScrollToBottom: Boolean(this.props.stickyBottom),
   };
@@ -173,8 +177,26 @@ class ManagedTable extends React.Component<
       });
     }
 
+    //
+    if (this.props.highlightedRows !== nextProps.highlightedRows) {
+      this.setState({highlightedRows: nextProps.highlightedRows});
+      const {current} = this.tableRef;
+      const {highlightedRows} = nextProps;
+      if (current && highlightedRows && highlightedRows.size > 0) {
+        const index = nextProps.rows.findIndex(
+          ({key}) => key === Array.from(highlightedRows)[0],
+        );
+        current.scrollToItem(index);
+      }
+    }
+
     // if columnOrder has changed
-    if (nextProps.columnOrder !== this.props.columnOrder) {
+    if (
+      nextProps.columnOrder !== this.props.columnOrder &&
+      this.tableRef &&
+      this.tableRef.current
+    ) {
+      this.tableRef.current.resetAfterIndex(0);
       this.setState({
         columnOrder: nextProps.columnOrder,
       });