Use real stats of a tree after all filters

swiftyfinch · May 5, 2024 · 8c9dad1 · 8c9dad1
1 parent a39a536
commit 8c9dad1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/Sources/XTreeKit/Common/Tree/TreeFilter.swift b/Sources/XTreeKit/Common/Tree/TreeFilter.swift
@@ -20,13 +20,18 @@ final class TreeFilter {
                 maxHeight: maxHeight
             )
         }
+        let children = filteredExplicitChildren.recursiveChildren()
         return TreeNode(
             icon: tree.icon,
             name: tree.name,
             info: tree.info,
             explicitChildren: filteredExplicitChildren,
-            children: filteredExplicitChildren.recursiveChildren(),
-            stats: tree.stats
+            children: children,
+            stats: TreeNode.Stats(
+                height: tree.stats.height,
+                explicitChildrenCount: filteredExplicitChildren.count,
+                childrenCount: children.count
+            )
         )
     }
 }
diff --git a/Sources/XTreeKit/Managers/TreeManager.swift b/Sources/XTreeKit/Managers/TreeManager.swift
@@ -86,16 +86,17 @@ extension TreeManager: IInternalTreeManager {
                 maxHeight: filter.maxHeight
             )
         }
+        let children = filteredForest.recursiveChildren()
         return TreeNode(
             icon: nil,
             name: ".",
             info: nil,
             explicitChildren: filteredForest,
-            children: filteredForest.recursiveChildren(),
+            children: children,
             stats: TreeNode.Stats(
                 height: forest.maxHeight(),
-                explicitChildrenCount: forest.count,
-                childrenCount: forest.recursiveChildren().count
+                explicitChildrenCount: filteredForest.count,
+                childrenCount: children.count
             )
         )
     }