add truncate warning

santisq · Sep 6, 2024 · 9fbaaca · 9fbaaca
1 parent 81edd27
commit 9fbaaca
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/PSADTree/PSADTreeCmdletBase.cs b/src/PSADTree/PSADTreeCmdletBase.cs
@@ -16,6 +16,8 @@ public abstract class PSADTreeCmdletBase : PSCmdlet, IDisposable
 
     private bool _disposed;
 
+    private bool _truncateWritten;
+
     protected readonly Stack<(GroupPrincipal? group, TreeGroup treeGroup)> _stack = new();
 
     internal readonly TreeCache _cache = new();
@@ -87,6 +89,14 @@ protected void Push(GroupPrincipal? groupPrincipal, TreeGroup treeGroup)
         {
             _stack.Push((groupPrincipal, treeGroup));
         }
+        else
+        {
+            if (!_truncateWritten)
+            {
+                this.WriteWarning($"Result is truncated as enumeration has exceeded the set depth of {Depth}.");
+                _truncateWritten = true;
+            }
+        }
     }
 
     private static bool MatchAny(