u-tree.vue
2.57 KB
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
<template>
<view class="u-tree">
<tree-node
v-for="node in treeData"
:key="node[props.nodeKey]"
:node="node"
:props="props"
:show-checkbox="showCheckbox"
:check-strictly="checkStrictly"
:expand-on-click-node="expandOnClickNode"
@node-click="handleNodeClick"
@check-change="$emit('check-change', $event)">
<template #default="{ nodeData, level }">
<slot :node="nodeData" :level="level"></slot>
</template>
</tree-node>
</view>
</template>
<script>
import TreeNode from './tree-node.vue';
export default {
name: 'u-tree',
components: { TreeNode },
props: {
data: {
type: Array,
required: true
},
props: {
type: Object,
default: () => ({
label: 'label',
children: 'children',
nodeKey: 'id'
})
},
showCheckbox: {
type: Boolean,
default: false
},
defaultExpandAll: {
type: Boolean,
default: false
},
expandOnClickNode: {
type: Boolean,
default: true
},
checkStrictly: {
type: Boolean,
default: false
}
},
data() {
return {
treeData: []
};
},
created() {
this.initTree();
},
watch: {
data: {
handler(newVal) {
this.treeData = JSON.parse(JSON.stringify(newVal));
this.initExpandedState(this.treeData, this.defaultExpandAll);
},
deep: true,
immediate: true
}
},
emits: ['node-click', 'check-change'],
methods: {
initTree() {
this.treeData = JSON.parse(JSON.stringify(this.data));
this.initExpandedState(this.treeData, this.defaultExpandAll);
},
initExpandedState(nodes, expanded) {
nodes.forEach(node => {
node.expanded = expanded;
if (node[this.props.children]) {
this.initExpandedState(node[this.props.children], expanded);
}
});
},
handleNodeClick(node) {
this.$emit('node-click', node);
},
/**
* 直接递归 treeData 获取所有 checked 的节点
*/
getCheckedNodes() {
const traverse = (nodes) => {
let result = [];
nodes.forEach(node => {
if (node.checked) {
result.push(node);
}
if (node[this.props.children] && node[this.props.children].length > 0) {
result = result.concat(traverse(node[this.props.children]));
}
});
return result;
};
return traverse(this.treeData);
}
}
};
</script>
<style scoped>
.u-tree {
font-size: 28rpx;
}
</style>