add scroll to index

Author: Avivhdr

index.d.ts

@@ -109,7 +109,8 @@ interface Selectors {
   deleteNodeFromTree: (nodes: Node[], nodeToDelete: FlattenedNode) => Node[],
   deleteNode: (node: FlattenedNode[]) => NodeAction,
   addNode: (node: FlattenedNode[]) => NodeAction,
-  updateNode: (node: FlattenedNode, state: { [stateKey: string]: any }) => NodeAction
-}
+  updateNode: (node: FlattenedNode, state: { [stateKey: string]: any }) => NodeAction,
+  getFlattenedTree: (node: node[]) => FlattenedNode,
+  getRowIndexFromId: (node: FlattenedNode, id: number) => number,}
 
 export const selectors: Selectors;

src/Tree.js

@@ -32,7 +32,7 @@ export default class Tree extends React.Component {
   };
 
   render() {
-    const {nodes} = this.props;
+    const {nodes, scrollToIndex} = this.props;
 
     return (
       <AutoSizer>
@@ -45,6 +45,7 @@ export default class Tree extends React.Component {
             rowHeight={this._cache.rowHeight}
             rowRenderer={this.measureRowRenderer(nodes)}
             width={width}
+            scrollToIndex={scrollToIndex}
           />
         )}
       </AutoSizer>

src/TreeContainer.js

@@ -49,6 +49,7 @@ export default class TreeContainer extends React.Component {
         nodes={getFlattenedTree(this.props.nodes)}
         onChange={this.handleChange}
         NodeRenderer={this.props.children}
+        scrollToIndex={this.props.scrollToIndex}
       />
     );
   }
@@ -62,6 +63,7 @@ TreeContainer.propTypes = {
   onChange: PropTypes.func,
   children: PropTypes.func.isRequired,
   nodeMarginLeft: PropTypes.number,
+  scrollToIndex: PropTypes.number,
 };
 
 TreeContainer.defaultProps = {

src/selectors/nodes.js

@@ -3,6 +3,8 @@ import omit from 'lodash.omit';
 
 import {UPDATE_TYPE} from '../contants';
 
+export { getFlattenedTree } from './getFlattenedTree';
+
 export const getNodeRenderOptions = createSelector(
   node => (node.state || {}).expanded,
   node => (node.state || {}).favorite,
@@ -86,3 +88,6 @@ export const addNode = node => ({
   node,
   type: UPDATE_TYPE.ADD,
 });
+
+export const getRowIndexFromId = (flattenedTree, id) => 
+  flattenedTree.findIndex((item, index) => item.id === id);