feat: 🎸 added node update support to TreeStateModifiers

Author: diogofcunha

index.d.ts

@@ -150,7 +150,7 @@ export interface TreeState {
 }
 
 export interface TreeStateModifiers {
-  editNodeAt: (state: State, index: number, setNode: (oldNode: Node) => Node) => State;
+  editNodeAt: (state: State, index: number, updateNode: ((oldNode: Node) => Node) | Node) => State;
   deleteNodeAt: (state: State, index: number) => State;
 }
 

src/state/TreeStateModifiers.js

@@ -21,12 +21,12 @@ export default class TreeStateModifiers {
    * Given a state, finds a node at a certain row index.
    * @param {State} state - The current state
    * @param {number} index - The visible row index
-   * @param {setNode} setNode - A function to update the node
+   * @param {setNode|Node} nodeUpdate - A function to update the node
    * @return {State} An internal state representation
    */
-  static editNodeAt = (state, index, setNode) => {
+  static editNodeAt = (state, index, nodeUpdate) => {
     const node = TreeState.getNodeAt(state, index);
-    const updatedNode = setNode(node);
+    const updatedNode = typeof nodeUpdate === 'function' ? nodeUpdate(node) : nodeUpdate;
     const flattenedTree = [...state.flattenedTree];
     const flattenedNodeMap = flattenedTree[index];
     const parents = flattenedNodeMap.slice(0, flattenedNodeMap.length - 1);