PowerShellUtils

1.0.1

PowerShellUtils/Commands/PrintTree/BfsImplFs.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Threading;
 
namespace PowerShellStandardModule1.Commands.PrintTree;
 
public class BfsImplFs(
    Func<FileSystemInfoTreeNode, bool> Where,
    Func<FileSystemInfo, IEnumerable<FileSystemInfo>> childProvider,
    FileSystemInfo startingDirectory,
    int height,
    int limit,
    CancellationToken cancellationToken)
{
    public FileSystemInfoTreeNodeEnumerable Invoke()
    {
        if (height <= 0 || limit <= 0) return [];
         
        return new BfsExecutor<FileSystemInfo>
        {
            Where = Where,
            ChildProvider = childProvider,
            ShouldBreak = ShouldBreak
        }.Invoke(startingDirectory);
    }
 
    private bool ShouldBreak(FileSystemInfoTreeNode node)
    {
        cancellationToken.ThrowIfCancellationRequested();
        return node.Height > height || node.Count > limit;
    }
}