mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-06 05:58:46 +08:00
e7aa014c31
* docs: init * chore: all types * docs: faq * chore: fix lint
112 lines
2.9 KiB
TypeScript
112 lines
2.9 KiB
TypeScript
import React, { useState } from 'react';
|
|
import { Tree } from 'antd';
|
|
import type { TreeDataNode, TreeProps } from 'antd';
|
|
|
|
const x = 3;
|
|
const y = 2;
|
|
const z = 1;
|
|
const defaultData: TreeDataNode[] = [];
|
|
|
|
const generateData = (_level: number, _preKey?: React.Key, _tns?: TreeDataNode[]) => {
|
|
const preKey = _preKey || '0';
|
|
const tns = _tns || defaultData;
|
|
|
|
const children: React.Key[] = [];
|
|
for (let i = 0; i < x; i++) {
|
|
const key = `${preKey}-${i}`;
|
|
tns.push({ title: key, key });
|
|
if (i < y) {
|
|
children.push(key);
|
|
}
|
|
}
|
|
if (_level < 0) {
|
|
return tns;
|
|
}
|
|
const level = _level - 1;
|
|
children.forEach((key, index) => {
|
|
tns[index].children = [];
|
|
return generateData(level, key, tns[index].children);
|
|
});
|
|
};
|
|
generateData(z);
|
|
|
|
const App: React.FC = () => {
|
|
const [gData, setGData] = useState(defaultData);
|
|
const [expandedKeys] = useState(['0-0', '0-0-0', '0-0-0-0']);
|
|
|
|
const onDragEnter: TreeProps['onDragEnter'] = (info) => {
|
|
console.log(info);
|
|
// expandedKeys, set it when controlled is needed
|
|
// setExpandedKeys(info.expandedKeys)
|
|
};
|
|
|
|
const onDrop: TreeProps['onDrop'] = (info) => {
|
|
console.log(info);
|
|
const dropKey = info.node.key;
|
|
const dragKey = info.dragNode.key;
|
|
const dropPos = info.node.pos.split('-');
|
|
const dropPosition = info.dropPosition - Number(dropPos[dropPos.length - 1]); // the drop position relative to the drop node, inside 0, top -1, bottom 1
|
|
|
|
const loop = (
|
|
data: TreeDataNode[],
|
|
key: React.Key,
|
|
callback: (node: TreeDataNode, i: number, data: TreeDataNode[]) => void,
|
|
) => {
|
|
for (let i = 0; i < data.length; i++) {
|
|
if (data[i].key === key) {
|
|
return callback(data[i], i, data);
|
|
}
|
|
if (data[i].children) {
|
|
loop(data[i].children!, key, callback);
|
|
}
|
|
}
|
|
};
|
|
const data = [...gData];
|
|
|
|
// Find dragObject
|
|
let dragObj: TreeDataNode;
|
|
loop(data, dragKey, (item, index, arr) => {
|
|
arr.splice(index, 1);
|
|
dragObj = item;
|
|
});
|
|
|
|
if (!info.dropToGap) {
|
|
// Drop on the content
|
|
loop(data, dropKey, (item) => {
|
|
item.children = item.children || [];
|
|
// where to insert. New item was inserted to the start of the array in this example, but can be anywhere
|
|
item.children.unshift(dragObj);
|
|
});
|
|
} else {
|
|
let ar: TreeDataNode[] = [];
|
|
let i: number;
|
|
loop(data, dropKey, (_item, index, arr) => {
|
|
ar = arr;
|
|
i = index;
|
|
});
|
|
if (dropPosition === -1) {
|
|
// Drop on the top of the drop node
|
|
ar.splice(i!, 0, dragObj!);
|
|
} else {
|
|
// Drop on the bottom of the drop node
|
|
ar.splice(i! + 1, 0, dragObj!);
|
|
}
|
|
}
|
|
setGData(data);
|
|
};
|
|
|
|
return (
|
|
<Tree
|
|
className="draggable-tree"
|
|
defaultExpandedKeys={expandedKeys}
|
|
draggable
|
|
blockNode
|
|
onDragEnter={onDragEnter}
|
|
onDrop={onDrop}
|
|
treeData={gData}
|
|
/>
|
|
);
|
|
};
|
|
|
|
export default App;
|