tree v2 Add the method of dynamically expanding and collapsing the tr…

…ee, and add the method of getting the node
sukney · Feb 10, 2022 · 948831c · 948831c
1 parent d953bd4
commit 948831c
Show file tree

Hide file tree

Showing 2 changed files with 69 additions and 1 deletion.
diff --git a/packages/components/tree-v2/src/composables/useTree.ts b/packages/components/tree-v2/src/composables/useTree.ts
@@ -228,11 +228,64 @@ export function useTree(props: TreeProps, emit) {
     emit(NODE_EXPAND, node.data, node)
   }
 
+  function expandByKeys(keys: Array<TreeKey>) {
+    if (keys && keys.length > 0) {
+      const keySet = expandedKeySet.value
+      if (tree?.value) {
+        // whether only one node among the same level can be expanded at one time
+        const { treeNodeMap } = tree.value
+
+        keys.forEach((key) => {
+          const node = treeNodeMap.get(key)
+          if (node) {
+            if (!keySet.has(node.key)) keySet.add(node.key)
+            emit(NODE_EXPAND, node.data, node)
+          }
+        })
+      }
+    }
+  }
+
   function collapse(node: TreeNode) {
     expandedKeySet.value.delete(node.key)
     emit(NODE_COLLAPSE, node.data, node)
   }
 
+  function collapseByKeys(keys: Array<TreeKey>) {
+    if (keys && keys.length > 0) {
+      const keySet = expandedKeySet.value
+      if (tree?.value) {
+        // whether only one node among the same level can be expanded at one time
+        const { treeNodeMap } = tree.value
+        keys.forEach((key) => {
+          const node = treeNodeMap.get(key)
+          if (node) {
+            if (keySet.has(node.key)) keySet.delete(node.key)
+            emit(NODE_COLLAPSE, node.data, node)
+          }
+        })
+      }
+    }
+  }
+
+   function getTreeNode(key: TreeKey) {
+    var nodes = getTreeNodes([key])
+    if (nodes && nodes.length > 0) return nodes[0]
+    else return undefined
+  }
+
+  function getTreeNodes(keys: Array<TreeKey>) {
+    let nodes: Array<TreeNode> = []
+    if (tree?.value) {
+      const { treeNodeMap } = tree.value
+      keys.forEach((x) => {
+        let node = treeNodeMap.get(x)
+        if (node) nodes.push(node)
+      })
+    }
+    return nodes
+  }
+
   function isExpanded(node: TreeNode): boolean {
     return expandedKeySet.value.has(node.key)
   }
@@ -291,5 +344,9 @@ export function useTree(props: TreeProps, emit) {
     setCheckedKeys,
     filter,
     setData,
+    expandByKeys,
+    collapseByKeys,
+    getTreeNode,
+    getTreeNodes,
   }
 }
diff --git a/packages/components/tree-v2/src/tree.vue b/packages/components/tree-v2/src/tree.vue
@@ -90,6 +90,12 @@ export default defineComponent({
       setCheckedKeys,
       filter,
       setData,
+      expand,
+      collapse,
+      expandByKeys,
+      collapseByKeys,
+      getTreeNode,
+      getTreeNodes,
     } = useTree(props, ctx.emit)
 
     ctx.expose({
@@ -104,7 +110,12 @@ export default defineComponent({
       setCheckedKeys,
       filter,
       setData,
-	  toggleCheckbox,
+      expand,
+      collapse,
+      expandByKeys,
+      collapseByKeys,
+      getTreeNode,
+      getTreeNodes,
     })
 
     return {