antd dragTree 控制只能在同层级拖拽

import React, { useState, useEffect } from 'react';
import { Tree, Popconfirm, message } from 'antd';
import { user } from '@pms/console';
import { MinusCircleOutlined } from '@ant-design/icons';
import AddPositionModal from '../AddPositionModal';
import { getTree, del } from '@/services/position';

//循环
const loop = (data, key, callback) => {
  data.forEach((item, index, arr) => {
    if (item.positionId === key) {
      return callback(item, index, arr);
    }
    if (item.childNode.length > 0) {
      return loop(item.childNode, key, callback);
    }
  });
};

// 判断被移动的节点与目标节点的父节点是否一致，不可跨父节点移动
const isDescendant = (node, target, dropPosition?) => {
  console.log('node?.parentId !== target?.parentId', node?.pos, target, dropPosition);
  if ((node.childNode || []).length > 0 && node.expanded && dropPosition === 0) {
    console.log(123132);
    if (node?.pos === '0-0') return true;
    if (node?.parentId !== target?.parentId) return true;
    return false;
  }
};

const DragTree = (props) => {
  const [treeData, setTreeData] = useState([]);
  const [selectedTreeKeys, setSelectTreeKeys] = useState(''); // 设置选中树节点

  const getTreeData = async () => {
    const res = await getTree({});
    if (res?.success) {
      setTreeData(res?.data);
      setSelectTreeKeys(res?.data?.[0]?.childNode?.[0]?.childNode?.[0]?.positionId);
    }
  };

  //删除
  const handleDel = (nodeData) => {
    del({
      id: nodeData?.id,
    }).then((res) => {
      if (res.success) {
        message.success(`删除成功`);
        getTreeData();
      } else {
        message.warning(res.errorMsg);
      }
    });
  };

  useEffect(() => {
    getTreeData();
  }, []);

  //点击拖拽
  const onDrop = (info) => {
    const dragKey = info.dragNode.key; //被拖拽节点的key
    const dropKey = info.node.key; //被拖拽节点要移动到的目标key
    const dropPos = info.node.pos.split('-');
    const dropPosition = info.dropPosition - Number(dropPos[dropPos.length - 1]); //被拖拽节点要移动到的目标index(子节点第一个index从0开始)
    const data = [...treeData];

    const isSameLevel = (a, b) => {
      const aLevel = a.props.pos.split('-').length;
      const bLevel = b.props.pos.split('-').length;
      console.log('isSameLevel',aLevel,bLevel)
      return aLevel === bLevel;
    };

    const isSameParent = (a, b) => {
      const aLevel = a.props.pos.split('-');
      const bLevel = b.props.pos.split('-');
      console.log('isSameParent',aLevel,bLevel)
      aLevel.pop();
      bLevel.pop();
      return aLevel.join('') === bLevel.join('');
    };

    const isDropToFirst = (a, b) => {
      const aLevel = a.props.pos.split('-');
      const bLevel = b.props.pos.split('-');
      console.log('isDropToFirst',aLevel,bLevel)
      aLevel.pop();
      return aLevel.join('') === bLevel.join('');
    };

    const dragNode = info.dragNode;
    const dropNode = info.node;
    const canDrop = (isDropToFirst(dragNode, dropNode) || (isSameParent(dragNode, dropNode) && isSameLevel(dragNode, dropNode))&&info.dropToGap) ;
    if (!canDrop) {
      return;
    }

    let dragObj; //拿到被拖拽的节点数据
    loop(data, dragKey, (item, index, arr) => {
      arr.splice(index, 1); //将被拖拽的节点从data中删除
      dragObj = item;
    });

    if (!info.dropToGap) {
      loop(data, dropKey, (item) => {
        item.childNode = item.childNode || [];
        // 将被拖拽的节点插入到目标节点上面
        item.childNode.unshift(dragObj);
      });
    } else {
      let ar; //拿到拖拽完成后的数据结构
      let i; //拿到被拖拽节点的index
      loop(data, dropKey, (item, index, arr) => {
        ar = arr;
        i = index;
      });
      if (dropPosition === -1) {
        ar.splice(i!, 0, dragObj!);
      } else {
        ar?.splice(i! + 1, 0, dragObj!);
      }
    }
    //修改tree的数据
    setTreeData(data);
  };

  //节点
  const TitleNode = (props) => {
    const { nodeData } = props;
    return (
      <div
        style={{
          display: 'flex',
          justifyContent: 'space-between',
          width: '150px',
          alignItems: 'center',
        }}
        onClick={() => {
          if (nodeData?.childNode?.length > 0) return; //父级节点不可选中
          setSelectTreeKeys(nodeData?.positionId);
        }}
      >
        <span>{nodeData?.positionName}</span>
      </div>
    );
  };

  return (
    treeData?.length > 0 && (
      <Tree
        selectedKeys={selectedTreeKeys ? [selectedTreeKeys] : []}
        defaultExpandAll
        draggable
        blockNode
        onDrop={onDrop}
        treeData={treeData}
        fieldNames={{
          title: 'positionName',
          key: 'positionId',
          children: 'childNode',
        }}
        titleRender={(nodeData: any) => <TitleNode nodeData={nodeData} />}
      />
    )
  );
};

export default DragTree;