Merge pull request #23 from tuj84257/master

#5 Add `expandAll` and `collapseAll` methods

README.md

@@ -202,6 +202,13 @@ const disabledNodes = myTree.disabledNodes;
 | loaded     | null         | invoke after the tree load data    |
 | onChange   | null         | invoke when the node status change |
 
+## Methods
+
+| Method      | Parameters   | Description                        |
+| ----------- | ------------ | ---------------------------------- |
+| expandAll   | null         | expand all tree nodes              |
+| collapseAll | null         | collapse all tree nodes            |
+
 ## License
 
 [MIT](./LICENSE)

src/index.js

@@ -34,6 +34,27 @@ function animation(duration, callback) {
   });
 }
 
+function collapseFromLeaf(tree, leafNode) {
+  try {
+    const nodeLiElement = tree.liElementsById[leafNode.parent.id];
+    if(!nodeLiElement.classList.contains('treejs-node__close'))
+      nodeLiElement.getElementsByClassName('treejs-switcher')[0].click();
+  } catch (error) {
+    return;
+  }
+  if(leafNode.hasOwnProperty('parent'))
+    collapseFromLeaf(tree, leafNode.parent);
+}
+
+function expandFromRoot(tree, root) {
+  const nodeLiElement = tree.liElementsById[root.id];
+  if(nodeLiElement.classList.contains('treejs-node__close'))
+    nodeLiElement.getElementsByClassName('treejs-switcher')[0].click();
+  if(root.hasOwnProperty('children'))
+    for(let child of root.children)
+      expandFromRoot(tree, child);
+}
+
 export default function Tree(container, options) {
   const defaultOptions = {
     selectMode: 'checkbox',
@@ -438,6 +459,18 @@ Tree.prototype.updateLiElement = function(node) {
   }
 };
 
+Tree.prototype.collapseAll = function() {
+  const leafNodesById = this.leafNodesById;
+  for(let id in leafNodesById) {
+    const leafNode = leafNodesById[id];
+    collapseFromLeaf(this, leafNode);
+  }
+}
+
+Tree.prototype.expandAll = function() {
+  expandFromRoot(this, this.treeNodes[0]);
+}
+
 Tree.parseTreeData = function(data) {
   const treeNodes = deepClone(data);
   const nodesById = {};