Ben Lin
2024-07-06 e23fdf87e6c3bc6d42198b9309617195f2efd6c0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
<template>
  <PageWrapper title="Tree函数操作示例" contentBackground contentClass="p-4">
    <div class="mb-4">
      <a-button @click="expandAll(true)" class="mr-2"> 展开全部 </a-button>
      <a-button @click="expandAll(false)" class="mr-2"> 折叠全部 </a-button>
      <a-button @click="checkAll(true)" class="mr-2"> 全选 </a-button>
      <a-button @click="checkAll(false)" class="mr-2"> 全不选 </a-button>
      <a-button @click="handleLevel(2)" class="mr-2"> 显示到第2级 </a-button>
      <a-button @click="handleLevel(1)" class="mr-2"> 显示到第1级 </a-button>
    </div>
    <div class="mb-4">
      <a-button @click="handleSetCheckData" class="mr-2"> 设置勾选数据 </a-button>
      <a-button @click="handleGetCheckData" class="mr-2"> 获取勾选数据 </a-button>
      <a-button @click="handleSetSelectData" class="mr-2"> 设置选中数据 </a-button>
      <a-button @click="handleGetSelectData" class="mr-2"> 获取选中数据 </a-button>
      <a-button @click="handleGetSelectNode" class="mr-2"> 获取选中节点 </a-button>
      <a-button @click="handleGetTreeData" class="mr-2"> 获取Tree数据 </a-button>
      <a-button @click="handleSetExpandData" class="mr-2"> 设置展开数据 </a-button>
      <a-button @click="handleGetExpandData" class="mr-2"> 获取展开数据 </a-button>
    </div>
    <div class="mb-4">
      <a-button @click="appendNodeByKey(null)" class="mr-2"> 添加根节点 </a-button>
      <a-button @click="appendNodeByKey('2-2')" class="mr-2"> 添加在parent3内添加节点 </a-button>
      <a-button @click="deleteNodeByKey('2-2')" class="mr-2"> 删除parent3节点 </a-button>
      <a-button @click="updateNodeByKey('1-1')" class="mr-2"> 更新parent2节点 </a-button>
    </div>
    <BasicTree :treeData="treeData" title="函数操作" ref="treeRef" :checkable="true" />
  </PageWrapper>
</template>
<script lang="ts" setup>
  import { ref, unref } from 'vue';
  import { BasicTree, TreeActionType } from '@/components/Tree';
  import { treeData } from './data';
  import { useMessage } from '@/hooks/web/useMessage';
  import { PageWrapper } from '@/components/Page';
  import { type Nullable } from '@vben/types';
 
  const treeRef = ref<Nullable<TreeActionType>>(null);
  const { createMessage } = useMessage();
 
  function getTree() {
    const tree = unref(treeRef);
    if (!tree) {
      throw new Error('tree is null!');
    }
    return tree;
  }
 
  function handleLevel(level: number) {
    getTree().filterByLevel(level);
  }
 
  function handleSetCheckData() {
    getTree().setCheckedKeys(['0-0']);
  }
 
  function handleGetCheckData() {
    const keys = getTree().getCheckedKeys();
    createMessage.success(JSON.stringify(keys));
  }
 
  function handleSetSelectData() {
    getTree().setSelectedKeys(['0-0']);
  }
 
  function handleGetSelectData() {
    const keys = getTree().getSelectedKeys();
    createMessage.success(JSON.stringify(keys));
  }
 
  function handleGetSelectNode() {
    const keys = getTree().getSelectedKeys();
    const node = getTree().getSelectedNode(keys[0]);
    createMessage.success(node !== null ? JSON.stringify(node) : null);
  }
 
  function handleSetExpandData() {
    getTree().setExpandedKeys(['0-0']);
  }
 
  function handleGetExpandData() {
    const keys = getTree().getExpandedKeys();
    createMessage.success(JSON.stringify(keys));
  }
 
  function checkAll(checkAll: boolean) {
    getTree().checkAll(checkAll);
  }
 
  function expandAll(checkAll: boolean) {
    getTree().expandAll(checkAll);
  }
 
  function appendNodeByKey(parentKey: string | null = null) {
    getTree().insertNodeByKey({
      parentKey: parentKey,
      node: {
        title: '新增节点',
        key: '2-2-2',
      },
      // 往后插入
      push: 'push',
      // 往前插入
      // push:'unshift'
    });
  }
 
  function deleteNodeByKey(key: string) {
    getTree().deleteNodeByKey(key);
  }
 
  function updateNodeByKey(key: string) {
    getTree().updateNodeByKey(key, {
      title: 'parent2-new',
    });
  }
 
  function handleGetTreeData() {
    createMessage.success(JSON.stringify(getTree().getTreeData()));
  }
</script>