98 lines
3.1 KiB
TypeScript
98 lines
3.1 KiB
TypeScript
import type { InsertNodeParams, ReplaceFields, TreeItem } from './types';
|
|
import type { Ref, ComputedRef } from 'vue';
|
|
|
|
import { cloneDeep } from 'lodash-es';
|
|
import { unref } from 'vue';
|
|
import { forEach } from '/@/utils/helper/treeHelper';
|
|
|
|
export function useTree(
|
|
treeDataRef: Ref<TreeItem[]>,
|
|
getReplaceFields: ComputedRef<ReplaceFields>
|
|
) {
|
|
// 更新节点
|
|
function updateNodeByKey(key: string, node: TreeItem, list?: TreeItem[]) {
|
|
if (!key) return;
|
|
const treeData = list || unref(treeDataRef);
|
|
const { key: keyField, children: childrenField } = unref(getReplaceFields);
|
|
|
|
if (!childrenField || !keyField) return;
|
|
|
|
for (let index = 0; index < treeData.length; index++) {
|
|
const element: any = treeData[index];
|
|
const children = element[childrenField];
|
|
|
|
if (element[keyField] === key) {
|
|
treeData[index] = { ...treeData[index], ...node };
|
|
break;
|
|
} else if (children && children.length) {
|
|
updateNodeByKey(key, node, element[childrenField]);
|
|
}
|
|
}
|
|
}
|
|
|
|
// 展开指定级别
|
|
function filterByLevel(level = 1, list?: TreeItem[], currentLevel = 1) {
|
|
if (!level) {
|
|
return [];
|
|
}
|
|
const res: (string | number)[] = [];
|
|
const data = list || unref(treeDataRef) || [];
|
|
for (let index = 0; index < data.length; index++) {
|
|
const item = data[index] as any;
|
|
|
|
const { key: keyField, children: childrenField } = unref(getReplaceFields);
|
|
const key = keyField ? item[keyField] : '';
|
|
const children = childrenField ? item[childrenField] : [];
|
|
res.push(key);
|
|
if (children && children.length && currentLevel < level) {
|
|
currentLevel += 1;
|
|
res.push(...filterByLevel(level, children, currentLevel));
|
|
}
|
|
}
|
|
return res as string[] | number[];
|
|
}
|
|
|
|
/**
|
|
* 添加节点
|
|
*/
|
|
function insertNodeByKey({ parentKey = null, node, push = 'push' }: InsertNodeParams) {
|
|
const treeData: any = cloneDeep(unref(treeDataRef));
|
|
if (!parentKey) {
|
|
treeData[push](node);
|
|
treeDataRef.value = treeData;
|
|
return;
|
|
}
|
|
const { key: keyField, children: childrenField } = unref(getReplaceFields);
|
|
if (!childrenField || !keyField) return;
|
|
|
|
forEach(treeData, (treeItem) => {
|
|
if (treeItem[keyField] === parentKey) {
|
|
treeItem[childrenField] = treeItem[childrenField] || [];
|
|
treeItem[childrenField][push](node);
|
|
}
|
|
});
|
|
treeDataRef.value = treeData;
|
|
}
|
|
|
|
// 删除节点
|
|
function deleteNodeByKey(key: string, list?: TreeItem[]) {
|
|
if (!key) return;
|
|
const treeData = list || unref(treeDataRef);
|
|
const { key: keyField, children: childrenField } = unref(getReplaceFields);
|
|
if (!childrenField || !keyField) return;
|
|
|
|
for (let index = 0; index < treeData.length; index++) {
|
|
const element: any = treeData[index];
|
|
const children = element[childrenField];
|
|
|
|
if (element[keyField] === key) {
|
|
treeData.splice(index, 1);
|
|
break;
|
|
} else if (children && children.length) {
|
|
deleteNodeByKey(key, element[childrenField]);
|
|
}
|
|
}
|
|
}
|
|
return { deleteNodeByKey, insertNodeByKey, filterByLevel, updateNodeByKey };
|
|
}
|