fix: reset row height on each re-computation

Author: Lodin

src/FixedSizeTree.tsx

@@ -77,8 +77,8 @@ export class FixedSizeTree<T extends FixedSizeNodeData = NodeData> extends Tree<
     return (
       <FixedSizeList
         {...rest}
-        itemData={this.state}
         itemCount={this.state.order!.length}
+        itemData={this.state}
         ref={this.list}
       >
         {rowComponent!}

src/Tree.tsx

@@ -12,6 +12,7 @@ import {
   ListProps,
   VariableSizeList,
 } from 'react-window';
+import {DefaultTreeProps, DefaultTreeState} from './utils';
 
 export type NodeData = Readonly<{
   /**
@@ -261,14 +262,14 @@ class Tree<
   >,
   TListComponent extends FixedSizeList | VariableSizeList
 > extends PureComponent<TProps, TState> {
-  public static defaultProps: Partial<TreeProps<any, any>> = {
+  public static defaultProps: Partial<DefaultTreeProps> = {
     rowComponent: Row,
   };
 
   public static getDerivedStateFromProps(
-    props: TreeProps<any, any>,
-    state: TreeState<any, any, any, any>,
-  ): Partial<TreeState<any, any, any, any>> {
+    props: DefaultTreeProps,
+    state: DefaultTreeState,
+  ): Partial<DefaultTreeState> {
     const {children: component, itemData: treeData, treeWalker} = props;
     const {computeTree, order, treeWalker: oldTreeWalker} = state;
 

src/VariableSizeTree.tsx

@@ -138,9 +138,7 @@ export class VariableSizeTree<T extends VariableSizeNodeData> extends Tree<
 
   public recomputeTree(options?: VariableSizeUpdateOptions): Promise<void> {
     return super.recomputeTree(options).then(() => {
-      if (options?.useDefaultHeight) {
-        this.list.current?.resetAfterIndex(0, true);
-      }
+      this.list.current?.resetAfterIndex(0, true);
     });
   }
 
@@ -150,8 +148,8 @@ export class VariableSizeTree<T extends VariableSizeNodeData> extends Tree<
     return (
       <VariableSizeList
         {...rest}
-        itemData={this.state}
         itemCount={this.state.order!.length}
+        itemData={this.state}
         // eslint-disable-next-line @typescript-eslint/unbound-method
         itemSize={itemSize ?? this.getItemSize}
         ref={this.list}

src/utils.ts

@@ -3,21 +3,29 @@ import {
   NodeData,
   NodeRecord,
   TreeCreatorOptions,
+  TreeProps,
   TreeState,
   UpdateOptions,
 } from './Tree';
 
+export type DefaultTreeProps = TreeProps<
+  NodeComponentProps<NodeData>,
+  NodeData
+>;
+
+export type DefaultTreeState = TreeState<
+  NodeComponentProps<NodeData>,
+  NodeRecord<NodeData>,
+  UpdateOptions,
+  NodeData
+>;
+
 export type DefaultTreeCreatorOptions = TreeCreatorOptions<
   NodeComponentProps<NodeData>,
   NodeRecord<NodeData>,
   UpdateOptions,
   NodeData,
-  TreeState<
-    NodeComponentProps<NodeData>,
-    NodeRecord<NodeData>,
-    UpdateOptions,
-    NodeData
-  >
+  DefaultTreeState
 >;
 
 export const identity = <T>(value: T): T => value;