删除react-sortable-tree中的节点

import React, { useState } from 'react';
import SortableTree from 'react-sortable-tree';
import 'react-sortable-tree/style.css';

const MyTreeComponent = () => {
  const [treeData, setTreeData] = useState([
    { title: 'Node 1', expanded: true, children: [{ title: 'Child 1' }] },
    { title: 'Node 2', expanded: true, children: [{ title: 'Child 2' }] },
  ]);

  const handleNodeDelete = (node) => {
    const newData = removeNode(treeData, node);
    setTreeData(newData);
  };

  const removeNode = (data, node) => {
    return data.filter((item) => {
      if (item.id === node.id) {
        return false;
      } else if (item.children) {
        item.children = removeNode(item.children, node);
      }
      return true;
    });
  };

  return (
    <div style={{ height: 400 }}>
      <SortableTree
        treeData={treeData}
        onChange={(data) => setTreeData(data)}
        generateNodeProps={({ node }) => ({
          buttons: [
            <button onClick={() => handleNodeDelete(node)}>Delete</button>,
          ],
        })}
      />
    </div>
  );
};

export default MyTreeComponent;