Blame view

uni_modules/uview-plus/components/u-tree/u-tree.vue 2.57 KB
a2702f6d   刘淇   巡查计划
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>