feat: Add expandTree() and collapseTree() methods to TreeState

JetBrains · Oct 30, 2023 · 776d751 · 776d751
1 parent 120bca5
commit 776d751
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/foundation/src/main/kotlin/org/jetbrains/jewel/foundation/lazy/tree/TreeState.kt b/foundation/src/main/kotlin/org/jetbrains/jewel/foundation/lazy/tree/TreeState.kt
@@ -39,4 +39,14 @@ class TreeState(
     fun openNodes(nodes: List<Any>) {
         openNodes += nodes
     }
+
+    fun expandTree() {
+        if (allNodes.size != openNodes.size) {
+            openNodes = allNodes.toSet()
+        }
+    }
+
+    fun collapseTree() {
+        if (openNodes.isNotEmpty()) openNodes = setOf()
+    }
 }