feat: add more parameters to rowHeight. Fixes #199

Added `treeIndex`, `node` and `path` to the parameters.
frontend-collective · Jan 14, 2018 · 8ff0ff2 · 8ff0ff2
1 parent a80d4cb
commit 8ff0ff2
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/src/react-sortable-tree.js b/src/react-sortable-tree.js
@@ -619,7 +619,17 @@ class ReactSortableTree extends Component {
               estimatedRowSize={
                 typeof rowHeight !== 'function' ? rowHeight : undefined
               }
-              rowHeight={rowHeight}
+              rowHeight={
+                typeof rowHeight !== 'function'
+                  ? rowHeight
+                  : ({ index }) =>
+                      rowHeight({
+                        index,
+                        treeIndex: index,
+                        node: rows[index].node,
+                        path: rows[index].path,
+                      })
+              }
               rowRenderer={({ index, style: rowStyle }) =>
                 this.renderRow(rows[index], {
                   listIndex: index,
@@ -645,7 +655,12 @@ class ReactSortableTree extends Component {
             height:
               typeof rowHeight !== 'function'
                 ? rowHeight
-                : rowHeight({ index }),
+                : rowHeight({
+                    index,
+                    treeIndex: index,
+                    node: row.node,
+                    path: row.path,
+                  }),
           },
           getPrevRow: () => rows[index - 1] || null,
           matchKeys,