From 007579f55e8f53f4f557b4fa11a02bf5634d9c79 Mon Sep 17 00:00:00 2001
From: Faris Ansari <netchamp.faris@gmail.com>
Date: Fri, 5 Apr 2019 17:02:31 +0530
Subject: [PATCH] feat: Add API for expand/collapse tree nodes

---
 src/rowmanager.js | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/rowmanager.js b/src/rowmanager.js
index 208fbb0..43a1260 100644
--- a/src/rowmanager.js
+++ b/src/rowmanager.js
@@ -218,6 +218,22 @@ export default class RowManager {
         this.showRows(rowsToShow);
     }
 
+    expandAllNodes() {
+        let rows = this.datamanager.getRows();
+        let rootNodes = rows.filter(row => !row.meta.isLeaf);
+        rootNodes.map(row => {
+            this.openSingleNode(row.meta.rowIndex);
+        });
+    }
+
+    collapseAllNodes() {
+        let rows = this.datamanager.getRows();
+        let rootNodes = rows.filter(row => row.meta.indent === 0);
+        rootNodes.map(row => {
+            this.closeSingleNode(row.meta.rowIndex);
+        });
+    }
+
     getRow$(rowIndex) {
         return $(this.selector(rowIndex), this.bodyScrollable);
     }